From 5ae23419d82dbea66c21db0432ca84ef2f0d8d19 Mon Sep 17 00:00:00 2001 From: J-How Huang Date: Sat, 16 Sep 2023 23:09:07 -0400 Subject: [PATCH] fix power_readings tag ambuguity in nvidia-smi xml --- experiment_impact_tracker/gpu/nvidia.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiment_impact_tracker/gpu/nvidia.py b/experiment_impact_tracker/gpu/nvidia.py index ff8a04d..3dd5292 100644 --- a/experiment_impact_tracker/gpu/nvidia.py +++ b/experiment_impact_tracker/gpu/nvidia.py @@ -189,7 +189,7 @@ def get_nvidia_gpu_power(pid_list, logger=None, **kwargs): gpu_data["utilization"] = {"gpu_util": gpu_util, "memory_util": memory_util} # get power - power_readings = gpu.findall("power_readings")[0] + power_readings = gpu.findall("power_readings")[0] if gpu.findall("power_readings") else gpu.findall("gpu_power_readings")[0] power_draw = power_readings.findall("power_draw")[0].text gpu_data["power_readings"] = {"power_draw": power_draw}