#1838: lb: add getModeledComm to composed model

DARMA-tasking · Aug 4, 2022 · 73418b2 · 73418b2
1 parent b758bfa
commit 73418b2
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
@@ -59,6 +59,11 @@ ComposedModel::getModeledLoad(ElementIDStruct object, PhaseOffset when) {
   return base_->getModeledLoad(object, when);
 }
 
+TimeType
+ComposedModel::getModeledComm(ElementIDStruct object, PhaseOffset when) {
+  return base_->getModeledComm(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
@@ -70,6 +70,7 @@ class ComposedModel : public LoadModel
   void updateLoads(PhaseType last_completed_phase) override;
 
   TimeType getModeledLoad(ElementIDStruct object, PhaseOffset when) override;
+  TimeType getModeledComm(ElementIDStruct object, PhaseOffset when) override;
   bool hasRawLoad() const override;
   TimeType getRawLoad(ElementIDStruct object, PhaseOffset when) override;
   unsigned int getNumPastPhasesNeeded(unsigned int look_back) override;