Fix potential deadlock

PTsolvers · Sep 14, 2023 · 85883d0 · 85883d0
1 parent 66f102f
commit 85883d0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/scripts_future_API/mpi_utils2.jl b/scripts_future_API/mpi_utils2.jl
@@ -63,10 +63,14 @@ mutable struct Exchanger
                         copyto!(send_buf, border)
                         KernelAbstractions.synchronize(backend)
                         send = MPI.Isend(send_buf, comm; dest=rank)
-                        wait(recv)
-                        copyto!(halo, recv_buf)
+                        while true
+                            test_recv = MPI.Test(recv)
+                            test_send = MPI.Test(send)
+                            if test_recv copyto!(halo, recv_buf) end
+                            if test_recv && test_send break end
+                            yield()
+                        end
                         KernelAbstractions.synchronize(backend)
-                        wait(send)
                     end
                     notify(bottom)
                 end