megatron.core.parallel_state.get_global_memory_buffer#

megatron.core.parallel_state.get_global_memory_buffer()#

Return the global GlobalMemoryBuffer object