24a7e7aa34
To support the new mpool two changes were made to the mpool infrastructure:
1) Added an mpool flag to indicate that an mpool does not need the memory
hooks to use the leave pinned protocols. This flag is checked in the
mpool lookup.
2) Add a mpool context to the base registration. This new member is used
by the udreg mpool to store the udreg context associated with the
particular registration. The new member will not break the ABI
compatibility as the new member is only currently used by the udreg
mpool.
Dynamics support for Cray systems makes use of the global rank provided by
orte to give the ugni library a unique rank for each process. Dynamics
support is not available under direct-launch (srun.)
cmr=v1.7.4
This commit was SVN r29719.
118 строки
2.9 KiB
C
118 строки
2.9 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2011-2013 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* Copyright (c) 2011 UT-Battelle, LLC. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include "opal/util/output.h"
|
|
#include "opal/class/opal_list.h"
|
|
#include "opal/include/opal/prefetch.h"
|
|
#include "opal_stdint.h"
|
|
|
|
#include "ompi/include/ompi/constants.h"
|
|
#include "ompi/runtime/ompi_module_exchange.h"
|
|
#include "ompi/proc/proc.h"
|
|
|
|
#include <errno.h>
|
|
#include <stdint.h>
|
|
#include <sys/types.h>
|
|
#include <assert.h>
|
|
#include <sys/time.h>
|
|
#include <gni_pub.h>
|
|
|
|
#include "common_ugni_ep.h"
|
|
|
|
#if !defined(MPI_COMMON_UGNI_H)
|
|
#define MPI_COMMON_UGNI_H
|
|
|
|
struct ompi_common_ugni_modex_t {
|
|
uint32_t addr;
|
|
int id;
|
|
};
|
|
typedef struct ompi_common_ugni_modex_t ompi_common_ugni_modex_t;
|
|
|
|
struct ompi_common_ugni_device_t {
|
|
opal_object_t super;
|
|
|
|
gni_nic_handle_t dev_handle;
|
|
|
|
/* Minor number of the Gemini NIC */
|
|
int32_t dev_id;
|
|
uint32_t dev_pe_addr;
|
|
uint32_t dev_addr;
|
|
uint32_t dev_cpu_id;
|
|
|
|
size_t dev_ep_count;
|
|
void *btl_ctx;
|
|
};
|
|
typedef struct ompi_common_ugni_device_t ompi_common_ugni_device_t;
|
|
|
|
struct ompi_common_ugni_module_t {
|
|
/* protection tag */
|
|
uint8_t ptag;
|
|
|
|
/* unique id for this process assigned by the system */
|
|
uint32_t cookie;
|
|
|
|
/* communication domain handle */
|
|
gni_cdm_handle_t cd_handle;
|
|
|
|
/* device count. to be used if we have more than 1 common per ugni device */
|
|
int device_count;
|
|
ompi_common_ugni_device_t *devices;
|
|
|
|
int rdma_max_retries;
|
|
};
|
|
typedef struct ompi_common_ugni_module_t ompi_common_ugni_module_t;
|
|
|
|
struct ompi_common_ugni_post_desc_t {
|
|
gni_post_descriptor_t base;
|
|
|
|
ompi_common_ugni_endpoint_t *endpoint;
|
|
int tries;
|
|
};
|
|
typedef struct ompi_common_ugni_post_desc_t ompi_common_ugni_post_desc_t;
|
|
|
|
extern ompi_common_ugni_module_t ompi_common_ugni_module;
|
|
extern mca_base_component_t ompi_common_ugni_component;
|
|
|
|
static inline int
|
|
ompi_common_rc_ugni_to_ompi (gni_return_t rc)
|
|
{
|
|
int codes[] = {OMPI_SUCCESS,
|
|
OMPI_ERR_RESOURCE_BUSY,
|
|
OMPI_ERR_BAD_PARAM,
|
|
OMPI_ERR_OUT_OF_RESOURCE,
|
|
OMPI_ERR_TIMEOUT,
|
|
OMPI_ERR_PERM,
|
|
OMPI_ERROR,
|
|
OMPI_ERR_BAD_PARAM,
|
|
OMPI_ERR_BAD_PARAM,
|
|
OMPI_ERR_NOT_FOUND,
|
|
OMPI_ERR_VALUE_OUT_OF_BOUNDS,
|
|
OMPI_ERROR,
|
|
OMPI_ERR_NOT_SUPPORTED,
|
|
OMPI_ERR_OUT_OF_RESOURCE};
|
|
return codes[rc];
|
|
}
|
|
|
|
/*
|
|
* Initialize uGNI communication domain and device(s).
|
|
*/
|
|
int ompi_common_ugni_init (void);
|
|
|
|
/*
|
|
* Finalize uGNI communication domain and device(s).
|
|
*/
|
|
int ompi_common_ugni_fini (void);
|
|
|
|
#endif /* MPI_COMMON_UGNI_H */
|