Merge pull request #2262 from DARMA-tasking/2261-add-settime-to-elmlbdata

2261 Add support code for simulating load from within an application

Author: nlslatt

src/vt/elm/elm_lb_data.cc

@@ -133,11 +133,27 @@ void ElementLBData::addTime(LoadType const timeLoad) {
   vt_debug_print(
     verbose,lb,
     "ElementLBData: addTime: time={}, cur_load={}\n",
-    time,
+    timeLoad,
     phase_timings_[cur_phase_]
   );
 }
 
+void ElementLBData::setTime(
+  LoadType const timeLoad,
+  std::vector<LoadType> const& subphaseLoads
+) {
+  // warning: this will override any existing time that might be there
+  phase_timings_[cur_phase_] = timeLoad;
+
+  subphase_timings_[cur_phase_] = subphaseLoads;
+
+  vt_debug_print(
+    verbose,lb,
+    "ElementLBData: setTime: time={}\n",
+    timeLoad
+  );
+}
+
 void ElementLBData::setPhase(PhaseType const& new_phase) {
   cur_phase_ = new_phase;
 

src/vt/elm/elm_lb_data.h

@@ -64,6 +64,10 @@ struct ElementLBData {
   void start(TimeType time);
   void stop(TimeType time);
   void addTime(LoadType const timeLoad);
+  /// warning: this will override any existing time that might be there
+  void setTime(
+    LoadType const timeLoad, std::vector<LoadType> const& subphaseLoads
+  );
 
   void sendToEntity(ElementIDStruct to, ElementIDStruct from, double bytes);
   void sendComm(elm::CommKey key, double bytes);