code: def run_dist(rank, world_size, port):
      colossalai.launch(
          config=dict(
              parallel=dict(
                  data=1, pipeline=2, tensor=dict(mode="1d", size=2)
              )
          ),
          rank=rank,
          world_size=world_size,
          port=port,
          host="localhost",
      )
      # TODO(ver217): this function should be removed when gpc is removed
      # check_process_group_mesh_with_gpc()
      check_process_group_mesh_with_cases()
  
file_path: ColossalAI/tests/test_cluster/test_process_group_mesh.py
imports: import pytest
  import torch.distributed
  import colossalai
  from colossalai.cluster import ProcessGroupMesh
  from colossalai.testing import spawn
label_desc: Use image node_id 3 for calling a global function with example usage: run_dist(rank, world_size, port) without return types