未验证 提交 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: ...@@ -850,7 +850,7 @@ class DeviceWrapper:
if target_abi == ABIType.host: if target_abi == ABIType.host:
runtime_list.append(RuntimeType.cpu) runtime_list.append(RuntimeType.cpu)
elif model_runtime == RuntimeType.cpu_gpu: elif model_runtime == RuntimeType.cpu_gpu:
runtime_list.extend([RuntimeType.cpu, RuntimeType.cpu_gpu]) runtime_list.extend([RuntimeType.cpu, RuntimeType.gpu])
else: else:
runtime_list.append(model_runtime) runtime_list.append(model_runtime)
for runtime in runtime_list: for runtime in runtime_list:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册