Lunderberg commented on a change in pull request #8010:
URL: https://github.com/apache/tvm/pull/8010#discussion_r639963942
##########
File path: python/tvm/testing.py
##########
@@ -366,24 +368,37 @@ def _check_forward(constraints1, constraints2, varmap,
backvarmap):
)
-def _get_targets():
- target_str = os.environ.get("TVM_TEST_TARGETS", "")
+def _get_targets(target_str=None):
+ if target_str is None:
+ target_str = os.environ.get("TVM_TEST_TARGETS", "")
+
if len(target_str) == 0:
target_str = DEFAULT_TEST_TARGETS
- targets = set()
- for dev in target_str.split(";"):
- if len(dev) == 0:
- continue
- target_kind = dev.split()[0]
- if tvm.runtime.enabled(target_kind) and tvm.device(target_kind,
0).exist:
- targets.add(dev)
- if len(targets) == 0:
+
+ target_names = set(t.strip() for t in target_str.split(";") if t.strip())
+
+ targets = []
+ for target in target_names:
+ target_kind = target.split()[0]
+ is_enabled = tvm.runtime.enabled(target_kind)
+ is_runnable = is_enabled and tvm.device(target_kind).exist
+ targets.append(
+ {
+ "target": target,
+ "target_kind": target_kind,
+ "is_enabled": is_enabled,
+ "is_runnable": is_runnable,
+ }
+ )
+
+ if all(not t["is_runnable"] for t in targets):
logging.warning(
"None of the following targets are supported by this build of TVM:
%s."
" Try setting TVM_TEST_TARGETS to a supported target. Defaulting
to llvm.",
target_str,
)
- return {"llvm"}
+ return _get_targets("llvm")
Review comment:
Good catch, it would. Updating to check `tvm.runtime.enabled('llvm')`.
If enabled, maintain current behavior. Otherwise, raise an exception.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
