Chaitanya Sri Krishna Lolla 1a820e38a2 Remove dependency of pynvml on ROCm (#2995)

2025-01-20 13:00:35 +08:00

..

__init__.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

cuda_wrapper.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

custom_all_reduce_utils.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

custom_all_reduce.py

Remove dependency of pynvml on ROCm (#2995)

2025-01-20 13:00:35 +08:00

hpu_communicator.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

pynccl_wrapper.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

pynccl.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

shm_broadcast.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00

xpu_communicator.py

adapt vllm distributed module to sglang (#2244)

2024-12-01 15:54:52 +08:00