code: def set_moe_group(moe_group):
      global _moe_group
      _moe_group = moe_group
  
file_path: thu-pacman-faster-moe/fmoe/megatron/distributed.py
imports: from fmoe.distributed import DistributedGroupedDataParallel
label_desc: Use image node_id 1 for calling a global function with example usage: set_moe_group(moe_group) without return types