zbian e94c79f15b improved allgather & reducescatter for 3d

2023-01-03 17:46:08 +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_v2.py

[communication] add p2p_v2.py to support communication with List[Any] (#1407)

2022-08-09 11:40:04 +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