Skip to content

Commit e21681a

Browse files
committed
Update to new PMT Python interface
1 parent c24217a commit e21681a

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

kernel_tuner/observers/pmt.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def __init__(self, observable=None):
4848
if not obs in supported:
4949
raise ValueError(f"Observable {obs} not in supported: {supported}")
5050

51-
self.pms = [pmt.get_pmt(obs[0], obs[1]) for obs in observable.items()]
51+
self.pms = [pmt.create(obs[0], obs[1]) for obs in observable.items()]
5252
self.pm_names = list(observable.keys())
5353

5454
self.begin_states = [None] * len(self.pms)
@@ -70,9 +70,8 @@ def after_finish(self):
7070
for i in range(len(self.pms)):
7171
begin_state = self.begin_states[i]
7272
end_state = end_states[i]
73-
measured_energy = pmt.pypmt.PMT.joules(begin_state, end_state)
74-
measured_power = pmt.pypmt.PMT.watts(begin_state, end_state)
75-
print(measured_energy, measured_power)
73+
measured_energy = pmt.joules(begin_state, end_state)
74+
measured_power = pmt.watts(begin_state, end_state)
7675
pm_name = self.pm_names[i]
7776
energy_result_name = f"{pm_name}_energy"
7877
power_result_name = f"{pm_name}_power"

0 commit comments

Comments
 (0)