b8cb6e1c64
be used in MPI_Alloc_mem operations. Note that we found an interesting bug in which if memory was allocated by the sm mpool (via mmap) and then registered via the mvapi mpool, the registration would fail on certain systems. Added mca param mpool_base_use_mem_hooks, set to 1 to enable the memory hooks so that memory is deregistered if the user frees it behind our back. This is only useful if the mca param mpi_leave_pinned is also set to 1. Otherwise all registrations are deregistered within the MPI library or via MPI_Free_buf. After testing we should probably set both mpi_leave_pinned and mpool_base_use_mem_hooks to default to 1. This commit was SVN r7415.
67 строки
2.0 KiB
C
67 строки
2.0 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$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
#include "ompi/include/constants.h"
|
|
#include "mca/mca.h"
|
|
#include "mca/base/base.h"
|
|
#include "mca/mpool/mpool.h"
|
|
#include "mca/mpool/base/base.h"
|
|
#include "mpool_base_mem_cb.h"
|
|
|
|
extern int mca_mpool_base_use_mem_hooks;
|
|
|
|
int mca_mpool_base_close(void)
|
|
{
|
|
opal_list_item_t *item;
|
|
mca_mpool_base_selected_module_t *sm;
|
|
|
|
/* Finalize all the mpool components and free their list items */
|
|
|
|
for (item = opal_list_remove_first(&mca_mpool_base_modules);
|
|
NULL != item;
|
|
item = opal_list_remove_first(&mca_mpool_base_modules)) {
|
|
sm = (mca_mpool_base_selected_module_t *) item;
|
|
|
|
/* Blatently ignore the return code (what would we do to recover,
|
|
anyway? This component is going away, so errors don't matter
|
|
anymore). Note that it's legal for the module to have NULL for
|
|
the finalize function. */
|
|
|
|
if (NULL != sm->mpool_module->mpool_finalize) {
|
|
sm->mpool_module->mpool_finalize(sm->mpool_module);
|
|
}
|
|
OBJ_RELEASE(sm);
|
|
}
|
|
|
|
/* Close all remaining available components (may be one if this is a
|
|
OMPI RTE program, or [possibly] multiple if this is ompi_info) */
|
|
|
|
mca_base_components_close(mca_mpool_base_output,
|
|
&mca_mpool_base_components, NULL);
|
|
|
|
/* deregister memory free callback */
|
|
if(mca_mpool_base_use_mem_hooks && opal_mem_free_is_supported()) {
|
|
opal_mem_free_unregister_handler(mca_mpool_base_mem_cb);
|
|
}
|
|
/* All done */
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|