code: def print_rank_0(message):
      if torch.distributed.is_initialized():
          if torch.distributed.get_rank() == 0:
              print(message, flush=True)
      else:
          print(message, flush=True)
  
file_path: deepspeedexamples/training/bing_bert/timer.py
imports: import time
  import logging
  import psutil
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: print_rank_0(message) without return types