Skip to content

Commit 0b5ee2e

Browse files
committed
#1934: Cleanup of NodeLBData interface
1 parent 9962e6a commit 0b5ee2e

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

src/vt/vrt/collection/balance/lb_invoke/lb_manager.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,7 @@ void LBManager::finishedLB(PhaseType phase) {
469469
"finishedLB\n"
470470
);
471471

472-
theNodeLBData()->startIterCleanup(phase, min_hist_lb_data_);
472+
theNodeLBData()->startIterCleanup(phase);
473473
theNodeLBData()->outputLBDataForPhase(phase);
474474

475475
destroyLB();

src/vt/vrt/collection/balance/node_lb_data.cc

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -116,11 +116,11 @@ void NodeLBData::clearLBData() {
116116
next_elm_ = 1;
117117
}
118118

119-
void NodeLBData::startIterCleanup(PhaseType phase, unsigned int look_back) {
120-
if (phase >= look_back) {
121-
lb_data_->node_data_.erase(phase - look_back);
122-
lb_data_->node_comm_.erase(phase - look_back);
123-
lb_data_->node_subphase_comm_.erase(phase - look_back);
119+
void NodeLBData::startIterCleanup(PhaseType phase) {
120+
if (phase >= min_hist_lb_data_) {
121+
lb_data_->node_data_.erase(phase - min_hist_lb_data_);
122+
lb_data_->node_comm_.erase(phase - min_hist_lb_data_);
123+
lb_data_->node_subphase_comm_.erase(phase - min_hist_lb_data_);
124124
}
125125

126126
// Clear migrate lambdas and proxy lookup since LB is complete

src/vt/vrt/collection/balance/node_lb_data.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ struct NodeLBData : runtime::component::Component<NodeLBData> {
146146
/**
147147
* \internal \brief Cleanup after LB runs
148148
*/
149-
void startIterCleanup(PhaseType phase, unsigned int look_back);
149+
void startIterCleanup(PhaseType phase);
150150

151151
/**
152152
* \internal \brief Load and broadcast the LB specification file

0 commit comments

Comments
 (0)