Function _device_exists

python/tvm/testing/env.py:105–111

Return whether ``tvm.device(kind, index)`` is present and usable.

(kind: str, index: int = 0)

@functools.cache
def _device_exists(kind: str, index: int = 0) -> bool:
    """Return whether ``tvm.device(kind, index)`` is present and usable."""
    try:
        return bool(tvm.device(kind, index).exist)
    except Exception:  # pylint: disable=broad-except
        # A missing backend / driver must skip the test, not crash collection.
        return False


@functools.cache