WIP

Author: mathoudebine

library/sensors/sensors_python.py

@@ -275,35 +275,31 @@ class GpuAmd(sensors.Gpu):
     def stats() -> Tuple[float, float, float, float]:  # load (%) / used mem (%) / used mem (Mb) / temp (°C)
         if pyamdgpuinfo:
             # Unlike other sensors, AMD GPU with pyamdgpuinfo pulls in all the stats at once
-            i = 0
-            amd_gpus = []
-            while i < pyamdgpuinfo.detect_gpus():
-                amd_gpus.append(pyamdgpuinfo.get_gpu(i))
-                i = i + 1
+            amd_gpu = pyamdgpuinfo.get_gpu(0)
 
             try:
-                memory_used_all = [item.query_vram_usage() for item in amd_gpus]
+                memory_used_all = amd_gpu.query_vram_usage()
                 memory_used_bytes = sum(memory_used_all) / len(memory_used_all)
                 memory_used = memory_used_bytes / 1000000
             except:
                 memory_used_bytes = math.nan
                 memory_used = math.nan
 
             try:
-                memory_total_all = [item.memory_info["vram_size"] for item in amd_gpus]
+                memory_total_all = amd_gpu.memory_info["vram_size"]
                 memory_total_bytes = sum(memory_total_all) / len(memory_total_all)
                 memory_percentage = (memory_used_bytes / memory_total_bytes) * 100
             except:
                 memory_percentage = math.nan
 
             try:
-                load_all = [item.query_load() for item in amd_gpus]
+                load_all = amd_gpu.query_load()
                 load = (sum(load_all) / len(load_all)) * 100
             except:
                 load = math.nan
 
             try:
-                temperature_all = [item.query_temperature() for item in amd_gpus]
+                temperature_all = amd_gpu.query_temperature()
                 temperature = sum(temperature_all) / len(temperature_all)
             except:
                 temperature = math.nan
@@ -348,8 +344,9 @@ def fan_percent() -> float:
 
     @staticmethod
     def frequency() -> float:
-        # Not supported by Python libraries
-        return math.nan
+        if pyamdgpuinfo:
+            # Unlike other sensors, AMD GPU with pyamdgpuinfo pulls in all the stats at once
+            return pyamdgpuinfo.get_gpu(0)
 
     @staticmethod
     def is_available() -> bool: