From 6cc5fd924fe2d02dcde40e6b2f142d1e1742695d Mon Sep 17 00:00:00 2001
From: Jonathan Lifflander <jliffla@sandia.gov>
Date: Thu, 19 Dec 2024 14:15:25 -0800
Subject: [PATCH] #2382: ccm-lb: fix bug

---
 src/vt/vrt/collection/balance/temperedlb/temperedlb.cc | 2 +-
 src/vt/vrt/collection/balance/temperedlb/temperedlb.h  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc b/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc
index 9a4b2e8eab..3eecfa2ba9 100644
--- a/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc
+++ b/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc
@@ -2722,7 +2722,7 @@ void TemperedLB::swapClusters() {
     auto remote_block_count = getRemoteBlockCountHere();
     runInEpochCollective("TemperedLB::swapClusters -> compute rejection", [=] {
       proxy_.allreduce<&TemperedLB::rejectionStatsHandler, collective::PlusOp>(
-        n_rejected, n_transfers_swap_, remote_block_count, cycle_locks_, iter_time_
+        n_rejected, n_transfers_swap_, remote_block_count, cycle_locks_
       );
       proxy_.allreduce<&TemperedLB::maxIterTime, collective::MaxOp>(iter_time_);
     });
diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.h b/src/vt/vrt/collection/balance/temperedlb/temperedlb.h
index 55bdf6ba84..ebf2614a43 100644
--- a/src/vt/vrt/collection/balance/temperedlb/temperedlb.h
+++ b/src/vt/vrt/collection/balance/temperedlb/temperedlb.h
@@ -127,7 +127,7 @@ struct TemperedLB : BaseLB {
   void loadStatsHandler(std::vector<balance::LoadData> const& vec);
   void workStatsHandler(std::vector<balance::LoadData> const& vec);
   void rejectionStatsHandler(
-    int n_rejected, int n_transfers, int n_unhomed_blocks
+    int n_rejected, int n_transfers, int n_unhomed_blocks, int cycle_count
   );
   void maxIterTime(double max_iter_time);
   void remoteBlockCountHandler(int n_unhomed_blocks);