modified mpirun to longer fork/exec prterun but now to instead call the function directly in the process

Signed-off-by: Gerardo Garcia <[email protected]>

Author: Gerardo Garcia

ompi/tools/mpirun/Makefile.am

@@ -21,7 +21,8 @@ mpirun_SOURCES = \
     main.c
 
 mpirun_LDADD = \
-	$(top_builddir)/opal/libopen-pal_core.la
+	$(top_builddir)/opal/libopen-pal_core.la \
+	$(top_builddir)/3rd-party/src/libprrte.la
 
 mpirun_CPPFLAGS = \
     -DMCA_oshmem_FRAMEWORKS="\"$(MCA_oshmem_FRAMEWORKS)\"" \

ompi/tools/mpirun/main.c

@@ -28,46 +28,8 @@
 #include "opal/util/printf.h"
 #include "opal/util/show_help.h"
 #include "ompi/constants.h"
+#include "3rd-party/prrte/include/prte.h"
 
-static char *find_prterun(void)
-{
-    char *filename = NULL;
-#if !OMPI_USING_INTERNAL_PRRTE
-    char *prrte_prefix = NULL;
-#endif
-
-    /* 1) Did the user tell us exactly where to find prterun? */
-    filename = getenv("OMPI_PRTERUN");
-    if (NULL != filename) {
-        return filename;
-    }
-
-#if OMPI_USING_INTERNAL_PRRTE
-    /* 2) If using internal PRRTE, use our bindir.  Note that this
-     * will obey OPAL_PREFIX and OPAL_DESTDIR */
-    opal_asprintf(&filename, "%s%sprterun", opal_install_dirs.bindir, OPAL_PATH_SEP);
-    return filename;
-#else
-
-    /* 3) Look in ${PRTE_PREFIX}/bin */
-    prrte_prefix = getenv("PRTE_PREFIX");
-    if (NULL != prrte_prefix) {
-        opal_asprintf(&filename, "%s%sbin%sprterun", prrte_prefix, OPAL_PATH_SEP, OPAL_PATH_SEP);
-        return filename;
-    }
-
-    /* 4) See if configure told us where to look, if set */
-#if defined(OMPI_PRTERUN_PATH)
-    return strdup(OMPI_PRTERUN_PATH);
-#else
-
-    /* 5) Use path search */
-    filename = opal_find_absolute_path("prterun");
-
-    return filename;
-#endif
-#endif
-}
 
 static void append_prefixes(char ***out, const char *in)
 {
@@ -119,10 +81,7 @@ static void setup_mca_prefixes(void)
 int main(int argc, char *argv[])
 {
     char *opal_prefix = getenv("OPAL_PREFIX");
-    char *full_prterun_path = NULL;
-    char **prterun_args = NULL;
     int ret;
-    size_t i;
 
     ret = opal_init_util(&argc, &argv);
     if (OMPI_SUCCESS != ret) {
@@ -154,12 +113,6 @@ int main(int argc, char *argv[])
 #endif
     }
 
-    full_prterun_path = find_prterun();
-    if (NULL == full_prterun_path) {
-        opal_show_help("help-mpirun.txt", "no-prterun-found", 1);
-        exit(1);
-    }
-
     /*
      * set environment variable for our install location
      * used within the OMPI prrte schizo component
@@ -171,24 +124,14 @@ int main(int argc, char *argv[])
     // to Open MPI.
     setup_mca_prefixes();
 
-    /* calling mpirun (and now prterun) with a full path has a special
-     * meaning in terms of -prefix behavior, so copy that behavior
-     * into prterun */
-    if (opal_path_is_absolute(argv[0])) {
-        opal_argv_append_nosize(&prterun_args, full_prterun_path);
-    } else {
-        opal_argv_append_nosize(&prterun_args, "prterun");
+    
+    ret = prte_launch(argc, argv);
+    if (OMPI_SUCCESS != ret) {
+        opal_show_help("help-mpirun.txt", "prte-launch-failed", 1, strerror(errno));
+        exit(1);
     }
 
-    /* Copy all the mpirun arguments to prterun.
-     * TODO: Need to handle --prefix rationally here. */
-    for (i = 1; NULL != argv[i]; i++) {
-        opal_argv_append_nosize(&prterun_args, argv[i]);
-    }
-    ret = execv(full_prterun_path, prterun_args);
-    opal_show_help("help-mpirun.txt", "prterun-exec-failed",
-                   1, full_prterun_path, strerror(errno));
-    exit(1);
+    return 0;
 }
 
 /*
@@ -214,4 +157,4 @@ int main(int argc, char *argv[])
  * Additional copyrights may follow
  *
  * $HEADER$
- */
+ */