xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-07-09 16:07:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix][Intel GPU] Use refactored API for dist_backend in V1 worker (#20596)

Browse Source 
Signed-off-by: ratnampa <ratnam.parikh@intel.com>
This commit is contained in:
Ratnam Parikh 2025-07-08 19:44:23 -07:00 committed by GitHub
parent baed180aa0
commit c40784c794
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/v1/worker/xpu_worker.py
View File

@ -148,11 +148,11 @@ class XPUWorker(Worker):
os.environ["CCL_ATL_TRANSPORT"] = ENV_CCL_ATL_TRANSPORT os.environ["CCL_ATL_TRANSPORT"] = ENV_CCL_ATL_TRANSPORT
os.environ["LOCAL_WORLD_SIZE"] = ENV_LOCAL_WORLD_SIZE os.environ["LOCAL_WORLD_SIZE"] = ENV_LOCAL_WORLD_SIZE
os.environ["LOCAL_RANK"] = str(self.local_rank) os.environ["LOCAL_RANK"] = str(self.local_rank)
dist_backend = "ccl"
init_worker_distributed_environment(self.vllm_config, self.rank, init_worker_distributed_environment(self.vllm_config, self.rank,
self.distributed_init_method, self.distributed_init_method,
self.local_rank, dist_backend) self.local_rank,
current_platform.dist_backend)
# global all_reduce needed for overall oneccl warm up # global all_reduce needed for overall oneccl warm up
torch.distributed.all_reduce(torch.zeros(1).xpu()) torch.distributed.all_reduce(torch.zeros(1).xpu())