160 строки
4.7 KiB
C
160 строки
4.7 KiB
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$
|
|
*/
|
|
|
|
#ifndef OMPI_PROC
|
|
#define OMPI_PROC
|
|
|
|
#include "include/types.h"
|
|
#include "opal/class/opal_list.h"
|
|
#include "dps/dps_types.h"
|
|
#include "opal/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 {
|
|
/** allow proc to be placed on a list */
|
|
opal_list_item_t super;
|
|
/** this process' name */
|
|
orte_process_name_t proc_name;
|
|
/** PML specific proc data */
|
|
struct mca_pml_proc_t* proc_pml;
|
|
/** MCA module exchange data */
|
|
opal_object_t* proc_modex;
|
|
/** architecture of this process */
|
|
uint32_t proc_arch;
|
|
/** process-wide convertor */
|
|
struct ompi_convertor_t* proc_convertor;
|
|
/** process-wide lock */
|
|
opal_mutex_t proc_lock;
|
|
/** flags for this proc */
|
|
uint8_t proc_flags;
|
|
};
|
|
/**
|
|
* Convenience typedef
|
|
*/
|
|
typedef struct ompi_proc_t ompi_proc_t;
|
|
OMPI_DECLSPEC extern ompi_proc_t* ompi_proc_local_proc;
|
|
|
|
/*
|
|
* Flags
|
|
*/
|
|
|
|
/**
|
|
* Flag to indicate that the proc is on the same node as the local proc
|
|
*/
|
|
#define OMPI_PROC_FLAG_LOCAL 0x01
|
|
|
|
|
|
/**
|
|
* 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 */
|
|
|