#2171: Update LBDataHolder's constructor to pupulate user_per_phase_j…

…son_
DARMA-tasking · Oct 24, 2023 · d54ee9b · d54ee9b
1 parent 6b7d21a
commit d54ee9b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/vt/vrt/collection/balance/lb_data_holder.cc b/src/vt/vrt/collection/balance/lb_data_holder.cc
@@ -370,6 +370,12 @@ LBDataHolder::LBDataHolder(nlohmann::json const& j)
           }
         }
       }
+
+      if (phase.find("user_defined") != phase.end()) {
+        auto userDefined = phase["user_defined"];
+        user_per_phase_json_[phase] = std::make_shared<nlohmann::json>();
+        *(user_per_phase_json_[phase]) = userDefined;
+      }
     }
   }
 

diff --git a/tests/unit/collection/test_lb.extended.cc b/tests/unit/collection/test_lb.extended.cc
@@ -51,7 +51,6 @@
 #include "data_message.h"
 
 #include "vt/phase/phase_manager.h"
-#include "vt/timing/timing_type.h"
 #include "vt/vrt/collection/manager.h"
 #include "vt/vrt/collection/balance/lb_data_holder.h"
 #include "vt/vrt/collection/balance/node_lb_data.h"