code: def _reshape_pp_dimension(old_2d_map, new_pp_degree):
      old_tp_degree = old_2d_map.tp_degree
      new_2d_map = meg_2d_parallel_map(new_pp_degree, old_tp_degree)
      for i in range(old_tp_degree):
          ranks_for_tp_index = old_2d_map.get_data(
              pp_index=None, tp_index=i
          )
          split_ranks = partition_data(
              ranks_for_tp_index, new_pp_degree
          )
          for j in range(new_pp_degree):
              new_2d_map.add_data(j, i, split_ranks[j])
  
      return new_2d_map
  
file_path: DeepSpeed/deepspeed/checkpoint/reshape_meg_2d.py
imports: from .reshape_utils import partition_data
label_desc: Use image node_id 2 for calling a global function with example usage: _reshape_pp_dimension(old_2d_map, new_pp_degree) and returns: new_2d_map