44 #if !defined INCLUDED_VT_VRT_COLLECTION_BALANCE_LB_DATA_HOLDER_H 45 #define INCLUDED_VT_VRT_COLLECTION_BALANCE_LB_DATA_HOLDER_H 51 #include <unordered_map> 56 #include <nlohmann/json_fwd.hpp> 58 namespace vt {
namespace vrt {
namespace collection {
namespace balance {
76 template <
typename SerializerT>
142 nlohmann::json
const& field, nlohmann::json&
object,
143 bool& is_bitpacked,
bool& is_collection);
172 std::unordered_map<
PhaseType, std::unordered_map<
182 std::unordered_map<ElementIDStruct, std::tuple<VirtualProxyType, std::vector<uint64_t>>>
node_idx_;
std::unique_ptr< nlohmann::json > metadataToJson() const
Output a LB phase's metadata to JSON.
Definition: lb_data_holder.cc:112
+
std::unordered_map< PhaseType, std::unordered_map< SubphaseType, CommMapType > > node_subphase_comm_
Node communication graph for each subphase.
Definition: lb_data_holder.h:170
+
std::unordered_map< PhaseType, std::unordered_map< ElementIDStruct, std::shared_ptr< nlohmann::json > > > user_defined_json_
User-defined data from each phase for JSON output.
Definition: lb_data_holder.h:174
void serialize(SerializerT &s)
Definition: lb_data_holder.h:77
-
std::unordered_map< ElementIDStruct, ObjGroupProxyType > node_objgroup_
Map from id to objgroup proxy.
Definition: lb_data_holder.h:159
-
std::unordered_map< PhaseType, CommMapType > node_comm_
Node communication graph for each local object.
Definition: lb_data_holder.h:143
-
void outputEntity(nlohmann::json &j, ElementIDStruct const &elm_id) const
Output an entity to json.
Definition: lb_data_holder.cc:52
+
std::unordered_map< ElementIDStruct, ObjGroupProxyType > node_objgroup_
Map from id to objgroup proxy.
Definition: lb_data_holder.h:184
+
std::unordered_map< PhaseType, CommMapType > node_comm_
Node communication graph for each local object.
Definition: lb_data_holder.h:168
+
void outputEntity(nlohmann::json &j, ElementIDStruct const &elm_id) const
Output an entity to json.
Definition: lb_data_holder.cc:92
Definition: activefn.h:51
-
std::unordered_map< PhaseType, DataMapType > user_defined_lb_info_
User-defined data from each phase for LB.
Definition: lb_data_holder.h:153
+
std::unordered_map< PhaseType, DataMapType > user_defined_lb_info_
User-defined data from each phase for LB.
Definition: lb_data_holder.h:178
Definition: debug_config.h:80
-
std::unordered_map< PhaseType, LoadMapType > node_data_
Node timings for each local object.
Definition: lb_data_holder.h:141
+
std::unordered_map< PhaseType, LoadMapType > node_data_
Node timings for each local object.
Definition: lb_data_holder.h:166
-
ElmUserDataType rank_attributes_
Node attributes for the current rank.
Definition: lb_data_holder.h:139
+
ElmUserDataType rank_attributes_
Node attributes for the current rank.
Definition: lb_data_holder.h:164
Data structure that holds LB data for a set of phases. Can output them as JSON.
Definition: lb_data_holder.h:66
-
std::unordered_map< PhaseType, std::shared_ptr< nlohmann::json > > user_per_phase_json_
Definition: lb_data_holder.h:151
+
std::unordered_map< PhaseType, std::shared_ptr< nlohmann::json > > user_per_phase_json_
Definition: lb_data_holder.h:176
Definition: debug_config.h:73
-
std::set< PhaseType > identical_phases_
Definition: lb_data_holder.h:163
-
void readMetadata(nlohmann::json const &j)
Read the LB phase's metadata.
Definition: lb_data_holder.cc:471
+
std::set< PhaseType > identical_phases_
Definition: lb_data_holder.h:188
+
PhysicalResourceType NodeType
Used to hold the current node/rank or the number of nodes.
Definition: types_type.h:57
+
void readMetadata(nlohmann::json const &j)
Read the LB phase's metadata.
Definition: lb_data_holder.cc:496
std::unordered_map< std::string, UserDataValueType > ElmUserDataType
Definition: lb_common.h:117
-
std::set< PhaseType > skipped_phases_
Definition: lb_data_holder.h:161
-
void clear()
Clear all LB data.
Definition: lb_data_holder.cc:524
+
ElementIDStruct getElmFromCommObject_(nlohmann::json const &field) const
Create an ElementIDStruct for the communication object.
Definition: lb_data_holder.cc:71
+
std::set< PhaseType > skipped_phases_
Definition: lb_data_holder.h:186
+
void clear()
Clear all LB data.
Definition: lb_data_holder.cc:549
A general identifier for a task context. The id is unique in the system.
Definition: elm_id.h:64
-
std::unordered_map< PhaseType, DataMapType > node_user_attributes_
User-defined attributes from each phase.
Definition: lb_data_holder.h:155
-
std::unique_ptr< nlohmann::json > toJson(PhaseType phase) const
Output a phase's LB data to JSON.
Definition: lb_data_holder.cc:157
+
NodeType this_node_
The current node.
Definition: lb_data_holder.h:162
+
std::unordered_map< PhaseType, DataMapType > node_user_attributes_
User-defined attributes from each phase.
Definition: lb_data_holder.h:180
+
std::unique_ptr< nlohmann::json > toJson(PhaseType phase) const
Output a phase's LB data to JSON.
Definition: lb_data_holder.cc:197
uint64_t PhaseType
Used for hold a phase for load balancing.
Definition: types_type.h:103
-
void addInitialTask(nlohmann::json &j, std::size_t n) const
Definition: lb_data_holder.cc:148
+
void addInitialTask(nlohmann::json &j, std::size_t n) const
Definition: lb_data_holder.cc:188
elm::ElementIDStruct ElementIDStruct
Definition: lb_common.h:63
-
std::unordered_map< ElementIDStruct, std::tuple< VirtualProxyType, std::vector< uint64_t > > > node_idx_
Node indices for each ID along with the proxy ID.
Definition: lb_data_holder.h:157
-
std::unique_ptr< nlohmann::json > rankAttributesToJson() const
Output a LB rank attributes metadata to JSON.
Definition: lb_data_holder.cc:128
+
std::unordered_map< ElementIDStruct, std::tuple< VirtualProxyType, std::vector< uint64_t > > > node_idx_
Node indices for each ID along with the proxy ID.
Definition: lb_data_holder.h:182
+
static void getObjectFromJsonField_(nlohmann::json const &field, nlohmann::json &object, bool &is_bitpacked, bool &is_collection)
Determine the object ID from the tasks or communication field of input JSON.
Definition: lb_data_holder.cc:52
+
std::unique_ptr< nlohmann::json > rankAttributesToJson() const
Output a LB rank attributes metadata to JSON.
Definition: lb_data_holder.cc:168