94160da4c0
- Change ompi_proc_world() to only return the procs in this job (as
opposed to all of them)
- Add a subscription that fires during MPI_INIT (stg1) for figuring
out which procs are on my local node. Need to figure out what to do
in the esoteric cases -- but the obvious one (Red Storm), where
subscriptions are never fired, is ok, because by definition, no
other procs will be on my node, so their default value (not on my
node) is ok.
--> Need to have RHC check this code; it seems to work, but I think
I'm getting too much data back from the subscription.
- End result is that any proc that is on my node will have its
OMPI_PROC_FLAG_LOCAL bit set on its proc->proc_flags field.
- Added/corrected a few comments in proc.h.
This commit was SVN r6507.
158 строки
4.5 KiB
C
158 строки
4.5 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 "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 */
|
|
|