KaMinPar: reduce memory consumption during initial partitioning.

KaHIP · Sep 4, 2023 · 5a26395 · 5a26395
1 parent a03a364
commit 5a26395
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/kaminpar/partitioning/deep_multilevel.cc b/kaminpar/partitioning/deep_multilevel.cc
@@ -155,16 +155,20 @@ PartitionedGraph DeepMultilevelPartitioner::initial_partition(const Graph *graph
   // initial partitioning.
   DISABLE_TIMERS();
   PartitionedGraph p_graph = [&] {
+    Context input_ctx = _input_ctx;
+    input_ctx.partition.n = graph->n();
+    input_ctx.partition.m = graph->m();
+
     switch (_input_ctx.initial_partitioning.mode) {
     case InitialPartitioningMode::SEQUENTIAL:
       return helper::bipartition(graph, _input_ctx.partition.k, _input_ctx, _ip_m_ctx_pool);
 
     case InitialPartitioningMode::SYNCHRONOUS_PARALLEL:
-      return SyncInitialPartitioner(_input_ctx, _ip_m_ctx_pool, _ip_extraction_pool)
+      return SyncInitialPartitioner(input_ctx, _ip_m_ctx_pool, _ip_extraction_pool)
           .partition(_coarsener.get(), _current_p_ctx);
 
     case InitialPartitioningMode::ASYNCHRONOUS_PARALLEL:
-      return AsyncInitialPartitioner(_input_ctx, _ip_m_ctx_pool, _ip_extraction_pool)
+      return AsyncInitialPartitioner(input_ctx, _ip_m_ctx_pool, _ip_extraction_pool)
           .partition(_coarsener.get(), _current_p_ctx);
     }
 

diff --git a/kaminpar/refinement/fm_refiner.cc b/kaminpar/refinement/fm_refiner.cc
@@ -426,11 +426,11 @@ LocalizedFMRefiner::LocalizedFMRefiner(
       _shared(shared),
       _d_graph(&_p_graph),
       _d_gain_cache(_shared.gain_cache),
-      _block_pq(_p_ctx.k),
+      _block_pq(_p_graph.k()),
       _stopping_policy(_fm_ctx.alpha) {
   _stopping_policy.init(_p_graph.n());
   for (const BlockID b : _p_graph.blocks()) {
-    _node_pqs.emplace_back(_p_ctx.n, _shared.shared_pq_handles.data());
+    _node_pqs.emplace_back(_p_graph.n(), _shared.shared_pq_handles.data());
   }
 }