Skip to content

Commit 29591ab

Browse files
code quality fixes
1 parent 4ccecdf commit 29591ab

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

kernel_tuner/observers/pmt.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import numpy as np
22

3-
from kernel_tuner.observers.observer import BenchmarkObserver
3+
from kernel_tuner.observers.observer import BenchmarkObserver, ContinuousObserver
44

55
# check if pmt is installed
66
try:
@@ -94,12 +94,14 @@ class PMTContinuousObserver(ContinuousObserver):
9494
a .read_power() method, which the ContinuousObserver can call to read power in Watt
9595
"""
9696
def before_start(self):
97+
""" Override default method in ContinuousObserver """
9798
pass
9899

99100
def after_start(self):
100101
self.parent.after_start()
101102

102103
def during(self):
104+
""" Override default method in ContinuousObserver """
103105
pass
104106

105107
def after_finish(self):
@@ -113,7 +115,7 @@ def get_results(self):
113115

114116
# correct energy measurement, because current _energy number is collected over the entire duration
115117
# we estimate energy as the average power over the continuous duration times the kernel execution time
116-
for pm_name in pm_names:
118+
for pm_name in self.parent.pm_names:
117119
energy_result_name = f"{pm_name}_energy"
118120
power_result_name = f"{pm_name}_power"
119121
averages[energy_result_name] = averages[power_result_name] * (average_kernel_execution_time_ms / 1e3)

0 commit comments

Comments
 (0)