#276: lb: update zoltanlb for new reduce interface

DARMA-tasking · Jun 13, 2023 · c7da00b · c7da00b
1 parent 59b93d6
commit c7da00b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git 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<ReduceMsg>(total_ids);
-  proxy.reduce<collective::MaxOp<int>>(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
@@ -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);