ports/openmpi
ports/openmpi
1
1
Форкнуть 0
Код Релизы Активность

This commit closes trac:1300.

Просмотр исходного кода 
This commit was SVN r18473.

The following Trac tickets were found above:
  Ticket 1300 --> https://svn.open-mpi.org/trac/ompi/ticket/1300
Этот коммит содержится в:
Pak Lui 2008-05-21 22:35:04 +00:00
родитель c31cc5b270
Коммит 7b3d7dcac4
9 изменённых файлов: 42 добавлений и 1010 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

47
orte/mca/plm/gridengine/Makefile.am
Просмотреть файл

@ -1,47 +0,0 @@
#
# Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
# University Research and Technology
# Corporation. All rights reserved.
# Copyright (c) 2004-2005 The University of Tennessee and The University
# of Tennessee Research Foundation. All rights
# reserved.
# Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
# University of Stuttgart. All rights reserved.
# Copyright (c) 2004-2005 The Regents of the University of California.
# All rights reserved.
# Copyright (c) 2006 Sun Microsystems, Inc. All rights reserved.
# Use is subject to license terms.
# $COPYRIGHT$
#
# Additional copyrights may follow
#
# $HEADER$
#
dist_pkgdata_DATA = help-plm-gridengine.txt
sources = \
plm_gridengine.h \
plm_gridengine_component.c \
plm_gridengine_module.c
# Make the output library in this directory, and name it either
# mca_<type>_<name>.la (for DSO builds) or libmca_<type>_<name>.la
# (for static builds).
if OMPI_BUILD_plm_gridengine_DSO
component_noinst =
component_install = mca_plm_gridengine.la
else
component_noinst = libmca_plm_gridengine.la
component_install =
endif
mcacomponentdir = $(pkglibdir)
mcacomponent_LTLIBRARIES = $(component_install)
mca_plm_gridengine_la_SOURCES = $(sources)
mca_plm_gridengine_la_LDFLAGS = -module -avoid-version
noinst_LTLIBRARIES = $(component_noinst)
libmca_plm_gridengine_la_SOURCES =$(sources)
libmca_plm_gridengine_la_LDFLAGS = -module -avoid-version

26
orte/mca/plm/gridengine/configure.m4
Просмотреть файл

@ -1,26 +0,0 @@
# -*- shell-script -*-
#
# Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
# University Research and Technology
# Corporation. All rights reserved.
# Copyright (c) 2004-2005 The University of Tennessee and The University
# of Tennessee Research Foundation. All rights
# reserved.
# Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
# University of Stuttgart. All rights reserved.
# Copyright (c) 2004-2005 The Regents of the University of California.
# All rights reserved.
# Copyright (c) 2006 Sun Microsystems, Inc. All rights reserved.
# Use is subject to license terms.
# $COPYRIGHT$
#
# Additional copyrights may follow
#
# $HEADER$
#
# MCA_plm_sge_CONFIG([action-if-found], [action-if-not-found])
# -----------------------------------------------------------
AC_DEFUN([MCA_plm_gridengine_CONFIG],[
OMPI_CHECK_GRIDENGINE([plm_gridengine], [$1], [$2])
])dnl

26
orte/mca/plm/gridengine/configure.params
Просмотреть файл

@ -1,26 +0,0 @@
# -*- shell-script -*-
#
# Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
# University Research and Technology
# Corporation. All rights reserved.
# Copyright (c) 2004-2005 The University of Tennessee and The University
# of Tennessee Research Foundation. All rights
# reserved.
# Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
# University of Stuttgart. All rights reserved.
# Copyright (c) 2004-2005 The Regents of the University of California.
# All rights reserved.
# Copyright (c) 2006 Sun Microsystems, Inc. All rights reserved.
# Use is subject to license terms.
# Copyright (c) 2007 Los Alamos National Security, LLC. All rights
# reserved.
# $COPYRIGHT$
#
# Additional copyrights may follow
#
# $HEADER$
#
# Specific to this module
PARAM_CONFIG_FILES="Makefile"

47
orte/mca/plm/gridengine/help-plm-gridengine.txt
Просмотреть файл

