Skip to content

Commit

Permalink
DO NOT MERGE debug print statements
Browse files Browse the repository at this point in the history
Signed-off-by: Christian Glusa <[email protected]>
  • Loading branch information
cgcgcg committed Nov 12, 2024
1 parent 62febbb commit 699119a
Showing 1 changed file with 5 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -266,6 +266,8 @@ void AggregationPhase2bAlgorithm<LocalOrdinal, GlobalOrdinal, Node>::
// which will then be subtracted from numNonAggregatedNodes
LO numAggregated = 0;

std::cout << "pre " << iter << " " << color << std::endl;

if constexpr (deterministic) {
if (matchMLbehavior) {
auto functor = ExpansionFunctor<decltype(aggStat), decltype(procWinner), decltype(vertex2AggId), decltype(colors), decltype(graph), decltype(aggPenalties), LO, true, true>(aggStat, procWinner, vertex2AggId, colors, graph, aggPenalties, aggPenaltyUpdates, connectWeight, penaltyConnectWeight, color, myRank);
Expand Down Expand Up @@ -300,6 +302,9 @@ void AggregationPhase2bAlgorithm<LocalOrdinal, GlobalOrdinal, Node>::
}
}

Kokkos::fence();
std::cout << "post " << iter << " " << color << std::endl;

if constexpr (deterministic) {
Kokkos::parallel_for(
"Aggregation Phase 2b: updating agg penalties",
Expand Down

0 comments on commit 699119a

Please sign in to comment.