79ad6d983e
a standalone library named libopenmpi-malloc. Users wanting to use leave_pinned with ptmalloc2 will now need to link the library into their application explicitly. All other users will use the libc-provided allocator instead of Open MPI's ptmalloc2. This change may be overriden with the configure option enable-ptmalloc2-internal - The leave_pinned options will now default to using mallopt on Linux in the cases where ptmalloc2 was not linked in. mallopt will also only be available if munmap can be intercepted (the default whenever Open MPI is not compiled with --without-memory- manager. - Open MPI will now complain and refuse to use leave_pinned if no memory intercept / mallopt option is available. This commit was SVN r18654.
110 строки
3.3 KiB
C
110 строки
3.3 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2006 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$
|
|
*/
|
|
/**
|
|
* @file
|
|
*/
|
|
#ifndef MCA_MEM_BASE_H
|
|
#define MCA_MEM_BASE_H
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include "opal/class/opal_list.h"
|
|
#include "ompi/class/ompi_rb_tree.h"
|
|
#include "opal/mca/mca.h"
|
|
#include "ompi/mca/mpool/mpool.h"
|
|
#include "opal/threads/mutex.h"
|
|
|
|
#if defined(HAVE_MALLOPT) && defined(M_TRIM_THRESHOLD) && defined(M_MMAP_MAX)
|
|
#define OMPI_MPOOL_BASE_HAVE_LINUX_MALLOPT 1
|
|
#else
|
|
#define OMPI_MPOOL_BASE_HAVE_LINUX_MALLOPT 0
|
|
#endif
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
static inline unsigned int my_log2(unsigned long val) {
|
|
unsigned int count = 0;
|
|
while(val > 0) {
|
|
val = val >> 1;
|
|
count++;
|
|
}
|
|
return count > 0 ? count-1: 0;
|
|
}
|
|
static inline void *down_align_addr(void* addr, unsigned int shift) {
|
|
return (void*) (((intptr_t) addr) & (~(intptr_t) 0) << shift);
|
|
}
|
|
|
|
static inline void *up_align_addr(void*addr, unsigned int shift) {
|
|
return (void*) ((((intptr_t) addr) | ~((~(intptr_t) 0) << shift)));
|
|
}
|
|
|
|
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;
|
|
|
|
OMPI_DECLSPEC 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;
|
|
extern opal_list_t mca_mpool_base_modules;
|
|
OMPI_DECLSPEC extern uint32_t mca_mpool_base_page_size;
|
|
OMPI_DECLSPEC extern uint32_t mca_mpool_base_page_size_log;
|
|
|
|
/* only used within base -- no need to DECLSPEC */
|
|
extern int mca_mpool_base_use_mem_hooks;
|
|
extern int mca_mpool_base_mallopt_disable_free;
|
|
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif /* MCA_MEM_BASE_H */
|
|
|
|
|
|
|