@ -1,47 +0,0 @@
# -*- text -*-
#
# Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
# University Research and Technology
# Corporation. All rights reserved.
# Copyright (c) 2004-2005 The University of Tennessee and The University
# of Tennessee Research Foundation. All rights
# reserved.
# Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
# University of Stuttgart. All rights reserved.
# Copyright (c) 2004-2005 The Regents of the University of California.
# All rights reserved.
# Copyright (c) 2006-2007 Sun Microsystems, Inc. All rights reserved.
# Use is subject to license terms.
# $COPYRIGHT$
#
# Additional copyrights may follow
#
# $HEADER$
#
[no-local-orted]
The gridengine pls component was not able to find the executable "orted" in
your PATH or in the directory where Open MPI was initially installed,
and therefore cannot continue.
For reference, your current PATH is:
%s
We also looked for orted in the following directory:
%s
[bad-qrsh-path]
The gridengine pls component is not able to find the
executable "qrsh" in $SGE_ROOT/bin/$ARC/qrsh, and therefore cannot
continue.
For reference, your current path to qrsh is:
%s
The $SGE_ROOT environment variable points to:
%s
The $ARC environment variable points to:
%s

66
orte/mca/plm/gridengine/plm_gridengine.h
Просмотреть файл

@ -1,66 +0,0 @@
/*
* Copyright (c) 2004-2008 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2006-2007 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
/**
* @file:
* Process launcher for the Grid Engine
*
* See FAQ on how it works:
* http://www.open-mpi.org/faq/?category=running#run-n1ge-or-sge
*/
#ifndef ORTE_PLM_GRIDENGINE_EXPORT_H
#define ORTE_PLM_GRIDENGINE_EXPORT_H
#include "orte_config.h"
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
#include "orte/mca/plm/plm.h"
#include "opal/mca/mca.h"
BEGIN_C_DECLS
/*
* Module open / close
*/
int orte_plm_gridengine_component_open(void);
int orte_plm_gridengine_component_close(void);
int orte_plm_gridengine_component_query(mca_base_module_t **module, int *priority);
/**
* PLM Component
*/
struct orte_plm_gridengine_component_t {
orte_plm_base_component_t super;
bool verbose;
int priority;
char* orted;
};
typedef struct orte_plm_gridengine_component_t orte_plm_gridengine_component_t;
ORTE_MODULE_DECLSPEC extern orte_plm_gridengine_component_t mca_plm_gridengine_component;
extern orte_plm_base_module_t orte_plm_gridengine_module;
END_C_DECLS
#endif /* ORTE_PLM_GRIDENGINE_EXPORT_H */

152
orte/mca/plm/gridengine/plm_gridengine_component.c
Просмотреть файл

