Fixed broadcast error for spaces

bempp · Feb 6, 2025 · d9850f5 · d9850f5
1 parent 3ea90f0
commit d9850f5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/function.rs b/src/function.rs
@@ -459,11 +459,12 @@ where
         // on the last process
 
         let global_size = {
-            let mut tmp = 0;
+            let mut tmp = dof_n;
             if rank == comm.size() - 1 {
-                comm.this_process().scatter_into_root(&dof_n, &mut tmp);
+                comm.this_process().broadcast_into(&mut tmp);
             } else {
-                comm.process_at_rank(comm.size() - 1).scatter_into(&mut tmp);
+                comm.process_at_rank(comm.size() - 1)
+                    .broadcast_into(&mut tmp);
             }
 
             tmp