code: def set_moe_tensor_info(
      tensor: torch.Tensor, moe_info: MoeParallelInfo
  ) -> None:
      """
      Set moe info for the given tensor.
  
      Args:
          tensor (torch.Tensor): The tensor to be set.
          moe_info (dict): The moe info to be set.
  
      """
      tensor.__setattr__("moe_info", moe_info)
  
file_path: ColossalAI/colossalai/tensor/moe_tensor/api.py
imports: from typing import List
  import torch
  import torch.distributed
  from torch.distributed import ProcessGroup
  from .moe_info import MoeParallelInfo
label_desc: Use image node_id 2 for calling a global function with example usage: set_moe_tensor_info(tensor, moe_info) without return types