142 строки
4.3 KiB
C
142 строки
4.3 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 (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$
|
|
*/
|
|
|
|
#ifndef OMPI_PROC
|
|
#define OMPI_PROC
|
|
|
|
#include "include/types.h"
|
|
#include "opal/class/opal_list.h"
|
|
#include "dps/dps_types.h"
|
|
#include "threads/mutex.h"
|
|
|
|
#include "mca/ns/ns_types.h"
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
OMPI_DECLSPEC extern opal_class_t ompi_proc_t_class;
|
|
|
|
struct ompi_proc_t {
|
|
opal_list_item_t super; /* allow proc to be placed on a list */
|
|
orte_process_name_t proc_name;
|
|
struct mca_pml_proc_t* proc_pml; /* PML specific proc data */
|
|
opal_object_t* proc_modex; /* MCA module exchange data */
|
|
uint32_t proc_arch;
|
|
struct ompi_convertor_t* proc_convertor;
|
|
ompi_mutex_t proc_lock;
|
|
|
|
/* JMS: need to have the following information:
|
|
- how am i [mpi] connected (bitmap): spawn (parent/child),
|
|
connect, accept, joint
|
|
*/
|
|
};
|
|
typedef struct ompi_proc_t ompi_proc_t;
|
|
OMPI_DECLSPEC extern ompi_proc_t* ompi_proc_local_proc;
|
|
|
|
|
|
/**
|
|
* Query the run-time environment and build list of available proc instances.
|
|
*/
|
|
int ompi_proc_init(void);
|
|
|
|
/**
|
|
* Release the processes at the end of the application
|
|
*/
|
|
int ompi_proc_finalize(void);
|
|
|
|
/**
|
|
* Returns the list of proc instances associated with this job.
|
|
*/
|
|
ompi_proc_t** ompi_proc_world(size_t* size);
|
|
|
|
/**
|
|
* Returns the list of all known proc instances.
|
|
*/
|
|
ompi_proc_t** ompi_proc_all(size_t* size);
|
|
|
|
/**
|
|
* Returns a list (of one) proc instances.
|
|
*/
|
|
ompi_proc_t** ompi_proc_self(size_t* size);
|
|
|
|
/**
|
|
* Returns the proc instance corresponding to the local proc.
|
|
*/
|
|
static inline ompi_proc_t* ompi_proc_local(void)
|
|
{
|
|
extern ompi_proc_t* ompi_proc_local_proc;
|
|
return ompi_proc_local_proc;
|
|
}
|
|
|
|
/**
|
|
* Returns the proc instance for a given name
|
|
*/
|
|
ompi_proc_t * ompi_proc_find ( const orte_process_name_t* name );
|
|
|
|
|
|
/**
|
|
* Returns the proc instance for a given name. If process is not known,
|
|
* we add it to the process list.
|
|
*/
|
|
ompi_proc_t * ompi_proc_find_and_add ( const orte_process_name_t* name, bool* isnew );
|
|
|
|
/**
|
|
* INPUT: ompi_proc_t **proclist : list of process pointers
|
|
* INPUT: int proclistsize : lenght of the proclist array
|
|
* IN/OUT: orte_buffer_t *buf : an orte_buffer containing the packed names
|
|
*
|
|
* The function takes a list of ompi_proc_t pointers (e.g. as given in groups)
|
|
* and returns a buffer, which contains a list of 'packed' ompi_process_name_t
|
|
* objects, in the same order as provided in proclist.
|
|
* By 'packed' we mean, that the buffer should be able to be sent across
|
|
* heterogeneous environments (e.g. it has to be big endian, ilp32 ).
|
|
* This buffer could be sent using MPI_Send using MPI_BYTE or OOB_send_packed.
|
|
*
|
|
* Return values:
|
|
* OMPI_SUCCESS on success
|
|
* OMPI_ERROR: other errors
|
|
*/
|
|
int ompi_proc_get_namebuf ( ompi_proc_t **proclist, int proclistsize,
|
|
orte_buffer_t *buf);
|
|
|
|
|
|
|
|
/**
|
|
* INPUT: orte_buffer_t *buf : orte_buffer containing the packed names
|
|
* INPUT: int proclistsize : number of expected proc-pointres
|
|
* OUTPUT: ompi_proc_t ***proclist : list of process pointers
|
|
*
|
|
* This function 'unpacks' the ompi_process_name_t structures and looks
|
|
* the according ompi_proc_t structure up. If the 'environment' does not
|
|
* find a proc-structure, it tries to look it up from the name_service or
|
|
* any other service involved in such an operation (this is required for
|
|
* the dynamic MPI-2 scenarios). The buffer allocated by
|
|
* ompi_proc_get_proclist will not be returned to the 'environment'.
|
|
*
|
|
* Return value:
|
|
* OMPI_SUCCESS on success
|
|
* OMPI_ERROR else
|
|
*/
|
|
|
|
int ompi_proc_get_proclist (orte_buffer_t *buf, int proclistsize, ompi_proc_t ***proclist);
|
|
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
}
|
|
#endif
|
|
#endif /* OMPI_PROC */
|
|
|