ColossalAI/tests
Frank Lee 53297330c0
[test] fixed hybrid parallel test case on 8 GPUs (#1106)
2022-06-14 10:30:54 +08:00
..
components_to_test [test] ignore 8 gpu test (#1080) 2022-06-08 23:14:18 +08:00
test_amp
test_comm
test_config [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_context
test_data
test_data_pipeline_tensor_parallel [test] fixed hybrid parallel test case on 8 GPUs (#1106) 2022-06-14 10:30:54 +08:00
test_engine
test_gemini
test_layers [test] skip tests when not enough GPUs are detected (#1090) 2022-06-09 17:19:13 +08:00
test_moe
test_optimizer
test_pipeline [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_tensor [gemini] zero supports gemini (#1093) 2022-06-10 14:48:28 +08:00
test_trainer [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_utils [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_zero [test] ignore 8 gpu test (#1080) 2022-06-08 23:14:18 +08:00
__init__.py