ColossalAI/colossalai/communication
YuliangLiu0306 b167258b6a
[pipeline]refactor ppschedule to support tensor list (#1050)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* refactor ppschedule to support tensor list

* polish
2022-06-02 13:48:59 +08:00
..
__init__.py [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +08:00
collective.py updated collective ops api (#1054) 2022-06-02 12:52:27 +08:00
p2p.py [p2p]add object list send/recv (#1024) 2022-05-26 14:28:46 +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