Skip to content

GitLab

Unverified Commit 529cea34 authored by Yan Ma's avatar Yan Ma Committed by GitHub
Browse files

use default CCL_ZE_IPC_EXCHANGE (#28700) 


Signed-off-by: default avatarYan Ma <yan.ma@intel.com>
parent 93103575
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
vllm/v1/worker/xpu_worker.py
View file @ 529cea34
...@@ -159,12 +159,10 @@ class XPUWorker(Worker): ...@@ -159,12 +159,10 @@ class XPUWorker(Worker):
else: else:
raise RuntimeError(f"Not support device type: {self.device_config.device}") raise RuntimeError(f"Not support device type: {self.device_config.device}")
ENV_CCL_ZE_IPC_EXCHANGE = os.getenv("CCL_ZE_IPC_EXCHANGE", "pidfd")
ENV_CCL_ATL_TRANSPORT = os.getenv("CCL_ATL_TRANSPORT", "ofi") ENV_CCL_ATL_TRANSPORT = os.getenv("CCL_ATL_TRANSPORT", "ofi")
ENV_LOCAL_WORLD_SIZE = os.getenv( ENV_LOCAL_WORLD_SIZE = os.getenv(
"LOCAL_WORLD_SIZE", str(self.parallel_config.world_size) "LOCAL_WORLD_SIZE", str(self.parallel_config.world_size)
) )
os.environ["CCL_ZE_IPC_EXCHANGE"] = ENV_CCL_ZE_IPC_EXCHANGE
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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment