xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 02:44:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

[XPU] Make pp group initilized for pipeline-parallelism (#11648)

Browse Source 
Signed-off-by: yisheng <yi.sheng@intel.com>
This commit is contained in:
YiSheng5 2025-01-07 11:09:58 +08:00 committed by GitHub
parent d0169e1b0f
commit d93d2d74fd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/worker/xpu_worker.py
View File

@ -11,6 +11,7 @@ import torch.distributed
from vllm.config import VllmConfig
from vllm.distributed import (ensure_model_parallel_initialized,
init_distributed_environment)
from vllm.distributed.parallel_state import get_pp_group
from vllm.logger import init_logger
from vllm.model_executor import set_random_seed
from vllm.platforms import current_platform
@ -176,3 +177,8 @@ class XPUWorker(LoraNotSupportedWorkerBase, Worker):
parallel_config.pipeline_parallel_size)
# global all_reduce needed for overall oneccl warm up
torch.distributed.all_reduce(torch.zeros(1).xpu())
if parallel_config.pipeline_parallel_size > 1:
# Add pp group init to avoid
# p2p communication as the first call
get_pp_group().all_reduce(torch.zeros(1).xpu())