1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-02-03 01:22:44 +01:00

dont stuck on core_0 on rk3588 but untested

This commit is contained in:
yoni13 2025-01-25 19:14:34 +08:00
parent 3ed5d3dbac
commit 6e08f1f371

View file

@ -32,7 +32,7 @@ except (FileNotFoundError, ImportError):
soc_name = None
def init_rknn(rknnModel, id) -> Callable:
def init_rknn(rknnModel) -> Callable:
if not is_available:
raise RuntimeError("rknn is not available!")
rknn_lite = RKNNLite()
@ -41,16 +41,7 @@ def init_rknn(rknnModel, id) -> Callable:
raise RuntimeError("Load RKNN rknnModel failed")
if soc_name in coremask_supported_socs:
if id == 0:
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
elif id == 1:
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_1)
elif id == 2:
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_2)
elif id == -1:
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0_1_2)
else:
ret = rknn_lite.init_runtime()
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_AUTO)
else:
ret = rknn_lite.init_runtime() # Please do not set this parameter on other platforms.
@ -63,7 +54,7 @@ def init_rknn(rknnModel, id) -> Callable:
def init_rknns(rknnModel, tpes) -> list[Callable]:
rknn_list = []
for i in range(tpes):
rknn_list.append(init_rknn(rknnModel, i % 3))
rknn_list.append(init_rknn(rknnModel))
return rknn_list