ColossalAI/tests
Ziyue Jiang dfaff4e243
[Tensor] fix test_model (#916)
* polish test_model

* polish
2022-05-06 18:06:22 +08:00
..
components_to_test [Tensor] fix test_model (#916) 2022-05-06 18:06:22 +08:00
test_amp [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
test_comm [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +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 [refactor] moving grad acc logic to engine (#804) 2022-04-19 14:03:21 +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 [Tensor] fix test_model (#916) 2022-05-06 18:06:22 +08:00
test_trainer [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00
test_utils [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
test_zero [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
__init__.py