add more granular callouts to PH

Author: bknueven

mpisppy/cylinders/hub.py

@@ -532,6 +532,26 @@ def sync(self):
     def sync_with_spokes(self):
         self.sync()
 
+    def sync_bounds(self):
+        if self.has_outerbound_spokes:
+            self.receive_outerbounds()
+        if self.has_innerbound_spokes:
+            self.receive_innerbounds()
+        if self.has_bounds_only_spokes:
+            self.send_boundsout()
+
+    def sync_extensions(self):
+        if self.opt.extensions is not None:
+            self.opt.extobject.sync_with_spokes()
+
+    def sync_nonants(self):
+        if self.has_nonant_spokes:
+            self.send_nonants()
+
+    def sync_Ws(self):
+        if self.has_w_spokes:
+            self.send_ws()
+
     def is_converged(self):
         if self.opt.best_bound_obj_val is not None:
             self.BestOuterBound = self.OuterBoundUpdate(self.opt.best_bound_obj_val)

mpisppy/phbase.py

@@ -936,7 +936,8 @@ def _vb(msg):
             self.best_bound_obj_val = self.trivial_bound
 
         if self.spcomm is not None:
-            self.spcomm.sync()
+            self.spcomm.sync_nonants()
+            self.spcomm.sync_extensions()
 
         if have_extensions:
             self.extobject.post_iter0_after_sync()
@@ -1022,6 +1023,9 @@ def iterk_loop(self):
             self.Update_W(verbose)
             #global_toc('Rank: {} - After Update_W'.format(self.cylinder_rank), True)
 
+            if self.spcomm is not None:
+                self.spcomm.sync_Ws()
+
             if smoothed:
                 self.Update_z(verbose)
 
@@ -1067,7 +1071,9 @@ def iterk_loop(self):
                 self.extobject.enditer()
 
             if self.spcomm is not None:
-                self.spcomm.sync()
+                self.spcomm.sync_nonants()
+                self.spcomm.sync_bounds()
+                self.spcomm.sync_extensions()
                 if self.spcomm.is_converged():
                     global_toc("Cylinder convergence", self.cylinder_rank == 0)
                     break