From 5a493061bce0a230f753acd0a6f473dfe4c666c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cezary=20Skrzy=C5=84ski?=
 <cezary.skrzynski@ng-analytics.com>
Date: Thu, 4 Aug 2022 14:44:38 +0200
Subject: [PATCH] #1838: lb: add `getModeledWork` to composed model

---
 src/vt/vrt/collection/balance/model/composed_model.cc | 5 +++++
 src/vt/vrt/collection/balance/model/composed_model.h  | 1 +
 2 files changed, 6 insertions(+)

diff --git a/src/vt/vrt/collection/balance/model/composed_model.cc b/src/vt/vrt/collection/balance/model/composed_model.cc
index f4967337df..0f7dd0b58b 100644
--- 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
index e145c2b14d..7a3f878fa2 100644
--- 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;