You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_auto_parallel
Baizhou Zhang 0bb0b481b4
[gemini] fix argument naming during chunk configuration searching
1 year ago
..
test_ckpt_solvers [test] refactor tests with spawn (#3452) 2 years ago
test_offload [gemini] fix argument naming during chunk configuration searching 1 year ago
test_pass [test] refactor tests with spawn (#3452) 2 years ago
test_tensor_shard [gemini] fix argument naming during chunk configuration searching 1 year ago
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2 years ago