File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 1
1
import numpy as np
2
2
3
- from kernel_tuner .observers .observer import BenchmarkObserver
3
+ from kernel_tuner .observers .observer import BenchmarkObserver , ContinuousObserver
4
4
5
5
# check if pmt is installed
6
6
try :
@@ -94,12 +94,14 @@ class PMTContinuousObserver(ContinuousObserver):
94
94
a .read_power() method, which the ContinuousObserver can call to read power in Watt
95
95
"""
96
96
def before_start (self ):
97
+ """ Override default method in ContinuousObserver """
97
98
pass
98
99
99
100
def after_start (self ):
100
101
self .parent .after_start ()
101
102
102
103
def during (self ):
104
+ """ Override default method in ContinuousObserver """
103
105
pass
104
106
105
107
def after_finish (self ):
@@ -113,7 +115,7 @@ def get_results(self):
113
115
114
116
# correct energy measurement, because current _energy number is collected over the entire duration
115
117
# 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 :
117
119
energy_result_name = f"{ pm_name } _energy"
118
120
power_result_name = f"{ pm_name } _power"
119
121
averages [energy_result_name ] = averages [power_result_name ] * (average_kernel_execution_time_ms / 1e3 )
You can’t perform that action at this time.
0 commit comments