Test for has_gpu

JuliaParallel · Jun 22, 2024 · 25c361a · 25c361a
1 parent 98a5c10
commit 25c361a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test/test_basic.jl b/test/test_basic.jl
@@ -8,16 +8,22 @@ MPI.Init()
 
 @test MPI.has_cuda() isa Bool
 
-if get(ENV,"JULIA_MPI_TEST_ARRAYTYPE","") == "CuArray"
+if get(ENV, "JULIA_MPI_TEST_ARRAYTYPE", "") == "CuArray"
     @test MPI.has_cuda()
 end
 
 @test MPI.has_rocm() isa Bool
 
-if get(ENV,"JULIA_MPI_TEST_ARRAYTYPE","") == "ROCArray"
+if get(ENV, "JULIA_MPI_TEST_ARRAYTYPE", "") == "ROCArray"
     @test MPI.has_rocm()
 end
 
+@test MPI.has_gpu() isa Bool
+
+if get(ENV, "JULIA_MPI_TEST_ARRAYTYPE", "") == "CuArray" || get(ENV, "JULIA_MPI_TEST_ARRAYTYPE", "") == "ROCArray"
+    @test MPI.has_gpu()
+end
+
 @test !MPI.Finalized()
 MPI.Finalize()
 @test MPI.Finalized()