code: def max_non_model_data(self, device_type: str) -> float:
      if device_type == "cuda":
          return max(self._non_model_data_cuda_list)
      elif device_type == "cpu":
          return max(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 10 for calling the MemStats obj's underlying member method code with example usage: obj.max_non_model_data(device_type) and returns: max, max