megatron.core.utils.GlobalMemoryBuffer#

class megatron.core.utils.GlobalMemoryBuffer#

Bases: object

Global buffer to avoid dynamic memory allocations. Caller should ensure that buffers of the same name are not used concurrently.