megatron.memory.allocate_mem_buff#

megatron.memory.allocate_mem_buff(name, numel, dtype, track_usage)#

Allocate a memory buffer.