Function print_rank_0

deepspeed/utils/comms_logging.py:15–18

(message)

def print_rank_0(message):
    import deepspeed.comm as dist
    if dist.get_rank() == 0:
        print(message)


# Helper function to pretty-print message sizes