未验证 提交 2944de86 编写于 作者: L Liu Qi 提交者: GitHub

Merge pull request #288 from newway1995/bugfix_benchmark_gp_testing

bugfix: cannot run gpu when benchmark model with runtime cpu_gpu.
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册