ColossalAI/colossalai/communication
ver217 116e3d0b8f [NFC] polish communication/p2p_v2.py code style (#2303) 2023-01-04 15:09:57 +08:00
..
__init__.py
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
utils.py