diff --git a/tools/device.py b/tools/device.py index 298b997fe72de4de18d8357f76b2bb7c961ede98..8c78a09f8195ca8f577d9f1ed016f065fc50aa65 100644 --- a/tools/device.py +++ b/tools/device.py @@ -850,7 +850,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: