code: def mark_module_parallel_comm(module, comm):
      r"""
      Mark all parameters in `module` as doing data parallel in `comm`, where
      `comm` may be one of `'world', 'dp', 'none'`.
      """
      for p in module.parameters():
          setattr(p, "dp_comm", comm)
  
file_path: smart-moe/fmoe/layers.py
imports: import tree
  import os
  import torch
  import torch.nn
  from .functions import prepare_forward, ensure_comm
  from .functions import MOEScatter, MOEGather
  from .functions import AllGather, Slice
  from .gates import NaiveGate
  from .fastermoe.config import switch_from_env
label_desc: Use image node_id 1 for calling a global function with example usage: mark_module_parallel_comm(module, comm) without return types