code: def mlp_mp(self):
      gpu_index = dist.get_rank()
      for ep_index in range(self.local_ep_size):
          # mlp inter
          self.module.mlp[ep_index].inter_w.data = self._h4h_w[
              gpu_index * self.local_ep_size + ep_index
          ].to(get_accelerator().current_device_name())
          self.module.mlp[ep_index].inter_b.data = self._h4h_b[
              gpu_index * self.local_ep_size + ep_index
          ].to(get_accelerator().current_device_name())
  
          # mlp output
          self.module.mlp[ep_index].output_w.data = self._4hh_w[
              gpu_index * self.local_ep_size + ep_index
          ].to(get_accelerator().current_device_name())
          self.module.mlp[ep_index].output_b.data = self._4hh_b[
              gpu_index * self.local_ep_size + ep_index
          ].to(get_accelerator().current_device_name())
  
file_path: deepspeed/deepspeed/module_inject/containers/base_moe.py
imports: from .base import *
  from deepspeed import comm
  import deepspeed.ops.transformer
  from deepspeed.accelerator import get_accelerator
label_desc: Use image node_id 9 for calling the BaseTransformerMoEContainer obj's underlying member method code with example usage: obj.mlp_mp() without return types