code: def suppress_output(rank):
      """Suppress printing on the current device. Force printing with `force=True`."""
      import builtins as __builtin__
  
      builtin_print = __builtin__.print
  
      def print(*args, **kwargs):
          force = kwargs.pop("force", True)
          if force:
              builtin_print(
                  "rank #%d:" % rank, *args, **kwargs, flush=True
              )
          elif rank == 0:
              builtin_print(*args, **kwargs)
  
      __builtin__.print = print
  
imports: import torch
  import torch.distributed
label_desc: Use image node_id 5 for calling a global function with example usage: suppress_output(rank) without return types