megengine.reset_max_memory_stats¶

reset_max_memory_stats(device=None)[源代码]¶

Resets the maximum stats on the computing device.

Due to the asynchronous execution of MegEngine, please call megengine._full_sync before calling this function in order to properly reset memory stats.

megengine.get_max_allocated_memory

megengine.set_prealloc_config