diff --git a/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.cc b/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.cc index e092aaa4e7..e414b3dada 100644 --- a/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.cc +++ b/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.cc @@ -336,13 +336,11 @@ void ZoltanLB::countEdges() { total_ids ); - auto cb = theCB()->makeBcast<&ZoltanLB::reduceCount>(proxy); - auto msg = makeMessage(total_ids); - proxy.reduce>(msg.get(),cb); + proxy.allreduce<&ZoltanLB::reduceCount, collective::MaxOp>(total_ids); } -void ZoltanLB::reduceCount(ReduceMsg* msg) { - max_edges_per_node_ = msg->getVal(); +void ZoltanLB::reduceCount(int max_edges_per_node) { + max_edges_per_node_ = max_edges_per_node; vt_debug_print( normal, lb, diff --git a/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.h b/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.h index cf14270370..40f49ab497 100644 --- a/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.h +++ b/src/vt/vrt/collection/balance/zoltanlb/zoltanlb.h @@ -99,7 +99,7 @@ struct ZoltanLB : BaseLB { void combineEdges(); void countEdges(); void allocateEdges(); - void reduceCount(ReduceMsg* msg); + void reduceCount(int max_edges_per_node); void allocateShareEdgeGIDs(); void recvSharedEdges(CommMsg* msg);