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