Function verify_min_gpu_count

distributed/tensor_parallelism/log_utils.py:18–22

verification that we have at least 2 gpus to run dist examples

(min_gpus: int = 2)

def verify_min_gpu_count(min_gpus: int = 2) -> bool:
    """ verification that we have at least 2 gpus to run dist examples """
    has_gpu = torch.accelerator.is_available()
    gpu_count = torch.accelerator.device_count()
    return has_gpu and gpu_count >= min_gpus