Merge pull request #136 from cwpearson/cmake/mpi-vendor

CMakeLists.txt

@@ -29,6 +29,8 @@ project(
   HOMEPAGE_URL "https://kokkos.org/kokkos-comm/"
 )
 
+include(cmake/mpi-vendor.cmake)
+
 option(KokkosComm_ENABLE_PERFTESTS "Build KokkosComm perf tests" OFF)
 option(KokkosComm_ENABLE_TESTS "Build KokkosComm tests" OFF)
 option(KokkosComm_ENABLE_MPI "Build KokkosComm with MPI transport" ON)
@@ -42,6 +44,7 @@ find_package(Kokkos REQUIRED)
 if(KOKKOSCOMM_ENABLE_MPI)
   find_package(MPI REQUIRED)
 endif()
+kokkoscomm_set_mpi_vendor_variables()
 
 add_subdirectory(src)
 if(KOKKOSCOMM_ENABLE_TESTS)

cmake/KokkosComm_config.hpp.in

@@ -21,3 +21,5 @@
 #define KOKKOSCOMM_VERSION_PATCH @KOKKOSCOMM_VERSION_PATCH@
 
 #cmakedefine KOKKOSCOMM_ENABLE_MPI
+#cmakedefine KOKKOSCOMM_IMPL_MPI_IS_MPICH
+#cmakedefine KOKKOSCOMM_IMPL_MPI_IS_OPENMPI

cmake/mpi-vendor.cmake

@@ -0,0 +1,29 @@
+function(kokkoscomm_set_mpi_vendor_variables)
+    # Initialize the variables to false
+    set(KOKKOSCOMM_IMPL_MPI_IS_MPICH FALSE CACHE BOOL "Is MPI MPICH")
+    set(KOKKOSCOMM_IMPL_MPI_IS_OPENMPI FALSE CACHE BOOL "Is MPI OpenMPI")
+
+    if(KOKKOSCOMM_ENABLE_MPI)
+        if(NOT MPIEXEC_EXECUTABLE)
+            message(WARNING "MPIEXEC_EXECUTABLE is not set - unable to determine MPI vendor")
+            return()
+        endif()
+
+        # Get the directory of the MPI executable
+        get_filename_component(MPIEXEC_DIR ${MPIEXEC_EXECUTABLE} DIRECTORY)
+
+        # Check for mpichversion
+        find_program(MPICHVERSION_EXECUTABLE mpichversion HINTS ${MPIEXEC_DIR})
+        find_program(OMPI_INFO_EXECUTABLE ompi_info HINTS ${MPIEXEC_DIR})
+
+        if (MPICHVERSION_EXECUTABLE AND OPENMPI_INFO_EXECUTABLE)
+            message(WARNING "Both MPICHVERSION_EXECUTABLE and OMPI_INFO_EXECUTABLE are set - unable to determine MPI vendor")
+        elseif(MPICHVERSION_EXECUTABLE)
+            message(STATUS "Detected MPI as MPICH")
+            set(KOKKOSCOMM_IMPL_MPI_IS_MPICH TRUE CACHE BOOL "Is MPI MPICH" FORCE)
+        elseif(OMPI_INFO_EXECUTABLE)
+            message(STATUS "Detected MPI as Open MPI")
+            set(KOKKOSCOMM_IMPL_MPI_IS_OPENMPI TRUE CACHE BOOL "Is MPI OpenMPI" FORCE)
+        endif()
+    endif()
+endfunction()