8736a1c138
http://www.open-mpi.org/community/lists/devel/2014/05/14822.php Revamp the ORTE global data structures to reduce memory footprint and add new features. Add ability to control/set cpu frequency, though this can only be done if the sys admin has setup the system to support it (or you run as root). This commit was SVN r31916.
140 строки
3.5 KiB
C
140 строки
3.5 KiB
C
/*
|
|
* Copyright (c) 2014 Intel, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
|
|
#include "orte_config.h"
|
|
|
|
#include "opal/util/fd.h"
|
|
#include "opal/util/show_help.h"
|
|
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/odls/odls_types.h"
|
|
|
|
#include "orte/mca/rtc/base/base.h"
|
|
|
|
void orte_rtc_base_assign(orte_job_t *jdata)
|
|
{
|
|
orte_rtc_base_selected_module_t *active;
|
|
|
|
OPAL_LIST_FOREACH(active, &orte_rtc_base.actives, orte_rtc_base_selected_module_t) {
|
|
if (NULL != active->module->assign) {
|
|
/* give this module a chance to operate on it */
|
|
active->module->assign(jdata);
|
|
}
|
|
}
|
|
}
|
|
|
|
void orte_rtc_base_set(orte_job_t *jdata, orte_proc_t *proc,
|
|
char ***environ_copy, int error_fd)
|
|
{
|
|
orte_rtc_base_selected_module_t *active;
|
|
|
|
OPAL_LIST_FOREACH(active, &orte_rtc_base.actives, orte_rtc_base_selected_module_t) {
|
|
if (NULL != active->module->set) {
|
|
/* give this module a chance to operate on it */
|
|
active->module->set(jdata, proc, environ_copy, error_fd);
|
|
}
|
|
}
|
|
}
|
|
|
|
void orte_rtc_base_get_avail_vals(opal_list_t *vals)
|
|
{
|
|
orte_rtc_base_selected_module_t *active;
|
|
|
|
OPAL_LIST_FOREACH(active, &orte_rtc_base.actives, orte_rtc_base_selected_module_t) {
|
|
if (NULL != active->module->get_available_values) {
|
|
/* give this module a chance to operate on it */
|
|
active->module->get_available_values(vals);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
static int write_help_msg(int fd, orte_odls_pipe_err_msg_t *msg, const char *file,
|
|
const char *topic, va_list ap)
|
|
{
|
|
int ret;
|
|
char *str;
|
|
|
|
if (NULL == file || NULL == topic) {
|
|
return OPAL_ERR_BAD_PARAM;
|
|
}
|
|
|
|
str = opal_show_help_vstring(file, topic, true, ap);
|
|
|
|
msg->file_str_len = (int) strlen(file);
|
|
if (msg->file_str_len > ORTE_ODLS_MAX_FILE_LEN) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
msg->topic_str_len = (int) strlen(topic);
|
|
if (msg->topic_str_len > ORTE_ODLS_MAX_TOPIC_LEN) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
msg->msg_str_len = (int) strlen(str);
|
|
|
|
/* Only keep writing if each write() succeeds */
|
|
if (OPAL_SUCCESS != (ret = opal_fd_write(fd, sizeof(*msg), msg))) {
|
|
goto out;
|
|
}
|
|
if (msg->file_str_len > 0 &&
|
|
OPAL_SUCCESS != (ret = opal_fd_write(fd, msg->file_str_len, file))) {
|
|
goto out;
|
|
}
|
|
if (msg->topic_str_len > 0 &&
|
|
OPAL_SUCCESS != (ret = opal_fd_write(fd, msg->topic_str_len, topic))) {
|
|
goto out;
|
|
}
|
|
if (msg->msg_str_len > 0 &&
|
|
OPAL_SUCCESS != (ret = opal_fd_write(fd, msg->msg_str_len, str))) {
|
|
goto out;
|
|
}
|
|
|
|
out:
|
|
free(str);
|
|
return ret;
|
|
}
|
|
|
|
int orte_rtc_base_send_warn_show_help(int fd, const char *file,
|
|
const char *topic, ...)
|
|
{
|
|
int ret;
|
|
va_list ap;
|
|
orte_odls_pipe_err_msg_t msg;
|
|
|
|
msg.fatal = false;
|
|
msg.exit_status = 0; /* ignored */
|
|
|
|
/* Send it */
|
|
va_start(ap, topic);
|
|
ret = write_help_msg(fd, &msg, file, topic, ap);
|
|
va_end(ap);
|
|
|
|
return ret;
|
|
}
|
|
|
|
void orte_rtc_base_send_error_show_help(int fd, int exit_status,
|
|
const char *file, const char *topic, ...)
|
|
{
|
|
va_list ap;
|
|
orte_odls_pipe_err_msg_t msg;
|
|
|
|
msg.fatal = true;
|
|
msg.exit_status = exit_status;
|
|
|
|
/* Send it */
|
|
va_start(ap, topic);
|
|
write_help_msg(fd, &msg, file, topic, ap);
|
|
va_end(ap);
|
|
|
|
exit(exit_status);
|
|
}
|
|
|