ColossalAI/colossalai/gemini/memory_tracer
Jiarui Fang b3b89865e2
[Gemini] ParamOpHook -> ColoParamOpHook (#2080)
2022-12-05 17:11:06 +08:00
..
__init__.py [Gemini] remove not used MemtracerWrapper (#2072) 2022-12-05 11:57:59 +08:00
chunk_memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00
memory_monitor.py [Gemini] independent runtime tracer (#1974) 2022-11-18 10:53:42 +08:00
memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00
model_data_memtracer.py [Gemini] fix grad unreleased issue and param recovery issue (#2052) 2022-12-02 16:04:19 +08:00
runtime_mem_tracer.py [Gemini] ParamOpHook -> ColoParamOpHook (#2080) 2022-12-05 17:11:06 +08:00
static_memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00