Skip to content

Commit 2976892

Browse files
committed
#2435: lb: add another helper
1 parent 11debd9 commit 2976892

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,14 @@ struct ClusterInfo {
7878
}
7979
}
8080

81+
void addIntraVolume(bool is_send, double volume) {
82+
if (is_send) {
83+
intra_send_vol += volume;
84+
} else {
85+
intra_recv_vol += volume;
86+
}
87+
}
88+
8189
void addObjEdge(bool is_send, elm::ElementIDStruct obj, double volume) {
8290
if (is_send) {
8391
obj_send_vol[obj] += volume;

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

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -670,13 +670,9 @@ void TemperedLB::makeClusterSummaryAddEdges(
670670

671671
if (cluster_objs.find(send_or_recv_obj) != cluster_objs.end()) {
672672
// intra-cluster edge
673-
if (is_send) {
674-
info.intra_send_vol += volume;
675-
} else {
676-
info.intra_recv_vol += volume;
677-
}
673+
info.addIntraVolume(is_send, volume);
678674

679-
// iter-cluster
675+
// inter-cluster edge
680676
info.addInterClusterEdge(is_send, shared_id, volume);
681677
} else if (
682678
auto it2 = obj_shared_block_.find(send_or_recv_obj);

0 commit comments

Comments
 (0)