mirror of https://gitee.com/openkylin/openmpi.git
123 lines
3.9 KiB
C
123 lines
3.9 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 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) 2016 Research Organization for Information Science
|
|
* and Technology (RIST). All rights reserved.
|
|
* Copyright (c) 2016 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
|
|
#include "opal_config.h"
|
|
|
|
#include "opal/constants.h"
|
|
#include "opal/mca/mca.h"
|
|
#include "opal/mca/base/base.h"
|
|
#include "opal/mca/memory/memory.h"
|
|
#include "opal/mca/memory/base/base.h"
|
|
#include "opal/mca/memory/base/empty.h"
|
|
|
|
|
|
/*
|
|
* The following file was created by configure. It contains extern
|
|
* statements and the definition of an array of pointers to each
|
|
* component's public mca_base_component_t struct.
|
|
*/
|
|
#include "opal/mca/memory/base/static-components.h"
|
|
|
|
static int empty_process(void)
|
|
{
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
static int empty_query (int *priority)
|
|
{
|
|
*priority = 0;
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
/*
|
|
* Local variables
|
|
*/
|
|
static opal_memory_base_component_2_0_0_t empty_component = {
|
|
/* Empty / safe functions to call if no memory componet is selected */
|
|
.memoryc_query = empty_query,
|
|
.memoryc_process = empty_process,
|
|
.memoryc_register = opal_memory_base_component_register_empty,
|
|
.memoryc_deregister = opal_memory_base_component_deregister_empty,
|
|
.memoryc_set_alignment = opal_memory_base_component_set_alignment_empty,
|
|
};
|
|
|
|
|
|
/*
|
|
* Globals
|
|
*/
|
|
opal_memory_base_component_2_0_0_t *opal_memory = &empty_component;
|
|
|
|
|
|
void opal_memory_base_malloc_init_hook (void)
|
|
{
|
|
if (opal_memory->memoryc_init_hook) {
|
|
opal_memory->memoryc_init_hook ();
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Function for finding and opening either all MCA components, or the one
|
|
* that was specifically requested via a MCA parameter.
|
|
*/
|
|
static int opal_memory_base_open(mca_base_open_flag_t flags)
|
|
{
|
|
mca_base_component_list_item_t *item, *next;
|
|
opal_memory_base_component_2_0_0_t *tmp;
|
|
int priority, highest_priority = 0;
|
|
int ret;
|
|
|
|
/* can only be zero or one */
|
|
OPAL_LIST_FOREACH(item, &opal_memory_base_framework.framework_components, mca_base_component_list_item_t) {
|
|
tmp = (opal_memory_base_component_2_0_0_t *) item->cli_component;
|
|
ret = tmp->memoryc_query (&priority);
|
|
if (OPAL_SUCCESS != ret || priority < highest_priority) {
|
|
continue;
|
|
}
|
|
|
|
highest_priority = priority;
|
|
opal_memory = tmp;
|
|
}
|
|
|
|
OPAL_LIST_FOREACH_SAFE(item, next, &opal_memory_base_framework.framework_components, mca_base_component_list_item_t) {
|
|
if ((void *) opal_memory != (void *) item->cli_component) {
|
|
mca_base_component_unload (item->cli_component, opal_memory_base_framework.framework_output);
|
|
opal_list_remove_item (&opal_memory_base_framework.framework_components, &item->super);
|
|
}
|
|
}
|
|
|
|
/* open remaining component */
|
|
ret = mca_base_framework_components_open (&opal_memory_base_framework, flags);
|
|
if (ret != OPAL_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
/* All done */
|
|
return OPAL_SUCCESS;
|
|
}
|
|
|
|
/* Use default register/close functions */
|
|
MCA_BASE_FRAMEWORK_DECLARE(opal, memory, "memory hooks", NULL, opal_memory_base_open, NULL,
|
|
mca_memory_base_static_components, 0);
|