ColossalAI/tests
Jiarui Fang 49832b2344
[refactory] add nn.parallel module (#1068)
2022-06-06 15:34:41 +08:00
..
components_to_test [unit test] refactor test tensor (#1005) 2022-05-19 18:57:56 +08:00
test_amp [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
test_comm [p2p]add object list send/recv (#1024) 2022-05-26 14:28:46 +08:00
test_config
test_context [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
test_data [unittest] refactored unit tests for change in dependency (#838) 2022-04-22 15:39:07 +08:00
test_data_pipeline_tensor_parallel [titans]remove model zoo (#1042) 2022-05-31 10:40:47 +08:00
test_engine [refactor] moving grad acc logic to engine (#804) 2022-04-19 14:03:21 +08:00
test_gemini [gemini] accelerate adjust_layout() (#878) 2022-04-26 18:08:31 +08:00
test_layers [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
test_moe [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
test_optimizer
test_tensor [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
test_trainer [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00
test_utils [refactory] add nn.parallel module (#1068) 2022-06-06 15:34:41 +08:00
test_zero [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
__init__.py