mirror of https://github.com/hpcaitech/ColossalAI
ffa025e120
* dist spec s2s uses all-to-all * update unit test * add sanity check * polish unitest test with titans * add sanity check for DistMgr * add sanity check Co-authored-by: jiaruifang <fangjiarui123@gmail.com> |
||
---|---|---|
.. | ||
components_to_test | ||
test_amp | ||
test_comm | ||
test_config | ||
test_context | ||
test_data | ||
test_data_pipeline_tensor_parallel | ||
test_ddp | ||
test_engine | ||
test_fx | ||
test_gemini | ||
test_layers | ||
test_moe | ||
test_optimizer | ||
test_pipeline | ||
test_tensor | ||
test_trainer | ||
test_utils | ||
test_zero | ||
__init__.py |