ColossalAI/colossalai/gemini
HELSON 0d212183c4
[zero] add has_inf_or_nan in AgChunk; enhance the unit test of AgChunk (#1426)
2022-08-10 11:37:28 +08:00
..
memory_tracer
ophooks
paramhooks
__init__.py
ag_chunk.py [zero] add has_inf_or_nan in AgChunk; enhance the unit test of AgChunk (#1426) 2022-08-10 11:37:28 +08:00
chunk.py [hotfix] fix a running error in test_colo_checkpoint.py (#1387) 2022-07-29 15:58:06 +08:00
chunk_mgr.py [zero] chunk manager allows filtering ex-large params (#1393) 2022-08-02 10:40:27 +08:00
gemini_context.py
gemini_mgr.py
placement_policy.py
stateful_tensor.py
stateful_tensor_container.py
stateful_tensor_mgr.py
tensor_placement_policy.py
tensor_utils.py