#991: Norm Model: Fix to actually loop over subphases, rather than re…

…peatedly over WHOLE_PHASE
DARMA-tasking · Aug 31, 2020 · 20c9288 · 20c9288
1 parent 0e91408
commit 20c9288
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/model/norm.cc b/src/vt/vrt/collection/balance/model/norm.cc
@@ -65,6 +65,7 @@ TimeType Norm::getWork(ElementIDType object, PhaseOffset offset)
     double sum = 0.0;
 
     for (int i = 0; i < getNumSubphases(); ++i) {
+      offset.subphase = i;
       auto t = ComposedModel::getWork(object, offset);
       sum += std::pow(t, power_);
     }
@@ -75,6 +76,7 @@ TimeType Norm::getWork(ElementIDType object, PhaseOffset offset)
     double max = 0.0;
 
     for (int i = 0; i < getNumSubphases(); ++i) {
+      offset.subphase = i;
       auto t = ComposedModel::getWork(object, offset);
       max = std::max(max, t);
     }