ColossalAI/tests/test_auto_parallel/test_tensor_shard
Baizhou Zhang 0bb0b481b4 [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
..
test_gpt [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_metainfo [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_node_handler fix typo tests/ (#3936) 2023-06-09 09:49:41 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00
test_bias_addition_forward.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_broadcast.py [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
test_checkpoint.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_compatibility_with_ddp.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_compatibility_with_gemini.py [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
test_find_repeat_block.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_liveness_analysis.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_solver_with_resnet_v2.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00