From e521438f37a7f9bdfb48ff7ed2eaab12583f9760 Mon Sep 17 00:00:00 2001 From: Niuwei Date: Tue, 25 Dec 2018 00:11:31 -0500 Subject: [PATCH] =?UTF-8?q?bugfix:=20=E4=BF=AE=E5=A4=8D=E4=BD=BF=E7=94=A8b?= =?UTF-8?q?enchmark=E5=91=BD=E4=BB=A4=E6=98=AF=EF=BC=8C=E4=BC=A0=E5=85=A5?= =?UTF-8?q?=E7=9A=84device=E6=98=AFcpu+gpu=EF=BC=8C=E8=80=8Cmace=E7=9A=84?= =?UTF-8?q?=E6=9F=90=E4=BA=9B=E6=96=B9=E6=B3=95=E5=8F=AA=E8=83=BD=E6=A3=80?= =?UTF-8?q?=E6=B5=8Bcpu=E6=88=96=E8=80=85gpu=E8=BF=99=E7=A7=8D=E5=BD=A2?= =?UTF-8?q?=E5=BC=8F(=E8=80=8C=E4=B8=8D=E6=98=AFcpu+gpu)=EF=BC=8C=E5=9B=A0?= =?UTF-8?q?=E6=AD=A4=E9=9C=80=E8=A6=81=E5=B0=86cpu=E5=92=8Cgpu=E5=88=86?= =?UTF-8?q?=E5=BC=80=E5=8A=A0=E5=85=A5=E5=88=B0runtime=5Flist=E4=B8=AD?= =?UTF-8?q?=EF=BC=8C=E5=90=A6=E5=88=99=E6=97=A0=E6=B3=95=E6=89=A7=E8=A1=8C?= =?UTF-8?q?GPU?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/device.py b/tools/device.py index f0a91a56..7a84b599 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: -- GitLab