ColossalAI/colossalai/utils/memory_tracer
Jiarui Fang 8d8c5407c0
[zero] refactor model data tracing (#522)
2022-03-25 18:03:32 +08:00
..
__init__.py move async memory to an individual directory (#345) 2022-03-11 15:50:28 +08:00
async_memtracer.py [zero] fix init device bug in zero init context unittest (#516) 2022-03-25 12:24:18 +08:00
memstats_collector.py [zero] fix init device bug in zero init context unittest (#516) 2022-03-25 12:24:18 +08:00
model_data_memtracer.py [zero] refactor model data tracing (#522) 2022-03-25 18:03:32 +08:00
test_async_memtracer.py move async memory to an individual directory (#345) 2022-03-11 15:50:28 +08:00
test_memstats_collector.py [polish] use GLOBAL_MODEL_DATA_TRACER (#417) 2022-03-15 11:29:46 +08:00