Update MPI init

PTsolvers · Sep 14, 2023 · 330f43a · 330f43a
1 parent d8ac8ff
commit 330f43a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts_future_API/mpi_utils2.jl b/scripts_future_API/mpi_utils2.jl
@@ -1,6 +1,6 @@
 # MPI
 function init_distributed(dims::Tuple=(0, 0, 0); init_MPI=true)
-    init_MPI && MPI.Init()
+    init_MPI && (Threads.nthreads() > 1 ? MPI.Init(threadlevel=:multiple) : MPI.Init())
     nprocs = MPI.Comm_size(MPI.COMM_WORLD)
     dims = Tuple(MPI.Dims_create(nprocs, dims))
     # create MPI communicator