mirror of https://github.com/hpcaitech/ColossalAI
b0936e4a44
* add DAG to split_module * add comment * add test case for DAG * remove print * add DAG middleware in scheduler * add test case for scheduler * remove break * recover old lifecycle Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com> |
||
|---|---|---|
| .. | ||
| components_to_test | ||
| test_amp | ||
| test_auto_parallel | ||
| test_comm | ||
| test_config | ||
| test_context | ||
| test_data | ||
| test_data_pipeline_tensor_parallel | ||
| test_ddp | ||
| test_device | ||
| test_engine | ||
| test_fx | ||
| test_gemini | ||
| test_layers | ||
| test_moe | ||
| test_ops | ||
| test_optimizer | ||
| test_pipeline | ||
| test_tensor | ||
| test_trainer | ||
| test_utils | ||
| test_zero | ||
| __init__.py | ||