3c97b3f722
searching the tree. Modified the memory callback to search the tree at each page boundary for registrations. This is necessary as an application may malloc memory and send out of any portion of that memory, even discontiguous regions. This commit was SVN r7510.
95 строки
2.7 KiB
C
95 строки
2.7 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_MEM_BASE_H
|
|
#define MCA_MEM_BASE_H
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include "opal/class/opal_list.h"
|
|
#include "class/ompi_rb_tree.h"
|
|
#include "mca/mca.h"
|
|
#include "mca/mpool/mpool.h"
|
|
#include "opal/threads/mutex.h"
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
static inline unsigned long my_log2(unsigned long val) {
|
|
unsigned long count = -1;
|
|
while(val > 0) {
|
|
val = val >> 1;
|
|
count++;
|
|
}
|
|
return count;
|
|
}
|
|
static inline void *down_align_addr(void* addr, unsigned int shift) {
|
|
return (void*) (((unsigned long) addr) & (~(unsigned long) 0) << shift);
|
|
}
|
|
|
|
static inline void *up_align_addr(void*addr, unsigned int shift) {
|
|
return (void*) ((((unsigned long) addr) | ~((~(unsigned long) 0) << shift)) + 1);
|
|
}
|
|
|
|
struct mca_mpool_base_selected_module_t {
|
|
opal_list_item_t super;
|
|
mca_mpool_base_component_t *mpool_component;
|
|
mca_mpool_base_module_t *mpool_module;
|
|
void* user_data;
|
|
struct mca_mpool_base_resources_t *mpool_resources;
|
|
};
|
|
typedef struct mca_mpool_base_selected_module_t mca_mpool_base_selected_module_t;
|
|
|
|
OBJ_CLASS_DECLARATION(mca_mpool_base_selected_module_t);
|
|
|
|
/*
|
|
* Data structures for the tree of allocated memory
|
|
*/
|
|
|
|
/*
|
|
* Global functions for MCA: overall mpool open and close
|
|
*/
|
|
|
|
OMPI_DECLSPEC int mca_mpool_base_open(void);
|
|
OMPI_DECLSPEC int mca_mpool_base_init(bool enable_progress_threads, bool enable_mpi_threads);
|
|
OMPI_DECLSPEC int mca_mpool_base_close(void);
|
|
OMPI_DECLSPEC mca_mpool_base_component_t* mca_mpool_base_component_lookup(const char* name);
|
|
OMPI_DECLSPEC mca_mpool_base_module_t* mca_mpool_base_module_create(
|
|
const char* name,
|
|
void* user_data,
|
|
struct mca_mpool_base_resources_t* mpool_resources);
|
|
OMPI_DECLSPEC mca_mpool_base_module_t* mca_mpool_base_module_lookup(const char* name);
|
|
OMPI_DECLSPEC int mca_mpool_base_module_destroy(mca_mpool_base_module_t *module);
|
|
|
|
/*
|
|
* Globals
|
|
*/
|
|
OMPI_DECLSPEC extern int mca_mpool_base_output;
|
|
OMPI_DECLSPEC extern opal_list_t mca_mpool_base_components;
|
|
OMPI_DECLSPEC extern opal_list_t mca_mpool_base_modules;
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif /* MCA_MEM_BASE_H */
|
|
|
|
|
|
|