From b53b2ba8dd250be72f3b464a712552348e6f1534 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Wed, 22 Dec 2021 10:37:15 +0900 Subject: [PATCH] skip testing if target is not enabled --- tests/python/frontend/pytorch/test_forward.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index dca1f3f25947e..86970bfb6d4d9 100644 --- a/tests/python/frontend/pytorch/test_forward.py +++ b/tests/python/frontend/pytorch/test_forward.py @@ -213,6 +213,8 @@ def verify_model( with tvm.transform.PassContext(opt_level=3): for target in ["llvm", "cuda"]: + if not tvm.runtime.enabled(target): + continue dev = tvm.device(target, 0) relay_graph, relay_lib, relay_params = relay.build(mod, target=target, params=params) relay_model = graph_executor.create(relay_graph, relay_lib, dev) @@ -2270,7 +2272,10 @@ def verify_model_vm(input_model, ishapes, idtype=None, idata=None, targets=["llv mod, params = relay.frontend.from_pytorch(input_model, input_shapes) for tgt in targets: + if not tvm.runtime.enabled(tgt): + continue print("Running on target", tgt) + dev = tvm.device(tgt, 0) evaluator = relay.create_executor("vm", mod=mod, device=dev, target=tgt).evaluate()