MueLu: Fix for phase 2b

Compute aggWeight locally instead of globally. Signed-off-by: Christian Glusa <caglusa@sandia.gov>
trilinos · Nov 12, 2024 · 90487aa · 90487aa
1 parent d7d7ee2
commit 90487aa
Show file tree

Hide file tree

Showing 4 changed files with 163 additions and 206 deletions.
diff --git a/packages/muelu/src/Graph/UncoupledAggregation/MueLu_AggregationPhase2bAlgorithm_decl.hpp b/packages/muelu/src/Graph/UncoupledAggregation/MueLu_AggregationPhase2bAlgorithm_decl.hpp
@@ -73,17 +73,13 @@ class AggregationPhase2bAlgorithm : public MueLu::AggregationAlgorithmBase<Local
                        typename AggregationAlgorithmBase<LocalOrdinal, GlobalOrdinal, Node>::AggStatType& aggStat,
                        LO& numNonAggregatedNodes) const;
 
-  void BuildAggregatesRandom(const ParameterList& params,
-                             const LWGraph_kokkos& graph,
-                             Aggregates& aggregates,
-                             typename AggregationAlgorithmBase<LocalOrdinal, GlobalOrdinal, Node>::AggStatType& aggStat,
-                             LO& numNonAggregatedNodes) const;
-
-  void BuildAggregatesDeterministic(const ParameterList& params,
-                                    const LWGraph_kokkos& graph,
-                                    Aggregates& aggregates,
-                                    typename AggregationAlgorithmBase<LocalOrdinal, GlobalOrdinal, Node>::AggStatType& aggStat,
-                                    LO& numNonAggregatedNodes) const;
+  template <bool deterministic>
+  void BuildAggregates(const ParameterList& params,
+                       const LWGraph_kokkos& graph,
+                       Aggregates& aggregates,
+                       typename AggregationAlgorithmBase<LocalOrdinal, GlobalOrdinal, Node>::AggStatType& aggStat,
+                       LO& numNonAggregatedNodes) const;
+
   //@}
 
   std::string description() const { return "Phase 2b (expansion)"; }