[ROCm][CI] Fix test_ray_v2_executor (#54249)
Signed-off-by: charlifu <charlifu@amd.com>
Charlie Fu committed
Aug 28, 2026 at 17:11 UTC
fd98d32f732674b63705494ba3965f21aa449aba
1 file changed
+1
-1
vllm/v1/executor/ray_utils.py
+1
-1
@@ -590,7 +590,7 @@ def initialize_ray_cluster(
590
)
591
ray.init(
592
address=ray_address,
593
- num_gpus=parallel_config.world_size,
593
+ num_gpus=current_platform.device_count(),
594
runtime_env=parallel_config.ray_runtime_env,
595
)
596
else: