mirror of https://github.com/hpcaitech/ColossalAI
09d69e1c25
* add bwd and step for PP middleware * pre-commit 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 |