Function has_multi_gpu

python/tvm/testing/env.py:226–231

True if at least ``count`` devices of a single GPU backend exist.

(count: int = 2)

@functools.cache
def has_multi_gpu(count: int = 2) -> bool:
    """True if at least ``count`` devices of a single GPU backend exist."""
    for kind in ("cuda", "rocm", "opencl", "metal", "vulkan"):
        if all(_device_exists(kind, index) for index in range(count)):
            return True
    return False


# --- build-support probes --------------------------------------------------