class_docstr: Device mesh manager is responsible for creating and managing device meshes.
docstr: Create a device mesh and store it in the manager.
  
  Args:
      name (str): name of the device mesh
      device_mesh_info (DeviceMeshInfo): the information used to initialize the device mesh
docstr_tok: ["Create","a","device","mesh","and","store","it","in","the","manager",".","Args",":","name","(","str",")",":","name","of","the","device","mesh","device_mesh_info","(","DeviceMeshInfo",")",":","the","information","used","to","initialize","the","device","mesh"]
code: def create_device_mesh(
      self, name, device_mesh_info: DeviceMeshInfo
  ) -> DeviceMesh:
      """
      Create a device mesh and store it in the manager.
  
      Args:
          name (str): name of the device mesh
          device_mesh_info (DeviceMeshInfo): the information used to initialize the device mesh
      """
      if name not in self.device_mesh_store:
          device_mesh = initialize_device_mesh(device_mesh_info)
          self.device_mesh_store[name] = device_mesh
          return device_mesh
      else:
          raise ValueError(f"Device mesh {name} already exists.")
  
file_path: ColossalAI/colossalai/cluster/device_mesh_manager.py
imports: from dataclasses import dataclass
  from typing import Dict, List, Tuple, Union
  import torch
  import torch.distributed
  from colossalai.device.alpha_beta_profiler import AlphaBetaProfiler
  from colossalai.device.device_mesh import DeviceMesh
label_desc: Use image node_id 2 for calling the DeviceMeshManager obj's underlying member method code with example usage: obj.create_device_mesh(name, device_mesh_info) and returns: device_mesh