@ -1,152 +0,0 @@
/*
* Copyright (c) 2004-2008 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2006-2007 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*
* These symbols are in a file by themselves to provide nice linker
* semantics. Since linkers generally pull in symbols by object
* files, keeping these symbols as the only symbols in this file
* prevents utility programs such as "ompi_info" from having to import
* entire components just to query their version and parameters.
*/
/**
* @file:
* Part of the gridengine launcher.
* See plm_gridengine.h for an overview of how it works.
*/
#include "orte_config.h"
#include "orte/constants.h"
#include "opal/util/path.h"
#include "opal/util/argv.h"
#include "orte/util/output.h"
#include "opal/mca/base/mca_base_param.h"
#include "orte/util/proc_info.h"
#include "orte/mca/errmgr/errmgr.h"
#include "orte/mca/plm/plm.h"
#include "orte/mca/plm/base/base.h"
#include "orte/mca/plm/base/plm_private.h"
#include "plm_gridengine.h"
/**
* Public string showing the plm ompi_gridengine component version number
*/
const char *mca_plm_gridengine_component_version_string =
"Open MPI gridengine plm MCA component version " ORTE_VERSION;
/**
* Local functions
*/
/**
* Instantiate the public struct with all of our public information
* and pointers to our public functions in it
*/
orte_plm_gridengine_component_t mca_plm_gridengine_component = {
{
/* First, the mca_component_t struct containing meta information
about the component itself */
{
/* Indicate that we are a plm v1.0.0 component (which also
implies a specific MCA version) */
ORTE_PLM_BASE_VERSION_1_0_0,
/* Component name and version */
"gridengine",
ORTE_MAJOR_VERSION,
ORTE_MINOR_VERSION,
ORTE_RELEASE_VERSION,
/* Component open and close functions */
orte_plm_gridengine_component_open,
orte_plm_gridengine_component_close,
orte_plm_gridengine_component_query
},
/* Next the MCA v1.0.0 component meta data */
{
/* The component is checkpoint ready */
MCA_BASE_METADATA_PARAM_CHECKPOINT
}
}
};
/**
orte_plm_gridengine_component_open - open component and register all parameters
@return error number
*/
int orte_plm_gridengine_component_open(void)
{
int tmp;
mca_base_component_t *c = &mca_plm_gridengine_component.super.base_version;
mca_base_param_reg_int(c, "verbose",
"Enable verbose output of the gridengine qrsh -inherit command",
false, false, false, &tmp);
mca_plm_gridengine_component.verbose = OPAL_INT_TO_BOOL(tmp);
mca_base_param_reg_int(c, "priority",
"Priority of the gridengine plm component",
false , false, 100, &mca_plm_gridengine_component.priority);
mca_base_param_reg_string(c, "orted",
"The command name that the gridengine plm component will invoke for the ORTE daemon",
false, false, "orted", &mca_plm_gridengine_component.orted);
return ORTE_SUCCESS;
}
/**
orte_plm_gridengine_component_close - close component and register all parameters
@return error number
*/
int orte_plm_gridengine_component_close(void)
{
/* cleanup state */
if (NULL != mca_plm_gridengine_component.orted) {
free(mca_plm_gridengine_component.orted);
}
return ORTE_SUCCESS;
}
/**
orte_plm_gridengine_component_init - initialize component, check if we can run on this machine.
@return error number
*/
int orte_plm_gridengine_component_query(mca_base_module_t **module, int *priority)
{
if (NULL != getenv("SGE_ROOT") && NULL != getenv("ARC") &&
NULL != getenv("PE_HOSTFILE") && NULL != getenv("JOB_ID")) {
orte_output_verbose(10, orte_plm_globals.output,
"plm:gridengine: available for selection");
*priority = mca_plm_gridengine_component.priority;
*module = (mca_base_module_t *) &orte_plm_gridengine_module;
return ORTE_SUCCESS;
}
orte_output_verbose(10, orte_plm_globals.output,
"plm:gridengine: NOT available for selection");
*module = NULL;
return ORTE_ERROR;
}

645
orte/mca/plm/gridengine/plm_gridengine_module.c
Просмотреть файл

