Skip to content

Commit 41ae1d2

Browse files
Revert "Update PMTObserver for latest PMT changes"
This reverts commit 3737052.
1 parent 3737052 commit 41ae1d2

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

examples/cuda/vector_add_observers_pmt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@ def tune():
3131
tune_params = dict()
3232
tune_params["block_size_x"] = [128+64*i for i in range(15)]
3333

34-
pmtobserver = PMTObserver(["nvidia", "rapl"])
34+
pmtobserver = PMTObserver(["nvml", "rapl"])
3535

3636
metrics = OrderedDict()
37-
metrics["GPU W"] = lambda p: p["nvidia_power"]
37+
metrics["GPU W"] = lambda p: p["nvml_power"]
3838
metrics["CPU W"] = lambda p: p["rapl_power"]
3939

4040
results, env = tune_kernel("vector_add", kernel_string, size, args, tune_params, observers=[pmtobserver], metrics=metrics, iterations=32)

kernel_tuner/observers/pmt.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,12 +43,11 @@ def __init__(self, observable=None):
4343
else:
4444
# User specifices a string (single platform) as observable
4545
observable = {observable: None}
46-
supported = ["powersensor2", "powersensor3", "nvidia", "likwid", "rapl", "rocm", "xilinx"]
46+
supported = ["arduino", "jetson", "likwid", "nvml", "rapl", "rocm", "xilinx"]
4747
for obs in observable.keys():
4848
if not obs in supported:
4949
raise ValueError(f"Observable {obs} not in supported: {supported}")
5050

51-
print(observable.items())
5251
self.pms = [pmt.get_pmt(obs[0], obs[1]) for obs in observable.items()]
5352
self.pm_names = list(observable.keys())
5453

@@ -71,9 +70,8 @@ def after_finish(self):
7170
for i in range(len(self.pms)):
7271
begin_state = self.begin_states[i]
7372
end_state = end_states[i]
74-
measured_energy = pmt.pypmt.PMT.joules(begin_state, end_state)
75-
measured_power = pmt.pypmt.PMT.watts(begin_state, end_state)
76-
print(measured_energy, measured_power)
73+
measured_energy = pmt.joules(begin_state, end_state)
74+
measured_power = pmt.watts(begin_state, end_state)
7775
pm_name = self.pm_names[i]
7876
energy_result_name = f"{pm_name}_energy"
7977
power_result_name = f"{pm_name}_power"

0 commit comments

Comments
 (0)