ColossalAI/tests
zbian e94c79f15b improved allgather & reducescatter for 3d 2023-01-03 17:46:08 +08:00
..
components_to_test [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_amp
test_auto_parallel [autoparallel] fix spelling error (#2270) 2023-01-03 16:13:00 +08:00
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_ddp
test_device
test_engine
test_fx
test_gemini [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_layers improved allgather & reducescatter for 3d 2023-01-03 17:46:08 +08:00
test_moe
test_ops
test_optimizer
test_pipeline
test_tensor [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_trainer
test_utils
test_zero [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
__init__.py