1
1

corrections to handle host specifications

This commit was SVN r7441.
Этот коммит содержится в:
Tim Woodall 2005-09-20 13:32:08 +00:00
родитель f0cec8ac0c
Коммит 6c885acb91
2 изменённых файлов: 27 добавлений и 9 удалений

Просмотреть файл

@ -20,6 +20,7 @@
#include "include/orte_constants.h" #include "include/orte_constants.h"
#include "opal/util/output.h" #include "opal/util/output.h"
#include "opal/util/argv.h"
#include "mca/errmgr/errmgr.h" #include "mca/errmgr/errmgr.h"
#include "mca/soh/soh_types.h" #include "mca/soh/soh_types.h"
@ -137,7 +138,11 @@ int orte_ras_base_node_query(opal_list_t* nodes)
* Query the registry for all available nodes * Query the registry for all available nodes
*/ */
int orte_ras_base_node_query_context(opal_list_t* nodes, orte_app_context_t** context, size_t num_context) int orte_ras_base_node_query_context(
opal_list_t* nodes,
orte_app_context_t** context,
size_t num_context,
bool *constrained)
{ {
size_t i, cnt; size_t i, cnt;
orte_gpr_value_t** values; orte_gpr_value_t** values;
@ -148,15 +153,23 @@ int orte_ras_base_node_query_context(opal_list_t* nodes, orte_app_context_t** co
/* expand the list of node/host specifications on the context structure into /* expand the list of node/host specifications on the context structure into
* a list of node names * a list of node names
*/ */
*constrained = false;
OBJ_CONSTRUCT(&required, opal_list_t); OBJ_CONSTRUCT(&required, opal_list_t);
for(i=0; i<num_context; i++) { for(i=0; i<num_context; i++) {
orte_app_context_map_t** map = context[i]->map_data; orte_app_context_map_t** map = context[i]->map_data;
size_t m, num_map = context[i]->num_map; size_t m, num_map = context[i]->num_map;
for(m=0; m<num_map; m++) { for(m=0; m<num_map; m++) {
if(map[m]->map_type == ORTE_APP_CONTEXT_MAP_HOSTNAME) { if(map[m]->map_type == ORTE_APP_CONTEXT_MAP_HOSTNAME) {
orte_ras_node_t* node = OBJ_NEW(orte_ras_node_t); char** hosts = opal_argv_split(map[m]->map_data, ',');
node->node_name = strdup(map[m]->map_data); char** ptr = hosts;
opal_list_append(&required, (opal_list_item_t*)node); while(*ptr) {
orte_ras_node_t* node = OBJ_NEW(orte_ras_node_t);
node->node_name = strdup(*ptr);
opal_list_append(&required, (opal_list_item_t*)node);
ptr++;
}
opal_argv_free(hosts);
*constrained = true;
} }
} }
} }
@ -221,7 +234,7 @@ int orte_ras_base_node_query_context(opal_list_t* nodes, orte_app_context_t** co
for(item = opal_list_get_first(&required); for(item = opal_list_get_first(&required);
item != opal_list_get_end(&required); item != opal_list_get_end(&required);
item = opal_list_get_next(item)) { item = opal_list_get_next(item)) {
if(strcmp(((orte_ras_node_t*)item)->node_name,node->node_name)) { if(0 == strcmp(((orte_ras_node_t*)item)->node_name,node->node_name)) {
opal_list_remove_item(&required, item); opal_list_remove_item(&required, item);
OBJ_RELEASE(item); OBJ_RELEASE(item);
found = true; found = true;
@ -239,11 +252,16 @@ int orte_ras_base_node_query_context(opal_list_t* nodes, orte_app_context_t** co
/* append any remaining specified nodes to the list with /* append any remaining specified nodes to the list with
* with default settings for slots,etc. * with default settings for slots,etc.
*/ */
while(NULL != (item = opal_list_remove_first(&required))) { if(opal_list_get_size(&required)) {
opal_list_append(&required, item); if(ORTE_SUCCESS != (rc = orte_ras_base_node_insert(&required))) {
ORTE_ERROR_LOG(rc);
return rc;
}
while(NULL != (item = opal_list_remove_first(&required))) {
opal_list_append(nodes, item);
}
} }
OBJ_DESTRUCT(&required); OBJ_DESTRUCT(&required);
if (NULL != values) free(values); if (NULL != values) free(values);
return ORTE_SUCCESS; return ORTE_SUCCESS;
} }

Просмотреть файл

@ -42,7 +42,7 @@ int orte_ras_base_node_query(opal_list_t*);
* Query the registry for all available nodes that satisfy any * Query the registry for all available nodes that satisfy any
* constraints specified on the app_context(s) * constraints specified on the app_context(s)
*/ */
int orte_ras_base_node_query_context(opal_list_t*, orte_app_context_t**, size_t num_context); int orte_ras_base_node_query_context(opal_list_t*, orte_app_context_t**, size_t num_context, bool* constrained);
/* /*
* Query the registry for all nodes allocated to a specific job * Query the registry for all nodes allocated to a specific job