707e594d13
Authored-by: Reese Faucette <rfaucett@cisco.com> Reviewed-by: Jeff Squyres <jsquyres@cisco.com> cmr=v1.7.5:ticket=trac:4253 This commit was SVN r30833. The following Trac tickets were found above: Ticket 4253 --> https://svn.open-mpi.org/trac/ompi/ticket/4253
184 строки
6.1 KiB
C
184 строки
6.1 KiB
C
/*
|
|
* Copyright (c) 2004-2008 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2011 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) 2006 Sandia National Laboratories. All rights
|
|
* reserved.
|
|
* Copyright (c) 2008-2013 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2012 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
|
|
#include <infiniband/verbs.h>
|
|
|
|
#include "opal_stdint.h"
|
|
|
|
#include "ompi/constants.h"
|
|
#include "ompi/mca/btl/btl.h"
|
|
#include "ompi/mca/btl/base/base.h"
|
|
#include "ompi/mca/common/verbs/common_verbs.h"
|
|
|
|
#include "btl_usnic.h"
|
|
#include "btl_usnic_frag.h"
|
|
#include "btl_usnic_util.h"
|
|
#include "btl_usnic_send.h"
|
|
#include "btl_usnic_ack.h"
|
|
|
|
|
|
/*
|
|
* This function is called when a send of a full-fragment segment completes
|
|
* Return the WQE and also return the segment if no ACK pending
|
|
*/
|
|
void
|
|
ompi_btl_usnic_frag_send_complete(ompi_btl_usnic_module_t *module,
|
|
ompi_btl_usnic_send_segment_t *sseg)
|
|
{
|
|
ompi_btl_usnic_send_frag_t *frag;
|
|
|
|
frag = sseg->ss_parent_frag;
|
|
|
|
/* Reap a frag that was sent */
|
|
--sseg->ss_send_posted;
|
|
--frag->sf_seg_post_cnt;
|
|
|
|
/* checks for returnability made inside */
|
|
ompi_btl_usnic_send_frag_return_cond(module, frag);
|
|
|
|
/* do bookkeeping */
|
|
++frag->sf_endpoint->endpoint_send_credits;
|
|
++module->mod_channels[sseg->ss_channel].sd_wqe;
|
|
|
|
/* see if this endpoint needs to be made ready-to-send */
|
|
ompi_btl_usnic_check_rts(frag->sf_endpoint);
|
|
}
|
|
|
|
/*
|
|
* This function is called when a send segment completes
|
|
* Return the WQE and also return the segment if no ACK pending
|
|
*/
|
|
void
|
|
ompi_btl_usnic_chunk_send_complete(ompi_btl_usnic_module_t *module,
|
|
ompi_btl_usnic_send_segment_t *sseg)
|
|
{
|
|
ompi_btl_usnic_send_frag_t *frag;
|
|
|
|
frag = sseg->ss_parent_frag;
|
|
|
|
/* Reap a frag that was sent */
|
|
--sseg->ss_send_posted;
|
|
--frag->sf_seg_post_cnt;
|
|
|
|
if (sseg->ss_send_posted == 0 && !sseg->ss_ack_pending) {
|
|
ompi_btl_usnic_release_send_segment(module, frag, sseg);
|
|
}
|
|
|
|
/* done with whole fragment? */
|
|
/* checks for returnability made inside */
|
|
ompi_btl_usnic_send_frag_return_cond(module, frag);
|
|
|
|
/* do bookkeeping */
|
|
++frag->sf_endpoint->endpoint_send_credits;
|
|
++module->mod_channels[sseg->ss_channel].sd_wqe;
|
|
|
|
/* see if this endpoint needs to be made ready-to-send */
|
|
ompi_btl_usnic_check_rts(frag->sf_endpoint);
|
|
}
|
|
|
|
/* Responsible for completing non-fastpath parts of a put or send operation,
|
|
* including initializing any large frag bookkeeping fields and enqueuing the
|
|
* frag on the endpoint.
|
|
*
|
|
* This routine lives in this file to help prevent automatic inlining by the
|
|
* compiler.
|
|
*
|
|
* The "tag" only applies to sends. */
|
|
int
|
|
ompi_btl_usnic_finish_put_or_send(
|
|
ompi_btl_usnic_module_t *module,
|
|
ompi_btl_usnic_endpoint_t *endpoint,
|
|
ompi_btl_usnic_send_frag_t *frag,
|
|
mca_btl_base_tag_t tag)
|
|
{
|
|
int rc;
|
|
ompi_btl_usnic_small_send_frag_t *sfrag;
|
|
ompi_btl_usnic_send_segment_t *sseg;
|
|
|
|
/*
|
|
* If this is small, need to do the copyin now.
|
|
* We don't do this earlier in case we got lucky and were
|
|
* able to do an inline send. We did not, so here we are...
|
|
*/
|
|
if (frag->sf_base.uf_type == OMPI_BTL_USNIC_FRAG_SMALL_SEND) {
|
|
|
|
sfrag = (ompi_btl_usnic_small_send_frag_t *)frag;
|
|
sseg = &sfrag->ssf_segment;
|
|
|
|
/* Copy in user data if there is any, collapsing 2 segments into 1.
|
|
* We already packed via the convertor if necessary, so we only need to
|
|
* handle the simple memcpy case here.
|
|
*/
|
|
if (frag->sf_base.uf_base.des_src_cnt > 1) {
|
|
/* no convertor */
|
|
assert(NULL != frag->sf_base.uf_src_seg[1].seg_addr.pval);
|
|
|
|
memcpy(((char *)(intptr_t)frag->sf_base.uf_src_seg[0].seg_addr.lval +
|
|
frag->sf_base.uf_src_seg[0].seg_len),
|
|
frag->sf_base.uf_src_seg[1].seg_addr.pval,
|
|
frag->sf_base.uf_src_seg[1].seg_len);
|
|
|
|
/* update 1st segment length */
|
|
frag->sf_base.uf_base.des_src_cnt = 1;
|
|
frag->sf_base.uf_src_seg[0].seg_len +=
|
|
frag->sf_base.uf_src_seg[1].seg_len;
|
|
}
|
|
|
|
sseg->ss_base.us_sg_entry[0].length =
|
|
sizeof(ompi_btl_usnic_btl_header_t) + frag->sf_size;
|
|
|
|
/* use standard channel */
|
|
sseg->ss_channel = USNIC_DATA_CHANNEL;
|
|
sseg->ss_base.us_btl_header->tag = tag;
|
|
|
|
if (frag->sf_base.uf_src_seg[0].seg_len < module->tiny_mtu) {
|
|
sseg->ss_send_desc.send_flags |= IBV_SEND_INLINE;
|
|
}
|
|
} else {
|
|
ompi_btl_usnic_large_send_frag_t *lfrag;
|
|
|
|
/* Save info about the frag so that future invocations of
|
|
* usnic_handle_large_send can generate segments to put on the wire. */
|
|
lfrag = (ompi_btl_usnic_large_send_frag_t *)frag;
|
|
lfrag->lsf_tag = tag;
|
|
lfrag->lsf_cur_offset = 0;
|
|
lfrag->lsf_cur_ptr = lfrag->lsf_des_src[0].seg_addr.pval;
|
|
lfrag->lsf_cur_sge = 0;
|
|
lfrag->lsf_bytes_left_in_sge = lfrag->lsf_des_src[0].seg_len;
|
|
lfrag->lsf_bytes_left = frag->sf_size;
|
|
|
|
if (lfrag->lsf_pack_on_the_fly) {
|
|
lfrag->lsf_pack_bytes_left = frag->sf_size;
|
|
} else {
|
|
/* we pre-packed the convertor into a chain in prepare_src */
|
|
lfrag->lsf_pack_bytes_left = 0;
|
|
}
|
|
}
|
|
|
|
/* queue this fragment into the send engine */
|
|
rc = ompi_btl_usnic_endpoint_enqueue_frag(endpoint, frag);
|
|
return rc;
|
|
}
|