Skip to content

Commit

Permalink
Merge pull request #2259 from DARMA-tasking/2258-time-lb-and-mig
Browse files Browse the repository at this point in the history
#2258: lb: add timers for lb proper and migration
  • Loading branch information
nlslatt authored Mar 26, 2024
2 parents 0f18b6d + 8189160 commit 26f702c
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions src/vt/vrt/collection/balance/lb_invoke/lb_manager.cc
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,14 @@ void LBManager::defaultPostLBWork(ReassignmentMsg* msg) {
commitPhaseStatistics(phase);
}

auto const start_time = timing::getCurrentTime();
applyReassignment(reassignment);
auto const mig_time = timing::getCurrentTime() - start_time;
vt_debug_print(
terse, phase,
"phase={}: mig_time={}\n",
phase, mig_time
);

// Inform the collection manager to rebuild spanning trees if needed
if (reassignment->global_migration_count != 0) {
Expand Down Expand Up @@ -258,10 +265,17 @@ LBManager::runLB(PhaseType phase, vt::Callback<ReassignmentMsg> cb) {

vt_debug_print(terse, lb, "LBManager: running strategy\n");

auto const start_time = timing::getCurrentTime();
auto reassignment = strat->startLB(
phase, base_proxy, model_.get(), stats, *comm, total_load_from_model,
*data_map
);
auto const lb_time = timing::getCurrentTime() - start_time;
vt_debug_print(
terse, phase,
"phase={}: lb_time={}\n",
phase, lb_time
);
cb.send(reassignment, phase);
}

Expand Down

0 comments on commit 26f702c

Please sign in to comment.