Add support for rockchip GPU stats (#17812)

* Set rockchip GPU name

* Add support for calculating rockchip GPU load

* Format consistently

* Fix access error
This commit is contained in:
Nicolas Mowen
2025-04-19 16:34:05 -05:00
committed by GitHub
parent 80a13e43e9
commit 38ffe21b69
4 changed files with 35 additions and 2 deletions
+22 -1
View File
@@ -382,8 +382,29 @@ def get_intel_gpu_stats(sriov: bool) -> dict[str, str]:
return results
def get_rockchip_gpu_stats() -> dict[str, str]:
"""Get GPU stats using rk."""
try:
with open("/sys/kernel/debug/rkrga/load", "r") as f:
content = f.read()
except FileNotFoundError:
return None
load_values = []
for line in content.splitlines():
match = re.search(r"load = (\d+)%", line)
if match:
load_values.append(int(match.group(1)))
if not load_values:
return None
average_load = f"{round(sum(load_values) / len(load_values), 2)}%"
return {"gpu": average_load, "mem": "-"}
def get_rockchip_npu_stats() -> dict[str, str]:
"""Get stats using rk."""
"""Get NPU stats using rk."""
try:
with open("/sys/kernel/debug/rknpu/load", "r") as f:
npu_output = f.read()