avoid potential endless loop in deterministic LP

kahypar · Nov 26, 2024 · fab2b5e · fab2b5e
1 parent ad13a1c
commit fab2b5e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mt-kahypar/partition/refinement/deterministic/deterministic_label_propagation.cpp b/mt-kahypar/partition/refinement/deterministic/deterministic_label_propagation.cpp
@@ -240,7 +240,8 @@ namespace mt_kahypar {
       }
     }
 
-    if (num_overloaded_blocks > 0) {
+    // only try again if the input was balanced, otherwise we probably can't achieve balance
+    if (num_overloaded_blocks > 0 && num_overloaded_before_round == 0) {
       DBG << "still overloaded" << num_overloaded_blocks << V(num_moves) << V(num_reverted_moves)
           << V(num_overloaded_before_round) << V(num_overloaded_before_first_pass) << "trigger second run";