Function has_gpu

python/tvm/testing/env.py:214–222

True if any GPU backend (cuda/rocm/opencl/metal/vulkan) is present.

()

def has_gpu() -> bool:
    """True if any GPU backend (cuda/rocm/opencl/metal/vulkan) is present."""
    return (
        _device_exists("cuda")
        or _device_exists("rocm")
        or _device_exists("opencl")
        or _device_exists("metal")
        or _device_exists("vulkan")
    )


@functools.cache