Aurick Qiao 0437492ea9

PP comm optimization: replace send with partial send + allgather (#6695) ...

Co-authored-by: Aurick Qiao <aurick.qiao@snowflake.com>

2024-07-31 20:15:42 -07:00

..

device_communicators

[TPU] Reduce compilation time & Upgrade PyTorch XLA version (#6856)

2024-07-27 10:28:33 -07:00

__init__.py

[Core][Refactor] move parallel_utils into vllm/distributed (#3950)

2024-04-10 15:33:30 -07:00

communication_op.py

[Core][Distributed] code deduplication in tp&pp with coordinator(#5293)

2024-06-12 17:27:08 -07:00

parallel_state.py

PP comm optimization: replace send with partial send + allgather (#6695)

2024-07-31 20:15:42 -07:00

utils.py

[MISC] Introduce pipeline parallelism partition strategies (#6920)

2024-07-31 12:02:17 -07:00