ColossalAI/colossalai/communication
Michelle 86cf6aed5b Fix/format (#4261)
* revise shardformer readme (#4246)

* [example] add llama pretraining (#4257)

* [NFC] polish colossalai/communication/p2p.py code style

---------

Co-authored-by: Jianghai <72591262+CjhHa1@users.noreply.github.com>
Co-authored-by: binmakeswell <binmakeswell@gmail.com>
Co-authored-by: Qianran Ma <qianranm@luchentech.com>
2023-07-26 14:12: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 Fix/format (#4261) 2023-07-26 14:12:57 +08:00
p2p_v2.py [CI] fix some spelling errors (#3707) 2023-05-10 17:12:03 +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