add queries test

Author: vchuravy

test/integration/MPI/queries.jl

@@ -0,0 +1,15 @@
+using MPI
+using Enzyme
+using Test
+
+if !MPI.Initialized()
+    MPI.Init()
+end
+
+comm = MPI.COMM_WORLD
+
+@test autodiff(ForwardWithPrimal, MPI.Comm_size, Const(comm)) == (MPI.Comm_size(comm),)
+@test autodiff(ForwardWithPrimal, MPI.Comm_rank, Const(comm)) == (MPI.Comm_rank(comm),)
+
+@test autodiff(ReverseWithPrimal, MPI.Comm_size, Const(comm)) == ((nothing,), MPI.Comm_size(comm))
+@test autodiff(ReverseWithPrimal, MPI.Comm_rank, Const(comm)) == ((nothing,), MPI.Comm_rank(comm))

test/integration/MPI/runtests.jl

@@ -12,6 +12,11 @@ using Test
 # - MPI_Gather/MPI_Scatter
 # - MPI_Allgather
 
+# Query functions MPI_Comm_size/MPI_Comm_rank
+@testset "queries" for np in (1, 2, 4)
+    run(`$(mpiexec()) -n $np $(Base.julia_cmd()) --project=$(@__DIR__) $(joinpath(@__DIR__, "queries.jl"))`)
+end
+
 # Test MPI_Recv/MPI_Send with a blocking ring communication pattern
 @testset "blocking_ring" for np in (1, 2, 4)
     run(`$(mpiexec()) -n $np $(Base.julia_cmd()) --project=$(@__DIR__) $(joinpath(@__DIR__, "blocking_ring.jl"))`)