@@ -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