add cli option ofr kN with spanning tree

KaHIP · Jun 3, 2024 · d6e411b · d6e411b
1 parent f0a9582
commit d6e411b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/kaminpar-shm/context_io.cc b/kaminpar-shm/context_io.cc
@@ -78,7 +78,9 @@ std::unordered_map<std::string, SparsificationAlgorithm> get_sparsification_algo
       {"random", SparsificationAlgorithm::UNIFORM_RANDOM_SAMPLING},
       {"rn", SparsificationAlgorithm::UNIFORM_RANDOM_SAMPLING},
       {"k-neighbour", SparsificationAlgorithm::K_NEIGHBOUR},
-      {"kn", SparsificationAlgorithm::K_NEIGHBOUR}
+      {"kn", SparsificationAlgorithm::K_NEIGHBOUR},
+      {"k-neighbour-spanning-tree", SparsificationAlgorithm::K_NEIGHBOUR_SPANNING_TREE},
+      {"kn-st", SparsificationAlgorithm::K_NEIGHBOUR_SPANNING_TREE}
   };
 }
 

diff --git a/kaminpar-shm/factories.cc b/kaminpar-shm/factories.cc
@@ -100,6 +100,8 @@ std::unique_ptr<sparsification::Sampler> create_sampler(const Context &ctx) {
     return std::make_unique<sparsification::UniformRandomSampler>();
   case SparsificationAlgorithm::K_NEIGHBOUR:
     return std::make_unique<sparsification::kNeighbourSampler>();
+  case SparsificationAlgorithm::K_NEIGHBOUR_SPANNING_TREE:
+    return std::make_unique<sparsification::kNeighbourSampler>(true);
   }
 
   __builtin_unreachable();