youkaichao 99aa4eddaf [torch.compile] register allreduce operations as custom ops (#8526)

2024-09-16 22:57:57 -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] register allreduce operations as custom ops (#8526)

2024-09-16 22:57:57 -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][distributed] fix zmq hang (#6759)

2024-07-24 17:37:12 -07:00

tpu_communicator.py

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

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