bnellnm 5467ac3196 [Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

..

__init__.py

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

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

custom_all_reduce_utils.py

[Core][Distributed] improve p2p access check (#4992)

2024-05-29 11:29:07 +00:00

custom_all_reduce.py

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

pynccl_wrapper.py

[Core][Optimization] remove vllm-nccl (#5091)

2024-05-29 05:13:52 +00:00

pynccl.py

[Core][1/N] Support send/recv in PyNCCL Groups (#4988)

2024-05-23 09:54:48 -07:00