ColossalAI/tests
アマデウス 9ee197d0e9 moved env variables to global variables; (#215)
added branch context;
added vocab parallel layers;
moved split_batch from load_batch to tensor parallel embedding layers;
updated gpt model;
updated unit test cases;
fixed few collective communicator bugs
2022-02-15 11:31:13 +08:00
..
test_comm Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
test_config add pytorch hooks (#179) 2022-01-25 22:20:54 +08:00
test_context Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00
test_data added CI for unit testing (#69) 2021-12-16 10:32:08 +08:00
test_data_pipeline_tensor_parallel Optimize pipeline schedule (#94) 2021-12-30 15:56:46 +08:00
test_engine/test_engine Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
test_layers moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
test_trainer pipeline last stage supports multi output (#151) 2022-01-17 15:57:47 +08:00
test_utils Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
test_zero_data_parallel Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
test_zero_tensor_parallel Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00