code: def allocate_mem_buff(name, numel, dtype, track_usage):
      """Allocate a memory buffer."""
      assert (
          name not in _MEM_BUFFS
      ), "memory buffer {} already allocated.".format(name)
      _MEM_BUFFS[name] = MemoryBuffer(name, numel, dtype, track_usage)
      return _MEM_BUFFS[name]
  
file_path: apex/apex/transformer/tensor_parallel/memory.py
imports: import torch
label_desc: Use image node_id 1 for calling a global function with example usage: allocate_mem_buff(name, numel, dtype, track_usage) and returns: _MEM_BUFFS