Skip to content

Commit

Permalink
rknn: add more cpus (#1451)
Browse files Browse the repository at this point in the history
* add additional cpus supported by rknn model converter

* use queue-based approach

* bump 0.0.3

* Revert "use queue-based approach"

This reverts commit 4ec7749.

* bump 0.0.4
  • Loading branch information
bjia56 authored Apr 28, 2024
1 parent 40c7132 commit 6d520dc
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 3 deletions.
4 changes: 2 additions & 2 deletions plugins/rknn/package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion plugins/rknn/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -45,5 +45,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.0.2"
"version": "0.0.4"
}
4 changes: 4 additions & 0 deletions plugins/rknn/src/rknn/plugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 6d520dc

Please sign in to comment.