mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-08-03 01:22:17 +03:00
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:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user