ColossalAI/tests/test_pipeline
Ziyue Jiang b0936e4a44
[rpc] split with dag (#2028)
* 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>
2022-11-29 11:36:28 +08:00
..
rpc_test_utils.py [rpc] split with dag (#2028) 2022-11-29 11:36:28 +08:00
test_cuda_rpc_chimera.py
test_cuda_rpc_optimizer.py
test_cuda_rpc_performance.py
test_cuda_rpc_pipeline.py
test_cuda_rpc_value_correctness.py
test_middleware_1f1b.py [rpc] split with dag (#2028) 2022-11-29 11:36:28 +08:00
test_pipelinable.py
test_pipeline_process_group.py