Skip to content

Commit 30d8d08

Browse files
committed
#2435: lb: add helper and sentinel to header file
1 parent 21b59b1 commit 30d8d08

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

src/vt/vrt/collection/balance/temperedlb/temperedlb.h

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,22 @@ struct TemperedLB : BaseLB {
214214
*/
215215
ClusterInfo makeClusterSummary(SharedIDType shared_id);
216216

217+
/**
218+
* \brief Helper to add edges to cluster summary
219+
*
220+
* \param[in] shared_id the shared ID
221+
* \param[in] info cluster info
222+
* \param[in] cluster_objs cluster objs
223+
* \param[in] obj the sending or receiving object
224+
* \param[in] is_send whether it's a send or recv edge
225+
* \param[in] iter edge iterator
226+
*/
227+
void makeClusterSummaryAddEdges(
228+
SharedIDType shared_id, ClusterInfo& info,
229+
std::set<ObjIDType> const& cluster_objs,
230+
ObjIDType obj, bool is_send, typename EdgeMapType::iterator iter
231+
);
232+
217233
/**
218234
* \brief Try to lock a rank
219235
*
@@ -529,6 +545,8 @@ struct TemperedLB : BaseLB {
529545
}
530546
};
531547

548+
/// Whether a cluster does not have a shared ID
549+
SharedIDType const no_shared_id = -1;
532550
/// Whether we have memory information
533551
bool has_memory_data_ = false;
534552
/// Working bytes for this rank

0 commit comments

Comments
 (0)