code: def get_mem_space(x):
      try:
          ret = dtype_memory_size_dict[x]
      except KeyError:
          print(f"dtype {x} is not supported!")
      return ret
  
file_path: h2ogpt/models/gpu_mem_track.py
imports: import gc
  import datetime
  import inspect
  import torch
  import numpy
  import contextlib, sys
label_desc: Use image node_id 1 for calling a global function with example usage: get_mem_space(x) and returns: ret