code: def print_rank_0(message, debug=False, force=False):
      if dist.get_rank() == 0 and (debug or force):
          print(message)
  
imports: import os
  import shutil
  from enum import Enum
  import torch
  from deepspeed import comm
  from deepspeed.ops.aio import AsyncIOBuilder
  from .constants import *
  from .utils import swap_in_tensors, swap_out_tensors, MIN_AIO_BYTES, AIO_ALIGNED_BYTES, print_object, SwapBufferPool
label_desc: Use image node_id 1 for calling a global function with example usage: print_rank_0(message, debug, force) without return types