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_test.jl
include("common.jl")
MPI.Init()
comm = MPI.COMM_WORLD
size = MPI.Comm_size(comm)
rank = MPI.Comm_rank(comm)
dst = mod(rank+1, size)
src = mod(rank-1, size)
N = 32
send_mesg = ArrayType{Float64}(undef,N)
recv_mesg = ArrayType{Float64}(undef,N)
recv_mesg_expected = ArrayType{Float64}(undef,N)
fill!(send_mesg, Float64(rank))
fill!(recv_mesg_expected, Float64(src))
synchronize()
rreq = MPI.Irecv!(recv_mesg, comm; source=src, tag=src+32)
sreq = MPI.Isend(send_mesg, comm; dest=dst, tag=rank+32, )
reqs = [sreq,rreq]
inds = MPI.Waitsome(reqs)
@test !isempty(inds)
for ind in inds
(onedone,stat) = MPI.Test(reqs[ind], MPI.Status)
@test onedone
@test MPI.Get_tag(stat) == MPI.ANY_TAG
@test MPI.Get_source(stat) == MPI.ANY_SOURCE
@test MPI.Get_error(stat) == MPI.SUCCESS
@test MPI.Get_count(stat, Float64) == 0
end
done, ind = MPI.Testany(reqs)
if done && !isnothing(ind)
(onedone,stat) = MPI.Test(reqs[ind], MPI.Status)
@test onedone
@test MPI.Get_tag(stat) == MPI.ANY_TAG
@test MPI.Get_source(stat) == MPI.ANY_SOURCE
@test MPI.Get_error(stat) == MPI.SUCCESS
@test MPI.Get_count(stat, Float64) == 0
end
MPI.Waitall(reqs)
(inds, stats) = MPI.Waitsome(reqs, MPI.Status)
@test isnothing(inds)
@test isempty(stats)
(inds, stats) = MPI.Testsome(reqs, MPI.Status)
@test isnothing(inds)
@test isempty(stats)
reqs = MPI.MultiRequest(2)
MPI.Irecv!(recv_mesg, comm, reqs[1]; source=src, tag=src+32)
@test !MPI.Testall(reqs)
MPI.Barrier(comm)
MPI.Isend(send_mesg, comm, reqs[2]; dest=dst, tag=rank+32, )
inds = MPI.Waitsome(reqs)
@test !isempty(inds)
MPI.Waitall(reqs)
@test MPI.Testall(reqs)
reqs = MPI.UnsafeMultiRequest(2)
GC.@preserve send_mesg recv_mesg begin
MPI.Irecv!(recv_mesg, comm, reqs[1]; source=src, tag=src+32)
@test !MPI.Testall(reqs)
MPI.Barrier(comm)
MPI.Isend(send_mesg, comm, reqs[2]; dest=dst, tag=rank+32, )
inds = MPI.Waitsome(reqs)
@test !isempty(inds)
MPI.Waitall(reqs)
@test MPI.Testall(reqs)
end
MPI.Finalize()
@test MPI.Finalized()