File tree Expand file tree Collapse file tree 1 file changed +18
-0
lines changed
src/vt/vrt/collection/balance/temperedlb Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments