From 51dd14ac2b9a17cbc8eb91fe0f39311f1d594f63 Mon Sep 17 00:00:00 2001
From: Kebe <mail@kebe7jun.com>
Date: Fri, 24 Oct 2025 05:16:51 +0900
Subject: [PATCH] [Bugfix][DP] Fix creating too many DP Placement Groups
 (#26880)

Signed-off-by: Kebe <mail@kebe7jun.com>
Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
Co-authored-by: Rui Qiao <ruisearch42@gmail.com>
---
 vllm/v1/engine/utils.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/vllm/v1/engine/utils.py b/vllm/v1/engine/utils.py
index f97973808e3d9..5b4655354ed7d 100644
--- a/vllm/v1/engine/utils.py
+++ b/vllm/v1/engine/utils.py
@@ -497,6 +497,8 @@ class CoreEngineActorManager:
                 )
                 placement_groups.append(pg)
                 local_dp_ranks.append(i)
+                if len(placement_groups) == dp_size:
+                    break
 
         if len(placement_groups) < dp_size:
             raise ValueError(
@@ -506,6 +508,13 @@ class CoreEngineActorManager:
                 "Available resources: "
                 f"{available_resources}"
             )
+        assert len(placement_groups) == dp_size, (
+            f"Created {len(placement_groups)} DP placement groups, expected {dp_size}"
+        )
+        assert len(local_dp_ranks) == dp_size, (
+            f"local_dp_ranks length {len(local_dp_ranks)} does not match "
+            f"expected {dp_size}"
+        )
         return placement_groups, local_dp_ranks
 
     @staticmethod