
add misc asserts to check for proper reference counting. ugly hack 1 -- use mallopt to never release memory ala sbrk - this is commented out in mca_btl_mvapi_component_init ugly hack 2 -- test registrations comming out of the tree via rcache_find, for an unknown reason the tree is returning registrations where the address is not within the base or bound of the registration. If this happens, we return NULL. comment out code to enable mem hooks if leave_pinned is set, note we can do this via an mca param and will default it to leave_pinned with mem_hooks when we iron out these issues. I am adding a unit test for the rcache. Note that we have a unit test for the rb tree but the compare function is significantly different than that used for registrations. After we have tracked down the issues with rcache_rb we will remove the above hacks. This commit was SVN r7499.
63 строки
2.1 KiB
C
63 строки
2.1 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
|
|
*/
|
|
#include "mpool_base_mem_cb.h"
|
|
#include "base.h"
|
|
|
|
/*
|
|
* memory hook callback, called when memory is free'd out from under us
|
|
*/
|
|
void mca_mpool_base_mem_cb(void* base, size_t size, void* cbdata)
|
|
{
|
|
uint32_t i, cnt;
|
|
ompi_pointer_array_t regs;
|
|
mca_mpool_base_registration_t* reg;
|
|
mca_mpool_base_selected_module_t* current;
|
|
int rc;
|
|
opal_list_item_t* item;
|
|
|
|
OBJ_CONSTRUCT(®s, ompi_pointer_array_t);
|
|
for(item = opal_list_get_first(&mca_mpool_base_modules);
|
|
item != opal_list_get_end(&mca_mpool_base_modules);
|
|
item = opal_list_get_next(item)) {
|
|
|
|
current = (mca_mpool_base_selected_module_t*) item;
|
|
if(NULL != current->mpool_module->mpool_find) {
|
|
rc = current->mpool_module->mpool_find(
|
|
current->mpool_module,
|
|
base,
|
|
size,
|
|
®s,
|
|
&cnt
|
|
);
|
|
if(OMPI_SUCCESS != rc) {
|
|
continue;
|
|
}
|
|
for(i = 0; i < cnt; i++) {
|
|
reg = (mca_mpool_base_registration_t*)ompi_pointer_array_get_item(®s, i);
|
|
assert(reg->base == base);
|
|
current->mpool_module->mpool_deregister(current->mpool_module, reg);
|
|
}
|
|
}
|
|
}
|
|
OBJ_DESTRUCT(®s);
|
|
}
|
|
|
|
|
|
|