Function has_cuda_compute

python/tvm/testing/env.py:316–329

True if the CUDA compute capability satisfies ``(major, minor)``. When ``exact`` is False (default) the check is ``compute >= (major, minor)``; when True it requires an exact match. Returns False when no CUDA device is present, so it implies :func:`has_cuda`.

(major: int, minor: int = 0, exact: bool = False)

def has_cuda_compute(major: int, minor: int = 0, exact: bool = False) -> bool:
    """True if the CUDA compute capability satisfies ``(major, minor)``.

    When ``exact`` is False (default) the check is ``compute >= (major,
    minor)``; when True it requires an exact match.  Returns False when no
    CUDA device is present, so it implies :func:`has_cuda`.
    """
    if not has_cuda():
        return False
    compute = _cuda_compute_version()
    want = (major, minor)
    if exact:
        return compute == want
    return compute >= want


@functools.cache