#2171: Update LBDataHolder::toJson to correctly store user defined data

DARMA-tasking · Oct 24, 2023 · d97a5c9 · d97a5c9
1 parent a756ae5
commit d97a5c9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/lb_data_holder.cc b/src/vt/vrt/collection/balance/lb_data_holder.cc
@@ -218,6 +218,14 @@ std::unique_ptr<nlohmann::json> LBDataHolder::toJson(PhaseType phase) const {
     }
   }
 
+  if (user_per_phase_json_.find(phase) != user_per_phase_json_.end()) {
+    auto& user_def_this_phase = user_per_phase_json_.at(phase);
+
+    if (!user_def_this_phase->empty()) {
+      j["user_defined"] = *user_def_this_phase;
+    }
+  }
+
   return std::make_unique<json>(std::move(j));
 }