#1838: lb: add getModeledWork to composed model

DARMA-tasking · Nov 8, 2022 · 49d552e · 49d552e
1 parent 44d898a
commit 49d552e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/model/composed_model.cc b/src/vt/vrt/collection/balance/model/composed_model.cc
@@ -64,6 +64,11 @@ ComposedModel::getModeledComm(ElementIDStruct object, PhaseOffset when) {
   return base_->getModeledComm(object, when);
 }
 
+TimeType
+ComposedModel::getModeledWork(ElementIDStruct object, PhaseOffset when) {
+  return base_->getModeledWork(object, when);
+}
+
 bool ComposedModel::hasRawLoad() const {
   return base_->hasRawLoad();
 }

diff --git a/src/vt/vrt/collection/balance/model/composed_model.h b/src/vt/vrt/collection/balance/model/composed_model.h
@@ -71,6 +71,7 @@ class ComposedModel : public LoadModel
 
   TimeType getModeledLoad(ElementIDStruct object, PhaseOffset when) const override;
   TimeType getModeledComm(ElementIDStruct object, PhaseOffset when) const override;
+  TimeType getModeledWork(ElementIDStruct object, PhaseOffset when) const override;
   bool hasRawLoad() const override;
   TimeType getRawLoad(ElementIDStruct object, PhaseOffset when) const override;
   unsigned int getNumPastPhasesNeeded(unsigned int look_back) const override;