code: def get_colo_module(module: torch.nn.Module):
      global _COLOSSAL_MODULES
      if is_colo_module(module):
          for module_type, colo_module in _COLOSSAL_MODULES.items():
              if isinstance(module, module_type):
                  return colo_module
      else:
          return None
  
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 3 for calling a global function with example usage: get_colo_module(module) and returns: None, colo_module