2006-01-28 18:38:37 +03:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University.
|
|
|
|
* All rights reserved.
|
2006-08-24 20:38:08 +04:00
|
|
|
* Copyright (c) 2004-2006 The Trustees of the University of Tennessee.
|
2006-01-28 18:38:37 +03:00
|
|
|
* 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$
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "ompi_config.h"
|
|
|
|
|
2006-02-07 15:16:23 +03:00
|
|
|
#include <string.h>
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
#include "osc_pt2pt.h"
|
|
|
|
#include "osc_pt2pt_sendreq.h"
|
|
|
|
#include "osc_pt2pt_replyreq.h"
|
|
|
|
#include "osc_pt2pt_header.h"
|
|
|
|
#include "osc_pt2pt_obj_convert.h"
|
|
|
|
#include "osc_pt2pt_data_move.h"
|
2006-08-03 04:10:19 +04:00
|
|
|
#include "osc_pt2pt_buffer.h"
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
#include "opal/threads/mutex.h"
|
|
|
|
#include "ompi/info/info.h"
|
|
|
|
#include "ompi/communicator/communicator.h"
|
|
|
|
#include "ompi/mca/osc/osc.h"
|
2006-08-17 18:52:20 +04:00
|
|
|
#include "ompi/mca/osc/base/base.h"
|
2006-08-03 04:10:19 +04:00
|
|
|
#include "ompi/mca/pml/pml.h"
|
2006-02-27 21:47:00 +03:00
|
|
|
#include "ompi/datatype/dt_arch.h"
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-03-16 21:40:42 +03:00
|
|
|
static int ompi_osc_pt2pt_component_open(void);
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
ompi_osc_pt2pt_component_t mca_osc_pt2pt_component = {
|
|
|
|
{ /* ompi_osc_base_component_t */
|
|
|
|
{ /* ompi_base_component_t */
|
|
|
|
OMPI_OSC_BASE_VERSION_1_0_0,
|
|
|
|
"pt2pt",
|
2006-08-03 04:10:19 +04:00
|
|
|
OMPI_MAJOR_VERSION, /* MCA component major version */
|
|
|
|
OMPI_MINOR_VERSION, /* MCA component minor version */
|
|
|
|
OMPI_RELEASE_VERSION, /* MCA component release version */
|
2006-03-16 21:40:42 +03:00
|
|
|
ompi_osc_pt2pt_component_open,
|
2006-01-28 18:38:37 +03:00
|
|
|
NULL
|
|
|
|
},
|
|
|
|
{ /* mca_base_component_data */
|
|
|
|
false /* checkpointable? */
|
|
|
|
},
|
|
|
|
ompi_osc_pt2pt_component_init,
|
|
|
|
ompi_osc_pt2pt_component_query,
|
|
|
|
ompi_osc_pt2pt_component_select,
|
|
|
|
ompi_osc_pt2pt_component_finalize
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
ompi_osc_pt2pt_module_t ompi_osc_pt2pt_module_template = {
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_module_free,
|
|
|
|
|
|
|
|
ompi_osc_pt2pt_module_put,
|
|
|
|
ompi_osc_pt2pt_module_get,
|
|
|
|
ompi_osc_pt2pt_module_accumulate,
|
|
|
|
|
|
|
|
ompi_osc_pt2pt_module_fence,
|
|
|
|
|
|
|
|
ompi_osc_pt2pt_module_start,
|
|
|
|
ompi_osc_pt2pt_module_complete,
|
|
|
|
ompi_osc_pt2pt_module_post,
|
|
|
|
ompi_osc_pt2pt_module_wait,
|
|
|
|
ompi_osc_pt2pt_module_test,
|
|
|
|
|
|
|
|
ompi_osc_pt2pt_module_lock,
|
|
|
|
ompi_osc_pt2pt_module_unlock,
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2006-03-24 21:56:59 +03:00
|
|
|
/* look up parameters for configuring this window. The code first
|
|
|
|
looks in the info structure passed by the user, then through mca
|
|
|
|
parameters. */
|
2006-01-28 18:38:37 +03:00
|
|
|
static bool
|
2006-03-24 21:56:59 +03:00
|
|
|
check_config_value_bool(char *key, ompi_info_t *info)
|
2006-01-28 18:38:37 +03:00
|
|
|
{
|
2006-03-24 21:56:59 +03:00
|
|
|
char *value_string;
|
|
|
|
int value_len, ret, flag, param;
|
|
|
|
bool result;
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-03-24 21:56:59 +03:00
|
|
|
ret = ompi_info_get_valuelen(info, key, &value_len, &flag);
|
|
|
|
if (OMPI_SUCCESS != ret) goto info_not_found;
|
|
|
|
if (flag == 0) goto info_not_found;
|
|
|
|
value_len++;
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-08-24 20:38:08 +04:00
|
|
|
value_string = (char*)malloc(sizeof(char) * value_len);
|
2006-03-24 21:56:59 +03:00
|
|
|
if (NULL == value_string) goto info_not_found;
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-03-24 21:56:59 +03:00
|
|
|
ret = ompi_info_get(info, key, value_len, value_string, &flag);
|
2006-01-28 18:38:37 +03:00
|
|
|
if (OMPI_SUCCESS != ret) {
|
2006-03-24 21:56:59 +03:00
|
|
|
free(value_string);
|
|
|
|
goto info_not_found;
|
2006-01-28 18:38:37 +03:00
|
|
|
}
|
|
|
|
assert(flag != 0);
|
2006-03-24 21:56:59 +03:00
|
|
|
ret = ompi_info_value_to_bool(value_string, &result);
|
|
|
|
free(value_string);
|
|
|
|
if (OMPI_SUCCESS != ret) goto info_not_found;
|
|
|
|
return result;
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-03-24 21:56:59 +03:00
|
|
|
info_not_found:
|
|
|
|
param = mca_base_param_find("osc", "pt2pt", key);
|
|
|
|
if (param == OPAL_ERROR) return false;
|
|
|
|
|
|
|
|
ret = mca_base_param_lookup_int(param, &flag);
|
|
|
|
if (OMPI_SUCCESS != ret) return false;
|
|
|
|
|
2006-08-28 22:59:16 +04:00
|
|
|
return OPAL_INT_TO_BOOL(flag);
|
2006-01-28 18:38:37 +03:00
|
|
|
}
|
|
|
|
|
2006-03-24 21:56:59 +03:00
|
|
|
|
2006-03-16 21:40:42 +03:00
|
|
|
static int
|
|
|
|
ompi_osc_pt2pt_component_open(void)
|
|
|
|
{
|
2006-08-03 04:10:19 +04:00
|
|
|
int tmp;
|
|
|
|
|
|
|
|
mca_base_param_reg_int(&mca_osc_pt2pt_component.super.osc_version,
|
|
|
|
"no_locks",
|
|
|
|
"Enable optimizations available only if MPI_LOCK is not used.",
|
|
|
|
false, false, 0, NULL);
|
|
|
|
|
|
|
|
mca_base_param_reg_int(&mca_osc_pt2pt_component.super.osc_version,
|
|
|
|
"eager_limit",
|
|
|
|
"Max size of eagerly sent data",
|
|
|
|
false, false, 16 * 1024,
|
|
|
|
&tmp);
|
|
|
|
|
|
|
|
mca_osc_pt2pt_component.p2p_c_eager_size = tmp;
|
2006-03-24 21:56:59 +03:00
|
|
|
|
2006-03-16 21:40:42 +03:00
|
|
|
return OMPI_SUCCESS;
|
|
|
|
}
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
int
|
|
|
|
ompi_osc_pt2pt_component_init(bool enable_progress_threads,
|
|
|
|
bool enable_mpi_threads)
|
|
|
|
{
|
|
|
|
/* we can run with either threads or not threads (may not be able
|
|
|
|
to do win locks)... */
|
|
|
|
mca_osc_pt2pt_component.p2p_c_have_progress_threads =
|
|
|
|
enable_progress_threads;
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&mca_osc_pt2pt_component.p2p_c_lock, opal_mutex_t);
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&mca_osc_pt2pt_component.p2p_c_modules,
|
|
|
|
opal_hash_table_t);
|
|
|
|
opal_hash_table_init(&mca_osc_pt2pt_component.p2p_c_modules, 2);
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&mca_osc_pt2pt_component.p2p_c_sendreqs, opal_free_list_t);
|
|
|
|
opal_free_list_init(&mca_osc_pt2pt_component.p2p_c_sendreqs,
|
|
|
|
sizeof(ompi_osc_pt2pt_sendreq_t),
|
|
|
|
OBJ_CLASS(ompi_osc_pt2pt_sendreq_t),
|
|
|
|
1, -1, 1);
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&mca_osc_pt2pt_component.p2p_c_replyreqs, opal_free_list_t);
|
|
|
|
opal_free_list_init(&mca_osc_pt2pt_component.p2p_c_replyreqs,
|
|
|
|
sizeof(ompi_osc_pt2pt_replyreq_t),
|
|
|
|
OBJ_CLASS(ompi_osc_pt2pt_replyreq_t),
|
|
|
|
1, -1, 1);
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&mca_osc_pt2pt_component.p2p_c_longreqs, opal_free_list_t);
|
|
|
|
opal_free_list_init(&mca_osc_pt2pt_component.p2p_c_longreqs,
|
|
|
|
sizeof(ompi_osc_pt2pt_longreq_t),
|
|
|
|
OBJ_CLASS(ompi_osc_pt2pt_longreq_t),
|
|
|
|
1, -1, 1);
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
OBJ_CONSTRUCT(&mca_osc_pt2pt_component.p2p_c_buffers, opal_free_list_t);
|
|
|
|
opal_free_list_init(&mca_osc_pt2pt_component.p2p_c_buffers,
|
|
|
|
mca_osc_pt2pt_component.p2p_c_eager_size +
|
|
|
|
sizeof(ompi_osc_pt2pt_buffer_t),
|
|
|
|
OBJ_CLASS(ompi_osc_pt2pt_buffer_t),
|
|
|
|
1, -1, 1);
|
|
|
|
|
2006-05-03 22:17:00 +04:00
|
|
|
return OMPI_SUCCESS;
|
2006-01-28 18:38:37 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
ompi_osc_pt2pt_component_finalize(void)
|
|
|
|
{
|
|
|
|
size_t num_modules;
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
if (0 !=
|
2006-01-28 18:38:37 +03:00
|
|
|
(num_modules = opal_hash_table_get_size(&mca_osc_pt2pt_component.p2p_c_modules))) {
|
2006-08-17 18:52:20 +04:00
|
|
|
opal_output(ompi_osc_base_output,
|
2006-08-24 20:38:08 +04:00
|
|
|
"WARNING: There were %d Windows created but not freed.",
|
|
|
|
num_modules);
|
2006-08-03 04:10:19 +04:00
|
|
|
opal_progress_unregister(ompi_osc_pt2pt_progress);
|
2006-01-28 18:38:37 +03:00
|
|
|
}
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
OBJ_DESTRUCT(&mca_osc_pt2pt_component.p2p_c_buffers);
|
2006-01-28 18:38:37 +03:00
|
|
|
OBJ_DESTRUCT(&mca_osc_pt2pt_component.p2p_c_longreqs);
|
|
|
|
OBJ_DESTRUCT(&mca_osc_pt2pt_component.p2p_c_replyreqs);
|
|
|
|
OBJ_DESTRUCT(&mca_osc_pt2pt_component.p2p_c_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&mca_osc_pt2pt_component.p2p_c_modules);
|
|
|
|
OBJ_DESTRUCT(&mca_osc_pt2pt_component.p2p_c_lock);
|
|
|
|
|
|
|
|
return OMPI_SUCCESS;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
ompi_osc_pt2pt_component_query(ompi_win_t *win,
|
|
|
|
ompi_info_t *info,
|
|
|
|
ompi_communicator_t *comm)
|
|
|
|
{
|
2006-02-01 00:40:12 +03:00
|
|
|
/* we can always run - return a low priority */
|
2006-08-03 04:10:19 +04:00
|
|
|
return 5;
|
2006-01-28 18:38:37 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int
|
|
|
|
ompi_osc_pt2pt_component_select(ompi_win_t *win,
|
|
|
|
ompi_info_t *info,
|
|
|
|
ompi_communicator_t *comm)
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_module_t *module;
|
|
|
|
int ret, i;
|
2006-03-16 21:40:42 +03:00
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
/* create module structure */
|
2006-08-24 20:38:08 +04:00
|
|
|
module = (ompi_osc_pt2pt_module_t*)malloc(sizeof(ompi_osc_pt2pt_module_t));
|
2006-01-28 18:38:37 +03:00
|
|
|
if (NULL == module) return OMPI_ERROR;
|
|
|
|
|
|
|
|
/* fill in the function pointer part */
|
|
|
|
memcpy(module, &ompi_osc_pt2pt_module_template,
|
|
|
|
sizeof(ompi_osc_base_module_t));
|
|
|
|
|
|
|
|
/* initialize the p2p part */
|
|
|
|
OBJ_CONSTRUCT(&(module->p2p_lock), opal_mutex_t);
|
|
|
|
OBJ_CONSTRUCT(&(module->p2p_acc_lock), opal_mutex_t);
|
|
|
|
|
|
|
|
module->p2p_win = win;
|
|
|
|
|
|
|
|
ret = ompi_comm_dup(comm, &(module->p2p_comm));
|
|
|
|
if (ret != OMPI_SUCCESS) {
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
module->p2p_control_buffer = malloc(mca_osc_pt2pt_component.p2p_c_eager_size);
|
|
|
|
if (NULL == module->p2p_control_buffer) {
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return OMPI_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
module->p2p_cb_request = NULL;
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&module->p2p_pending_control_sends, opal_list_t);
|
|
|
|
|
2006-02-07 15:16:23 +03:00
|
|
|
OBJ_CONSTRUCT(&module->p2p_pending_sendreqs, opal_list_t);
|
2006-08-24 20:38:08 +04:00
|
|
|
module->p2p_num_pending_sendreqs = (short*)malloc(sizeof(short) *
|
|
|
|
ompi_comm_size(module->p2p_comm));
|
2006-02-07 15:16:23 +03:00
|
|
|
if (NULL == module->p2p_num_pending_sendreqs) {
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
2006-01-28 18:38:37 +03:00
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return ret;
|
|
|
|
}
|
2006-02-07 15:16:23 +03:00
|
|
|
memset(module->p2p_num_pending_sendreqs, 0,
|
|
|
|
sizeof(short) * ompi_comm_size(module->p2p_comm));
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
module->p2p_num_pending_out = 0;
|
|
|
|
module->p2p_num_pending_in = 0;
|
2006-09-21 23:57:57 +04:00
|
|
|
module->p2p_num_post_msgs = 0;
|
|
|
|
module->p2p_num_complete_msgs = 0;
|
2006-01-28 18:38:37 +03:00
|
|
|
module->p2p_tag_counter = 0;
|
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&(module->p2p_long_msgs), opal_list_t);
|
2006-02-07 15:16:23 +03:00
|
|
|
|
|
|
|
OBJ_CONSTRUCT(&(module->p2p_copy_pending_sendreqs), opal_list_t);
|
2006-08-24 20:38:08 +04:00
|
|
|
module->p2p_copy_num_pending_sendreqs = (short*)malloc(sizeof(short) *
|
|
|
|
ompi_comm_size(module->p2p_comm));
|
2006-02-07 15:16:23 +03:00
|
|
|
if (NULL == module->p2p_copy_num_pending_sendreqs) {
|
|
|
|
OBJ_DESTRUCT(&module->p2p_copy_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_long_msgs);
|
|
|
|
free(module->p2p_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
memset(module->p2p_num_pending_sendreqs, 0,
|
|
|
|
sizeof(short) * ompi_comm_size(module->p2p_comm));
|
|
|
|
|
|
|
|
/* fence data */
|
2006-08-24 20:38:08 +04:00
|
|
|
module->p2p_fence_coll_counts = (int*)malloc(sizeof(int) *
|
|
|
|
ompi_comm_size(module->p2p_comm));
|
2006-02-07 15:16:23 +03:00
|
|
|
if (NULL == module->p2p_fence_coll_counts) {
|
|
|
|
free(module->p2p_copy_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_copy_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_long_msgs);
|
|
|
|
free(module->p2p_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
for (i = 0 ; i < ompi_comm_size(module->p2p_comm) ; ++i) {
|
|
|
|
module->p2p_fence_coll_counts[i] = 1;
|
|
|
|
}
|
|
|
|
|
2006-08-24 20:38:08 +04:00
|
|
|
module->p2p_fence_coll_results = (short*)malloc(sizeof(short) *
|
|
|
|
ompi_comm_size(module->p2p_comm));
|
2006-09-22 00:49:15 +04:00
|
|
|
if (NULL == module->p2p_fence_coll_results) {
|
2006-03-16 21:40:42 +03:00
|
|
|
free(module->p2p_fence_coll_counts);
|
|
|
|
free(module->p2p_copy_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_copy_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_long_msgs);
|
|
|
|
free(module->p2p_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2006-02-07 15:16:23 +03:00
|
|
|
/* pwsc data */
|
|
|
|
module->p2p_pw_group = NULL;
|
|
|
|
module->p2p_sc_group = NULL;
|
2006-09-22 00:49:15 +04:00
|
|
|
module->p2p_sc_remote_active_ranks =
|
|
|
|
malloc(sizeof(bool) * ompi_comm_size(module->p2p_comm));
|
|
|
|
if (NULL == module->p2p_sc_remote_active_ranks) {
|
|
|
|
free(module->p2p_fence_coll_results);
|
|
|
|
free(module->p2p_fence_coll_counts);
|
|
|
|
free(module->p2p_copy_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_copy_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_long_msgs);
|
|
|
|
free(module->p2p_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return OMPI_ERROR;
|
|
|
|
}
|
|
|
|
module->p2p_sc_remote_ranks =
|
|
|
|
malloc(sizeof(int) * ompi_comm_size(module->p2p_comm));
|
|
|
|
if (NULL == module->p2p_sc_remote_ranks) {
|
|
|
|
free(module->p2p_sc_remote_active_ranks);
|
|
|
|
free(module->p2p_fence_coll_results);
|
|
|
|
free(module->p2p_fence_coll_counts);
|
|
|
|
free(module->p2p_copy_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_copy_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_long_msgs);
|
|
|
|
free(module->p2p_num_pending_sendreqs);
|
|
|
|
OBJ_DESTRUCT(&module->p2p_pending_sendreqs);
|
|
|
|
ompi_comm_free(&comm);
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_acc_lock));
|
|
|
|
OBJ_DESTRUCT(&(module->p2p_lock));
|
|
|
|
free(module);
|
|
|
|
return OMPI_ERROR;
|
|
|
|
}
|
2006-02-07 15:16:23 +03:00
|
|
|
|
|
|
|
/* lock data */
|
2006-02-07 21:45:18 +03:00
|
|
|
module->p2p_lock_status = 0;
|
|
|
|
module->p2p_shared_count = 0;
|
|
|
|
OBJ_CONSTRUCT(&(module->p2p_locks_pending), opal_list_t);
|
|
|
|
module->p2p_lock_received_ack = 0;
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
/* update component data */
|
|
|
|
OPAL_THREAD_LOCK(&mca_osc_pt2pt_component.p2p_c_lock);
|
|
|
|
opal_hash_table_set_value_uint32(&mca_osc_pt2pt_component.p2p_c_modules,
|
|
|
|
module->p2p_comm->c_contextid,
|
|
|
|
module);
|
2006-08-03 04:10:19 +04:00
|
|
|
|
|
|
|
if (1 == opal_hash_table_get_size(&mca_osc_pt2pt_component.p2p_c_modules)) {
|
|
|
|
/* start progress thread */
|
|
|
|
opal_progress_register(ompi_osc_pt2pt_progress);
|
|
|
|
}
|
2006-01-28 18:38:37 +03:00
|
|
|
OPAL_THREAD_UNLOCK(&mca_osc_pt2pt_component.p2p_c_lock);
|
2006-02-21 21:43:28 +03:00
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
/* fill in window information */
|
|
|
|
win->w_osc_module = (ompi_osc_base_module_t*) module;
|
2006-03-24 21:56:59 +03:00
|
|
|
if (check_config_value_bool("no_locks", info)) {
|
2006-02-21 21:43:28 +03:00
|
|
|
win->w_flags |= OMPI_WIN_NO_LOCKS;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* sync memory - make sure all initialization completed */
|
|
|
|
opal_atomic_mb();
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
/* start up receive for protocol headers */
|
|
|
|
ret = MCA_PML_CALL(irecv(module->p2p_control_buffer,
|
|
|
|
mca_osc_pt2pt_component.p2p_c_eager_size,
|
|
|
|
MPI_BYTE,
|
|
|
|
MPI_ANY_SOURCE,
|
|
|
|
-200,
|
|
|
|
module->p2p_comm,
|
|
|
|
&module->p2p_cb_request));
|
2006-03-24 21:56:59 +03:00
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* dispatch for callback on message completion */
|
2006-08-03 04:10:19 +04:00
|
|
|
static void
|
|
|
|
ompi_osc_pt2pt_component_fragment_cb(ompi_osc_pt2pt_module_t *module,
|
|
|
|
void *buffer,
|
|
|
|
size_t buffer_len)
|
2006-01-28 18:38:37 +03:00
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
void *payload;
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(buffer_len >=
|
2006-02-07 15:16:23 +03:00
|
|
|
sizeof(ompi_osc_pt2pt_base_header_t));
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
/* handle message */
|
2006-08-03 04:10:19 +04:00
|
|
|
switch (((ompi_osc_pt2pt_base_header_t*) buffer)->hdr_type) {
|
2006-01-28 18:38:37 +03:00
|
|
|
case OMPI_OSC_PT2PT_HDR_PUT:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_send_header_t *header;
|
|
|
|
|
|
|
|
/* get our header and payload */
|
|
|
|
header = (ompi_osc_pt2pt_send_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-01-28 18:38:37 +03:00
|
|
|
payload = (void*) (header + 1);
|
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_SEND_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module == ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-09-01 01:07:52 +04:00
|
|
|
if (!ompi_win_exposure_epoch(module->p2p_win)) {
|
|
|
|
opal_output(0, "Invalid MPI_PUT on Window %s. Window not in exposure epoch",
|
|
|
|
module->p2p_win->w_name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
ret = ompi_osc_pt2pt_sendreq_recv_put(module, header, payload);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case OMPI_OSC_PT2PT_HDR_ACC:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_send_header_t *header;
|
|
|
|
|
|
|
|
/* get our header and payload */
|
|
|
|
header = (ompi_osc_pt2pt_send_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-01-28 18:38:37 +03:00
|
|
|
payload = (void*) (header + 1);
|
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_SEND_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module == ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-09-01 01:07:52 +04:00
|
|
|
if (!ompi_win_exposure_epoch(module->p2p_win)) {
|
|
|
|
opal_output(0, "Invalid MPI_ACCUMULATE on Window %s. Window not in exposure epoch",
|
|
|
|
module->p2p_win->w_name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
/* receive into temporary buffer */
|
|
|
|
ret = ompi_osc_pt2pt_sendreq_recv_accum(module, header, payload);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case OMPI_OSC_PT2PT_HDR_GET:
|
|
|
|
{
|
|
|
|
ompi_datatype_t *datatype;
|
|
|
|
ompi_osc_pt2pt_send_header_t *header;
|
|
|
|
ompi_osc_pt2pt_replyreq_t *replyreq;
|
2006-02-07 15:16:23 +03:00
|
|
|
ompi_proc_t *proc;
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
/* get our header and payload */
|
|
|
|
header = (ompi_osc_pt2pt_send_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-01-28 18:38:37 +03:00
|
|
|
payload = (void*) (header + 1);
|
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_SEND_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module == ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-01-28 18:38:37 +03:00
|
|
|
|
2006-09-01 01:07:52 +04:00
|
|
|
if (!ompi_win_exposure_epoch(module->p2p_win)) {
|
|
|
|
opal_output(0, "Invalid MPI_GET on Window %s. Window not in exposure epoch",
|
|
|
|
module->p2p_win->w_name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
/* create or get a pointer to our datatype */
|
2006-09-21 02:14:46 +04:00
|
|
|
proc = ompi_comm_peer_lookup( module->p2p_comm, header->hdr_origin );
|
2006-02-07 15:16:23 +03:00
|
|
|
datatype = ompi_osc_pt2pt_datatype_create(proc, &payload);
|
2006-01-28 18:38:37 +03:00
|
|
|
|
|
|
|
/* create replyreq sendreq */
|
|
|
|
ret = ompi_osc_pt2pt_replyreq_alloc_init(module,
|
|
|
|
header->hdr_origin,
|
|
|
|
header->hdr_origin_sendreq,
|
|
|
|
header->hdr_target_disp,
|
|
|
|
header->hdr_target_count,
|
|
|
|
datatype,
|
|
|
|
&replyreq);
|
|
|
|
|
|
|
|
/* send replyreq */
|
|
|
|
ompi_osc_pt2pt_replyreq_send(module, replyreq);
|
|
|
|
|
|
|
|
/* sendreq does the right retain, so we can release safely */
|
|
|
|
OBJ_RELEASE(datatype);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case OMPI_OSC_PT2PT_HDR_REPLY:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_reply_header_t *header;
|
|
|
|
ompi_osc_pt2pt_sendreq_t *sendreq;
|
|
|
|
|
|
|
|
/* get our header and payload */
|
|
|
|
header = (ompi_osc_pt2pt_reply_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-01-28 18:38:37 +03:00
|
|
|
payload = (void*) (header + 1);
|
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_REPLY_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
/* get original sendreq pointer */
|
|
|
|
sendreq = (ompi_osc_pt2pt_sendreq_t*) header->hdr_origin_sendreq.pval;
|
|
|
|
module = sendreq->req_module;
|
|
|
|
|
|
|
|
/* receive data */
|
|
|
|
ompi_osc_pt2pt_replyreq_recv(module, sendreq, header, payload);
|
|
|
|
}
|
|
|
|
break;
|
2006-01-31 05:44:08 +03:00
|
|
|
case OMPI_OSC_PT2PT_HDR_POST:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_control_header_t *header =
|
|
|
|
(ompi_osc_pt2pt_control_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-01-31 05:44:08 +03:00
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_CONTROL_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module == ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-01-31 05:44:08 +03:00
|
|
|
|
2006-09-21 23:57:57 +04:00
|
|
|
OPAL_THREAD_ADD32(&(module->p2p_num_post_msgs), -1);
|
2006-01-31 05:44:08 +03:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
case OMPI_OSC_PT2PT_HDR_COMPLETE:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_control_header_t *header =
|
|
|
|
(ompi_osc_pt2pt_control_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-01-31 05:44:08 +03:00
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_CONTROL_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module = ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-01-31 05:44:08 +03:00
|
|
|
|
|
|
|
/* we've heard from one more place, and have value reqs to
|
|
|
|
process */
|
2006-09-21 23:57:57 +04:00
|
|
|
OPAL_THREAD_ADD32(&(module->p2p_num_complete_msgs), -1);
|
2006-02-07 21:45:18 +03:00
|
|
|
OPAL_THREAD_ADD32(&(module->p2p_num_pending_in), header->hdr_value[0]);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case OMPI_OSC_PT2PT_HDR_LOCK_REQ:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_control_header_t *header =
|
|
|
|
(ompi_osc_pt2pt_control_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-02-07 21:45:18 +03:00
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_CONTROL_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module == ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-02-07 21:45:18 +03:00
|
|
|
|
|
|
|
if (header->hdr_value[1] > 0) {
|
|
|
|
ompi_osc_pt2pt_passive_lock(module, header->hdr_value[0],
|
|
|
|
header->hdr_value[1]);
|
|
|
|
} else {
|
|
|
|
OPAL_THREAD_ADD32(&(module->p2p_lock_received_ack), 1);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case OMPI_OSC_PT2PT_HDR_UNLOCK_REQ:
|
|
|
|
{
|
|
|
|
ompi_osc_pt2pt_control_header_t *header =
|
|
|
|
(ompi_osc_pt2pt_control_header_t*)
|
2006-08-03 04:10:19 +04:00
|
|
|
buffer;
|
2006-02-07 21:45:18 +03:00
|
|
|
|
2006-02-27 21:47:00 +03:00
|
|
|
#if !defined(WORDS_BIGENDIAN) && OMPI_ENABLE_HETEROGENEOUS_SUPPORT
|
|
|
|
if (header->hdr_base.hdr_flags & OMPI_OSC_PT2PT_HDR_FLAG_NBO) {
|
|
|
|
OMPI_OSC_PT2PT_CONTROL_HDR_NTOH(*header);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2006-08-03 04:10:19 +04:00
|
|
|
assert(module == ompi_osc_pt2pt_windx_to_module(header->hdr_windx));
|
2006-02-07 21:45:18 +03:00
|
|
|
|
|
|
|
ompi_osc_pt2pt_passive_unlock(module, header->hdr_value[0],
|
|
|
|
header->hdr_value[1]);
|
2006-01-31 05:44:08 +03:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
2006-01-28 18:38:37 +03:00
|
|
|
default:
|
2006-08-17 18:52:20 +04:00
|
|
|
opal_output_verbose(5, ompi_osc_base_output,
|
|
|
|
"received packet for Window with unknown type");
|
2006-01-28 18:38:37 +03:00
|
|
|
}
|
|
|
|
}
|
2006-08-03 04:10:19 +04:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static int
|
|
|
|
ompi_osc_pt2pt_request_test(ompi_request_t ** rptr,
|
|
|
|
int *completed,
|
|
|
|
ompi_status_public_t * status )
|
|
|
|
{
|
|
|
|
ompi_request_t *request = *rptr;
|
|
|
|
int ret = OMPI_SUCCESS;
|
|
|
|
|
|
|
|
#if OMPI_ENABLE_PROGRESS_THREADS == 0
|
|
|
|
if (request->req_state == OMPI_REQUEST_INACTIVE ||
|
|
|
|
request->req_complete) {
|
|
|
|
ret = ompi_request_test(rptr, completed, status);
|
|
|
|
} else {
|
|
|
|
*completed = 0;
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
ret = ompi_request_test(rptr, completed, status);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
|
|
|
ompi_osc_pt2pt_progress(void)
|
|
|
|
{
|
|
|
|
int ret, done, count = 0;
|
|
|
|
ompi_status_public_t status;
|
|
|
|
void *node;
|
|
|
|
uint32_t key;
|
|
|
|
ompi_osc_pt2pt_module_t *module;
|
|
|
|
opal_list_item_t *item;
|
|
|
|
|
|
|
|
ret = opal_hash_table_get_first_key_uint32(&mca_osc_pt2pt_component.p2p_c_modules,
|
|
|
|
&key,
|
|
|
|
(void**) &module,
|
|
|
|
&node);
|
|
|
|
if (OMPI_SUCCESS != ret) return 0;
|
|
|
|
|
|
|
|
do {
|
|
|
|
ret = ompi_osc_pt2pt_request_test(&module->p2p_cb_request, &done, &status);
|
|
|
|
if (OMPI_SUCCESS == ret && done) {
|
|
|
|
/* process message */
|
|
|
|
ompi_osc_pt2pt_component_fragment_cb(module,
|
|
|
|
module->p2p_control_buffer,
|
|
|
|
status._count);
|
|
|
|
|
|
|
|
/* repost receive */
|
|
|
|
ret = MCA_PML_CALL(irecv(module->p2p_control_buffer,
|
|
|
|
mca_osc_pt2pt_component.p2p_c_eager_size,
|
|
|
|
MPI_BYTE,
|
|
|
|
MPI_ANY_SOURCE,
|
|
|
|
-200,
|
|
|
|
module->p2p_comm,
|
|
|
|
&module->p2p_cb_request));
|
|
|
|
assert(OMPI_SUCCESS == ret);
|
|
|
|
count++;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* loop through sends */
|
|
|
|
for (item = opal_list_get_first(&module->p2p_pending_control_sends) ;
|
|
|
|
item != opal_list_get_end(&module->p2p_pending_control_sends) ;
|
|
|
|
item = opal_list_get_next(item)) {
|
|
|
|
ompi_osc_pt2pt_buffer_t *buffer =
|
|
|
|
(ompi_osc_pt2pt_buffer_t*) item;
|
|
|
|
|
|
|
|
ret = ompi_osc_pt2pt_request_test(&buffer->request, &done, &status);
|
|
|
|
if (OMPI_SUCCESS == ret && done) {
|
|
|
|
item = opal_list_remove_item(&module->p2p_pending_control_sends,
|
|
|
|
item);
|
|
|
|
buffer->cbfunc(buffer);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
} while (OMPI_SUCCESS ==
|
|
|
|
opal_hash_table_get_next_key_uint32(&mca_osc_pt2pt_component.p2p_c_modules,
|
|
|
|
&key,
|
|
|
|
(void**) &module,
|
|
|
|
node,
|
|
|
|
&node));
|
|
|
|
|
|
|
|
return count;
|
|
|
|
}
|
|
|
|
|
|
|
|
|