@ -1,645 +0,0 @@
/*
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2008 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2006 Cisco Systems, Inc. All rights reserved.
* Copyright (c) 2006-2007 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
* Copyright (c) 2007 Los Alamos National Security, LLC. All rights
* reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*
* These symbols are in a file by themselves to provide nice linker
* semantics. Since linkers generally pull in symbols by object
* files, keeping these symbols as the only symbols in this file
* prevents utility programs such as "ompi_info" from having to import
* entire components just to query their version and parameters.
*/
/**
* @file:
* Part of the gridengine launcher.
* See plm_gridengine.h for an overview of how it works.
*/
#include "orte_config.h"
#include "orte/constants.h"
#include <stdlib.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <errno.h>
#include <string.h>
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
#ifdef HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
#ifdef HAVE_SYS_STAT_H
#include <sys/stat.h>
#endif
#ifdef HAVE_SYS_WAIT_H
#include <sys/wait.h>
#endif
#include <fcntl.h>
#include <signal.h>
#ifdef HAVE_PWD_H
#include <pwd.h>
#endif
#include "opal/mca/installdirs/installdirs.h"
#include "opal/mca/base/mca_base_param.h"
#include "opal/util/if.h"
#include "opal/util/os_path.h"
#include "opal/util/path.h"
#include "opal/event/event.h"
#include "opal/util/argv.h"
#include "opal/util/opal_environ.h"
#include "orte/util/output.h"
#include "opal/util/basename.h"
#include "opal/util/opal_environ.h"
#include "orte/util/name_fns.h"
#include "orte/runtime/orte_globals.h"
#include "orte/util/session_dir.h"
#include "orte/runtime/orte_wait.h"
#include "orte/runtime/orte_wakeup.h"
#include "orte/mca/rml/rml.h"
#include "orte/mca/errmgr/errmgr.h"
#include "orte/mca/rmaps/rmaps.h"
#include "orte/mca/plm/plm.h"
#include "orte/mca/plm/base/base.h"
#include "orte/mca/plm/base/plm_private.h"
#include "orte/mca/plm/gridengine/plm_gridengine.h"
static int plm_gridengine_init(void);
static int plm_gridengine_launch_job(orte_job_t *jdata);
static int plm_gridengine_terminate_job(orte_jobid_t jobid);
static int plm_gridengine_terminate_orteds(void);
static int plm_gridengine_signal_job(orte_jobid_t jobid, int32_t signal);
static int plm_gridengine_finalize(void);
orte_plm_base_module_t orte_plm_gridengine_module = {
plm_gridengine_init,
orte_plm_base_set_hnp_name,
plm_gridengine_launch_job,
NULL,
plm_gridengine_terminate_job,
plm_gridengine_terminate_orteds,
plm_gridengine_signal_job,
plm_gridengine_finalize
};
static void set_handler_default(int sig);
/* global storage of active jobid being launched */
static orte_jobid_t active_job=ORTE_JOBID_INVALID;
/**
* Init the module
*/
int plm_gridengine_init(void)
{
int rc;
if (ORTE_SUCCESS != (rc = orte_plm_base_comm_start())) {
ORTE_ERROR_LOG(rc);
}
return rc;
}
/**
* Fill the orted_path variable with the directory to the orted
*/
static int orte_plm_gridengine_fill_orted_path(char** orted_path)
{
struct stat buf;
asprintf(orted_path, "%s/orted", opal_install_dirs.bindir);
if (0 != stat(*orted_path, &buf)) {
char *path = getenv("PATH");
if (NULL == path) {
path = ("PATH is empty!");
}
orte_show_help("help-plm-gridengine.txt", "no-local-orted",
true, path, opal_install_dirs.bindir);
return ORTE_ERR_NOT_FOUND;
}
return ORTE_SUCCESS;
}
/**
* Callback on daemon exit.
*/
static void orte_plm_gridengine_wait_daemon(pid_t pid, int status, void* cbdata)
{
if (! WIFEXITED(status) || ! WEXITSTATUS(status) == 0) {
/* Need to catch SIGUSR1/2 for "qrsh/qsub -notify" to work.
* With "-notify" set, SIGUSR1/2 becomes the precursor for any pending
* SIGSTOP/SIGKILL. So just return and ignore the daemon_failed
* at the end as that would kill off the user processes */
if (SIGUSR1 == status || SIGUSR2 == status) {
orte_output(0, "The daemon received a signal %d", status);
return;
}
/* Otherwise, tell the user something went wrong. */
orte_output(0, "ERROR: A daemon failed to start as expected.");
orte_output(0, "ERROR: There may be more information available from");
orte_output(0, "ERROR: the 'qstat -t' command on the Grid Engine tasks.");
orte_output(0, "ERROR: If the problem persists, please restart the");
orte_output(0, "ERROR: Grid Engine PE job");
if (WIFEXITED(status)) {
orte_output(0, "ERROR: The daemon exited unexpectedly with status %d.",
WEXITSTATUS(status));
} else if (WIFSIGNALED(status)) {
#ifdef WCOREDUMP
if (WCOREDUMP(status)) {
orte_output(0, "The daemon received a signal %d (with core).",
WTERMSIG(status));
} else {
orte_output(0, "The daemon received a signal %d.", WTERMSIG(status));
}
#else
orte_output(0, "The daemon received a signal %d.", WTERMSIG(status));
#endif /* WCOREDUMP */
} else {
orte_output(0, "No extra status information is available: %d.", status);
}
/* report that the daemon has failed so we break out of the daemon
* callback receive and can exit
*/
orte_plm_base_launch_failed(active_job, true, pid, status, ORTE_JOB_STATE_FAILED_TO_START);
}
}
/**
* Launch a daemon (bootproxy) on each node. The daemon will be responsible
* for launching the application.
*/
/* When working in this function, ALWAYS jump to "cleanup" if
* you encounter an error so that orterun will be woken up and
* the job can cleanly terminate
*/
static int plm_gridengine_launch_job(orte_job_t *jdata)
{
orte_job_map_t *map;
int node_name_index1;
int node_name_index2;
int proc_vpid_index;
int orted_index;
char *prefix_dir;
char *param;
char **argv=NULL;
char **env=NULL;
int argc;
int rc;
sigset_t sigs;
char *lib_base = NULL, *bin_base = NULL;
char *sge_root, *sge_arch;
bool failed_launch = true;
orte_app_context_t **apps;
orte_node_t **nodes;
orte_std_cntr_t nnode;
/* create a jobid for this job */
if (ORTE_SUCCESS != (rc = orte_plm_base_create_jobid(&jdata->jobid))) {
ORTE_ERROR_LOG(rc);
goto cleanup;
}
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: launching job %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
ORTE_JOBID_PRINT(jdata->jobid)));
/* setup the job */
if (ORTE_SUCCESS != (rc = orte_plm_base_setup_job(jdata))) {
ORTE_ERROR_LOG(rc);
goto cleanup;
}
/* set the active jobid */
active_job = jdata->jobid;
/* Get the map for this job */
if (NULL == (map = orte_rmaps.get_job_map(active_job))) {
ORTE_ERROR_LOG(ORTE_ERR_NOT_FOUND);
rc = ORTE_ERR_NOT_FOUND;
goto cleanup;
}
apps = (orte_app_context_t**)jdata->apps->addr;
nodes = (orte_node_t**)map->nodes->addr;
if (map->num_new_daemons == 0) {
/* have all the daemons we need - launch app */
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: no new daemons to launch",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME)));
goto launch_apps;
}
/*
* Build argv array
*/
argv = opal_argv_split("qrsh", ' ');
argc = opal_argv_count(argv);
/* gridengine specific flags */
opal_argv_append(&argc, &argv, "-inherit");/*run tasks within curr job*/
opal_argv_append(&argc, &argv, "-noshell");/*execute w/o wrapping shell*/
opal_argv_append(&argc, &argv, "-nostdin");/*suppress input stream stdin*/
opal_argv_append(&argc, &argv, "-V"); /*task to have the env as job*/
if (mca_plm_gridengine_component.verbose) {
opal_argv_append(&argc, &argv, "-verbose");
}
node_name_index1 = argc;
opal_argv_append(&argc, &argv, "<template>");
/* add the orted daemon in command */
orted_index = argc;
opal_argv_append(&argc, &argv, mca_plm_gridengine_component.orted);
/* By default, orteds will not daemonize and will be forced to
* to stay in the same ptree as sge_shephard. The problem with
* --daemonize is that the qrsh -inherit connections will stay
* persistent for the whole duration of the task to the remote nodes,
* which may not be ideal for large number of nodes
*/
/* Add basic orted command line options, including
* all debug options
*/
orte_plm_base_orted_append_basic_args(&argc, &argv,
"env",
&proc_vpid_index,
&node_name_index2);
/* setup environment. The environment is common to all the daemons
* so we only need to do this once
*/
env = opal_argv_copy(environ);
if (0 < orte_output_get_verbosity(orte_plm_globals.output)) {
param = opal_argv_join(argv, ' ');
if (NULL != param) {
orte_output(0, "plm:gridengine: final template argv:");
orte_output(0, "plm:gridengine: %s", param);
free(param);
}
}
/* Figure out the basenames for the libdir and bindir. There is a
lengthy comment about this in plm_rsh_module.c explaining all
the rationale for how / why we're doing this.
*/
lib_base = opal_basename(opal_install_dirs.libdir);
bin_base = opal_basename(opal_install_dirs.bindir);
/* See the note about prefix_dir in the orte/mca/plm/slurm/plm_slurm.c
* module. Fo here, just note that we must have at least one app_context,
* and we take the prefix_dir from that first one.
*/
prefix_dir = apps[0]->prefix_dir;
/* If we have a prefix, then modify the PATH and
LD_LIBRARY_PATH environment variables.
*/
if (NULL != prefix_dir) {
char *oldenv, *newenv;
/* Reset PATH */
newenv = opal_os_path( false, prefix_dir, bin_base, NULL );
oldenv = getenv("PATH");
if (NULL != oldenv) {
char *temp;
asprintf(&temp, "%s:%s", newenv, oldenv);
free( newenv );
newenv = temp;
}
opal_setenv("PATH", newenv, true, &env);
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: reset PATH: %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
newenv));
free(newenv);
/* Reset LD_LIBRARY_PATH */
newenv = opal_os_path( false, prefix_dir, lib_base, NULL );
oldenv = getenv("LD_LIBRARY_PATH");
if (NULL != oldenv) {
char* temp;
asprintf(&temp, "%s:%s", newenv, oldenv);
free(newenv);
newenv = temp;
}
opal_setenv("LD_LIBRARY_PATH", newenv, true, &env);
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: reset LD_LIBRARY_PATH: %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
newenv));
free(newenv);
}
/*
* Iterate through the nodes.
*/
for(nnode=0; nnode < map->num_nodes; nnode++) {
pid_t pid;
char *exec_path, *orted_path;
char **exec_argv;
/* if this daemon already exists, don't launch it! */
if (nodes[nnode]->daemon_launched) {
continue;
}
/* setup node name */
free(argv[node_name_index1]);
if (NULL != nodes[nnode]->username &&
0 != strlen (nodes[nnode]->username)) {
asprintf(&argv[node_name_index1], "%s@%s",
nodes[nnode]->username, nodes[nnode]->name);
} else {
argv[node_name_index1] = strdup(nodes[nnode]->name);
}
free(argv[node_name_index2]);
argv[node_name_index2] = strdup(nodes[nnode]->name);
/* fork a child to do qrsh */
pid = fork();
if (pid < 0) {
ORTE_ERROR_LOG(ORTE_ERR_SYS_LIMITS_CHILDREN);
rc = ORTE_ERR_SYS_LIMITS_CHILDREN;
goto cleanup;
}
/* child */
if (pid == 0) {
char* var;
long fd, fdmax = sysconf(_SC_OPEN_MAX);
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: launching on node %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
nodes[nnode]->name));
/* setting exec_argv and exec_path for qrsh */
exec_argv = &argv[0];
sge_root = getenv("SGE_ROOT");
sge_arch = getenv("ARC");
asprintf(&exec_path, "%s/bin/%s/qrsh", sge_root, sge_arch);
exec_path = opal_path_findv(exec_path, X_OK, environ, NULL);
if (NULL == exec_path) {
orte_show_help("help-plm-gridengine.txt", "bad-qrsh-path",
true, exec_path, sge_root, sge_arch);
exit(-1); /* forked child must ALWAYS exit, not return */
}
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: exec_argv[0]=%s, exec_path=%s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
exec_argv[0], exec_path));
/* setting orted_path for orted */
orted_path = opal_path_findv(exec_argv[orted_index], 0, environ, NULL);
if (NULL == orted_path && NULL == prefix_dir) {
rc = orte_plm_gridengine_fill_orted_path(&orted_path);
if (ORTE_SUCCESS != rc) {
exit(-1); /* forked child must ALWAYS exit, not return */
}
} else {
if (NULL != prefix_dir) {
orted_path = opal_os_path( false, prefix_dir, bin_base, "orted", NULL );
}
/* If we yet did not fill up the orted_path, do so now */
if (NULL == orted_path) {
rc = orte_plm_gridengine_fill_orted_path(&orted_path);
if (ORTE_SUCCESS != rc) {
exit(-1); /* forked child must ALWAYS exit, not return */
}
}
}
asprintf(&argv[orted_index], orted_path);
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: orted_path=%s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
orted_path));
var = (char*)opal_home_directory();
if (NULL != var) {
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: changing to directory %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
var));
/* Ignore errors -- what are we going to do?
(and we ignore errors on the remote nodes
in the fork plm, so this is consistent) */
chdir(var);
}
/* pass the vpid */
rc = orte_util_convert_vpid_to_string(&param, nodes[nnode]->daemon->name.vpid);
if (ORTE_SUCCESS != rc) {
orte_output(0, "plm:gridengine: unable to get daemon vpid as string");
exit(-1);
}
free(argv[proc_vpid_index]);
argv[proc_vpid_index] = strdup(param);
free(param);
if (0 > orte_output_get_verbosity(orte_plm_globals.output)) {
/* setup stdin */
int fd = open("/dev/null", O_RDWR, 0);
dup2(fd, 0);
close(fd);
}
/* close all file descriptors w/ exception of stdin/stdout/stderr */
for(fd=3; fd<fdmax; fd++)
close(fd);
/* Set signal handlers back to the default. Do this close
to the execve() because the event library may (and likely
will) reset them. If we don't do this, the event
library may have left some set that, at least on some
OS's, don't get reset via fork() or exec(). Hence, the
orted could be unkillable (for example). */
set_handler_default(SIGTERM);
set_handler_default(SIGINT);
set_handler_default(SIGHUP);
set_handler_default(SIGPIPE);
set_handler_default(SIGCHLD);
/* Unblock all signals, for many of the same reasons that
we set the default handlers, above. This is noticable
on Linux where the event library blocks SIGTERM, but we
don't want that blocked by the orted (or, more
specifically, we don't want it to be blocked by the
orted and then inherited by the ORTE processes that it
forks, making them unkillable by SIGTERM). */
sigprocmask(0, 0, &sigs);
sigprocmask(SIG_UNBLOCK, &sigs, 0);
/* exec the daemon */
if (0 < orte_output_get_verbosity(orte_plm_globals.output)) {
param = opal_argv_join(exec_argv, ' ');
if (NULL != param) {
orte_output(0, "plm:gridengine: executing: %s", param);
free(param);
}
}
execve(exec_path, exec_argv, env);
orte_output(0, "plm:gridengine: execve failed with errno=%d\n", errno);
exit(-1);
} else { /* parent */
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: parent",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME)));
/* setup callback on sigchild - wait until setup above is complete
* as the callback can occur in the call to orte_wait_cb
*/
orte_wait_cb(pid, orte_plm_gridengine_wait_daemon, NULL);
}
}
/* wait for daemons to callback */
if (ORTE_SUCCESS != (rc = orte_plm_base_daemon_callback(map->num_new_daemons))) {
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: daemon launch failed for job %s on error %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
ORTE_JOBID_PRINT(active_job), ORTE_ERROR_NAME(rc)));
goto cleanup;
}
launch_apps:
if (ORTE_SUCCESS != (rc = orte_plm_base_launch_apps(active_job))) {
ORTE_OUTPUT_VERBOSE((1, orte_plm_globals.output,
"%s plm:gridengine: launch of apps failed for job %s on error %s",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
ORTE_JOBID_PRINT(active_job), ORTE_ERROR_NAME(rc)));
goto cleanup;
}
/* get here if launch went okay */
failed_launch = false;
cleanup:
if (NULL != lib_base) {
free(lib_base);
}
if (NULL != bin_base) {
free(bin_base);
}
if (NULL != argv) {
opal_argv_free(argv);
}
if (NULL != env) {
opal_argv_free(env);
}
/* check for failed launch - if so, force terminate */
if (failed_launch) {
orte_plm_base_launch_failed(jdata->jobid, false, -1, ORTE_ERROR_DEFAULT_EXIT_CODE, ORTE_JOB_STATE_FAILED_TO_START);
}
return rc;
}
/**
* Terminate the job
*/
static int plm_gridengine_terminate_job(orte_jobid_t jobid)
{
int rc;
/* order them to kill their local procs for this job */
if (ORTE_SUCCESS != (rc = orte_plm_base_orted_kill_local_procs(jobid))) {
ORTE_ERROR_LOG(rc);
}
return rc;
}
/**
* Terminate the orteds for a given job
*/
static int plm_gridengine_terminate_orteds(void)
{
int rc;
/* now tell them to die! */
if (ORTE_SUCCESS != (rc = orte_plm_base_orted_exit())) {
ORTE_ERROR_LOG(rc);
}
return rc;
}
/**
* Signal all processes associated with this job
*/
static int plm_gridengine_signal_job(orte_jobid_t jobid, int32_t signal)
{
int rc;
/* order them to pass this signal to their local procs */
if (ORTE_SUCCESS != (rc = orte_plm_base_orted_signal_local_procs(jobid, signal))) {
ORTE_ERROR_LOG(rc);
}
return rc;
}
/**
* Finalize
*/
static int plm_gridengine_finalize(void)
{
int rc;
/* cleanup any pending recvs */
if (ORTE_SUCCESS != (rc = orte_plm_base_comm_stop())) {
ORTE_ERROR_LOG(rc);
}
return ORTE_SUCCESS;
}
/**
* Set signal handler
*/
static void set_handler_default(int sig)
{
struct sigaction act;
act.sa_handler = SIG_DFL;
act.sa_flags = 0;
sigemptyset(&act.sa_mask);
sigaction(sig, &act, (struct sigaction *)0);
}

