141 строка
4.8 KiB
C
141 строка
4.8 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2013 Mellanox Technologies, Inc.
|
|
* All rights reserved.
|
|
* Copyright (c) 2015 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "oshmem_config.h"
|
|
#include "oshmem/runtime/params.h"
|
|
#include "oshmem/mca/spml/spml.h"
|
|
#include "spml_yoda_component.h"
|
|
#include "oshmem/mca/spml/yoda/spml_yoda_rdmafrag.h"
|
|
#include "oshmem/mca/spml/yoda/spml_yoda_putreq.h"
|
|
#include "oshmem/mca/spml/yoda/spml_yoda.h"
|
|
|
|
static int mca_spml_yoda_component_register(void);
|
|
static int mca_spml_yoda_component_open(void);
|
|
static int mca_spml_yoda_component_close(void);
|
|
static mca_spml_base_module_t*
|
|
mca_spml_yoda_component_init(int* priority,
|
|
bool enable_progress_threads,
|
|
bool enable_mpi_threads);
|
|
static int mca_spml_yoda_component_fini(void);
|
|
|
|
mca_spml_base_component_2_0_0_t mca_spml_yoda_component = {
|
|
|
|
/* First, the mca_base_component_t struct containing meta
|
|
information about the component itself */
|
|
|
|
.spmlm_version = {
|
|
MCA_SPML_BASE_VERSION_2_0_0,
|
|
|
|
.mca_component_name = "yoda",
|
|
MCA_BASE_MAKE_VERSION(component, OSHMEM_MAJOR_VERSION, OSHMEM_MINOR_VERSION,
|
|
OSHMEM_RELEASE_VERSION),
|
|
.mca_open_component = mca_spml_yoda_component_open,
|
|
.mca_close_component = mca_spml_yoda_component_close,
|
|
.mca_register_component_params = mca_spml_yoda_component_register,
|
|
},
|
|
.spmlm_data = {
|
|
/* The component is checkpoint ready */
|
|
MCA_BASE_METADATA_PARAM_CHECKPOINT
|
|
},
|
|
|
|
.spmlm_init = mca_spml_yoda_component_init,
|
|
.spmlm_finalize = mca_spml_yoda_component_fini,
|
|
};
|
|
|
|
static inline void mca_spml_yoda_param_register_int(const char *param_name,
|
|
int default_value,
|
|
const char *help_msg,
|
|
int *storage)
|
|
{
|
|
*storage = default_value;
|
|
(void) mca_base_component_var_register(&mca_spml_yoda_component.spmlm_version,
|
|
param_name,
|
|
help_msg,
|
|
MCA_BASE_VAR_TYPE_INT, NULL, 0, MCA_BASE_VAR_FLAG_SETTABLE,
|
|
OPAL_INFO_LVL_9,
|
|
MCA_BASE_VAR_SCOPE_READONLY,
|
|
storage);
|
|
}
|
|
|
|
static int mca_spml_yoda_component_register(void)
|
|
{
|
|
mca_spml_yoda_param_register_int("free_list_num", 1024,
|
|
0,
|
|
&mca_spml_yoda.free_list_num);
|
|
mca_spml_yoda_param_register_int("free_list_max", 1024,
|
|
0,
|
|
&mca_spml_yoda.free_list_max);
|
|
mca_spml_yoda_param_register_int("free_list_inc", 16,
|
|
0,
|
|
&mca_spml_yoda.free_list_inc);
|
|
mca_spml_yoda_param_register_int("bml_alloc_threshold", 3,
|
|
"number of puts to wait \
|
|
in case of put/get temporary buffer \
|
|
allocation failture",
|
|
&mca_spml_yoda.bml_alloc_threshold);
|
|
mca_spml_yoda_param_register_int("priority", 10,
|
|
"[integer] yoda priority",
|
|
&mca_spml_yoda.priority);
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
static int mca_spml_yoda_component_open(void)
|
|
{
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
static int mca_spml_yoda_component_close(void)
|
|
{
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
static mca_spml_base_module_t*
|
|
mca_spml_yoda_component_init(int* priority,
|
|
bool enable_progress_threads,
|
|
bool enable_mpi_threads)
|
|
{
|
|
SPML_VERBOSE( 10, "in yoda, my priority is %d\n", mca_spml_yoda.priority);
|
|
|
|
*priority = mca_spml_yoda.priority;
|
|
if ((*priority) > mca_spml_yoda.priority) {
|
|
return NULL ;
|
|
}
|
|
|
|
/* We use BML/BTL and need to start it */
|
|
if (!mca_bml_base_inited()) {
|
|
SPML_VERBOSE(10, "can not select yoda because ompi has no bml component");
|
|
return NULL;
|
|
}
|
|
|
|
mca_spml_yoda.n_active_puts = 0;
|
|
mca_spml_yoda.n_active_gets = 0;
|
|
|
|
return &mca_spml_yoda.super;
|
|
}
|
|
|
|
int mca_spml_yoda_component_fini(void)
|
|
{
|
|
if (!mca_spml_yoda.enabled) {
|
|
return OSHMEM_SUCCESS; /* never selected.. return success.. */
|
|
}
|
|
mca_spml_yoda.enabled = false; /* not anymore */
|
|
|
|
OBJ_DESTRUCT(&mca_spml_yoda.lock);
|
|
#if OSHMEM_WAIT_COMPLETION_DEBUG == 1
|
|
condition_dbg_finalize();
|
|
#endif
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|