ColossalAI/colossalai/communication
Zangwei Zheng 197a2c89e2 [NFC] polish colossalai/communication/collective.py (#1262) 2022-07-13 12:08:21 +08:00
..
__init__.py [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00
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
ring.py
utils.py [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00