ColossalAI/communication at 293fb40c4246dd4c800f8cb03292a02122edd24b - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-21 23:02:07 +00:00

Files

History

ver217 293fb40c42 add scatter/gather optim for pipeline (#123 )

2022-01-07 13:22:22 +08:00

..

__init__.py

add scatter/gather optim for pipeline (#123 )

2022-01-07 13:22:22 +08:00

collective.py

Layer integration (#83 )

2021-12-27 15:04:32 +08:00

p2p.py

add scatter/gather optim for pipeline (#123 )

2022-01-07 13:22:22 +08:00

ring.py

update examples and sphnix docs for the new api (#63 )

2021-12-13 22:07:01 +08:00

utils.py

add scatter/gather optim for pipeline (#123 )

2022-01-07 13:22:22 +08:00