youkaichao 663874e048

[torch.compile] improve allreduce registration (#9061)

2024-10-04 16:43:50 -07:00

..

__init__.py

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

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

cuda_wrapper.py

[core][misc] update libcudart finding (#7620)

2024-08-16 23:01:35 -07:00

custom_all_reduce_utils.py

[misc] fix custom allreduce p2p cache file generation (#7853)

2024-08-26 15:02:25 -07:00

custom_all_reduce.py

[torch.compile] improve allreduce registration (#9061)

2024-10-04 16:43:50 -07:00

pynccl_wrapper.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

pynccl.py

[Distributed] Add send and recv helpers (#5719)

2024-06-23 14:42:28 -07:00

shm_broadcast.py

[Core] Allow IPv6 in VLLM_HOST_IP with zmq (#8575)

2024-09-23 12:15:41 -07:00

tpu_communicator.py

[TPU] Support single and multi-host TPUs on GKE (#7613)

2024-08-30 00:27:40 -07:00