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> |
||
---|---|---|
.. | ||
rpc_test_utils.py | ||
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 | ||
test_pipelinable.py | ||
test_pipeline_process_group.py |