megatron.utils.report_memory#

megatron.utils.report_memory(name)#

Simple GPU memory report.