#1838: lb: add getModeledWork to composed model

Add `getModeledWork` to `ComposedModel` class. Return object's load as modeled work when there is no implementation in derived classes.
DARMA-tasking · Aug 4, 2022 · fe0e0a0 · fe0e0a0
1 parent 51dfe80
commit fe0e0a0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
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) override;
   TimeType getModeledComm(ElementIDStruct object, PhaseOffset when) override;
+  TimeType getModeledWork(ElementIDStruct object, PhaseOffset when) override;
   bool hasRawLoad() const override;
   TimeType getRawLoad(ElementIDStruct object, PhaseOffset when) override;
   unsigned int getNumPastPhasesNeeded(unsigned int look_back) override;

diff --git a/src/vt/vrt/collection/balance/model/load_model.h b/src/vt/vrt/collection/balance/model/load_model.h
@@ -276,7 +276,7 @@ struct LoadModel
    * this.
    */
   virtual TimeType getModeledWork(ElementIDStruct object, PhaseOffset when) {
-    return {};
+    return getModeledLoad(object, when);
   }
 
   /**
-Original file line number
+Diff line change
@@ Expand Up / @@ -276,7 +276,7 @@ struct LoadModel @@
        * this.
        */
       virtual TimeType getModeledWork(ElementIDStruct object, PhaseOffset when) {
-        return {};
+        return getModeledLoad(object, when);
       }
       /**
@@ Expand Down @@