ColossalAI/tests/test_legacy/test_pipeline
..
rpc_test_utils.py
test_cuda_rpc_chimera.py
test_cuda_rpc_optimizer.py
test_cuda_rpc_pipeline.py
test_cuda_rpc_value_correctness.py
test_middleware_1f1b.py
test_pipelinable.py
test_pipeline_process_group.py