YARN-11752: Global Scheduler: Improve the container allocation time (a…

…pache#7277) Contributed by Syed Shameerur Rahman. Signed-off-by: Shilun Fan <[email protected]>
slfan1989 · Jan 12, 2025 · c24d569 · c24d569
1 parent d2095fa
commit c24d569
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...a/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/CapacityScheduler.java b/...a/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/CapacityScheduler.java
@@ -604,7 +604,10 @@ static void schedule(CapacityScheduler cs) throws InterruptedException{
           if (candidates == null) {
             continue;
           }
-          cs.allocateContainersToNode(candidates, false);
+          int nodesPerPartitionCount = candidates.getAllNodes().size();
+          for (int i = 0; i < nodesPerPartitionCount; i++) {
+            cs.allocateContainersToNode(candidates, false);
+          }
         }
       }
 
@@ -620,7 +623,10 @@ static void schedule(CapacityScheduler cs) throws InterruptedException{
         if (candidates == null) {
           continue;
         }
-        cs.allocateContainersToNode(candidates, false);
+        int nodesPerPartitionCount = candidates.getAllNodes().size();
+        for (int i = 0; i < nodesPerPartitionCount; i++) {
+          cs.allocateContainersToNode(candidates, false);
+        }
       }
 
     }