Function _target_enabled

python/tvm/testing/env.py:130–146

True if ``kind`` is selected by ``TVM_TEST_TARGETS`` (or the default set). Honors the ``TVM_TEST_TARGETS`` opt-out, so CI can exclude a flaky backend (e.g. opencl) via ``TVM_TEST_TARGETS`` and have its tests skip even when a device is physically present.

(kind: str)

@functools.cache
def _target_enabled(kind: str) -> bool:
    """True if ``kind`` is selected by ``TVM_TEST_TARGETS`` (or the default set).

    Honors the ``TVM_TEST_TARGETS`` opt-out, so CI can exclude a flaky
    backend (e.g. opencl) via ``TVM_TEST_TARGETS`` and have its tests skip
    even when a device is physically present.
    """
    try:
        from tvm.testing.utils import _tvm_test_targets  # pylint: disable=import-outside-toplevel

        for target in _tvm_test_targets():
            k = target["kind"] if isinstance(target, dict) else str(target).split()[0]
            if k == kind:
                return True
        return False
    except Exception:  # pylint: disable=broad-except
        return True  # fail open: the device check still gates


@functools.cache