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