2
orte/mca/plm/rsh/plm_rsh.h
Просмотреть файл

@ -9,6 +9,7 @@
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2008 Sun Microsystems, Inc. All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
@ -65,6 +66,7 @@ struct orte_plm_rsh_component_t {
orte_plm_base_component_t super;
bool assume_same_shell;
bool force_rsh;
bool disable_qrsh;
int delay;
int priority;
char *agent_param;

41
orte/mca/plm/rsh/plm_rsh_component.c
Просмотреть файл

@ -11,6 +11,7 @@
* All rights reserved.
* Copyright (c) 2007 Los Alamos National Security, LLC. All rights
* reserved.
* Copyright (c) 2008 Sun Microsystems, Inc. All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
@ -131,7 +132,10 @@ int orte_plm_rsh_component_open(void)
"Force the launcher to always use rsh",
false, false, false, &tmp);
mca_plm_rsh_component.force_rsh = OPAL_INT_TO_BOOL(tmp);
mca_base_param_reg_int(c, "disable_qrsh",
"Disable the launcher to use qrsh when under the SGE parallel environment",
false, false, false, &tmp);
mca_plm_rsh_component.disable_qrsh = OPAL_INT_TO_BOOL(tmp);
mca_base_param_reg_string(c, "orted",
"The command name that the rsh plm component will invoke for the ORTE daemon",
false, false, "orted",
@ -176,6 +180,24 @@ int orte_plm_rsh_component_query(mca_base_module_t **module, int *priority)
mca_plm_rsh_component.agent_argc =
opal_argv_count(mca_plm_rsh_component.agent_argv);
mca_plm_rsh_component.agent_path = NULL;
/* To be absolutely sure that we are under an SGE parallel env */
if (!mca_plm_rsh_component.disable_qrsh &&
NULL != getenv("SGE_ROOT") && NULL != getenv("ARC") &&
NULL != getenv("PE_HOSTFILE") && NULL != getenv("JOB_ID")) {
/* setting exec_argv and exec_path for qrsh */
asprintf(&mca_plm_rsh_component.agent_param, "qrsh");
asprintf(&mca_plm_rsh_component.agent_path, "%s/bin/%s", getenv("SGE_ROOT"), getenv("ARC"));
asprintf(&mca_plm_rsh_component.agent_argv[0], "%s/bin/%s/qrsh", getenv("SGE_ROOT"), getenv("ARC"));
if (0 < orte_output_get_verbosity(orte_plm_globals.output)) {
orte_output_verbose(1, orte_plm_globals.output,
"%s plm:rsh: using %s for launching\n",
ORTE_NAME_PRINT(ORTE_PROC_MY_NAME),
mca_plm_rsh_component.agent_argv[0]);
}
}
if (mca_plm_rsh_component.agent_argc > 0) {
/* If the agent is ssh, and debug was not selected, then
automatically add "-x" */
@ -194,6 +216,23 @@ int orte_plm_rsh_component_query(mca_base_module_t **module, int *priority)
&mca_plm_rsh_component.agent_argv, "-x");
}
}
/* If the agent is qrsh, then automatically add -inherit
* and grid engine PE related flags */
if (NULL != bname && 0 == strcmp(bname, "qrsh")) {
opal_argv_append(&mca_plm_rsh_component.agent_argc,
&mca_plm_rsh_component.agent_argv, "-inherit");
/* Don't use the "-noshell" flag as qrsh would have a problem
* swallowing a long command */
opal_argv_append(&mca_plm_rsh_component.agent_argc,
&mca_plm_rsh_component.agent_argv, "-nostdin");
opal_argv_append(&mca_plm_rsh_component.agent_argc,
&mca_plm_rsh_component.agent_argv, "-V");
if (0 < orte_output_get_verbosity(orte_plm_globals.output)) {
opal_argv_append(&mca_plm_rsh_component.agent_argc,
&mca_plm_rsh_component.agent_argv, "-verbose");
}
}
if (NULL != bname) {
free(bname);
}