diff --git a/kernel_tuner/observers/nvml.py b/kernel_tuner/observers/nvml.py index d33327a3c..18945352a 100644 --- a/kernel_tuner/observers/nvml.py +++ b/kernel_tuner/observers/nvml.py @@ -298,7 +298,8 @@ def auto_boost(self, setting): def pwr_usage(self): """Return current power usage in milliwatts.""" - return pynvml.nvmlDeviceGetPowerUsage(self.dev) + NVML_FI_DEV_POWER_INSTANT = 186 + return pynvml.nvmlDeviceGetFieldValues(self.dev, [NVML_FI_DEV_POWER_INSTANT])[0].value.uiVal def gr_voltage(self): """Return current graphics voltage in millivolts."""