code: def print_rank_0(message):
      import deepspeed.comm as dist
  
      if dist.get_rank() == 0:
          print(message)
  
file_path: DeepSpeed/deepspeed/utils/comms_logging.py
imports: import math
  from deepspeed.utils import log_dist
label_desc: Use image node_id 2 for calling a global function with example usage: print_rank_0(message) without return types