fix(partitioner): update usage of mt-kahypar

Author: uulm-janbaudisch

CMakeLists.txt

@@ -1,4 +1,4 @@
-cmake_minimum_required(VERSION 3.27)
+cmake_minimum_required(VERSION 3.30)
 
 project(d4 VERSION 2.0.0 LANGUAGES CXX)
 
@@ -35,7 +35,6 @@ include_directories(${CMAKE_CURRENT_SOURCE_DIR}/3rdParty/glucose-3.0)
 include_directories(${GMP_INCLUDE_DIRS})
 include_directories(${GMPXX_INCLUDE_DIRS})
 include_directories(${Boost_INCLUDE_DIRS})
-include_directories(${MtKaHyPar_INCLUDE_DIR})
 include_directories(${arjun_INCLUDE_DIR})
 include_directories(${GMPC_INCLUDE_DIR})
 include_directories(${glucose_INCLUDE_DIRS})
@@ -156,7 +155,7 @@ add_library(core STATIC
     $<TARGET_OBJECTS:utils>
 )
 
-target_link_libraries(core MtKaHyPar::MtKaHyPar GPMC::GPMC arjun glucose::glucose GMP::GMPXX GMP::GMP cadiback cadical)
+target_link_libraries(core MtKaHyPar::mtkahypar GPMC::GPMC arjun glucose::glucose GMP::GMPXX GMP::GMP cadiback cadical)
 
 if(USE_GLUCOSE)
     target_link_libraries(core glucose::glucose)

cmake/FindMtKaHyPar.cmake

@@ -18,9 +18,9 @@
 #include "PartitionerKahyparMT.hpp"
 
 #include <iostream>
-#include <libmtkahypar.h>
 #include <vector>
 
+#include "mtkahypar.h"
 #include "src/exceptions/OptionException.hpp"
 
 namespace d4 {
@@ -34,6 +34,8 @@ namespace d4 {
 PartitionerKahyparMT::PartitionerKahyparMT(unsigned maxNodes, unsigned maxEdges,
                                            unsigned maxSumEdgeSize,
                                            std::ostream &out) {
+  mt_kahypar_error_t error{};
+
   m_pins = std::make_unique<mt_kahypar_hyperedge_id_t[]>(maxSumEdgeSize);
   m_xpins = std::make_unique<size_t[]>(maxEdges + 3);
   m_cwghts = std::make_unique<mt_kahypar_hyperedge_weight_t[]>(maxNodes + 3);
@@ -46,13 +48,13 @@ PartitionerKahyparMT::PartitionerKahyparMT(unsigned maxNodes, unsigned maxEdges,
   m_mapNodes.resize(maxNodes + 3, false);
   m_markedNodes.resize(maxNodes + 3, false);
 
-  context = mt_kahypar_context_new();
+  mt_kahypar_context_t* context = mt_kahypar_context_from_preset(DEFAULT);
   mt_kahypar_set_partitioning_parameters(context, 2 /* number of blocks */,
                                          0.05 /* imbalance parameter */,
                                          CUT /* objective function */);
 
-  mt_kahypar_set_context_parameter(context, VERBOSE, "0");
-  mt_kahypar_load_preset(context, mt_kahypar_preset_type_t::QUALITY);
+  mt_kahypar_status_t status = mt_kahypar_set_context_parameter(context, VERBOSE, "0", &error);
+  assert(status == SUCCESS);
 } // constructor
 
 /**
@@ -63,7 +65,7 @@ PartitionerKahyparMT::~PartitionerKahyparMT() {
 } // destructor
 
 void PartitionerKahyparMT::initPartitioner(Config &config) {
-  mt_kahypar_initialize_thread_pool(config.partitioning_threads, true);
+  mt_kahypar_initialize(config.partitioning_threads, true);
 }
 
 /**
@@ -74,6 +76,7 @@ void PartitionerKahyparMT::initPartitioner(Config &config) {
  */
 void PartitionerKahyparMT::computePartition(HyperGraph &hypergraph, Level level,
                                             std::vector<int> &partition) {
+  mt_kahypar_error_t error{};
   std::vector<unsigned> elts;
 
   // graph initialization and shift the hypergraph
@@ -109,15 +112,16 @@ void PartitionerKahyparMT::computePartition(HyperGraph &hypergraph, Level level,
   const mt_kahypar_hyperedge_id_t num_hyperedges = sizeXpins;
 
   auto hgraph =
-      mt_kahypar_create_hypergraph(DEFAULT, num_vertices, num_hyperedges,
-                                   m_xpins.get(), m_pins.get(), cost, nullptr);
+      mt_kahypar_create_hypergraph(context, num_vertices, num_hyperedges,
+                                   m_xpins.get(), m_pins.get(), cost, nullptr, &error);
 
-  auto p = mt_kahypar_partition(hgraph, context);
+  auto p = mt_kahypar_partition(hgraph, context, &error);
 
   mt_kahypar_get_partition(p, m_partition.data());
 
-  mt_kahypar_free_partitioned_hypergraph(p);
   mt_kahypar_free_hypergraph(hgraph);
+  mt_kahypar_free_partitioned_hypergraph(p);
+
   for (unsigned i = 0; i < elts.size(); i++)
     partition[elts[i]] = m_partition[i];
 } // computePartition

src/partitioner/PartitionerKahyparMT.cpp

@@ -19,7 +19,7 @@
 
 #include <vector>
 #include <memory>
-#include <libmtkahypar.h>
+#include <mtkahypar.h>
 
 #include "src/config/Config.hpp"
 
@@ -35,6 +35,7 @@ class PartitionerKahyparMT : public PartitionerManager {
   std::unique_ptr<mt_kahypar_hyperedge_id_t[]> m_pins;
   std::vector<mt_kahypar_partition_id_t> m_partition;
   mt_kahypar_context_s *context;
+  mt_kahypar_error_t error;
 
  public:
   PartitionerKahyparMT(unsigned maxNodes, unsigned maxEdges,