From d449808115ca61e8f0332af0bbae4bc2a984749c Mon Sep 17 00:00:00 2001 From: Zhang Hao Date: Wed, 4 Aug 2021 10:09:49 +0800 Subject: [PATCH] remove multiple targets test for sim --- vta/tutorials/frontend/deploy_classification.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vta/tutorials/frontend/deploy_classification.py b/vta/tutorials/frontend/deploy_classification.py index b2f909b9710aa..e724448417f39 100644 --- a/vta/tutorials/frontend/deploy_classification.py +++ b/vta/tutorials/frontend/deploy_classification.py @@ -191,7 +191,7 @@ env.WGT_WIDTH, start_name=pack_dict[model][0], stop_name=pack_dict[model][1], - device_annot=(env.TARGET == "intelfocl" or env.TARGET == "sim"), + device_annot=(env.TARGET == "intelfocl"), ) else: relay_prog = mod["main"] @@ -203,7 +203,7 @@ relay_prog, target=target, params=params, target_host=env.target_host ) else: - if env.TARGET == "intelfocl" or env.TARGET == "sim": + if env.TARGET == "intelfocl": # multiple targets to run both on cpu and vta target = {"cpu": env.target_vta_cpu, "ext_dev": target} with vta.build_config(opt_level=3, disabled_pass={"AlterOpLayout"}): @@ -221,7 +221,7 @@ remote.upload(temp.relpath("graphlib.tar")) lib = remote.load_module("graphlib.tar") - if env.TARGET == "intelfocl" or env.TARGET == "sim": + if env.TARGET == "intelfocl": ctxes = [remote.ext_dev(0), remote.cpu(0)] m = graph_runtime.create(graph, lib, ctxes) else: