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 [doc] improved docstring in the communication module (#863) 2022-04-25 13:41:43 +08:00
utils.py [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00