Add GPU sensors

backend/systembridgebackend/modules/gpu/__init__.py

@@ -31,6 +31,17 @@ def core_load(
                 return value
         return None
 
+    def fan_speed(
+        self,
+        database: Database,
+    ) -> float | None:
+        """GPU fan speed"""
+        for key, value in database.table_data_to_ordered_dict("sensors").items():
+            if "gpu" in key and "fan" in key:
+                self._logger.debug("Found GPU fan speed: %s = %s", key, value)
+                return value
+        return None
+
     def memory_clock(
         self,
         database: Database,
@@ -53,6 +64,39 @@ def memory_load(
                 return value
         return None
 
+    def memory_free(
+        self,
+        database: Database,
+    ) -> float | None:
+        """GPU memory free"""
+        for key, value in database.table_data_to_ordered_dict("sensors").items():
+            if "gpu" in key and "memory" in key and "free" in key:
+                self._logger.debug("Found GPU memory free: %s = %s", key, value)
+                return value
+        return None
+
+    def memory_used(
+        self,
+        database: Database,
+    ) -> float | None:
+        """GPU memory used"""
+        for key, value in database.table_data_to_ordered_dict("sensors").items():
+            if "gpu" in key and "memory" in key and "used" in key:
+                self._logger.debug("Found GPU memory used: %s = %s", key, value)
+                return value
+        return None
+
+    def memory_total(
+        self,
+        database: Database,
+    ) -> float | None:
+        """GPU memory total"""
+        for key, value in database.table_data_to_ordered_dict("sensors").items():
+            if "gpu" in key and "memory" in key and "total" in key:
+                self._logger.debug("Found GPU memory total: %s = %s", key, value)
+                return value
+        return None
+
     def power(
         self,
         database: Database,

backend/systembridgebackend/modules/gpu/update.py

@@ -24,6 +24,10 @@ async def update_core_load(self) -> None:
         """Update core load"""
         self._database.write("gpu", "core_load", self._gpu.core_load(self._database))
 
+    async def update_fan_speed(self) -> None:
+        """Update fan speed"""
+        self._database.write("gpu", "fan_speed", self._gpu.fan_speed(self._database))
+
     async def update_memory_clock(self) -> None:
         """Update memory clock"""
         self._database.write(
@@ -36,6 +40,24 @@ async def update_memory_load(self) -> None:
             "gpu", "memory_load", self._gpu.memory_load(self._database)
         )
 
+    async def update_memory_free(self) -> None:
+        """Update memory free"""
+        self._database.write(
+            "gpu", "memory_free", self._gpu.memory_free(self._database)
+        )
+
+    async def update_memory_used(self) -> None:
+        """Update memory used"""
+        self._database.write(
+            "gpu", "memory_used", self._gpu.memory_used(self._database)
+        )
+
+    async def update_memory_total(self) -> None:
+        """Update memory total"""
+        self._database.write(
+            "gpu", "memory_total", self._gpu.memory_total(self._database)
+        )
+
     async def update_power(self) -> None:
         """Update power"""
         self._database.write("gpu", "power", self._gpu.power(self._database))
@@ -50,7 +72,11 @@ async def update_all_data(self) -> None:
         """Update data"""
         await self.update_core_clock()
         await self.update_core_load()
+        await self.update_fan_speed()
         await self.update_memory_clock()
         await self.update_memory_load()
+        await self.update_memory_free()
+        await self.update_memory_used()
+        await self.update_memory_total()
         await self.update_power()
         await self.update_temperature()