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