From c9791f18138d1a11bfe68550b10673b493ec9330 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Thu, 30 Oct 2025 16:26:13 -0700 Subject: [PATCH] [BugFix] Fix broken import in initialize_ray_cluster() (#27838) Signed-off-by: Nick Hill --- vllm/v1/executor/ray_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/v1/executor/ray_utils.py b/vllm/v1/executor/ray_utils.py index 382f008266e6..9385e55b066f 100644 --- a/vllm/v1/executor/ray_utils.py +++ b/vllm/v1/executor/ray_utils.py @@ -322,7 +322,7 @@ def initialize_ray_cluster( # Prevalidate GPU requirements before Ray processing if current_platform.is_cuda() and parallel_config.world_size > 1: - from vllm.utils import cuda_device_count_stateless + from vllm.utils.torch_utils import cuda_device_count_stateless available_gpus = cuda_device_count_stateless() if parallel_config.world_size > available_gpus: