code: def get_gpu_mem():
      return torch.cuda.memory_allocated() / 1024**2
  
file_path: colossalai/examples/community/roberta/pretraining/utils/exp_util.py
imports: import functools
  import os
  import shutil
  import psutil
  import torch
  from colossalai.legacy.core import global_context
label_desc: Use image node_id 5 for calling a global function with example usage: get_gpu_mem() and returns: unknown