Elsa Granger d565df3821 [pipeline] A more general _communicate in p2p (#5062) ...

* A more general _communicate

* feat: finish tree_flatten version p2p

* fix: update p2p api calls

---------

Co-authored-by: Wenhao Chen <cwher@outlook.com>

2024-01-08 15:37:27 +08:00

..

schedule

[pipeline] A more general _communicate in p2p (#5062)

2024-01-08 15:37:27 +08:00

__init__.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

p2p.py

[pipeline] A more general _communicate in p2p (#5062)

2024-01-08 15:37:27 +08:00

stage_manager.py

[pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134)

2023-12-22 10:44:00 +08:00