exclude MPICH_jll

JuliaParallel · Dec 6, 2021 · d9bd9eb · d9bd9eb
1 parent d2b60e3
commit d9bd9eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/environment.jl b/src/environment.jl
@@ -286,8 +286,9 @@ or `false`.
 """
 function has_cuda()
     flag = get(ENV, "JULIA_MPI_HAS_CUDA", nothing)
-    if flag === nothing
-        # Only Open MPI provides a function to check CUDA support
+    if flag === nothing && binary != "MPICH_jll"
+        # Not every MPI implementation provides a function to check CUDA support
+        # It is broken on MPICH_jll: https://github.com/JuliaPackaging/Yggdrasil/issues/4039
         if dlsym_e(dlopen(libmpi), :MPIX_Query_cuda_support) != C_NULL
             return 0 != ccall((:MPIX_Query_cuda_support, libmpi), Cint, ())
         else