ColossalAI/colossalai/gemini/update
HELSON b80340168e
[zero] add chunk_managerV2 for all-gather chunk (#1441)
2022-08-11 19:17:24 +08:00
..
__init__.py [zero] add chunk_managerV2 for all-gather chunk (#1441) 2022-08-11 19:17:24 +08:00
chunk_mgrv2.py [zero] add chunk_managerV2 for all-gather chunk (#1441) 2022-08-11 19:17:24 +08:00
chunkv2.py [polish] add update directory in gemini; rename AgChunk to ChunkV2 (#1432) 2022-08-10 16:40:29 +08:00
search_utils.py [zero] add chunk size searching algorithm for parameters in different groups (#1436) 2022-08-11 13:32:19 +08:00