code: def register_memory_io():
      global MEMORY_IO_REGISTERED
      if not MEMORY_IO_REGISTERED:
          mem_io_path = os.getenv("TDE_MEMORY_IO_PATH")
          if mem_io_path is None:
              raise RuntimeError(
                  "env TDE_MEMORY_IO_PATH must set for unittest"
              )
  
          torch.ops.tde.register_io(mem_io_path)
          MEMORY_IO_REGISTERED = True
  
imports: import os
  import torch
  import torch.distributed
label_desc: Use image node_id 1 for calling a global function with example usage: register_memory_io() without return types