https://github.com/JuliaParallel/MPI.jl
Tip revision: 69ce0bf5c15e4a29cb4a4c3df3e8d02f3e9ba0ad authored by Simon Byrne on 23 January 2023, 18:49:26 UTC
Tag v0.20.8 (#712)
Tag v0.20.8 (#712)
Tip revision: 69ce0bf
test_wait.jl
# tests for the various kinds of waits
include("common.jl")
MPI.Init()
myrank = MPI.Comm_rank(MPI.COMM_WORLD)
commsize = MPI.Comm_rank(MPI.COMM_WORLD)
nsends = 2
send_arr = [ArrayType{Int}([i]) for i = 1:nsends]
recv_arr = [ArrayType{Int}(undef,1) for i = 1:nsends]
synchronize()
# send to self
send_reqs = [MPI.Isend(send_arr[i], MPI.COMM_WORLD; dest=myrank, tag=i) for i = 1:nsends]
recv_reqs = [MPI.Irecv!(recv_arr[i], MPI.COMM_WORLD; source=myrank, tag=i) for i = 1:nsends]
send_check = zeros(Int, nsends)
recv_check = zeros(Int, nsends)
for i = 1:nsends
idx = MPI.Waitany(send_reqs)
@test MPI.isnull(send_reqs[idx])
send_check[idx] += 1
end
@test send_check == ones(Int, nsends)
for i = 1:nsends
idx = MPI.Waitany(recv_reqs)
@test MPI.isnull(recv_reqs[idx])
recv_check[idx] += 1
end
@test recv_check == ones(Int, nsends)
MPI.Barrier(MPI.COMM_WORLD)
MPI.Finalize()
@test MPI.Finalized()