188d529beb
* Accidentally had the PLS LSF using the env SDS; switch it back to the LSF SDS This commit was SVN r15650.
130 строки
3.8 KiB
C
130 строки
3.8 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 (c) 2007 Cisco, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif /* HAVE_UNISTD_H */
|
|
#ifdef HAVE_STRING_H
|
|
#include <string.h>
|
|
#endif /* HAVE_STRING_H */
|
|
#include <ctype.h>
|
|
|
|
#include <lsf/lsbatch.h>
|
|
|
|
#include "orte/orte_constants.h"
|
|
#include "orte/mca/sds/sds.h"
|
|
#include "orte/mca/sds/base/base.h"
|
|
#include "orte/mca/sds/lsf/sds_lsf.h"
|
|
#include "orte/util/proc_info.h"
|
|
#include "opal/util/opal_environ.h"
|
|
#include "opal/mca/base/mca_base_param.h"
|
|
#include "orte/mca/ns/ns.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/ns/base/base.h"
|
|
#include "orte/util/sys_info.h"
|
|
#include "opal/util/argv.h"
|
|
|
|
orte_sds_base_module_t orte_sds_lsf_module = {
|
|
orte_sds_base_basic_contact_universe,
|
|
orte_sds_lsf_set_name,
|
|
orte_sds_lsf_finalize,
|
|
};
|
|
|
|
|
|
int orte_sds_lsf_set_name(void)
|
|
{
|
|
int rc;
|
|
int id;
|
|
char* name_string = NULL;
|
|
int lsf_nodeid;
|
|
|
|
/* start by getting our jobid, and vpid (which is the
|
|
starting vpid for the list of daemons) */
|
|
id = mca_base_param_register_string("ns", "nds", "name", NULL, NULL);
|
|
mca_base_param_lookup_string(id, &name_string);
|
|
|
|
if (name_string != NULL) {
|
|
if (ORTE_SUCCESS !=
|
|
(rc = orte_ns.convert_string_to_process_name(&(orte_process_info.my_name),
|
|
name_string))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
free(name_string);
|
|
return rc;
|
|
}
|
|
free(name_string);
|
|
} else {
|
|
orte_jobid_t jobid;
|
|
orte_vpid_t vpid;
|
|
char* jobid_string;
|
|
char* vpid_string;
|
|
|
|
id = mca_base_param_register_string("ns", "nds", "jobid", NULL, NULL);
|
|
mca_base_param_lookup_string(id, &jobid_string);
|
|
if (NULL == jobid_string) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_NOT_FOUND);
|
|
return ORTE_ERR_NOT_FOUND;
|
|
}
|
|
if (ORTE_SUCCESS !=
|
|
(rc = orte_ns.convert_string_to_jobid(&jobid, jobid_string))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return(rc);
|
|
}
|
|
|
|
id = mca_base_param_register_string("ns", "nds", "vpid", NULL, NULL);
|
|
mca_base_param_lookup_string(id, &vpid_string);
|
|
if (NULL == vpid_string) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_NOT_FOUND);
|
|
return ORTE_ERR_NOT_FOUND;
|
|
}
|
|
if (ORTE_SUCCESS !=
|
|
(rc = orte_ns.convert_string_to_vpid(&vpid, vpid_string))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return(rc);
|
|
}
|
|
|
|
if (ORTE_SUCCESS !=
|
|
(rc = orte_ns.create_process_name(&(orte_process_info.my_name),
|
|
jobid, vpid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
}
|
|
|
|
/* fix up the base name and make it the "real" name */
|
|
lsf_nodeid = atoi(getenv("LSF_PM_TASKID"));
|
|
orte_process_info.my_name->vpid = lsf_nodeid;
|
|
|
|
/* get the non-name common environmental variables */
|
|
if (ORTE_SUCCESS != (rc = orte_sds_env_get())) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
int orte_sds_lsf_finalize(void)
|
|
{
|
|
return ORTE_SUCCESS;
|
|
}
|