code: def test_reshape_222_to_211():
      src_3d = model_3d_desc(pp_degree=2, tp_degree=2, dp_degree=2)
      tgt_3d = model_3d_desc(pp_degree=2, tp_degree=1, dp_degree=1)
  
      new_3d_map = _do_reshape(src_3d, tgt_3d)
  
      assert new_3d_map[0].get_data(pp_index=0, tp_index=0) == [
          0,
          4,
          1,
          5,
      ]
      assert new_3d_map[0].get_data(pp_index=1, tp_index=0) == [
          2,
          6,
          3,
          7,
      ]
  
imports: from deepspeed.checkpoint import model_3d_desc
label_desc: Use image node_id 5 for calling a global function with example usage: test_reshape_222_to_211() without return types