ColossalAI/colossalai/gemini/memory_tracer
Jiarui Fang 3712ac7f90
[Gemini] add bert for MemtracerWrapper unintests (#1982)
2022-11-18 14:58:28 +08:00
..
__init__.py [Gemini] independent runtime tracer (#1974) 2022-11-18 10:53:42 +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 [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
module_tracer_wrapper.py [Gemini] add bert for MemtracerWrapper unintests (#1982) 2022-11-18 14:58:28 +08:00
static_memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00