Skip to content

Commit a8d6a3a

Browse files
fix typos
1 parent f090f8d commit a8d6a3a

File tree

3 files changed

+10
-8
lines changed

3 files changed

+10
-8
lines changed

kernel_tuner/observers/nvml.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,7 @@ def __init__(
352352
save_all=False,
353353
nvidia_smi_fallback=None,
354354
use_locked_clocks=False,
355-
continous_duration=1,
355+
continuous_duration=1,
356356
):
357357
"""Create an NVMLObserver."""
358358
if nvidia_smi_fallback:
@@ -384,7 +384,7 @@ def __init__(
384384
if any([obs in self.needs_power for obs in observables]):
385385
self.measure_power = True
386386
power_observables = [obs for obs in observables if obs in self.needs_power]
387-
self.continuous_observer = ContinuousObserver("nvml", self, power_observables, continous_duration=continuous_duration)
387+
self.continuous_observer = ContinuousObserver("nvml", power_observables, self, continuous_duration=continuous_duration)
388388

389389
# remove power observables
390390
self.observables = [obs for obs in observables if obs not in self.needs_power]
@@ -407,7 +407,8 @@ def __init__(
407407
self.iteration = {obs: [] for obs in self.during_obs}
408408

409409
def read_power(self):
410-
return self.nvml.pwr_usage()
410+
""" Return power in Watt """
411+
return self.nvml.pwr_usage() / 1e3
411412

412413
def before_start(self):
413414
# clear results of the observables for next measurement

kernel_tuner/observers/observer.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from abc import ABC, abstractmethod
2-
2+
import time
3+
import numpy as np
34

45
class BenchmarkObserver(ABC):
56
"""Base class for Benchmark Observers"""
@@ -47,9 +48,9 @@ class ContinuousObserver(BenchmarkObserver):
4748
"""Generic observer that measures power while and continuous benchmarking.
4849
4950
To support continuous benchmarking an Observer should support:
50-
a .read_power() method, which the ContinuousObserver can call to read power
51+
a .read_power() method, which the ContinuousObserver can call to read power in Watt
5152
"""
52-
def __init__(self, name, observables, parent, continous_duration=1):
53+
def __init__(self, name, observables, parent, continuous_duration=1):
5354
self.parent = parent
5455
self.name = name
5556

@@ -60,7 +61,7 @@ def __init__(self, name, observables, parent, continous_duration=1):
6061
self.observables = observables
6162

6263
# duration in seconds
63-
self.continuous_duration = continous_duration
64+
self.continuous_duration = continuous_duration
6465

6566
self.power = 0
6667
self.energy = 0

kernel_tuner/observers/tegra.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ def __init__(
224224
if any([obs in self.needs_power for obs in observables]):
225225
self.measure_power = True
226226
power_observables = [obs for obs in observables if obs in self.needs_power]
227-
self.continuous_observer = ContinuousObserver("tegra", power_observables, self, continous_duration=3)
227+
self.continuous_observer = ContinuousObserver("tegra", power_observables, self, continuous_duration=3)
228228

229229
# remove power observables
230230
self.observables = [obs for obs in observables if obs not in self.needs_power]

0 commit comments

Comments
 (0)