From 6d520dc4b24f2f20378ab7f7cbfd5d1d57fe6510 Mon Sep 17 00:00:00 2001 From: Brett Jia Date: Sun, 28 Apr 2024 09:52:16 -0400 Subject: [PATCH] rknn: add more cpus (#1451) * add additional cpus supported by rknn model converter * use queue-based approach * bump 0.0.3 * Revert "use queue-based approach" This reverts commit 4ec77495e898da1d3086940d398a4b53533e853e. * bump 0.0.4 --- plugins/rknn/package-lock.json | 4 ++-- plugins/rknn/package.json | 2 +- plugins/rknn/src/rknn/plugin.py | 4 ++++ 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/plugins/rknn/package-lock.json b/plugins/rknn/package-lock.json index 88a84a615c..d7ed02b201 100644 --- a/plugins/rknn/package-lock.json +++ b/plugins/rknn/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/rknn", - "version": "0.0.2", + "version": "0.0.4", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/rknn", - "version": "0.0.2", + "version": "0.0.4", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/rknn/package.json b/plugins/rknn/package.json index a42d3ec151..d1f07515ce 100644 --- a/plugins/rknn/package.json +++ b/plugins/rknn/package.json @@ -45,5 +45,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.0.2" + "version": "0.0.4" } diff --git a/plugins/rknn/src/rknn/plugin.py b/plugins/rknn/src/rknn/plugin.py index f949f36843..f6ec2fadbc 100644 --- a/plugins/rknn/src/rknn/plugin.py +++ b/plugins/rknn/src/rknn/plugin.py @@ -37,6 +37,10 @@ def ensure_compatibility_and_get_cpu(): device_compatible_str = f.read() if 'rk3562' in device_compatible_str: return 'rk3562' + elif 'rk3566' in device_compatible_str: + return 'rk3566' + elif 'rk3568' in device_compatible_str: + return 'rk3568' elif 'rk3576' in device_compatible_str: return 'rk3576' elif 'rk3588' in device_compatible_str: