Ziyue Jiang 09d69e1c25 [PP Middleware] Add bwd and step for PP middleware (#2111) ...

* add bwd and step for PP middleware

* pre-commit

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

2022-12-12 12:40:03 +08:00

..

rpc_test_utils.py

[PP Middleware] Add bwd and step for PP middleware (#2111)

2022-12-12 12:40:03 +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

[PP Middleware] Add bwd and step for PP middleware (#2111)

2022-12-12 12:40:03 +08:00

test_pipelinable.py

…

test_pipeline_process_group.py

…