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: