Use allreduce instead of allgather

marchirschvogel · Oct 3, 2024 · c906216 · c906216
1 parent 8995369
commit c906216
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/ambit_fe/mor/mor_main.py b/src/ambit_fe/mor/mor_main.py
@@ -151,10 +151,7 @@ def prepare_rob(self):
         self.partition_pod_space()
 
         # gather Phi
-        phi_arr = self.pb.comm.allgather(self.Phi)
-        self.Phi = np.zeros(self.Phi.shape)
-        for i in range(len(phi_arr)):
-            self.Phi += phi_arr[i]
+        self.Phi = self.pb.comm.allreduce(self.Phi)
 
         if self.orthogonalize_rom_basis:
             ts = time.time()