vllm.distributed.device_communicators
Modules:
| Name | Description |
|---|---|
all2all |
|
base_device_communicator |
|
cpu_communicator |
|
cuda_communicator |
|
cuda_wrapper |
This file is a pure Python wrapper for the cudart library. |
custom_all_reduce |
|
custom_all_reduce_utils |
|
hpu_communicator |
|
neuron_communicator |
|
pynccl |
|
pynccl_wrapper |
|
quick_all_reduce |
|
shm_broadcast |
|
tpu_communicator |
|
xpu_communicator |
|