ColossalAI/colossalai/gemini
Zihao 20e255d4e8
MemStatsCollectorStatic (#1765)
2022-11-07 16:49:03 +08:00
..
chunk [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2022-11-02 16:11:34 +08:00
memory_tracer MemStatsCollectorStatic (#1765) 2022-11-07 16:49:03 +08:00
ophooks
paramhooks
__init__.py
gemini_context.py
gemini_mgr.py MemStatsCollectorStatic (#1765) 2022-11-07 16:49:03 +08:00
placement_policy.py
stateful_tensor.py
stateful_tensor_mgr.py
tensor_placement_policy.py
tensor_utils.py