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
Frank Lee e2089c5c15
adapted for sequence parallel (#163)
3 years ago
..
test_comm Hotfix/Colossalai layers (#92) 3 years ago
test_config Develop/experiments (#59) 3 years ago
test_context Optimize pipeline schedule (#94) 3 years ago
test_data added CI for unit testing (#69) 3 years ago
test_data_pipeline_tensor_parallel Optimize pipeline schedule (#94) 3 years ago
test_engine/test_engine Hotfix/Colossalai layers (#92) 3 years ago
test_layers adapted for sequence parallel (#163) 3 years ago
test_trainer pipeline last stage supports multi output (#151) 3 years ago
test_utils Hotfix/Colossalai layers (#92) 3 years ago
test_zero_data_parallel Hotfix/Colossalai layers (#92) 3 years ago
test_zero_tensor_parallel Hotfix/Colossalai layers (#92) 3 years ago