5dfd54c778
Clean up the remainder of the size_t references in the runtime itself. Convert to orte_std_cntr_t wherever it makes sense (only avoid those places where the actual memory size is referenced). Remove the obsolete oob barrier function (we actually obsoleted it a long time ago - just never bothered to clean it up). I have done my best to go through all the components and catch everything, even if I couldn't test compile them since I wasn't on that type of system. Still, I cannot guarantee that problems won't show up when you test this on specific systems. Usually, these will just show as "warning: comparison between signed and unsigned" notes which are easily fixed (just change a size_t to orte_std_cntr_t). In some places, people didn't use size_t, but instead used some other variant (e.g., I found several places with uint32_t). I tried to catch all of them, but... Once we get all the instances caught and fixed, this should once and for all resolve many of the heterogeneity problems. This commit was SVN r11204.
374 строки
9.3 KiB
C
374 строки
9.3 KiB
C
/* -*- 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$
|
|
*/
|
|
/** @file:
|
|
*
|
|
* Convenience functions for accessing the General Purpose Registry
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* includes
|
|
*/
|
|
#include "orte_config.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "orte/orte_constants.h"
|
|
|
|
#include "opal/util/output.h"
|
|
|
|
#include "orte/dss/dss.h"
|
|
#include "orte/util/proc_info.h"
|
|
#include "orte/util/sys_info.h"
|
|
#include "orte/mca/ns/ns.h"
|
|
#include "orte/mca/gpr/gpr.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
|
|
#include "orte/mca/schema/base/base.h"
|
|
|
|
|
|
int orte_schema_base_get_proc_tokens(char ***proc_tokens, orte_std_cntr_t* num_tokens, orte_process_name_t *proc)
|
|
{
|
|
int rc;
|
|
char** tokens;
|
|
char* vpid_string;
|
|
|
|
tokens = (char**)malloc(3 * sizeof(char*));
|
|
if (NULL == tokens) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_proc_name_string(&tokens[0], proc))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
goto CLEANUP;
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_vpid_string(&vpid_string, proc))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
goto CLEANUP;
|
|
}
|
|
|
|
asprintf(&tokens[1], "%s-%s", ORTE_VPID_KEY, vpid_string);
|
|
free(vpid_string);
|
|
tokens[2] = NULL;
|
|
*proc_tokens = tokens;
|
|
if(num_tokens != NULL)
|
|
*num_tokens = 2;
|
|
return ORTE_SUCCESS;
|
|
|
|
CLEANUP:
|
|
if (NULL != tokens) {
|
|
if (NULL != tokens[0])
|
|
free(tokens[0]);
|
|
if (NULL != tokens[1])
|
|
free(tokens[1]);
|
|
free(tokens);
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
int orte_schema_base_get_node_tokens(char ***node_tokens, orte_std_cntr_t* num_tokens, orte_cellid_t cellid, char *nodename)
|
|
{
|
|
int rc;
|
|
char** tokens;
|
|
char* cellid_string;
|
|
|
|
tokens = (char**)malloc(3 * sizeof(char*));
|
|
if (NULL == tokens) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_cellid_to_string(&cellid_string, cellid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
goto CLEANUP;
|
|
}
|
|
|
|
asprintf(&tokens[0], "%s-%s", ORTE_CELLID_KEY, cellid_string);
|
|
free(cellid_string);
|
|
tokens[1] = strdup(nodename);
|
|
tokens[2] = NULL;
|
|
*node_tokens = tokens;
|
|
if(num_tokens != NULL)
|
|
*num_tokens = 2;
|
|
return ORTE_SUCCESS;
|
|
|
|
CLEANUP:
|
|
if (NULL != tokens) {
|
|
if (NULL != tokens[0])
|
|
free(tokens[0]);
|
|
if (NULL != tokens[1])
|
|
free(tokens[1]);
|
|
free(tokens);
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
int orte_schema_base_get_job_tokens(char ***job_tokens, orte_std_cntr_t* num_tokens, orte_jobid_t jobid)
|
|
{
|
|
int rc;
|
|
char** tokens;
|
|
char* jobid_string;
|
|
|
|
tokens = (char**)malloc(2 * sizeof(char*));
|
|
if (NULL == tokens) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_jobid_to_string(&jobid_string, jobid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
goto CLEANUP;
|
|
}
|
|
|
|
asprintf(&tokens[0], "%s-%s", ORTE_JOBID_KEY, jobid_string);
|
|
free(jobid_string);
|
|
tokens[1] = NULL;
|
|
*job_tokens = tokens;
|
|
if(num_tokens != NULL)
|
|
*num_tokens = 1;
|
|
return ORTE_SUCCESS;
|
|
|
|
CLEANUP:
|
|
if (NULL != tokens) {
|
|
if (NULL != tokens[0]) free(tokens[0]);
|
|
free(tokens);
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
int orte_schema_base_get_cell_tokens(char ***cell_tokens, orte_std_cntr_t* num_tokens, orte_cellid_t cellid)
|
|
{
|
|
int rc;
|
|
char **tokens;
|
|
char *site, *resource, *cellstr;
|
|
orte_std_cntr_t n, i;
|
|
|
|
n = 1;
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_cell_info(cellid, &site, &resource))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_cellid_to_string(&cellstr, cellid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
if (NULL != site) n++;
|
|
if (NULL != resource) n++;
|
|
|
|
tokens = (char**)malloc(n * sizeof(char*));
|
|
if (NULL == tokens) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
tokens[0] = cellstr;
|
|
i=1;
|
|
if (NULL != site) {
|
|
tokens[i] = site;
|
|
i++;
|
|
}
|
|
if (NULL != resource) {
|
|
tokens[i] = resource;
|
|
}
|
|
|
|
*num_tokens = n;
|
|
*cell_tokens = tokens;
|
|
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
int orte_schema_base_get_job_segment_name(char **name, orte_jobid_t jobid)
|
|
{
|
|
char *jobidstring;
|
|
int rc;
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_jobid_to_string(&jobidstring, jobid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
if (0 > asprintf(name, "%s-%s", ORTE_JOB_SEGMENT, jobidstring)) {
|
|
free(jobidstring);
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
free(jobidstring);
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
int orte_schema_base_extract_jobid_from_segment_name(orte_jobid_t *jobid, char *name)
|
|
{
|
|
char *jobstring;
|
|
orte_jobid_t job;
|
|
int rc;
|
|
|
|
jobstring = strrchr(name, '-');
|
|
if (NULL == jobstring) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
jobstring++;
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_string_to_jobid(&job, jobstring))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
opal_output(0, "[%lu,%lu,%lu] %s\n", ORTE_NAME_ARGS(orte_process_info.my_name), jobstring);
|
|
return rc;
|
|
}
|
|
*jobid = job;
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
/**
|
|
* Set the process mapping in the registry.
|
|
*/
|
|
|
|
int orte_schema_base_store_my_info(void)
|
|
{
|
|
int rc;
|
|
orte_gpr_value_t *value;
|
|
orte_jobid_t jobid;
|
|
char *segment;
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_ns.get_jobid(&jobid, orte_process_info.my_name))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_schema_base_get_job_segment_name(&segment, jobid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.create_value(&value, ORTE_GPR_OVERWRITE, segment, 2, 0))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
free(segment);
|
|
return rc;
|
|
}
|
|
free(segment);
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_schema_base_get_proc_tokens(&(value->tokens),
|
|
&(value->num_tokens), orte_process_info.my_name))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
OBJ_RELEASE(value);
|
|
return rc;
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.create_keyval(&(value->keyvals[0]), ORTE_PROC_LOCAL_PID_KEY, ORTE_PID, &(orte_process_info.pid)))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
OBJ_RELEASE(value);
|
|
return rc;
|
|
}
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.create_keyval(&(value->keyvals[1]), ORTE_NODE_NAME_KEY, ORTE_STRING, orte_system_info.nodename))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
OBJ_RELEASE(value);
|
|
return rc;
|
|
}
|
|
|
|
/* insert values into registry */
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.put(1, &value))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
}
|
|
|
|
/* cleanup memory */
|
|
OBJ_RELEASE(value);
|
|
|
|
return rc;
|
|
}
|
|
|
|
int orte_schema_base_get_std_trigger_name(char **name,
|
|
char *trigger,
|
|
orte_jobid_t jobid)
|
|
{
|
|
char *jobidstring;
|
|
int rc;
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_jobid_to_string(&jobidstring, jobid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
if (0 > asprintf(name, "%s-%s", trigger, jobidstring)) {
|
|
free(jobidstring);
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
free(jobidstring);
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
bool orte_schema_base_check_std_trigger_name(char *name, char *trig)
|
|
{
|
|
if (NULL == name || NULL == trig) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
|
|
return false;
|
|
}
|
|
|
|
if (0 == strncmp(name, trig, strlen(trig))) return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
int orte_schema_base_extract_jobid_from_std_trigger_name(orte_jobid_t *jobid, char *trig)
|
|
{
|
|
char *jobstring;
|
|
orte_jobid_t job;
|
|
int rc;
|
|
|
|
jobstring = strrchr(trig, '-');
|
|
if (NULL == jobstring) {
|
|
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
|
|
return ORTE_ERR_BAD_PARAM;
|
|
}
|
|
jobstring++;
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_string_to_jobid(&job, jobstring))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
*jobid = job;
|
|
return ORTE_SUCCESS;
|
|
}
|
|
|
|
|
|
int orte_schema_base_get_std_subscription_name(char **name,
|
|
char *subscription,
|
|
orte_jobid_t jobid)
|
|
{
|
|
char *jobidstring;
|
|
int rc;
|
|
|
|
if (ORTE_SUCCESS != (rc = orte_ns.convert_jobid_to_string(&jobidstring, jobid))) {
|
|
ORTE_ERROR_LOG(rc);
|
|
return rc;
|
|
}
|
|
|
|
if (0 > asprintf(name, "%s-%s", subscription, jobidstring)) {
|
|
free(jobidstring);
|
|
ORTE_ERROR_LOG(ORTE_ERR_OUT_OF_RESOURCE);
|
|
return ORTE_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
free(jobidstring);
|
|
return ORTE_SUCCESS;
|
|
}
|