ColossalAI/colossalai/communication
Kirigaya Kazuto 44fd3c83ab
[communication] add p2p_v2.py to support communication with List[Any] (#1407)
* support p2p communication with any type of object | pass test

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

* [communication] add p2p_v2.py to support communication with List[Any]

* Delete _pipeline_schedule_v2.py

* Delete test_cifar_with_data_pipeline_tensor_v2.py

* [engin/schedule] use p2p_v2 to recontruct pipeline_schedule

* [communication] remove print code

* [communication] remove print code
2022-08-09 11:40:04 +08:00
..
__init__.py
collective.py [NFC] polish colossalai/communication/collective.py (#1262) 2022-07-13 12:08:21 +08:00
p2p.py [hotfix]fixed p2p process send stuck (#1181) 2022-06-28 14:41:11 +08:00
p2p_v2.py [communication] add p2p_v2.py to support communication with List[Any] (#1407) 2022-08-09 11:40:04 +08:00
ring.py
utils.py