youkaichao 614aa51203 [misc][cuda] use nvml to avoid accidentally cuda initialization (#6007)

2024-06-30 20:07:34 -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][Distributed] improve p2p cache generation (#5528)

2024-06-14 14:47:45 -07:00

custom_all_reduce_utils.py

[Hardware][AMD][CI/Build][Doc] Upgrade to ROCm 6.1, Dockerfile improvements, test fixes (#5422)

2024-06-25 15:56:15 -07:00

custom_all_reduce.py

[misc][cuda] use nvml to avoid accidentally cuda initialization (#6007)

2024-06-30 20:07:34 -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

[CI/Build] Add TP test for vision models (#5892)

2024-06-29 15:45:54 +00:00