Function rank_log

distributed/tensor_parallelism/log_utils.py:12–15

helper function to log only on global rank 0

(_rank, logger, msg)

def rank_log(_rank, logger, msg):
    """helper function to log only on global rank 0"""
    if _rank == 0:
        logger.info(f" {msg}")


def verify_min_gpu_count(min_gpus: int = 2) -> bool: