diff --git a/tools/device.py b/tools/device.py index f0a91a56f3a4be9110c46b7e75efb2406c9f8a21..7a84b59949a75f7e95564342bc003cb19ae9d280 100644 --- a/tools/device.py +++ b/tools/device.py @@ -842,7 +842,7 @@ class DeviceWrapper: if target_abi == ABIType.host: runtime_list.append(RuntimeType.cpu) elif model_runtime == RuntimeType.cpu_gpu: - runtime_list.extend([RuntimeType.cpu, RuntimeType.cpu_gpu]) + runtime_list.extend([RuntimeType.cpu, RuntimeType.gpu]) else: runtime_list.append(model_runtime) for runtime in runtime_list: