153 строки
3.7 KiB
C
153 строки
3.7 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2004-2006 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$
|
||
|
|
*
|
||
|
|
* Additional copyrights may follow
|
||
|
|
*
|
||
|
|
* $HEADER$
|
||
|
|
*/
|
||
|
|
#include "orte_config.h"
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#ifdef HAVE_UNISTD_H
|
||
|
|
#include <unistd.h>
|
||
|
|
#endif /* HAVE_UNISTD_H */
|
||
|
|
#ifdef HAVE_STRING_H
|
||
|
|
#include <string.h>
|
||
|
|
#endif /* HAVE_STRING_H */
|
||
|
|
#ifdef HAVE_SIGNAL_H
|
||
|
|
#include <signal.h>
|
||
|
|
#endif
|
||
|
|
#ifdef HAVE_CNOS_PM_BARRIER
|
||
|
|
#include <catamount/cnos_mpi_os.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include "orte/orte_constants.h"
|
||
|
|
#include "orte/mca/rmgr/base/base.h"
|
||
|
|
#include "orte/mca/ns/ns.h"
|
||
|
|
#include "orte/util/proc_info.h"
|
||
|
|
#include "pls_cnos.h"
|
||
|
|
|
||
|
|
|
||
|
|
static int orte_pls_cnos_launch_job(orte_jobid_t jobid);
|
||
|
|
static int orte_pls_cnos_terminate_job(orte_jobid_t jobid);
|
||
|
|
static int orte_pls_cnos_terminate_orteds(orte_jobid_t jobid);
|
||
|
|
static int orte_pls_cnos_terminate_proc(const orte_process_name_t* proc_name);
|
||
|
|
static int orte_pls_cnos_signal_job(orte_jobid_t jobid, int32_t signal);
|
||
|
|
static int orte_pls_cnos_signal_proc(const orte_process_name_t* proc_name, int32_t signal);
|
||
|
|
static int orte_pls_cnos_finalize(void);
|
||
|
|
|
||
|
|
|
||
|
|
orte_pls_base_module_t orte_pls_cnos_module = {
|
||
|
|
orte_pls_cnos_launch_job,
|
||
|
|
orte_pls_cnos_terminate_job,
|
||
|
|
orte_pls_cnos_terminate_orteds,
|
||
|
|
orte_pls_cnos_terminate_proc,
|
||
|
|
orte_pls_cnos_signal_job,
|
||
|
|
orte_pls_cnos_signal_proc,
|
||
|
|
orte_pls_cnos_finalize
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
static int orte_pls_cnos_launch_job(orte_jobid_t jobid)
|
||
|
|
{
|
||
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef HAVE_KILLRANK
|
||
|
|
#include "catamount/types.h"
|
||
|
|
/* secret sauce on the Cray machine */
|
||
|
|
extern int killrank(rank_t RANK, int SIG);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
static int orte_pls_cnos_terminate_job(orte_jobid_t jobid)
|
||
|
|
{
|
||
|
|
#ifdef HAVE_KILLRANK
|
||
|
|
orte_jobid_t my_jobid;
|
||
|
|
|
||
|
|
orte_ns.get_jobid(&my_jobid, orte_process_info.my_name);
|
||
|
|
|
||
|
|
/* make sure it's my job */
|
||
|
|
if (jobid == my_jobid) {
|
||
|
|
killrank(-1, SIGKILL);
|
||
|
|
} else {
|
||
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
exit(0);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return ORTE_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static int orte_pls_cnos_terminate_orteds(orte_jobid_t jobid)
|
||
|
|
{
|
||
|
|
#ifdef HAVE_KILLRANK
|
||
|
|
orte_jobid_t my_jobid;
|
||
|
|
|
||
|
|
orte_ns.get_jobid(&my_jobid, orte_process_info.my_name);
|
||
|
|
|
||
|
|
/* make sure it's my job */
|
||
|
|
if (jobid == my_jobid) {
|
||
|
|
killrank(-1, SIGKILL);
|
||
|
|
} else {
|
||
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
exit(0);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return ORTE_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int orte_pls_cnos_terminate_proc(const orte_process_name_t* proc_name)
|
||
|
|
{
|
||
|
|
#ifdef HAVE_KILLRANK
|
||
|
|
orte_jobid_t my_jobid;
|
||
|
|
orte_jobid_t his_jobid;
|
||
|
|
orte_vpid_t his_vpid;
|
||
|
|
|
||
|
|
orte_ns.get_jobid(&my_jobid, orte_process_info.my_name);
|
||
|
|
orte_ns.get_jobid(&his_jobid, proc_name);
|
||
|
|
|
||
|
|
orte_ns.get_vpid(&his_vpid, proc_name);
|
||
|
|
|
||
|
|
/* make sure it's my job. This may end up killing me, but what
|
||
|
|
the heck. */
|
||
|
|
if (his_jobid == my_jobid) {
|
||
|
|
killrank((int) his_vpid, SIGKILL);
|
||
|
|
} else {
|
||
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
exit(0);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return ORTE_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static int orte_pls_cnos_signal_job(orte_jobid_t jobid, int32_t signal)
|
||
|
|
{
|
||
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int orte_pls_cnos_signal_proc(const orte_process_name_t* proc_name, int32_t signal)
|
||
|
|
{
|
||
|
|
return ORTE_ERR_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int orte_pls_cnos_finalize(void)
|
||
|
|
{
|
||
|
|
return ORTE_SUCCESS;
|
||
|
|
}
|