diff --git a/src/vt/vrt/collection/balance/lb_data_holder.cc b/src/vt/vrt/collection/balance/lb_data_holder.cc index 3f0da67f2a..67361209ea 100644 --- 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 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(std::move(j)); }