323e4418b9
- similar to opal/shmem - next step is some refactoring and merge into opal/shmem Developed by Igor, reviewed by AlexM, MikeD This commit fixes trac:4261. This commit was SVN r30855. The following Trac tickets were found above: Ticket 4261 --> https://svn.open-mpi.org/trac/ompi/ticket/4261
141 строка
4.7 KiB
C
141 строка
4.7 KiB
C
/*
|
|
* Copyright (c) 2013 Mellanox Technologies, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "oshmem_config.h"
|
|
#include <stdio.h>
|
|
|
|
#include "opal/mca/mca.h"
|
|
#include "opal/util/output.h"
|
|
#include "opal/mca/base/base.h"
|
|
|
|
#include "oshmem/mca/memheap/memheap.h"
|
|
#include "oshmem/mca/memheap/base/base.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 "oshmem/mca/memheap/base/static-components.h"
|
|
|
|
|
|
int mca_memheap_base_output = -1;
|
|
int mca_memheap_base_key_exchange = 1;
|
|
char* mca_memheap_base_include = NULL;
|
|
char* mca_memheap_base_exclude = NULL;
|
|
opal_list_t mca_memheap_base_components_opened;
|
|
struct mca_memheap_base_module_t* mca_memheap_base_module_initialized = NULL;
|
|
int mca_memheap_base_already_opened = 0;
|
|
mca_memheap_map_t mca_memheap_base_map;
|
|
|
|
static int mca_memheap_base_register(mca_base_register_flag_t flags)
|
|
{
|
|
|
|
(void) mca_base_var_register("oshmem",
|
|
"memheap",
|
|
"base",
|
|
"key_exchange",
|
|
"0|1 - disabled, enabled(default) force memory keys exchange",
|
|
MCA_BASE_VAR_TYPE_INT,
|
|
NULL,
|
|
0,
|
|
MCA_BASE_VAR_FLAG_SETTABLE,
|
|
OPAL_INFO_LVL_9,
|
|
MCA_BASE_VAR_SCOPE_READONLY,
|
|
&mca_memheap_base_key_exchange);
|
|
|
|
(void) mca_base_var_register("oshmem",
|
|
"memheap",
|
|
"base",
|
|
"include",
|
|
"Specify a specific MEMHEAP implementation to use",
|
|
MCA_BASE_VAR_TYPE_STRING,
|
|
NULL,
|
|
0,
|
|
MCA_BASE_VAR_FLAG_SETTABLE,
|
|
OPAL_INFO_LVL_9,
|
|
MCA_BASE_VAR_SCOPE_READONLY,
|
|
&mca_memheap_base_include);
|
|
|
|
if (NULL == mca_memheap_base_include) {
|
|
mca_memheap_base_include = getenv(SHMEM_HEAP_TYPE);
|
|
if (NULL == mca_memheap_base_include)
|
|
mca_memheap_base_include = strdup("");
|
|
else
|
|
mca_memheap_base_include = strdup(mca_memheap_base_include);
|
|
}
|
|
|
|
(void) mca_base_var_register("oshmem",
|
|
"memheap",
|
|
"base",
|
|
"exclude",
|
|
"Specify excluded MEMHEAP implementations",
|
|
MCA_BASE_VAR_TYPE_STRING,
|
|
NULL,
|
|
0,
|
|
MCA_BASE_VAR_FLAG_SETTABLE,
|
|
OPAL_INFO_LVL_9,
|
|
MCA_BASE_VAR_SCOPE_READONLY,
|
|
&mca_memheap_base_exclude);
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
static int mca_memheap_base_close(void)
|
|
{
|
|
if (mca_memheap_base_already_opened <= 0) {
|
|
return OSHMEM_ERROR;
|
|
}
|
|
mca_memheap_base_already_opened--;
|
|
if (mca_memheap_base_already_opened > 0) {
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
memheap_oob_destruct();
|
|
|
|
mca_memheap_base_dereg(&mca_memheap_base_map);
|
|
|
|
mca_memheap_base_alloc_exit(&mca_memheap_base_map);
|
|
mca_memheap_base_static_exit(&mca_memheap_base_map);
|
|
|
|
/* Close all remaining available components */
|
|
return mca_base_framework_components_close(&oshmem_memheap_base_framework, NULL);
|
|
}
|
|
|
|
static int mca_memheap_base_open(mca_base_open_flag_t flags)
|
|
{
|
|
mca_memheap_base_already_opened = mca_memheap_base_already_opened + 1;
|
|
if (mca_memheap_base_already_opened > 1) {
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
memset(&mca_memheap_base_map, 0, sizeof(mca_memheap_base_map));
|
|
mca_memheap_base_map.n_segments = 0;
|
|
mca_memheap_base_map.num_transports = 0;
|
|
|
|
/* Open up all available components */
|
|
if (OPAL_SUCCESS !=
|
|
mca_base_framework_components_open(&oshmem_memheap_base_framework, flags)) {
|
|
return OSHMEM_ERROR;
|
|
}
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
MCA_BASE_FRAMEWORK_DECLARE(oshmem, memheap,
|
|
"OSHMEM MEMHEAP",
|
|
mca_memheap_base_register,
|
|
mca_memheap_base_open,
|
|
mca_memheap_base_close,
|
|
mca_memheap_base_static_components,
|
|
MCA_BASE_FRAMEWORK_FLAG_DEFAULT);
|