Fixing MPI tests

GalerkinToolkit · Feb 21, 2024 · 072307e · 072307e
1 parent 4724d2e
commit 072307e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/extensions/GalerkinToolkitExamples/test/runtests.jl b/extensions/GalerkinToolkitExamples/test/runtests.jl
@@ -7,6 +7,6 @@ using Test
     @testset "example002" begin include("example002_tests.jl") end
 end
 
-@test_broken include(joinpath("mpi_array","runtests.jl"))
+include(joinpath("mpi_array","runtests.jl"))
 
 end # module
diff --git a/src/geometry.jl b/src/geometry.jl
@@ -3382,8 +3382,11 @@ function scatter_mesh(pmeshes_on_main;source=MAIN)
     rcv = scatter(snd;source)
     mesh_partition, node_partition, face_partition_array = rcv |> tuple_of_arrays
     face_partition = face_partition_array |> tuple_of_arrays
-    snd2 = map_main(partition_strategy,pmeshes_on_main;main=source)
-    rcv2 = multicast(snd2)
+    np = length(snd)
+    snd2 = map_main(pmeshes_on_main;main=source) do pmesh
+        fill(partition_strategy(pmesh),np)
+    end
+    rcv2 = scatter(snd2)
     partition_strategy_mesh = PartitionedArrays.getany(rcv2)
     PMesh(mesh_partition,node_partition,face_partition,partition_strategy_mesh)
 end