Check OpenMPI version if >= 5.0.6

When using OpenMPI, add mpiexec command-line option --oversubscribe

Author: wkliao

darshan-runtime/configure.ac

@@ -690,8 +690,22 @@ if test "x$enable_darshan_runtime" = xyes ; then
          #error OPENMPI FOUND
          #endif
          ])],
-         [],
-      [AC_DEFINE(HAVE_OPEN_MPI, 1, Define if OpenMPI is being used)])
+         [is_ompi=0], [is_ompi=1])
+
+      if test "x$is_ompi" = x1 ; then
+         AC_DEFINE(HAVE_OPEN_MPI, 1, [Define if OpenMPI is being used])
+
+         dnl Check if version is 5.0.6 or later
+         AC_MSG_CHECKING([Check if OpenMPI version is 5.0.6 or later])
+         AC_COMPILE_IFELSE([AC_LANG_PROGRAM([[#include <mpi.h>
+#if (OMPI_MAJOR_VERSION*1000000 + OMPI_MINOR_VERSION*1000 + OMPI_RELEASE_VERSION < 5000006)
+      choke me
+#endif
+         ]])], [ompi_ge_5_0_6=1], [ompi_ge_5_0_6=0])
+         AC_MSG_RESULT([$ompi_ge_5_0_6])
+         AC_DEFINE_UNQUOTED(OMPI_GE_5_0_6, [$ompi_ge_5_0_6],
+                   [Define if OpenMPI version is 5.0.6 or later])
+      fi
 
       # determine if the MPI library includes MPI-IO functions or not
       AC_MSG_CHECKING(for MPI-IO support in MPI)
@@ -918,6 +932,8 @@ AC_SUBST(HDF5_PATH, ["$with_hdf5"])
 AC_SUBST(PNETCDF_PATH, ["$with_pnetcdf"])
 AC_SUBST(DAOS_PATH, ["$with_daos"])
 AC_SUBST(LDMS_PATH, ["$LDMS_HOME"])
+AC_SUBST(HAVE_OPEN_MPI, ["$is_ompi"])
+AC_SUBST(OMPI_GE_5_0_6, ["$ompi_ge_5_0_6"])
 
 AM_CONDITIONAL(ENABLE_MMAP_LOGS,    [test "x$enable_mmap_logs"   = xyes])
 AM_CONDITIONAL(ENABLE_LDPRELOAD,    [test "x$enable_ld_preload"  = xyes])

darshan-runtime/test/Makefile.am

@@ -17,6 +17,8 @@ AM_TESTS_ENVIRONMENT += export TST_DARSHAN_LOG_PATH="$(__DARSHAN_LOG_PATH)";
 AM_TESTS_ENVIRONMENT += export DARSHAN_INSTALL_DIR="$(prefix)";
 AM_TESTS_ENVIRONMENT += export USERNAME_ENV="$(USERNAME_ENV)";
 AM_TESTS_ENVIRONMENT += export HAVE_MPI_LARGE_COUNT="$(HAVE_MPI_LARGE_COUNT)";
+AM_TESTS_ENVIRONMENT += export HAVE_OPEN_MPI="$(HAVE_OPEN_MPI)";
+AM_TESTS_ENVIRONMENT += export OMPI_GE_5_0_6="$(OMPI_GE_5_0_6)";
 AM_TESTS_ENVIRONMENT += export check_PROGRAMS="$(check_PROGRAMS)";
 
 TESTS = tst_runs.sh

darshan-runtime/test/tst_runs.sh

@@ -34,12 +34,15 @@ echo "DARSHAN_CONFIG=$DARSHAN_CONFIG"
 
 $DARSHAN_CONFIG --all
 
-# run NP number of MPI processes
-# Note when using OpenMPI, setting NP > 2 will fail.
+# run NP number of MPI processes, default 2
 if test "x$NP" = x ; then
    NP=2
 fi
 
+if test "x$HAVE_OPEN_MPI" = x1 ; then
+   TESTMPIRUN="$TESTMPIRUN --oversubscribe"
+fi
+
 TEST_FILE=./testfile.dat
 
 # tst_mpi_io.c takes the following command-line options.