#1934: Reset ElementLBData containers after resetting the current phase

DARMA-tasking · Sep 19, 2024 · 509662f · 509662f
1 parent 4254d6d
commit 509662f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/vt/elm/elm_lb_data.cc b/src/vt/elm/elm_lb_data.cc
@@ -196,6 +196,11 @@ void ElementLBData::updatePhase(PhaseType const& inc) {
 
 void ElementLBData::resetPhase() {
   cur_phase_ = fst_lb_phase;
+  // Reset containers after resetting the current phase
+  phase_timings_.clear();
+  subphase_timings_.clear();
+  phase_comm_.clear();
+  subphase_comm_.clear();
 }
 
 PhaseType ElementLBData::getPhase() const {

diff --git a/tests/unit/collection/test_lb.extended.cc b/tests/unit/collection/test_lb.extended.cc
@@ -552,7 +552,8 @@ TEST_F(TestRestoreLBData, test_restore_lb_data_data_1) {
     );
   });
 
-  vt::vrt::collection::balance::LBDataHolder lbdh(num_phases);
+  vt::vrt::collection::balance::LBDataHolder lbdh;
+  lbdh.resizeHistory(num_phases);
   PhaseType write_phase = 0;
 
   using CommKey = vt::elm::CommKey;