6d6cebb4a7
Accordingly, there are new APIs to the name service to support the ability to get a job's parent, root, immediate children, and all its descendants. In addition, the terminate_job, terminate_orted, and signal_job APIs for the PLS have been modified to accept attributes that define the extent of their actions. For example, doing a "terminate_job" with an attribute of ORTE_NS_INCLUDE_DESCENDANTS will terminate the given jobid AND all jobs that descended from it. I have tested this capability on a MacBook under rsh, Odin under SLURM, and LANL's Flash (bproc). It worked successfully on non-MPI jobs (both simple and including a spawn), and MPI jobs (again, both simple and with a spawn). This commit was SVN r12597.
86 строки
2.0 KiB
C
86 строки
2.0 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2006 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$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
#include "orte/mca/ns/ns_types.h"
|
|
#include "orte/mca/iof/iof.h"
|
|
#include "ompi/constants.h"
|
|
|
|
#if defined(__WINDOWS__)
|
|
#include <io.h>
|
|
#endif /* defined(__WINDOWS__) */
|
|
|
|
int ompi_mpi_init_io(void)
|
|
{
|
|
int fds[2];
|
|
int rc;
|
|
|
|
/* We need to add a NULL component for iof. Until then, this will have to do */
|
|
/* setup stdin */
|
|
rc = pipe(fds);
|
|
if(rc < 0) {
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
dup2(fds[0], 0);
|
|
close(fds[0]);
|
|
|
|
rc = mca_iof.iof_publish(
|
|
OMPI_PROC_MY_NAME,
|
|
MCA_IOF_SINK,
|
|
MCA_IOF_STDIN,
|
|
fds[1]);
|
|
if(rc != OMPI_SUCCESS)
|
|
return rc;
|
|
|
|
/* setup stdout */
|
|
rc = pipe(fds);
|
|
if(rc < 0) {
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
dup2(fds[1], 1);
|
|
close(fds[1]);
|
|
|
|
rc = mca_iof.iof_publish(
|
|
OMPI_PROC_MY_NAME,
|
|
MCA_IOF_SOURCE,
|
|
MCA_IOF_STDOUT,
|
|
fds[0]);
|
|
if(rc != OMPI_SUCCESS)
|
|
return rc;
|
|
|
|
/* setup stderr */
|
|
rc = pipe(fds);
|
|
if(rc < 0) {
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
dup2(fds[1], 2);
|
|
close(fds[1]);
|
|
|
|
rc = mca_iof.iof_publish(
|
|
OMPI_PROC_MY_NAME,
|
|
MCA_IOF_SOURCE,
|
|
MCA_IOF_STDERR,
|
|
fds[0]);
|
|
if(rc != OMPI_SUCCESS)
|
|
return rc;
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|