From 43aa3892314f8336f83a9fbe614899ddcf0e1df8 Mon Sep 17 00:00:00 2001 From: Weiguang Li Date: Mon, 9 Mar 2026 11:07:29 +0800 Subject: [PATCH] [Bugfix] Fix CPU OMP autobind assertion to use local_world_size (#35815) Signed-off-by: liweiguang Co-authored-by: Claude Opus 4.6 Co-authored-by: Li, Jiang --- vllm/v1/worker/cpu_worker.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/vllm/v1/worker/cpu_worker.py b/vllm/v1/worker/cpu_worker.py index a72f450a7..c4e4783a6 100644 --- a/vllm/v1/worker/cpu_worker.py +++ b/vllm/v1/worker/cpu_worker.py @@ -144,12 +144,10 @@ class CPUWorker(Worker): allowed_numa_nodes, logical_cpu_list = ( CpuPlatform.get_allowed_cpu_core_node_list() ) - assert ( - len(allowed_numa_nodes) >= self.parallel_config.world_size - or sim_multi_numa_nodes - ), ( + local_world_size = self.parallel_config.local_world_size + assert len(allowed_numa_nodes) >= local_world_size or sim_multi_numa_nodes, ( f"Not enough allowed NUMA nodes to bind threads of " - f"{self.parallel_config.world_size} CPUWorkers. " + f"{local_world_size} local CPUWorkers. " f"Allowed NUMA nodes are {allowed_numa_nodes}. " "Please try to bind threads manually." )