administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-04 04:27:42 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Support multi and single core rknn npus

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-04 08:38:46 -06:00
parent 00261cb90c
commit 82cdb06a95
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/util/services.py
View File

@ -408,7 +408,13 @@ def get_rockchip_npu_stats() -> dict[str, str]:
try: try:
with open("/sys/kernel/debug/rknpu/load", "r") as f: with open("/sys/kernel/debug/rknpu/load", "r") as f:
npu_output = f.read() npu_output = f.read()
core_loads = re.findall(r"Core\d+:\s*(\d+)%", npu_output)
if "Core0:" in npu_output:
# multi core NPU
core_loads = re.findall(r"Core\d+:\s*(\d+)%", npu_output)
else:
# single core NPU
core_loads = re.findall(r"NPU load:\s+(\d+)%", npu_output)
except FileNotFoundError: except FileNotFoundError:
core_loads = None core_loads = None