ColossalAI/colossalai/communication
ver217 116e3d0b8f [NFC] polish communication/p2p_v2.py code style (#2303) 2023-01-04 15:09:57 +08:00
..
__init__.py [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00
collective.py improved allgather & reducescatter for 3d 2023-01-03 17:46:08 +08:00
p2p.py [hotfix]fixed p2p process send stuck (#1181) 2022-06-28 14:41:11 +08:00
p2p_v2.py [NFC] polish communication/p2p_v2.py code style (#2303) 2023-01-04 15:09:57 +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