254 строки
6.8 KiB
C
254 строки
6.8 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University.
|
|
* All rights reserved.
|
|
* Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "threads/mutex.h"
|
|
#include "util/output.h"
|
|
#include "proc/proc.h"
|
|
#include "mca/pcmclient/pcmclient.h"
|
|
#include "mca/pcmclient/base/base.h"
|
|
#include "mca/oob/oob.h"
|
|
#include "mca/ns/base/base.h"
|
|
#include "mca/pml/pml.h"
|
|
|
|
|
|
static ompi_list_t ompi_proc_list;
|
|
static ompi_mutex_t ompi_proc_lock;
|
|
ompi_proc_t* ompi_proc_local_proc = NULL;
|
|
|
|
static void ompi_proc_construct(ompi_proc_t* proc);
|
|
static void ompi_proc_destruct(ompi_proc_t* proc);
|
|
|
|
|
|
OBJ_CLASS_INSTANCE(
|
|
ompi_proc_t,
|
|
ompi_list_item_t,
|
|
ompi_proc_construct,
|
|
ompi_proc_destruct
|
|
);
|
|
|
|
|
|
void ompi_proc_construct(ompi_proc_t* proc)
|
|
{
|
|
proc->proc_pml = NULL;
|
|
proc->proc_modex = NULL;
|
|
OBJ_CONSTRUCT(&proc->proc_lock, ompi_mutex_t);
|
|
|
|
/* FIX - need to determine remote process architecture */
|
|
proc->proc_convertor = ompi_convertor_create(0, 0);
|
|
proc->proc_arch = 0;
|
|
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
ompi_list_append(&ompi_proc_list, (ompi_list_item_t*)proc);
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
}
|
|
|
|
|
|
void ompi_proc_destruct(ompi_proc_t* proc)
|
|
{
|
|
if(proc->proc_modex != NULL)
|
|
OBJ_RELEASE(proc->proc_modex);
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
ompi_list_remove_item(&ompi_proc_list, (ompi_list_item_t*)proc);
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
OBJ_DESTRUCT(&proc->proc_lock);
|
|
}
|
|
|
|
|
|
int ompi_proc_init(void)
|
|
{
|
|
ompi_process_name_t *peers;
|
|
ompi_process_name_t *self;
|
|
size_t i, npeers;
|
|
int rc;
|
|
|
|
OBJ_CONSTRUCT(&ompi_proc_list, ompi_list_t);
|
|
OBJ_CONSTRUCT(&ompi_proc_lock, ompi_mutex_t);
|
|
|
|
if(OMPI_SUCCESS != (rc = mca_pcmclient.pcmclient_get_peers(&peers,
|
|
&npeers))) {
|
|
ompi_output(0, "ompi_proc_init: mca_pcm.pcm_peers failed with errno=%d", rc);
|
|
return rc;
|
|
}
|
|
if(NULL == (self = mca_pcmclient.pcmclient_get_self())) {
|
|
ompi_output(0, "ompi_proc_init: mca_pcm.pcm_self failed with errno=%d", rc);
|
|
return rc;
|
|
}
|
|
|
|
for(i=0; i<npeers; i++) {
|
|
ompi_proc_t *proc = OBJ_NEW(ompi_proc_t);
|
|
proc->proc_name = peers[i];
|
|
if( peers + i == self ) {
|
|
ompi_proc_local_proc = proc;
|
|
}
|
|
}
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|
|
ompi_proc_t** ompi_proc_world(size_t *size)
|
|
{
|
|
ompi_proc_t **procs =
|
|
(ompi_proc_t**) malloc(ompi_list_get_size(&ompi_proc_list) * sizeof(ompi_proc_t*));
|
|
ompi_proc_t *proc;
|
|
size_t count = 0;
|
|
|
|
if(NULL == procs)
|
|
return NULL;
|
|
|
|
/* return only the procs that match this jobid */
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
for(proc = (ompi_proc_t*)ompi_list_get_first(&ompi_proc_list);
|
|
proc != (ompi_proc_t*)ompi_list_get_end(&ompi_proc_list);
|
|
proc = (ompi_proc_t*)ompi_list_get_next(proc)) {
|
|
/* TSW - FIX */
|
|
procs[count++] = proc;
|
|
}
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
*size = count;
|
|
return procs;
|
|
}
|
|
|
|
|
|
ompi_proc_t** ompi_proc_all(size_t* size)
|
|
{
|
|
ompi_proc_t **procs =
|
|
(ompi_proc_t**) malloc(ompi_list_get_size(&ompi_proc_list) * sizeof(ompi_proc_t*));
|
|
ompi_proc_t *proc;
|
|
size_t count = 0;
|
|
|
|
if(NULL == procs)
|
|
return NULL;
|
|
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
for(proc = (ompi_proc_t*)ompi_list_get_first(&ompi_proc_list);
|
|
proc != (ompi_proc_t*)ompi_list_get_end(&ompi_proc_list);
|
|
proc = (ompi_proc_t*)ompi_list_get_next(proc)) {
|
|
OBJ_RETAIN(proc);
|
|
procs[count++] = proc;
|
|
}
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
*size = count;
|
|
return procs;
|
|
}
|
|
|
|
|
|
ompi_proc_t** ompi_proc_self(size_t* size)
|
|
{
|
|
ompi_proc_t **procs = (ompi_proc_t**) malloc(sizeof(ompi_proc_t*));
|
|
if(NULL == procs)
|
|
return NULL;
|
|
OBJ_RETAIN(ompi_proc_local_proc);
|
|
*procs = ompi_proc_local_proc;
|
|
*size = 1;
|
|
return procs;
|
|
}
|
|
|
|
ompi_proc_t * ompi_proc_find ( const ompi_process_name_t * name )
|
|
{
|
|
ompi_proc_t *proc, *rproc=NULL;
|
|
ompi_ns_cmp_bitmask_t mask;
|
|
|
|
/* return the proc-struct which matches this jobid+process id */
|
|
|
|
mask = OMPI_NS_CMP_CELLID | OMPI_NS_CMP_JOBID | OMPI_NS_CMP_VPID;
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
for(proc = (ompi_proc_t*)ompi_list_get_first(&ompi_proc_list);
|
|
proc != (ompi_proc_t*)ompi_list_get_end(&ompi_proc_list);
|
|
proc = (ompi_proc_t*)ompi_list_get_next(proc)) {
|
|
if (0 == ompi_name_server.compare(mask, &proc->proc_name, name))
|
|
{
|
|
rproc = proc;
|
|
break;
|
|
}
|
|
}
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
return rproc;
|
|
}
|
|
|
|
|
|
ompi_proc_t * ompi_proc_find_and_add ( const ompi_process_name_t * name, bool* isnew )
|
|
{
|
|
ompi_proc_t *proc, *rproc=NULL;
|
|
ompi_ns_cmp_bitmask_t mask;
|
|
|
|
/* return the proc-struct which matches this jobid+process id */
|
|
mask = OMPI_NS_CMP_CELLID | OMPI_NS_CMP_JOBID | OMPI_NS_CMP_VPID;
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
for(proc = (ompi_proc_t*)ompi_list_get_first(&ompi_proc_list);
|
|
proc != (ompi_proc_t*)ompi_list_get_end(&ompi_proc_list);
|
|
proc = (ompi_proc_t*)ompi_list_get_next(proc)) {
|
|
if (0 == ompi_name_server.compare(mask, &proc->proc_name, name))
|
|
{
|
|
*isnew = false;
|
|
rproc = proc;
|
|
break;
|
|
}
|
|
}
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
|
|
if ( NULL == rproc ) {
|
|
ompi_proc_t *tproc = OBJ_NEW(ompi_proc_t);
|
|
rproc = tproc;
|
|
rproc->proc_name = *name;
|
|
*isnew = true;
|
|
}
|
|
return rproc;
|
|
}
|
|
|
|
int ompi_proc_get_namebuf ( ompi_proc_t **proclist, int proclistsize,
|
|
ompi_buffer_t buf)
|
|
{
|
|
int i;
|
|
|
|
/* buf has to be initiated in the calling function */
|
|
|
|
OMPI_THREAD_LOCK(&ompi_proc_lock);
|
|
for (i=0; i<proclistsize; i++) {
|
|
ompi_pack(buf, &(proclist[i]->proc_name), 1, OMPI_NAME);
|
|
}
|
|
OMPI_THREAD_UNLOCK(&ompi_proc_lock);
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|
|
int ompi_proc_get_proclist (ompi_buffer_t buf, int proclistsize, ompi_proc_t ***proclist)
|
|
{
|
|
int i;
|
|
ompi_proc_t **plist=NULL;
|
|
ompi_process_name_t name;
|
|
bool isnew = false;
|
|
|
|
/* do not free plist *ever*, since it is used in the remote group structure
|
|
of a communicator */
|
|
plist = (ompi_proc_t **) calloc (proclistsize, sizeof (ompi_proc_t *));
|
|
if ( NULL == plist ) {
|
|
return OMPI_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
for ( i=0; i<proclistsize; i++ ){
|
|
ompi_unpack(buf, &name, 1, OMPI_NAME);
|
|
plist[i] = ompi_proc_find_and_add ( &name, &isnew );
|
|
if(isnew) {
|
|
mca_pml.pml_add_procs(&plist[i], 1);
|
|
}
|
|
}
|
|
*proclist = plist;
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|