Merge c634dd7 into 157a109

OpenMDAO · Jan 16, 2024 · 50ce9c1 · 50ce9c1
2 parents 157a109 + c634dd7
commit 50ce9c1
Show file tree

Hide file tree

Showing 4 changed files with 126 additions and 97 deletions.
diff --git a/openmdao/core/group.py b/openmdao/core/group.py
@@ -5106,11 +5106,11 @@ def _setup_iteration_lists(self):
             graph.add_edge(opt_sys, response0)
             graph.add_edge(response0, opt_sys)
 
-        if grad_groups:
-            if self.comm.size > 1:
-                grad_groups = self.comm.allgather(grad_groups)
-                grad_groups = {g for grp in grad_groups for g in grp}
+        if self.comm.size > 1:
+            grad_groups = self.comm.allgather(grad_groups)
+            grad_groups = {g for grp in grad_groups for g in grp}
 
+        if grad_groups:
             remaining = set(grad_groups)
             for name in sorted(grad_groups, key=lambda x: x.count('.')):
                 prefix = name + '.'