99 строки
3.3 KiB
C
99 строки
3.3 KiB
C
/*
|
|
* 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$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include "orte/orte_constants.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "opal/util/output.h"
|
|
|
|
#include "orte/dss/dss.h"
|
|
#include "orte/mca/ns/ns.h"
|
|
#include "orte/mca/gpr/gpr.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
|
|
#include "orte/runtime/orte_wakeup.h"
|
|
|
|
/* orterun will only wakeup when all procs IN THE ROOT JOB report terminated. In some cases,
|
|
* such as when an orted fails to start, we have a situation where the root job's processes
|
|
* cannot report back as terminated or aborted. For those situations, we force the issue by
|
|
* deliberately causing the TERMINATE trigger on the root job to fire
|
|
*/
|
|
int orte_wakeup(orte_jobid_t job) {
|
|
int rc;
|
|
orte_vpid_t root, range;
|
|
char *segment;
|
|
orte_std_cntr_t num;
|
|
char *tokens[] = {
|
|
ORTE_JOB_GLOBALS,
|
|
NULL
|
|
};
|
|
orte_data_value_t dval = ORTE_DATA_VALUE_EMPTY;
|
|
|
|
/* first, let's ensure we cause any triggers on this job to fire */
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_vpid_range(job, &range))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_schema.get_job_segment_name(&segment, job))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
num = range;
|
|
if (ORTE_SUCCESS != (rc = orte_dss.set(&dval, (void*)&num, ORTE_STD_CNTR))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.put_1(ORTE_GPR_OVERWRITE | ORTE_GPR_TOKENS_AND | ORTE_GPR_KEYS_OR,
|
|
segment, tokens, ORTE_PROC_NUM_TERMINATED, &dval))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
/* clear the dval for future use */
|
|
dval.data = NULL;
|
|
|
|
/* now let's get the root job - note, this may not actually happen IF the
|
|
* prior trigger was enough to cause us to terminate the job
|
|
*/
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_root_job(&root, job))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
|
|
/* if the root and the specified job are the same, stop here */
|
|
if (root == job) {
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_vpid_range(root, &range))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_schema.get_job_segment_name(&segment, root))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
num = range;
|
|
if (ORTE_SUCCESS != (rc = orte_dss.set(&dval, (void*)&num, ORTE_STD_CNTR))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.put_1(ORTE_GPR_OVERWRITE | ORTE_GPR_TOKENS_AND | ORTE_GPR_KEYS_OR,
|
|
segment, tokens, ORTE_PROC_NUM_TERMINATED, &dval))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|