From 82cdb06a952812491062ed0eede2057c16b52bf1 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 4 May 2025 08:38:46 -0600 Subject: [PATCH] Support multi and single core rknn npus --- frigate/util/services.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/frigate/util/services.py b/frigate/util/services.py index 7210837ff..1c778bac4 100644 --- a/frigate/util/services.py +++ b/frigate/util/services.py @@ -408,7 +408,13 @@ def get_rockchip_npu_stats() -> dict[str, str]: try: with open("/sys/kernel/debug/rknpu/load", "r") as f: 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: core_loads = None