Various Fixes (#18035)

* Support multi and single core rknn npus * Update docs config to be more clear
2026-08-02 09:02:15 +03:00 · 2025-05-04 09:33:27 -06:00
parent da2636d6f7
commit 895afcdb0e
2 changed files with 10 additions and 4 deletions
@@ -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