ColossalAI/tests/test_auto_parallel
Baizhou Zhang 0bb0b481b4 [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
..
test_ckpt_solvers [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_offload [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
test_pass [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_tensor_shard [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
__init__.py