#2382: ccm-lb: fix iter time so it's not additive

DARMA-tasking · Dec 23, 2024 · 8e14f8d · 8e14f8d
1 parent 6cc5fd9
commit 8e14f8d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc b/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc
@@ -2147,6 +2147,7 @@ void TemperedLB::originalTransfer() {
   if (theConfig()->vt_debug_temperedlb) {
     // compute rejection rate because it will be printed
     runInEpochCollective("TemperedLB::originalTransfer -> compute rejection", [=] {
+      iter_time_ = MPI_Wtime() - iter_time_;
       proxy_.allreduce<&TemperedLB::rejectionStatsHandler, collective::PlusOp>(
         n_rejected, n_transfers, 0, 0
       );
@@ -2721,6 +2722,7 @@ void TemperedLB::swapClusters() {
     int n_rejected = 0;
     auto remote_block_count = getRemoteBlockCountHere();
     runInEpochCollective("TemperedLB::swapClusters -> compute rejection", [=] {
+      iter_time_ = MPI_Wtime() - iter_time_;
       proxy_.allreduce<&TemperedLB::rejectionStatsHandler, collective::PlusOp>(
         n_rejected, n_transfers_swap_, remote_block_count, cycle_locks_
       );