youkaichao 3cd9b5bb2d [Core][Distributed] use existing torch.cuda.device (#4318) ...

[Core][Distributed] use existing torch.cuda.device context manager (#4318)

2024-04-24 09:00:20 -07:00

..

__init__.py

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

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

custom_all_reduce.py

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

pynccl_utils.py

[Core][Distributed] use cpu/gloo to initialize pynccl (#4248)

2024-04-23 18:32:19 -07:00

pynccl.py

[Core][Distributed] use existing torch.cuda.device (#4318)

2024-04-24 09:00:20 -07:00