454fdff824
work on the configure.stub to correctly locate the ib libraries. This commit was SVN r6435.
160 строки
3.6 KiB
C
160 строки
3.6 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$
|
|
*/
|
|
/**
|
|
* @file
|
|
*/
|
|
#ifndef MCA_MPOOL_VAPI_H
|
|
#define MCA_MPOOL_VAPI_H
|
|
|
|
#include "opal/class/opal_list.h"
|
|
#include "class/ompi_free_list.h"
|
|
#include "opal/event/event.h"
|
|
#include "mca/mpool/mpool.h"
|
|
#include "mca/allocator/allocator.h"
|
|
#include "mca/common/vapi/vapi_mem_reg.h"
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
|
|
static inline void * DOWN_ALIGN_ADDR(void * addr, uint32_t cnt) {
|
|
return (void*)((MT_virt_addr_t)(addr) & (~((MT_virt_addr_t)0) << (cnt)));
|
|
}
|
|
|
|
static inline void* ALIGN_ADDR(void* addr, uint32_t cnt ) {
|
|
DOWN_ALIGN_ADDR(((addr) + ~(~((MT_virt_addr_t)0) << (cnt))), (cnt));
|
|
return addr;
|
|
}
|
|
|
|
|
|
struct mca_mpool_openib_component_t {
|
|
mca_mpool_base_component_t super;
|
|
char* vapi_allocator_name;
|
|
long page_size;
|
|
long page_size_log;
|
|
};
|
|
|
|
typedef struct mca_mpool_openib_component_t mca_mpool_openib_component_t;
|
|
|
|
OMPI_COMP_EXPORT extern mca_mpool_openib_component_t mca_mpool_openib_component;
|
|
|
|
|
|
|
|
|
|
|
|
struct mca_mpool_base_resources_t {
|
|
struct ibv_pd* ib_pd;
|
|
};
|
|
typedef struct mca_mpool_base_resources_t mca_mpool_base_resources_t;
|
|
|
|
struct mca_mpool_openib_module_t {
|
|
mca_mpool_base_module_t super;
|
|
mca_allocator_base_module_t * vapi_allocator;
|
|
struct mca_mpool_base_resources_t resources;
|
|
}; typedef struct mca_mpool_openib_module_t mca_mpool_openib_module_t;
|
|
|
|
|
|
struct mca_mpool_openib_registration_t {
|
|
mca_mpool_base_registration_t base_reg;
|
|
struct ibv_mr *mr;
|
|
bool is_leave_pinned;
|
|
|
|
};
|
|
typedef struct mca_mpool_openib_registration_t mca_mpool_openib_registration_t;
|
|
OMPI_DECLSPEC OBJ_CLASS_DECLARATION(mca_mpool_openib_registration_t);
|
|
|
|
|
|
|
|
|
|
/*
|
|
* Initializes the mpool module.
|
|
*/
|
|
void mca_mpool_openib_module_init(mca_mpool_openib_module_t* mpool);
|
|
|
|
|
|
/*
|
|
* Returns base address of shared memory mapping.
|
|
*/
|
|
void* mca_mpool_openib_base(mca_mpool_base_module_t*);
|
|
|
|
/**
|
|
* Allocate block of shared memory.
|
|
*/
|
|
void* mca_mpool_openib_alloc(
|
|
mca_mpool_base_module_t* mpool,
|
|
size_t size,
|
|
size_t align,
|
|
mca_mpool_base_registration_t** registration);
|
|
|
|
/**
|
|
* realloc function typedef
|
|
*/
|
|
void* mca_mpool_openib_realloc(
|
|
mca_mpool_base_module_t* mpool,
|
|
void* addr,
|
|
size_t size,
|
|
mca_mpool_base_registration_t** registration);
|
|
|
|
/**
|
|
* register function typedef
|
|
*/
|
|
int mca_mpool_openib_register(
|
|
mca_mpool_base_module_t* mpool,
|
|
void *addr,
|
|
size_t size,
|
|
mca_mpool_base_registration_t** registration);
|
|
|
|
int mca_mpool_openib_deregister(
|
|
mca_mpool_base_module_t* mpool,
|
|
void *addr,
|
|
size_t size,
|
|
mca_mpool_base_registration_t* );
|
|
|
|
|
|
/**
|
|
* free function typedef
|
|
*/
|
|
void mca_mpool_openib_free(mca_mpool_base_module_t* mpool,
|
|
void * addr,
|
|
mca_mpool_base_registration_t* registration);
|
|
|
|
void* mca_common_vapi_segment_alloc(
|
|
struct mca_mpool_base_module_t* module,
|
|
size_t* size,
|
|
mca_mpool_base_registration_t** registration);
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|