fix tools

kahypar · Dec 19, 2024 · ce2e7ba · ce2e7ba
1 parent 91f8817
commit ce2e7ba
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 8 deletions.
diff --git a/tools/CMakeLists.txt b/tools/CMakeLists.txt
@@ -18,6 +18,9 @@ target_link_libraries(BenchShuffle MtKaHyPar-ToolsMin)
 add_executable(MtxToGraph mtx_to_graph.cc)
 target_link_libraries(MtxToGraph MtKaHyPar-BuildFlags)
 
+add_executable(FixedVertexFileGenerator fixed_vertex_file_generator.cc)
+target_link_libraries(FixedVertexFileGenerator MtKaHyPar-ToolsMin)
+
 
 #################################################################
 ## Tools with more dependencies                                ##
@@ -66,6 +69,3 @@ target_link_libraries(EvaluatePartition MtKaHyPar-BuildTools)
 
 add_executable(VerifyPartition verify_partition.cc)
 target_link_libraries(VerifyPartition MtKaHyPar-BuildTools)
-
-add_executable(FixedVertexFileGenerator fixed_vertex_file_generator.cc)
-target_link_libraries(FixedVertexFileGenerator MtKaHyPar-BuildTools)
diff --git a/tools/fixed_vertex_file_generator.cc b/tools/fixed_vertex_file_generator.cc
@@ -32,7 +32,6 @@
 
 #include "mt-kahypar/macros.h"
 #include "mt-kahypar/datastructures/hypergraph_common.h"
-#include "mt-kahypar/io/hypergraph_io.h"
 #include "mt-kahypar/utils/randomize.h"
 
 namespace po = boost::program_options;
@@ -60,7 +59,16 @@ int main(int argc, char* argv[]) {
   po::notify(cmd_vm);
 
   std::vector<PartitionID> partition;
-  io::readPartitionFile(partition_file, partition);
+  std::ifstream file(partition_file);
+  if (file) {
+    int part;
+    while (file >> part) {
+      partition.push_back(part);
+    }
+    file.close();
+  } else {
+    ERR("File not found:" << partition_file);
+  }
 
   int threshold = percentage * 1000;
   std::string fixed_vertex_file = partition_file;

diff --git a/tools/one_to_one_mapping.cc b/tools/one_to_one_mapping.cc
@@ -98,7 +98,7 @@ int main(int argc, char* argv[]) {
 
   // Read Partition
   std::vector<PartitionID> partition;
-  io::readPartitionFile(context.partition.graph_partition_filename, partition);
+  io::readPartitionFile(context.partition.graph_partition_filename, hg.initialNumNodes(), partition);
   PartitionedHypergraph partitioned_hg(context.partition.k, hg, parallel_tag_t { });
   partitioned_hg.doParallelForAllNodes([&](const HypernodeID& hn) {
     partitioned_hg.setOnlyNodePart(hn, partition[hn]);

diff --git a/tools/verify_partition.cc b/tools/verify_partition.cc
@@ -51,7 +51,7 @@ using PartitionedHypergraph = ds::PartitionedHypergraph<Hypergraph, ds::Connecti
 bool readPartitionFile(const std::string& partition_file, PartitionedHypergraph& hypergraph) {
   bool success = true;
   std::vector<PartitionID> partition;
-  mt_kahypar::io::readPartitionFile(partition_file, partition);
+  mt_kahypar::io::readPartitionFile(partition_file, hypergraph.initialNumNodes(), partition);
   for ( const HypernodeID& hn : hypergraph.nodes() ) {
     if ( partition[hn] == kInvalidPartition ) {
       LOG << RED << "[ERROR]" << END << "Hypernode" << hn << "is not assigned to a block";

diff --git a/tools/verify_target_graph_partition.cc b/tools/verify_target_graph_partition.cc
@@ -102,7 +102,7 @@ int main(int argc, char* argv[]) {
 
   // Read Partition
   std::vector<PartitionID> partition;
-  io::readPartitionFile(context.partition.graph_partition_filename, partition);
+  io::readPartitionFile(context.partition.graph_partition_filename, hg.initialNumNodes(), partition);
   PartitionedHypergraph partitioned_hg(context.partition.k, hg, parallel_tag_t { });
   partitioned_hg.doParallelForAllNodes([&](const HypernodeID& hn) {
     partitioned_hg.setOnlyNodePart(hn, partition[hn]);