code: def add_data(self, pp_index, tp_index, data):
      self._validate_indices(pp_index, tp_index)
      assert type(data) is list
  
      key = self._make_key(pp_index, tp_index)
      if not key in self.map.keys():
          self.map[key] = []
      self.map[key] += data
  
file_path: DeepSpeed/deepspeed/checkpoint/reshape_meg_2d.py
imports: from .reshape_utils import partition_data
label_desc: Use image node_id 3 for calling the meg_2d_parallel_map obj's underlying member method code with example usage: obj.add_data(pp_index, tp_index, data) without return types