code: def print_separator(message):
      torch.distributed.barrier()
      filler_len = (78 - len(message)) // 2
      filler = "-" * filler_len
      string = "\n" + filler + " {} ".format(message) + filler
      if torch.distributed.get_rank() == 0:
          print(string, flush=True)
      torch.distributed.barrier()
  
file_path: megatron-lm/megatron/mpu/tests/commons.py
imports: import argparse
  import os
  import random
  import numpy
  import torch
  import mpu
label_desc: Use image node_id 3 for calling a global function with example usage: print_separator(message) without return types