added use fallback option

kahypar · Jul 1, 2024 · da69ef2 · da69ef2
1 parent ce4fcca
commit da69ef2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/mt-kahypar/io/command_line_options.cpp b/mt-kahypar/io/command_line_options.cpp
@@ -156,6 +156,9 @@ namespace mt_kahypar {
              ("assure-balance",
              po::value<bool>(&context.partition.assure_balance)->value_name("<bool>")->default_value(true),
              "max imbalance of refinement in md_rebalancer")
+             ("use-fallback",
+             po::value<bool>(&context.partition.use_fallback)->value_name("<bool>")->default_value(true),
+             "max imbalance of refinement in md_rebalancer")
              ("constraint-in-unconstraint",
              po::value<bool>(&context.partition.constraint_in_unconstraint)->value_name("<bool>")->default_value(true),
              "max imbalance of refinement in md_rebalancer")

diff --git a/mt-kahypar/partition/context.h b/mt-kahypar/partition/context.h
@@ -80,6 +80,7 @@ struct PartitioningParameters {
   bool use_l1_factor_decrease = false;
   double decrease_exponent = 1.0;
 
+  bool use_fallback = true;
   double L_threshold = 0.0;
   double fallback_large_node_threshold;
   bool fallback_extract_equally = false;

diff --git a/mt-kahypar/partition/refinement/rebalancing/md_rebalancer.cpp b/mt-kahypar/partition/refinement/rebalancing/md_rebalancer.cpp
@@ -1694,7 +1694,7 @@ namespace mt_kahypar{
             greedyRefiner(rebalance_moves, best_metrics, local_attributed_gain, factor);
           }
         }
-        else{
+        else if(_context->partition.use_fallback){
           HypernodeID size = rebalance_moves->size();
           vec<bool> L(phg->initialNumNodes(), false);
           auto start = std::chrono::high_resolution_clock::now();