megatron.memory.allocate\_mem\_buff
===================================

.. currentmodule:: megatron.memory

.. autofunction:: allocate_mem_buff