code: def print_rank_0(message):
      if dist.get_rank() == 0:
          print(message)
  
imports: import torch
  from deepspeed import comm
label_desc: Use image node_id 1 for calling a global function with example usage: print_rank_0(message) without return types