ColossalAI/tests/test_pipeline
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
..
test_pipeline_utils [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_schedule [pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134) 2023-12-22 10:44:00 +08:00
test_p2p_communication.py [pipeline] A more general _communicate in p2p (#5062) 2024-01-08 15:37:27 +08:00
test_stage_manager.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00