#1672: lb: use alpha and gamma when calculating load

DARMA-tasking · Mar 11, 2022 · 3c0408e · 3c0408e
1 parent c4c71ab
commit 3c0408e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc b/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc
@@ -515,9 +515,10 @@ void TemperedLB::doLBStages(TimeType start_imb) {
         cur_objs_.clear();
         for (auto obj : *load_model_) {
           if (obj.isMigratable()) {
-            cur_objs_[obj] = load_model_->getWork(
+            // TODO: `beta_ * communication` component is still missing here
+            cur_objs_[obj] = alpha_ * load_model_->getWork(
               obj, {balance::PhaseOffset::NEXT_PHASE, balance::PhaseOffset::WHOLE_PHASE}
-            );
+            ) + gamma_;
           }
         }
         this_new_load_ = this_load;