code: def non_model_data_list(self, device_type: str) -> List[int]:
      if device_type == "cuda":
          return self._non_model_data_cuda_list
      elif device_type == "cpu":
          return self._non_model_data_cpu_list
      else:
          raise TypeError
  
file_path: ColossalAI/colossalai/zero/gemini/memory_tracer/memory_stats.py
imports: from typing import List, Optional
  import torch
  from .param_runtime_order import OrderedParamGenerator
label_desc: Use image node_id 9 for calling the MemStats obj's underlying member method code with example usage: obj.non_model_data_list(device_type) and returns: self, self