189 строки
6.6 KiB
C
189 строки
6.6 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University.
|
|
* All rights reserved.
|
|
* Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
|
|
* 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) 2007 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* Copyright (c) 2010 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#ifndef OMPI_OSC_RDMA_SENDREQ_H
|
|
#define OMPI_OSC_RDMA_SENDREQ_H
|
|
|
|
#include "osc_rdma.h"
|
|
#include "osc_rdma_longreq.h"
|
|
|
|
#include "opal/class/opal_list.h"
|
|
#include "ompi/datatype/ompi_datatype.h"
|
|
#include "opal/datatype/opal_convertor.h"
|
|
#include "ompi/communicator/communicator.h"
|
|
#include "ompi/proc/proc.h"
|
|
#include "ompi/memchecker.h"
|
|
|
|
typedef enum {
|
|
OMPI_OSC_RDMA_GET,
|
|
OMPI_OSC_RDMA_ACC,
|
|
OMPI_OSC_RDMA_PUT
|
|
} ompi_osc_rdma_req_type_t;
|
|
|
|
|
|
struct ompi_osc_rdma_sendreq_t {
|
|
ompi_request_t super;
|
|
|
|
int req_refcount;
|
|
|
|
/** type of sendreq (from ompi_osc_rdma_req_type_t) */
|
|
ompi_osc_rdma_req_type_t req_type;
|
|
/** pointer to the module that created the sendreq */
|
|
ompi_osc_rdma_module_t *req_module;
|
|
|
|
/** Datatype for the origin side of the operation */
|
|
struct ompi_datatype_t *req_origin_datatype;
|
|
/** Convertor for the origin side of the operation. Setup for
|
|
either send (Put / Accumulate) or receive (Get) */
|
|
opal_convertor_t req_origin_convertor;
|
|
/** packed size of message on the origin side */
|
|
size_t req_origin_bytes_packed;
|
|
|
|
/** rank in module's communicator for target of operation */
|
|
int req_target_rank;
|
|
/** pointer to the proc structure for the target of the operation */
|
|
ompi_proc_t *req_target_proc;
|
|
|
|
/** displacement on target */
|
|
OPAL_PTRDIFF_TYPE req_target_disp;
|
|
/** datatype count on target */
|
|
int req_target_count;
|
|
/** datatype on target */
|
|
struct ompi_datatype_t *req_target_datatype;
|
|
|
|
/** op index on the target */
|
|
int req_op_id;
|
|
|
|
uint8_t remote_segs[MCA_BTL_SEG_MAX_SIZE];
|
|
};
|
|
typedef struct ompi_osc_rdma_sendreq_t ompi_osc_rdma_sendreq_t;
|
|
OBJ_CLASS_DECLARATION(ompi_osc_rdma_sendreq_t);
|
|
|
|
|
|
/** allocate and populate a sendreq structure. Both datatypes are
|
|
RETAINed for the life of the sendreq */
|
|
int
|
|
ompi_osc_rdma_sendreq_alloc_init(ompi_osc_rdma_req_type_t req_type,
|
|
void *origin_addr, int origin_count,
|
|
struct ompi_datatype_t *origin_dt,
|
|
int target, OPAL_PTRDIFF_TYPE target_disp,
|
|
int target_count,
|
|
struct ompi_datatype_t *target_datatype,
|
|
ompi_osc_rdma_module_t *module,
|
|
ompi_osc_rdma_sendreq_t **sendreq);
|
|
|
|
static inline int
|
|
ompi_osc_rdma_sendreq_alloc(ompi_osc_rdma_module_t *module,
|
|
int target_rank,
|
|
ompi_osc_rdma_sendreq_t **sendreq)
|
|
{
|
|
int ret;
|
|
opal_free_list_item_t *item;
|
|
ompi_proc_t *proc = ompi_comm_peer_lookup( module->m_comm, target_rank );
|
|
|
|
/* BWB - FIX ME - is this really the right return code? */
|
|
if (NULL == proc) return OMPI_ERR_OUT_OF_RESOURCE;
|
|
|
|
OPAL_FREE_LIST_GET(&mca_osc_rdma_component.c_sendreqs,
|
|
item, ret);
|
|
if (OMPI_SUCCESS != ret) return ret;
|
|
*sendreq = (ompi_osc_rdma_sendreq_t*) item;
|
|
|
|
(*sendreq)->req_module = module;
|
|
(*sendreq)->req_target_rank = target_rank;
|
|
(*sendreq)->req_target_proc = proc;
|
|
(*sendreq)->req_refcount = 1;
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|
|
static inline int
|
|
ompi_osc_rdma_sendreq_init_origin(ompi_osc_rdma_sendreq_t *sendreq,
|
|
ompi_osc_rdma_req_type_t req_type,
|
|
void *origin_addr,
|
|
int origin_count,
|
|
struct ompi_datatype_t *origin_dt)
|
|
{
|
|
OBJ_RETAIN(origin_dt);
|
|
sendreq->req_origin_datatype = origin_dt;
|
|
sendreq->req_type = req_type;
|
|
|
|
if (req_type != OMPI_OSC_RDMA_GET) {
|
|
opal_convertor_copy_and_prepare_for_send(sendreq->req_target_proc->proc_convertor,
|
|
&(origin_dt->super),
|
|
origin_count,
|
|
origin_addr,
|
|
0,
|
|
&(sendreq->req_origin_convertor));
|
|
opal_convertor_get_packed_size(&sendreq->req_origin_convertor,
|
|
&sendreq->req_origin_bytes_packed);
|
|
} else {
|
|
opal_convertor_copy_and_prepare_for_recv(sendreq->req_target_proc->proc_convertor,
|
|
&(origin_dt->super),
|
|
origin_count,
|
|
origin_addr,
|
|
0,
|
|
&(sendreq->req_origin_convertor));
|
|
opal_convertor_get_packed_size(&sendreq->req_origin_convertor,
|
|
&sendreq->req_origin_bytes_packed);
|
|
}
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|
|
static inline int
|
|
ompi_osc_rdma_sendreq_init_target(ompi_osc_rdma_sendreq_t *sendreq,
|
|
OPAL_PTRDIFF_TYPE target_disp,
|
|
int target_count,
|
|
struct ompi_datatype_t *target_datatype)
|
|
{
|
|
OBJ_RETAIN(target_datatype);
|
|
|
|
sendreq->req_target_disp = target_disp;
|
|
sendreq->req_target_count = target_count;
|
|
sendreq->req_target_datatype = target_datatype;
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
|
|
static inline int
|
|
ompi_osc_rdma_sendreq_free(ompi_osc_rdma_sendreq_t *sendreq)
|
|
{
|
|
if (0 == (--sendreq->req_refcount)) {
|
|
MEMCHECKER(
|
|
memchecker_convertor_call(&opal_memchecker_base_mem_defined,
|
|
&sendreq->req_origin_convertor);
|
|
);
|
|
opal_convertor_cleanup(&sendreq->req_origin_convertor);
|
|
|
|
OBJ_RELEASE(sendreq->req_target_datatype);
|
|
OBJ_RELEASE(sendreq->req_origin_datatype);
|
|
|
|
OPAL_FREE_LIST_RETURN(&mca_osc_rdma_component.c_sendreqs,
|
|
(opal_list_item_t*) sendreq);
|
|
}
|
|
|
|
return OMPI_SUCCESS;
|
|
}
|
|
|
|
#endif /* OMPI_OSC_RDMA_SENDREQ_H */
|