Function has_cudagraph

python/tvm/testing/env.py:390–403

True if a CUDA device is present and the toolkit supports CUDA Graphs. Implies :func:`has_cuda`: ``nvcc.have_cudagraph()`` only checks the toolkit version, so the device guard must be explicit. Gate a test with ``@pytest.mark.skipif(not tvm.testing.env.has_cudagraph(), reason=...)``

()

@functools.cache
def has_cudagraph() -> bool:
    """True if a CUDA device is present and the toolkit supports CUDA Graphs.

    Implies :func:`has_cuda`: ``nvcc.have_cudagraph()`` only checks the
    toolkit version, so the device guard must be explicit.  Gate a test with
    ``@pytest.mark.skipif(not tvm.testing.env.has_cudagraph(), reason=...)``
    (add ``@pytest.mark.gpu`` for CI selection).
    """
    try:
        from tvm.support import nvcc  # pylint: disable=import-outside-toplevel

        return has_cuda() and bool(nvcc.have_cudagraph())
    except Exception:  # pylint: disable=broad-except
        return False


# --- toolchain / environment probes ----------------------------------------