code: def register_colo_module(module_type: type, colo_module: ColoModule):
      global _COLOSSAL_MODULES
      _COLOSSAL_MODULES[module_type] = colo_module
  
file_path: ColossalAI/colossalai/legacy/nn/parallel/layers/module_utils.py
imports: from typing import Dict
  import torch
  from colossalai.legacy.tensor import ComputeSpec, ProcessGroup
  from colossalai.tensor import ColoParameter
  from .None import ColoModule
label_desc: Use image node_id 1 for calling a global function with example usage: register_colo_module(module_type, colo_module) without return types