Skip to content

Commit c531162

Browse files
committed
Remove code smells in comments.
1 parent 4cf1a68 commit c531162

File tree

1 file changed

+19
-10
lines changed

1 file changed

+19
-10
lines changed

kernel_tuner/observers/nvml.py

Lines changed: 19 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -27,15 +27,17 @@ def __init__(self, device_id=0, nvidia_smi_fallback='nvidia-smi', use_locked_clo
2727
self.pwr_constraints = pynvml.nvmlDeviceGetPowerManagementLimitConstraints(self.dev)
2828
except pynvml.NVMLError_NotSupported:
2929
self.pwr_limit_default = None
30-
self.pwr_constraints = [1, 0] # inverted range to make all range checks fail
30+
# inverted range to make all range checks fail
31+
self.pwr_constraints = [1, 0]
3132

3233
try:
3334
self._persistence_mode = pynvml.nvmlDeviceGetPersistenceMode(self.dev)
3435
except pynvml.NVMLError_NotSupported:
3536
self._persistence_mode = None
3637

3738
try:
38-
self._auto_boost = pynvml.nvmlDeviceGetAutoBoostedClocksEnabled(self.dev)[0] # returns [isEnabled, isDefaultEnabled]
39+
# returns [isEnabled, isDefaultEnabled]
40+
self._auto_boost = pynvml.nvmlDeviceGetAutoBoostedClocksEnabled(self.dev)[0]
3941
except pynvml.NVMLError_NotSupported:
4042
self._auto_boost = None
4143

@@ -100,7 +102,8 @@ def pwr_limit(self, new_limit):
100102
pynvml.nvmlDeviceSetPowerManagementLimit(self.dev, new_limit)
101103
except pynvml.NVMLError_NoPermission:
102104
if self.nvidia_smi:
103-
new_limit_watt = int(new_limit / 1000.0) # nvidia-smi expects Watts rather than milliwatts
105+
# nvidia-smi expects Watts rather than milliwatts
106+
new_limit_watt = int(new_limit / 1000.0)
104107
args = ["sudo", self.nvidia_smi, "-i", str(self.id), "--power-limit="+str(new_limit_watt)]
105108
subprocess.run(args, check=True)
106109

@@ -307,7 +310,8 @@ def before_start(self):
307310

308311
def after_start(self):
309312
self.t0 = time.perf_counter()
310-
self.during() # ensure during is called at least once
313+
# ensure during is called at least once
314+
self.during()
311315

312316
def during(self):
313317
if "temperature" in self.observables:
@@ -332,7 +336,8 @@ def after_finish(self):
332336
gr_voltage_readings = self.gr_voltage_readings
333337
gr_voltage_readings = [[0.0, gr_voltage_readings[0][1]]] + gr_voltage_readings
334338
gr_voltage_readings = gr_voltage_readings + [[execution_time, gr_voltage_readings[-1][1]]]
335-
self.results["gr_voltages"].append(np.average(gr_voltage_readings[:][:][1])) #time in s, graphics voltage in millivolts
339+
# time in s, graphics voltage in millivolts
340+
self.results["gr_voltages"].append(np.average(gr_voltage_readings[:][:][1]))
336341

337342
def get_results(self):
338343
averaged_results = {}
@@ -365,14 +370,16 @@ def __init__(self, observables, parent, nvml_instance, continous_duration=1):
365370
raise ValueError(f"Observable {obs} not in supported: {supported}")
366371
self.observables = observables
367372

368-
self.continuous_duration = continous_duration # seconds
373+
# duration in seconds
374+
self.continuous_duration = continous_duration
369375

370376
self.power = 0
371377
self.energy = 0
372378
self.power_readings = []
373379
self.t0 = 0
374380

375-
self.results = None # results from the last iteration-based benchmark
381+
# results from the last iteration-based benchmark
382+
self.results = None
376383

377384
def before_start(self):
378385
self.parent.before_start()
@@ -398,7 +405,8 @@ def after_finish(self):
398405
if not self.power_readings:
399406
return
400407

401-
execution_time = (self.results["time"]/1e3) # converted to seconds from milliseconds
408+
# convert to seconds from milliseconds
409+
execution_time = (self.results["time"]/1e3)
402410
self.power = np.median([d[1] / 1e3 for d in self.power_readings])
403411
self.energy = self.power * execution_time
404412

@@ -427,8 +435,9 @@ def get_nvml_pwr_limits(device, n=None, quiet=False):
427435
power_limits = d.pwr_constraints
428436
power_limit_min = power_limits[0]
429437
power_limit_max = power_limits[-1]
430-
power_limit_min *= 1e-3 # Convert to Watt
431-
power_limit_max *= 1e-3 # Convert to Watt
438+
# Min and Max converted to Watt
439+
power_limit_min *= 1e-3
440+
power_limit_max *= 1e-3
432441
power_limit_round = 5
433442
tune_params = OrderedDict()
434443
if n == None:

0 commit comments

Comments
 (0)