ColossalAI/tests/test_pipeline
Kirigaya Kazuto bb5f5289e0
[pipeline/rpc] implement a demo for PP with cuda rpc framework (#1470)
* support p2p communication with any type of object | pass test

* reconstruct pipeline schedule with p2p_v2.py(support communication with List[Any]) | pass test

* [engin/schedule] use p2p_v2 to recontruct pipeline_schedule

* [pipeline/rpc] implement a demo for PP with cuda rpc framework

* Delete p2p_v2.py

* Delete _pipeline_schedule_v2.py

* Delete test_object_list_p2p_v2.py

* Delete test_boardcast_send_recv_v2.py

* Delete test_cifar_with_data_pipeline_tensor_v2.py
2022-08-22 10:50:51 +08:00
..
test_cuda_rpc_pipeline.py [pipeline/rpc] implement a demo for PP with cuda rpc framework (#1470) 2022-08-22 10:50:51 +08:00
test_pipelinable.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00