ColossalAI/tests
YuliangLiu0306 7106bd671d
[p2p]add object list send/recv (#1024)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [p2p]add object list send recv

* refactor for code reusability

* polish
2022-05-26 14:28:46 +08:00
..
components_to_test [unit test] refactor test tensor (#1005) 2022-05-19 18:57:56 +08:00
test_amp
test_comm [p2p]add object list send/recv (#1024) 2022-05-26 14:28:46 +08:00
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_engine
test_gemini [gemini] accelerate adjust_layout() (#878) 2022-04-26 18:08:31 +08:00
test_layers
test_moe
test_optimizer
test_tensor [Tensor] add module handler for linear (#1021) 2022-05-26 11:50:44 +08:00
test_trainer
test_utils
test_zero
__init__.py