ColossalAI/colossalai/gemini/memory_tracer
Zihao 95c4532fff
[Gemini] paramWrapper paramTracerHook unitest (#2030)
2022-11-26 13:30:24 +08:00
..
__init__.py [hotfix] revert bug PRs (#2016) 2022-11-24 15:28:58 +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
param_tracer_wrapper.py [Gemini] paramWrapper paramTracerHook unitest (#2030) 2022-11-26 13:30:24 +08:00
static_memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00