Function has_tensorcore

python/tvm/testing/env.py:368–375

True if a CUDA device with Tensor Core support (compute >= 7) exists.

()

@functools.cache
def has_tensorcore() -> bool:
    """True if a CUDA device with Tensor Core support (compute >= 7) exists."""
    try:
        from tvm.support import nvcc  # pylint: disable=import-outside-toplevel

        return has_cuda() and bool(nvcc.have_tensorcore(tvm.cuda().compute_version))
    except Exception:  # pylint: disable=broad-except
        return False


@functools.cache