9d4a26f47d
Notes:
- This commit also eliminates the need for an available components list in use
in several frameworks. None of the code in question was making use of the
priority field of the priority component list item so these extra lists were
removed.
- Cleaned up selection code in several frameworks to sort lists using opal_list_sort.
- Cleans up the ompi/orte-info functions. Expose the functions that construct the
list of params so they can be used elsewhere.
patches for mtl/portals4 from brian
missed a few output variables in openib
This commit was SVN r28241.
102 строки
3.7 KiB
C
102 строки
3.7 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; -*- */
|
|
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2007 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 (c) 2009 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2012 Los Alamos National Security, LLC.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
/**
|
|
* @file
|
|
*/
|
|
#include "ompi_config.h"
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include "opal/util/show_help.h"
|
|
#include "ompi/mca/rte/rte.h"
|
|
#include "ompi/runtime/mpiruntime.h"
|
|
|
|
#include "ompi/mca/mpool/base/mpool_base_mem_cb.h"
|
|
#include "ompi/mca/mpool/base/base.h"
|
|
|
|
#include "opal/mca/mca.h"
|
|
#include "opal/memoryhooks/memory.h"
|
|
|
|
static char msg[512];
|
|
|
|
|
|
/*
|
|
* memory hook callback, called when memory is free'd out from under
|
|
* us. Be wary of the from_alloc flag -- if you're called with
|
|
* from_alloc==true, then you cannot call malloc (or any of its
|
|
* friends)!
|
|
*/
|
|
void mca_mpool_base_mem_cb(void* base, size_t size, void* cbdata,
|
|
bool from_alloc)
|
|
{
|
|
mca_mpool_base_selected_module_t* current;
|
|
int rc;
|
|
opal_list_item_t* item;
|
|
|
|
/* Only do anything meaningful if we're between MPI_INIT and
|
|
MPI_FINALIZE, and size != 0 */
|
|
if ((from_alloc && (!ompi_mpi_initialized || ompi_mpi_finalized)) ||
|
|
size == 0) {
|
|
return;
|
|
}
|
|
|
|
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(current->mpool_module->mpool_release_memory != NULL) {
|
|
rc = current->mpool_module->mpool_release_memory(current->mpool_module,
|
|
base, size);
|
|
|
|
if (rc != OMPI_SUCCESS) {
|
|
if (from_alloc) {
|
|
int len;
|
|
len = snprintf(msg, sizeof(msg), "[%s:%d] Attempt to free memory that is still in use by an ongoing MPI communication (buffer %p, size %lu). MPI job will now abort.\n",
|
|
ompi_process_info.nodename,
|
|
getpid(),
|
|
base, (unsigned long) size);
|
|
msg[sizeof(msg) - 1] = '\0';
|
|
write(2, msg, len);
|
|
} else {
|
|
opal_show_help("help-mpool-base.txt",
|
|
"cannot deregister in-use memory", true,
|
|
current->mpool_component->mpool_version.mca_component_name,
|
|
ompi_process_info.nodename,
|
|
base, (unsigned long) size);
|
|
}
|
|
|
|
/* We're in a callback from somewhere; we can't do
|
|
anything meaningful to pass an error back up. :-(
|
|
So just exit. Call _exit() so that we don't try to
|
|
call anything on the way out -- just exit!
|
|
(remember that we're in a callback, and state may
|
|
be very undefined at this point...) */
|
|
_exit(1);
|
|
}
|
|
}
|
|
}
|
|
}
|