code: def is_colo_module(module: torch.nn.Module):
      global _COLOSSAL_MODULES
      for module_type in _COLOSSAL_MODULES.keys():
          if isinstance(module, module_type):
              return True
      return False
  
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 2 for calling a global function with example usage: is_colo_module(module) and returns: False, True