1
1
openmpi/ompi/mca/btl/usnic/btl_usnic_module.c
Jeff Squyres b30ad28276 Remove some unused variables and an unused goto label.
This commit was SVN r29044.
2013-08-19 16:18:35 +00:00

1915 строки
65 KiB
C

/*
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2008 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) 2009-2013 Cisco Systems, Inc. All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "ompi_config.h"
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include "opal/class/opal_bitmap.h"
#include "opal/prefetch.h"
#include "opal/util/output.h"
#include "opal/datatype/opal_convertor.h"
#include "opal/include/opal_stdint.h"
#include "opal/util/show_help.h"
#include "ompi/mca/rte/rte.h"
#include "ompi/mca/btl/btl.h"
#include "ompi/mca/btl/base/btl_base_error.h"
#include "ompi/mca/mpool/base/base.h"
#include "ompi/mca/mpool/mpool.h"
#include "ompi/memchecker.h"
#include "btl_usnic.h"
#include "btl_usnic_frag.h"
#include "btl_usnic_proc.h"
#include "btl_usnic_endpoint.h"
#include "btl_usnic_module.h"
#include "btl_usnic_util.h"
#include "btl_usnic_send.h"
#include "btl_usnic_ack.h"
#include "btl_usnic_hwloc.h"
static void
ompi_btl_usnic_channel_finalize(
ompi_btl_usnic_module_t *module,
struct ompi_btl_usnic_channel_t *channel);
/*
* Add procs to this BTL module, receiving endpoint information from
* the modex.
*/
static int usnic_add_procs(struct mca_btl_base_module_t* base_module,
size_t nprocs,
struct ompi_proc_t **ompi_procs,
struct mca_btl_base_endpoint_t** endpoints,
opal_bitmap_t* reachable)
{
ompi_btl_usnic_module_t* module = (ompi_btl_usnic_module_t*) base_module;
ompi_proc_t* my_proc;
size_t i, count;
int rc;
/* get pointer to my proc structure */
my_proc = ompi_proc_local();
if (NULL == my_proc) {
return OMPI_ERR_OUT_OF_RESOURCE;
}
count = 0;
for (i = 0; i < nprocs; i++) {
struct ompi_proc_t* ompi_proc = ompi_procs[i];
ompi_btl_usnic_proc_t* usnic_proc;
mca_btl_base_endpoint_t* usnic_endpoint;
/* Do not create loopback usnic connections */
if (ompi_proc == my_proc) {
continue;
}
/* usNIC does not support loopback to the same machine */
if (OPAL_PROC_ON_LOCAL_NODE(ompi_proc->proc_flags)) {
continue;
}
/* Find (or create if it doesn't exist) this peer's proc.
This will receive the modex info for that proc. Note that
the proc is shared by all usnic modules that are trying
to reach this destination. */
usnic_proc = NULL;
rc = ompi_btl_usnic_proc_match(ompi_proc, module, &usnic_proc);
if (OMPI_SUCCESS != rc) {
return OMPI_ERR_OUT_OF_RESOURCE;
}
/* Create the endpoint for this proc/module combination. If we cannot
* reach this proc via this module, move on to the next proc. */
usnic_endpoint = NULL;
rc = ompi_btl_usnic_create_endpoint(module, usnic_proc,
&usnic_endpoint);
if (OMPI_SUCCESS != rc) {
OBJ_RELEASE(usnic_proc);
continue;
}
/* Add to array of all procs */
opal_pointer_array_add(&module->all_procs, usnic_proc);
opal_output_verbose(5, USNIC_OUT,
"btl:usnic: new usnic peer: subnet = 0x%016" PRIx64 ", interface = 0x%016" PRIx64,
ntoh64(usnic_endpoint->endpoint_remote_addr.gid.global.subnet_prefix),
ntoh64(usnic_endpoint->endpoint_remote_addr.gid.global.interface_id));
opal_bitmap_set_bit(reachable, i);
endpoints[i] = usnic_endpoint;
opal_output_verbose(15, USNIC_OUT,
"btl:usnic: made %p endpoint", (void*) usnic_endpoint);
count++;
}
opal_output_verbose(5, USNIC_OUT,
"btl:usnic: made %" PRIsize_t " endpoints", count);
return OMPI_SUCCESS;
}
/*
* Delete the proc as reachable from this module. If there are
* multiple usnic modules in a process, we'll come through here
* multiple times to remove each proc. The OBJ reference counts
* will make all the details work out.
*/
static int usnic_del_procs(struct mca_btl_base_module_t *base_module,
size_t nprocs,
struct ompi_proc_t **procs,
struct mca_btl_base_endpoint_t **peers)
{
size_t i, j;
ompi_btl_usnic_module_t *module;
ompi_btl_usnic_endpoint_t *endpoint;
int index;
module = (struct ompi_btl_usnic_module_t *)base_module;
for (i = 0; i < nprocs; i++) {
ompi_btl_usnic_proc_t* proc =
ompi_btl_usnic_proc_lookup_ompi(procs[i]);
if (NULL != proc) {
/* find endpoint for this module */
for (j = 0; j < proc->proc_endpoint_count; ++j) {
endpoint = proc->proc_endpoints[j];
if (NULL != endpoint &&
endpoint->endpoint_module == module) {
/* If no pending ACKs needed, all done with endpoint */
if (ENDPOINT_DRAINED(endpoint)) {
OBJ_RELEASE(endpoint);
/* still some ACKs needed, mark endpoint as "exiting" */
} else {
endpoint->endpoint_exiting = true;
}
break; /* done once we found match */
}
}
/* remove proc from this module */
for (index = 0; index < module->all_procs.size; ++index) {
if (opal_pointer_array_get_item(&module->all_procs, index) ==
proc) {
opal_pointer_array_set_item(&module->all_procs, index,
NULL);
break;
}
}
}
}
return OMPI_SUCCESS;
}
/*
* Let the PML register a callback function with me
*/
static int usnic_register_pml_err_cb(struct mca_btl_base_module_t* btl,
mca_btl_base_module_error_cb_fn_t cbfunc)
{
ompi_btl_usnic_module_t *module = (ompi_btl_usnic_module_t*) btl;
module->pml_error_callback = cbfunc;
return OMPI_SUCCESS;
}
/**
* Allocate PML control messages or eager frags if BTL does not have
* INPLACE flag. To be clear: max it will ever alloc is eager_limit.
* THEREFORE: eager_limit is the max that ALLOC must always be able to
* alloc.
* --> Contraction in the btl.h documentation.
*/
static mca_btl_base_descriptor_t*
usnic_alloc(struct mca_btl_base_module_t* btl,
struct mca_btl_base_endpoint_t* endpoint,
uint8_t order,
size_t size,
uint32_t flags)
{
ompi_btl_usnic_send_frag_t *frag;
ompi_btl_usnic_module_t *module = (ompi_btl_usnic_module_t*) btl;
mca_btl_base_descriptor_t *desc;
/* will this fit into a small send? */
if (size <= module->max_frag_payload) {
ompi_btl_usnic_small_send_frag_t *sfrag;
sfrag = ompi_btl_usnic_small_send_frag_alloc(module);
if (NULL == sfrag) {
return NULL;
}
frag = &sfrag->ssf_base;
} else {
ompi_btl_usnic_large_send_frag_t *lfrag;
lfrag = ompi_btl_usnic_large_send_frag_alloc(module);
if (NULL == lfrag) {
return NULL;
}
frag = &lfrag->lsf_base;
abort(); /* XXX - we don't ever want to see this... */
}
#if MSGDEBUG2
opal_output(0, "usnic_alloc: %s frag=%p, size=%d\n",
(size <= module->max_frag_payload)?"small":"large",
(void *)frag, (int)size);
#endif
/* set # of bytes remaining to be ACKed */
frag->sf_ack_bytes_left = size;
frag->sf_size = size;
/* set endpoint */
frag->sf_endpoint = endpoint;
/* no convertor */
frag->sf_convertor = NULL;
/* set up descriptor */
desc = &frag->sf_base.uf_base;
desc->des_src[0].seg_len = size;
desc->des_src_cnt = 1;
desc->des_flags = flags;
return desc;
}
/**
* Return a small send fragment
*
* Return the send fragment to the appropriate list
*/
static int usnic_free(struct mca_btl_base_module_t* btl,
mca_btl_base_descriptor_t* des)
{
ompi_btl_usnic_frag_t* frag = (ompi_btl_usnic_frag_t*)des;
#if MSGDEBUG1
opal_output(0, "usnic_free: %p\n", frag);
#endif
OMPI_FREE_LIST_RETURN_MT(frag->uf_freelist, &(frag->uf_base.super));
return OMPI_SUCCESS;
}
/*
* Notes from george:
*
* - BTL ALLOC: allocating PML control messages or eager frags if BTL
does not have INPLACE flag. To be clear: max it will ever alloc
is eager_limit. THEREFORE: eager_limit is the max that ALLOC
must always be able to alloc.
--> Contraction in the btl.h documentation.
*
* - BTL PREPARE SRC: max_send_size frags go through here. Can return
a smaller size than was asked for.
*
* - BTL PREPARE DEST: not used if you don't have PUT/GET
*
* - BTL SEND: will be used after ALLOC / PREPARE
*/
/**
* Pack data and return a descriptor that can be used for send (or
* put, but we don't do that here in usnic).
*/
static mca_btl_base_descriptor_t*
usnic_prepare_src(
struct mca_btl_base_module_t* base_module,
struct mca_btl_base_endpoint_t* endpoint,
struct mca_mpool_base_registration_t* registration,
struct opal_convertor_t* convertor,
uint8_t order,
size_t reserve,
size_t* size,
uint32_t flags)
{
ompi_btl_usnic_module_t *module = (ompi_btl_usnic_module_t*) base_module;
ompi_btl_usnic_send_frag_t *frag;
void *data_ptr;
uint32_t payload_len;
/*
* If data is contiguous, get the data pointer for it, else leave
* data_ptr == NULL to indicate converter in use.
*/
if (OPAL_UNLIKELY(opal_convertor_need_buffers(convertor))) {
data_ptr = NULL;
} else {
opal_convertor_get_current_pointer (convertor, &data_ptr);
}
/*
* if convertor is involved, or total payload len fits in one MTU,
* use small send, else large
*/
payload_len = *size + reserve;
if (data_ptr == NULL || payload_len <= module->max_frag_payload) {
ompi_btl_usnic_small_send_frag_t *sfrag;
/* Get holder for the msg */
sfrag = ompi_btl_usnic_small_send_frag_alloc(module);
if (OPAL_UNLIKELY(NULL == sfrag)) {
return NULL;
}
frag = &sfrag->ssf_base;
/* In the case of a convertor, we will copy the data in now, since
* that is the only way to discover how much we can actually send
* The reason we do not copy non-convertor pointer at this point is
* because we might still use INLINE for the send, and in that case
* we do not want to copy the data at all.
*/
if (data_ptr == NULL) {
struct iovec iov;
uint32_t iov_count;
size_t max_data;
int rc;
/* put user data just after end of 1st seg (PML header) */
if (payload_len > module->max_frag_payload) {
payload_len = module->max_frag_payload;
}
iov.iov_len = payload_len - reserve;
iov.iov_base = (IOVBASE_TYPE*)
(frag->sf_base.uf_src_seg[0].seg_addr.lval + reserve);
iov_count = 1;
max_data = iov.iov_len;
rc = opal_convertor_pack(convertor, &iov, &iov_count, &max_data);
if (OPAL_UNLIKELY(rc < 0)) {
ompi_btl_usnic_send_frag_return_cond(module, frag);
abort(); /* XXX */
}
*size = max_data;
payload_len = max_data + reserve;
}
} else {
ompi_btl_usnic_large_send_frag_t *lfrag;
/* Get holder for the msg */
lfrag = ompi_btl_usnic_large_send_frag_alloc(module);
if (OPAL_UNLIKELY(NULL == lfrag)) {
return NULL;
}
frag = &lfrag->lsf_base;
/* Save info about the frag */
lfrag->lsf_cur_offset = 0;
lfrag->lsf_bytes_left = payload_len;
/* make sure PML header small enough */
assert(reserve < sizeof(lfrag->lsf_pml_header));
}
/* fill in segment sizes */
frag->sf_base.uf_src_seg[0].seg_len = reserve;
frag->sf_base.uf_src_seg[1].seg_len = *size;
frag->sf_size = payload_len;
frag->sf_base.uf_base.des_src_cnt = 2;
if (data_ptr != NULL) {
frag->sf_base.uf_src_seg[1].seg_addr.pval = data_ptr;
frag->sf_convertor = NULL;
} else {
frag->sf_convertor = convertor;
}
/* set up common parts of frag */
frag->sf_base.uf_base.des_flags = flags;
frag->sf_endpoint = endpoint;
/* fragment accounting */
frag->sf_ack_bytes_left = payload_len;
#if MSGDEBUG2
opal_output(0, "prep_src: %s %s frag %p, size=%d+%d, src=%p\n",
module->device->name,
payload_len <= module->max_frag_payload?"small":"large",
(void *)frag, (int)reserve, (int)*size,
frag->sf_base.uf_base.des_src[0].seg_addr.pval);
opal_output(0, " data_ptr = %p, conv=%p\n",
data_ptr, (void *)frag->sf_convertor);
#endif
return &frag->sf_base.uf_base;
}
static mca_btl_base_descriptor_t*
usnic_prepare_dst(
struct mca_btl_base_module_t* base_module,
struct mca_btl_base_endpoint_t* endpoint,
struct mca_mpool_base_registration_t* registration,
struct opal_convertor_t* convertor,
uint8_t order,
size_t reserve,
size_t* size,
uint32_t flags)
{
ompi_btl_usnic_put_dest_frag_t *pfrag;
ompi_btl_usnic_module_t *module;
void *data_ptr;
module = (ompi_btl_usnic_module_t *)base_module;
/* allocate a fragment for this */
pfrag = (ompi_btl_usnic_put_dest_frag_t *)
ompi_btl_usnic_put_dest_frag_alloc(module);
if (NULL == pfrag) {
return NULL;
}
/* find start of the data */
opal_convertor_get_current_pointer (convertor, (void **) &data_ptr);
/* make a seg entry pointing at data_ptr */
pfrag->uf_dst_seg[0].seg_addr.pval = data_ptr;
pfrag->uf_dst_seg[0].seg_len = *size;
pfrag->uf_base.order = order;
pfrag->uf_base.des_flags = flags;
#if MSGDEBUG2
opal_output(0, "prep_dst size=%d, addr=%p, pfrag=%p\n", (int)*size,
data_ptr, (void *)pfrag);
#endif
return &pfrag->uf_base;
}
/*
* Emulate an RDMA put. We'll send the remote address
* across to the other side so it will know where to put the data
*/
static int
usnic_put(
struct mca_btl_base_module_t *btl,
struct mca_btl_base_endpoint_t *endpoint,
struct mca_btl_base_descriptor_t *des)
{
ompi_btl_usnic_send_frag_t *frag;
ompi_btl_usnic_send_segment_t *sseg;
frag = (ompi_btl_usnic_send_frag_t *)des;
#if MSGDEBUG2
opal_output(0, "usnic_put: %d bytes to %p\n",
des->des_dst->seg_len,
des->des_dst->seg_addr.pval);
opal_output(0, " des_dst=%p, frag->uf_dst_seg=%p\n",
(void *)des->des_dst,
(void *)frag->sf_base.uf_dst_seg);
#endif
/* copy out address - why does he not use ours? silly PML */
frag->sf_base.uf_dst_seg[0].seg_addr.pval = des->des_dst->seg_addr.pval;
/*
* 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) {
ompi_btl_usnic_small_send_frag_t *sfrag;
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
*/
if (frag->sf_base.uf_base.des_src_cnt > 1) {
/* If not convertor, copy now. Already copied in convertor case */
if (frag->sf_convertor == NULL) {
memcpy(((char *)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;
}
/* set up VERBS SG list */
sseg->ss_send_desc.num_sge = 1;
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;
}
ompi_btl_usnic_endpoint_enqueue_frag(endpoint, frag);
return OMPI_SUCCESS;
}
static inline void usnic_stats_reset(ompi_btl_usnic_module_t *module)
{
int i;
module->num_total_sends =
module->num_resends =
module->num_chunk_sends =
module->num_frag_sends =
module->num_ack_recvs =
module->num_total_recvs =
module->num_unk_recvs =
module->num_dup_recvs =
module->num_oow_low_recvs =
module->num_oow_high_recvs =
module->num_frag_recvs =
module->num_chunk_recvs =
module->num_badfrag_recvs =
module->num_ack_sends =
module->num_recv_reposts =
module->num_crc_errors =
module->num_old_dup_acks =
module->num_dup_acks =
module->num_fast_retrans =
module->num_timeout_retrans =
module->max_sent_window_size =
module->max_rcvd_window_size =
module->pml_module_sends =
module->pml_send_callbacks =
0;
for (i=0; i<USNIC_NUM_CHANNELS; ++i) {
module->mod_channels[i].num_channel_sends = 0;
}
}
static void usnic_stats_callback(int fd, short flags, void *arg)
{
ompi_btl_usnic_module_t *module = (ompi_btl_usnic_module_t*) arg;
char tmp[128], str[2048];
if (!mca_btl_usnic_component.stats_enabled) {
return;
}
if (module->final_stats) {
snprintf(tmp, sizeof(tmp), "final");
} else {
snprintf(tmp, sizeof(tmp), "%4lu", ++module->stats_report_num);
}
/* The usuals */
snprintf(str, sizeof(str), "%s:MCW:%3u, ST(P+D)/F/C/R(T+F)/A:%8lu(%8u+%8u)/%8lu/%8lu/%4lu(%4lu+%4lu)/%8lu, RcvTot/Chk/F/C/L/H/D/BF/A:%8lu/%c%c/%8lu/%8lu/%4lu+%2lu/%4lu/%4lu/%6lu OA/DA %4lu/%4lu CRC:%4lu ",
tmp,
ompi_proc_local()->proc_name.vpid,
module->num_total_sends,
module->mod_channels[USNIC_PRIORITY_CHANNEL].num_channel_sends,
module->mod_channels[USNIC_DATA_CHANNEL].num_channel_sends,
module->num_frag_sends,
module->num_chunk_sends,
module->num_resends,
module->num_timeout_retrans,
module->num_fast_retrans,
module->num_ack_sends,
module->num_total_recvs,
(module->num_total_recvs - module->num_recv_reposts) == 0 ? 'g' : 'B',
(module->num_total_recvs -
module->num_frag_recvs -
module->num_chunk_recvs -
module->num_badfrag_recvs -
module->num_oow_low_recvs -
module->num_oow_high_recvs -
module->num_dup_recvs -
module->num_ack_recvs -
module->num_unk_recvs) == 0 ? 'g' : 'B',
module->num_frag_recvs,
module->num_chunk_recvs,
module->num_oow_low_recvs,
module->num_oow_high_recvs,
module->num_dup_recvs,
module->num_badfrag_recvs,
module->num_ack_recvs,
module->num_old_dup_acks,
module->num_dup_acks,
module->num_crc_errors);
/* If our PML calls were 0, then show send and receive window
extents instead */
if (module->pml_module_sends +
module->pml_send_callbacks == 0) {
int64_t send_unacked, su_min = WINDOW_SIZE * 2, su_max = 0;
int64_t recv_depth, rd_min = WINDOW_SIZE * 2, rd_max = 0;
ompi_btl_usnic_endpoint_t *endpoint;
opal_list_item_t *item;
rd_min = su_min = WINDOW_SIZE * 2;
rd_max = su_max = 0;
item = opal_list_get_first(&module->all_endpoints);
while (item != opal_list_get_end(&(module->all_endpoints))) {
endpoint = container_of(item, mca_btl_base_endpoint_t,
endpoint_endpoint_li);
item = opal_list_get_next(item);
/* Number of un-acked sends (i.e., sends for which we're
still waiting for ACK) */
send_unacked =
endpoint->endpoint_next_seq_to_send -
endpoint->endpoint_ack_seq_rcvd - 1;
if (send_unacked > su_max) su_max = send_unacked;
if (send_unacked < su_min) su_min = send_unacked;
/* Receive window depth (i.e., difference between highest
seq received and the next message we haven't ACKed
yet) */
recv_depth =
endpoint->endpoint_highest_seq_rcvd -
endpoint->endpoint_next_contig_seq_to_recv;
if (recv_depth > rd_max) rd_max = recv_depth;
if (recv_depth < rd_min) rd_min = recv_depth;
}
snprintf(tmp, sizeof(tmp), "PML S:%1ld, Win!A/R:%4ld/%4ld %4ld/%4ld",
module->pml_module_sends,
su_min, su_max,
rd_min, rd_max);
} else {
snprintf(tmp, sizeof(tmp), "PML S/CB/Diff:%4lu/%4lu=%4ld",
module->pml_module_sends,
module->pml_send_callbacks,
module->pml_module_sends - module->pml_send_callbacks);
}
strncat(str, tmp, sizeof(str) - strlen(str) - 1);
opal_output(0, str);
if (mca_btl_usnic_component.stats_relative) {
usnic_stats_reset(module);
}
/* In OMPI v1.6, we have to re-add this event (because there's an
old libevent in OMPI v1.6) */
if (!module->final_stats) {
opal_event_add(&(module->stats_timer_event),
&(module->stats_timeout));
}
}
static int usnic_finalize(struct mca_btl_base_module_t* btl)
{
int i;
ompi_btl_usnic_proc_t *proc;
ompi_btl_usnic_module_t* module = (ompi_btl_usnic_module_t*)btl;
if (module->device_async_event_active) {
opal_event_del(&(module->device_async_event));
module->device_async_event_active = false;
}
ompi_btl_usnic_channel_finalize(module,
&module->mod_channels[USNIC_DATA_CHANNEL]);
ompi_btl_usnic_channel_finalize(module,
&module->mod_channels[USNIC_PRIORITY_CHANNEL]);
ibv_dealloc_pd(module->pd);
/* Disable the stats callback event, and then call the stats
callback manually to display the final stats */
if (mca_btl_usnic_component.stats_enabled) {
opal_event_del(&(module->stats_timer_event));
module->final_stats = true;
usnic_stats_callback(0, 0, module);
}
/* Empty the all_endpoints list so that we can destroy endpoints
* (i.e., so that their super/opal_list_item_t member isn't still
* in use)
* This call to usnic_finalize is actually an implicit ACK of
* every packet we have ever sent, so call handle_ack to do all
* the ACK processing and release all the data that needs releasing.
*
* If we don't care about releasing data, we could just set:
* endpoint->endpoint_ack_seq_rcvd =
* endpoint->endpoint_next_seq_to_send-1;
* as the main point is to make usnic_del_procs() happy to release
* the proc/endpoint.
*/
while (!opal_list_is_empty(&(module->all_endpoints))) {
opal_list_item_t *item;
ompi_btl_usnic_endpoint_t *endpoint;
item = opal_list_remove_first(&module->all_endpoints);
endpoint = container_of(item, mca_btl_base_endpoint_t,
endpoint_endpoint_li);
if (!ENDPOINT_DRAINED(endpoint)) {
ompi_btl_usnic_flush_endpoint(endpoint);
}
}
OBJ_DESTRUCT(&(module->all_endpoints));
/* Similarly, empty the endpoints_that_need_acks list so that
endpoints don't still have an endpoint_ack_li item still in
use */
while (!opal_list_is_empty(&(module->endpoints_that_need_acks))) {
(void) opal_list_remove_first(&(module->endpoints_that_need_acks));
}
OBJ_DESTRUCT(&module->endpoints_that_need_acks);
/*
* Use usnic_del_procs to remove each proc from this module.
* This is done to prevent duplicating the proc deletion code
*/
for (i = 0; i < opal_pointer_array_get_size(&module->all_procs); ++i) {
proc = opal_pointer_array_get_item(&module->all_procs, i);
if (NULL != proc) {
usnic_del_procs(&module->super, 1, &proc->proc_ompi, NULL);
}
}
OBJ_DESTRUCT(&module->all_procs);
OBJ_DESTRUCT(&module->ack_segs);
OBJ_DESTRUCT(&module->endpoints_with_sends);
OBJ_DESTRUCT(&module->small_send_frags);
OBJ_DESTRUCT(&module->large_send_frags);
OBJ_DESTRUCT(&module->put_dest_frags);
OBJ_DESTRUCT(&module->senders);
mca_mpool_base_module_destroy(module->super.btl_mpool);
return OMPI_SUCCESS;
}
static void
usnic_do_resends(
ompi_btl_usnic_module_t *module)
{
ompi_btl_usnic_send_segment_t *sseg;
ompi_btl_usnic_endpoint_t *endpoint;
struct ompi_btl_usnic_channel_t *data_channel;
int ret;
data_channel = &module->mod_channels[USNIC_DATA_CHANNEL];
while (((size_t)data_channel->sd_wqe > 0) &&
!opal_list_is_empty(&module->pending_resend_segs)) {
/*
* If a segment is on the re-send list, it will not
* be in the retransmit hotel. Post the segment, then check it in.
*/
sseg = (ompi_btl_usnic_send_segment_t *)
opal_list_remove_first(&module->pending_resend_segs);
endpoint = sseg->ss_parent_frag->sf_endpoint;
/* clobber any stale piggy-backed ACK */
sseg->ss_base.us_btl_header->ack_seq = 0;
/* Only post this segment if not already posted */
if (sseg->ss_send_posted == 0) {
/* resends are always standard segments */
sseg->ss_channel = USNIC_DATA_CHANNEL;
/* re-send the segment */
ompi_btl_usnic_post_segment(module, endpoint, sseg);
/* consume a send credit for this endpoint. May send us
* negative, oh well... This is because the completion routine
* always increments send credits, and we must balance.
* Alternative is to mark this as a retrans segment and check in
* completion, but this ugly way avoids extra checks in the
* critical path. And, really, respects the concept of send
* credits more.
*/
--endpoint->endpoint_send_credits;
++module->num_resends;
}
/* restart the retrans timer */
ret = opal_hotel_checkin(&endpoint->endpoint_hotel,
sseg, &sseg->ss_hotel_room);
if (OPAL_UNLIKELY(OPAL_SUCCESS != ret)) {
abort(); /* should not be possible */
}
}
}
static void
usnic_handle_large_send(
ompi_btl_usnic_module_t *module,
ompi_btl_usnic_endpoint_t *endpoint,
ompi_btl_usnic_send_frag_t *frag)
{
ompi_btl_usnic_large_send_frag_t *lfrag;
ompi_btl_usnic_btl_chunk_header_t *chp;
ompi_btl_usnic_send_segment_t *sseg;
size_t payload_len;
struct iovec iov;
size_t max_data;
lfrag = (ompi_btl_usnic_large_send_frag_t *)frag;
sseg = ompi_btl_usnic_chunk_segment_alloc(module);
if (OPAL_UNLIKELY(NULL == sseg)) {
/* XXX do something better here */
abort();
}
#if MSGDEBUG1
opal_output(0, "send large, frag=%p, addr=%p\n", lfrag, sseg->ss_base.us_payload.raw);
#endif
/* save back pointer to fragment */
sseg->ss_parent_frag = frag;
/* If this is the first chunk of the frag, need to insert
* the PML header at the start. On subsequent chunks,
* skip the PML header
*/
if (lfrag->lsf_cur_offset == 0) {
/* assign a fragment ID */
do {
lfrag->lsf_frag_id = endpoint->endpoint_next_frag_id++;
} while (lfrag->lsf_frag_id == 0);
/* copy in the PML header */
memcpy(sseg->ss_base.us_payload.raw, lfrag->lsf_pml_header,
lfrag->lsf_base.sf_base.uf_src_seg[0].seg_len);
/* adjust data pointer and len to skip PML */
iov.iov_base = sseg->ss_base.us_payload.raw +
lfrag->lsf_base.sf_base.uf_src_seg[0].seg_len;
/* payload so far, modified later */
payload_len = lfrag->lsf_base.sf_base.uf_src_seg[0].seg_len;
} else {
iov.iov_base = sseg->ss_base.us_payload.raw;
payload_len = 0;
}
/* payload_len is how much payload we have already used up
* with header. max_data is the amount of further data we
* can put into this segment
*/
max_data = payload_len + lfrag->lsf_bytes_left;
if (max_data > module->max_chunk_payload) {
max_data = module->max_chunk_payload;
}
max_data -= payload_len;
/* fill in destination for pack */
iov.iov_len = max_data;
/* pack the next bit of data (large frags never have convertors) */
memcpy(iov.iov_base,
lfrag->lsf_base.sf_base.uf_src_seg[1].seg_addr.pval,
max_data);
lfrag->lsf_base.sf_base.uf_src_seg[1].seg_addr.lval += max_data;
/* get real payload length */
payload_len += max_data;
/* fill in BTL header with frag info */
chp = sseg->ss_base.us_btl_chunk_header;
chp->ch_frag_id = lfrag->lsf_frag_id;
chp->ch_frag_size = lfrag->lsf_base.sf_size;
chp->ch_frag_offset = lfrag->lsf_cur_offset;
/* set actual packet length for verbs */
sseg->ss_base.us_sg_entry[0].length =
sizeof(ompi_btl_usnic_btl_chunk_header_t) + payload_len;
/* payload length into the header*/
sseg->ss_base.us_btl_header->payload_len = payload_len;
/* do the send */
ompi_btl_usnic_endpoint_send_segment(module, sseg);
/* do fragment bookkeeping */
lfrag->lsf_cur_offset += payload_len;
lfrag->lsf_bytes_left -= payload_len;
#if MSGDEBUG1
opal_output(0, "payload_len = %d, bytes_left=%d\n",
payload_len, lfrag->lsf_bytes_left);
#endif
/* done with fragment? */
if (lfrag->lsf_bytes_left == 0) {
/* only callback now if this was not a PUT, otherwise
* we need to wait until last byte is ACKed
*/
if (frag->sf_base.uf_dst_seg[0].seg_addr.pval == NULL) {
#if MSGDEBUG1
opal_output(0, " calling back %p, len=%d\n",
frag->sf_base.uf_base.des_cbfunc,
frag->sf_size);
#endif
frag->sf_base.uf_base.des_cbfunc(&module->super,
frag->sf_endpoint, &frag->sf_base.uf_base,
OMPI_SUCCESS);
++module->pml_send_callbacks;
}
opal_list_remove_item(&endpoint->endpoint_frag_send_queue,
&frag->sf_base.uf_base.super.super);
}
}
/*
* Progress the send engine.
*/
void
ompi_btl_usnic_module_progress_sends(
ompi_btl_usnic_module_t *module)
{
ompi_btl_usnic_send_frag_t *frag;
ompi_btl_usnic_send_segment_t *sseg;
ompi_btl_usnic_endpoint_t *endpoint;
struct ompi_btl_usnic_channel_t *data_channel;
struct ompi_btl_usnic_channel_t *prio_channel;
static bool progressing = false;
/*
* Since we call this routine from within enqueue_frag, it is possible
* that we will be called re-entrantly from a callback routine.
* Do nothing if re-entrantly called.
*/
if (progressing) {
return;
}
progressing = true;
/*
* Post all the sends that we can
* resends 1st priority
* ACKs 2nd priority
* new sends 3rd
*/
data_channel = &module->mod_channels[USNIC_DATA_CHANNEL];
prio_channel = &module->mod_channels[USNIC_PRIORITY_CHANNEL];
/*
* Handle all the retransmits we can
*/
if (OPAL_UNLIKELY(!opal_list_is_empty(&module->pending_resend_segs))) {
usnic_do_resends(module);
}
/*
* Keep sending as long as there are WQEs and something to do
*/
while (((size_t) data_channel->sd_wqe > 0) &&
!opal_list_is_empty(&module->endpoints_with_sends)) {
ompi_btl_usnic_small_send_frag_t *sfrag;
size_t payload_len;
/*
* Grab the first endpoint with a pending send. Presence on this
* list means there is a fragment with data ready to go and
* the endpoint's send window is open, and the endpoint has send
* credits.
*/
endpoint = (ompi_btl_usnic_endpoint_t *)
opal_list_get_first(&module->endpoints_with_sends);
frag = (ompi_btl_usnic_send_frag_t *)
opal_list_get_first(&endpoint->endpoint_frag_send_queue);
/*
* small send? (fragment fits in one segment)
* Send ptr and length will be in uf_src_seg[0]
*/
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;
/* get payload len from segment */
payload_len = sfrag->ssf_base.sf_size;
sseg->ss_base.us_btl_header->payload_len = payload_len;
#if MSGDEBUG1
opal_output(0, "send small, ptr=%p, payload=%d, len=%d\n",
sseg->ss_base.us_sg_entry[0].addr, payload_len,
sseg->ss_base.us_sg_entry[0].length);
#endif
/* post the send */
ompi_btl_usnic_endpoint_send_segment(module, sseg);
/* don't do callback yet if this is a put */
if (frag->sf_base.uf_dst_seg[0].seg_addr.pval == NULL) {
#if MSGDEBUG1
opal_output(0, " calling back %p, len=%d\n",
frag->sf_base.uf_base.des_cbfunc,
frag->sf_base.uf_src_seg[0].seg_len);
#endif
/* we have copied the data, proceed with callback */
/* could be done in usnic_send? XXX */
frag->sf_base.uf_base.des_cbfunc(&module->super,
frag->sf_endpoint, &frag->sf_base.uf_base,
OMPI_SUCCESS);
++module->pml_send_callbacks;
}
/* remove frag from sending list */
opal_list_remove_item(&endpoint->endpoint_frag_send_queue,
&sfrag->ssf_base.sf_base.uf_base.super.super);
/* Large sends... */
} else {
usnic_handle_large_send(module, endpoint, frag);
}
/* If no more sends or endpoint send window is closed,
* or no more send credits, remove from send list
*/
if (opal_list_is_empty(&endpoint->endpoint_frag_send_queue) ||
endpoint->endpoint_send_credits <= 0 ||
!WINDOW_OPEN(endpoint)) {
opal_list_remove_item(&module->endpoints_with_sends,
&endpoint->super);
endpoint->endpoint_ready_to_send = false;
}
}
/*
* Handle any ACKs that need to be sent
*/
endpoint = ompi_btl_usnic_get_first_endpoint_needing_ack(module);
while (((size_t)prio_channel->sd_wqe) > 0 && endpoint != NULL) {
ompi_btl_usnic_endpoint_t *next_endpoint;
/* get next in list */
next_endpoint = ompi_btl_usnic_get_next_endpoint_needing_ack(endpoint);
/* Is it time to send ACK? */
if (endpoint->endpoint_acktime == 0 ||
endpoint->endpoint_acktime <= get_nsec()) {
ompi_btl_usnic_ack_send(module, endpoint);
ompi_btl_usnic_remove_from_endpoints_needing_ack(endpoint);
}
endpoint = next_endpoint;
}
progressing = false;
}
/*
* Initiate a send.
*/
static int usnic_send(struct mca_btl_base_module_t* base_module,
struct mca_btl_base_endpoint_t* base_endpoint,
struct mca_btl_base_descriptor_t* descriptor,
mca_btl_base_tag_t tag)
{
int rc;
ompi_btl_usnic_send_frag_t *frag;
ompi_btl_usnic_small_send_frag_t *sfrag;
ompi_btl_usnic_endpoint_t *endpoint;
ompi_btl_usnic_module_t *module;
ompi_btl_usnic_send_segment_t *sseg;
endpoint = (ompi_btl_usnic_endpoint_t *)base_endpoint;
module = (ompi_btl_usnic_module_t *)base_module;
frag = (ompi_btl_usnic_send_frag_t*) descriptor;
#if MSGDEBUG2
opal_output(0, "usnic_send: frag=%p, endpoint=%p, tag=%d, sendreq=%p\n",
(void *)frag, (void *)endpoint,
tag, (void *)descriptor->des_cbdata);
opal_output(0, " data = %p\n", descriptor->des_src[0].seg_addr.pval);
#endif
assert(frag->sf_endpoint == endpoint);
frag->sf_base.uf_dst_seg[0].seg_addr.pval = NULL; /* not a PUT */
/* JMS From Dec OMPI meeting....
if PML doesn't set SEND_ALWAYS_CALLBACK, then we can return 1
here to say "the data is gone, PML can complete the request".
And then we don't need to do the PML callback (!). WE DON'T
NEED TO SET ALWAYS_CALLBACK! */
/*
* If this fragment is small enough to inline,
* and the data is contiguous (sf_convertor == NULL)
* and we have enough send WQEs,
* then inline and fastpath it
*/
if (frag->sf_ack_bytes_left < module->max_tiny_payload &&
frag->sf_convertor == NULL &&
WINDOW_OPEN(endpoint) &&
(module->mod_channels[USNIC_PRIORITY_CHANNEL].sd_wqe >=
module->mod_channels[USNIC_PRIORITY_CHANNEL].fastsend_wqe_thresh)) {
size_t payload_len;
sfrag = (ompi_btl_usnic_small_send_frag_t *)frag;
sseg = &sfrag->ssf_segment;
payload_len = frag->sf_ack_bytes_left;
sseg->ss_base.us_btl_header->payload_len = payload_len;
/* fix up verbs SG entries */
sseg->ss_base.us_sg_entry[0].length =
sizeof(ompi_btl_usnic_btl_header_t) +
frag->sf_base.uf_src_seg[0].seg_len;
if (frag->sf_base.uf_base.des_src_cnt > 1) {
sseg->ss_send_desc.num_sge = 2;
sseg->ss_base.us_sg_entry[1].addr =
frag->sf_base.uf_src_seg[1].seg_addr.lval,
sseg->ss_base.us_sg_entry[1].length =
frag->sf_base.uf_src_seg[1].seg_len;
} else {
sseg->ss_send_desc.num_sge = 1;
}
sseg->ss_send_desc.send_flags |= IBV_SEND_INLINE;
sseg->ss_channel = USNIC_PRIORITY_CHANNEL;
#if MSGDEBUG2
opal_output(0, " inline frag %d segs %p(%d) + %p(%d)\n",
(int)frag->sf_base.uf_base.des_src_cnt,
frag->sf_base.uf_src_seg[0].seg_addr.pval,
(int)frag->sf_base.uf_src_seg[0].seg_len,
frag->sf_base.uf_src_seg[1].seg_addr.pval,
(int)frag->sf_base.uf_src_seg[1].seg_len);
opal_output(0, " inline seg %d segs %p(%d) + %p(%d)\n",
sseg->ss_send_desc.num_sge,
(void *)sseg->ss_send_desc.sg_list[0].addr,
sseg->ss_send_desc.sg_list[0].length,
(void *)sseg->ss_send_desc.sg_list[1].addr,
sseg->ss_send_desc.sg_list[1].length);
#endif
/* post the segment now */
ompi_btl_usnic_endpoint_send_segment(module, sseg);
/* make a copy of the data for retrans */
if (frag->sf_base.uf_base.des_src_cnt > 1) {
memcpy(((char *)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;
/* set up VERBS SG list */
sseg->ss_send_desc.num_sge = 1;
sseg->ss_base.us_sg_entry[0].length =
sizeof(ompi_btl_usnic_btl_header_t) + frag->sf_size;
}
/* If requested, callback now, else just return 1 to show completion */
if (descriptor->des_flags & MCA_BTL_DES_SEND_ALWAYS_CALLBACK) {
frag->sf_base.uf_base.des_cbfunc(&module->super,
frag->sf_endpoint, &frag->sf_base.uf_base,
OMPI_SUCCESS);
rc = 0;
} else {
rc = 1;
}
++module->pml_module_sends;
++module->pml_send_callbacks; /* returning "1" is an implicit CB */
return rc;
} else {
/*
* We move this off to another function because having it inside
* this function seems to add a little latency, likely due to inlines
* making the function too big. In fact, the routine had to go to
* another file entirely, else the compiler tried to be helpful
* and inline all by itself.
*/
return ompi_btl_usnic_send_slower(module, endpoint, frag, tag);
}
}
#if 0
/*
* Initiate an immediate send
*/
static int usnic_sendi(struct mca_btl_base_module_t* btl,
struct mca_btl_base_endpoint_t* endpoint,
struct opal_convertor_t* convertor,
void* header,
size_t header_size,
size_t payload_size,
uint8_t order,
uint32_t flags,
mca_btl_base_tag_t tag,
mca_btl_base_descriptor_t** descriptor)
{
/* JMS write me */
return OMPI_ERROR;
}
#endif
/*
* RDMA Memory Pool (de)register callbacks
*/
static int usnic_reg_mr(void* reg_data, void* base, size_t size,
mca_mpool_base_registration_t* reg)
{
ompi_btl_usnic_module_t* mod = (ompi_btl_usnic_module_t*)reg_data;
ompi_btl_usnic_reg_t* ud_reg = (ompi_btl_usnic_reg_t*)reg;
ud_reg->mr = ibv_reg_mr(mod->pd, base, size, IBV_ACCESS_LOCAL_WRITE |
IBV_ACCESS_REMOTE_WRITE | IBV_ACCESS_REMOTE_READ);
if (NULL == ud_reg->mr) {
return OMPI_ERR_OUT_OF_RESOURCE;
}
return OMPI_SUCCESS;
}
static int usnic_dereg_mr(void* reg_data,
mca_mpool_base_registration_t* reg)
{
ompi_btl_usnic_reg_t* ud_reg = (ompi_btl_usnic_reg_t*)reg;
if (ud_reg->mr != NULL) {
if (ibv_dereg_mr(ud_reg->mr)) {
opal_output(0, "%s: error unpinning UD memory: %s\n",
__func__, strerror(errno));
return OMPI_ERROR;
}
}
ud_reg->mr = NULL;
return OMPI_SUCCESS;
}
/*
* Called back by libevent if an async event occurs on the device
*/
static void module_async_event_callback(int fd, short flags, void *arg)
{
bool got_event = false;
bool fatal = false;
ompi_btl_usnic_module_t *module = (ompi_btl_usnic_module_t*) arg;
struct ibv_async_event event;
/* Get the async event */
if (0 != ibv_get_async_event(module->device_context, &event)) {
/* This shouldn't happen. If it does, treat this as a fatal
error. */
fatal = true;
} else {
got_event = true;
}
/* Process the event */
if (got_event) {
switch (event.event_type) {
/* For the moment, these are the only cases usnic_verbs.ko
will report to us. However, they're only listed here
for completeness. We currently abort if any async
event occurs. */
case IBV_EVENT_QP_FATAL:
case IBV_EVENT_PORT_ERR:
case IBV_EVENT_PORT_ACTIVE:
#if BTL_USNIC_HAVE_IBV_EVENT_GID_CHANGE
case IBV_EVENT_GID_CHANGE:
#endif
default:
opal_show_help("help-mpi-btl-usnic.txt", "async event",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
ibv_event_type_str(event.event_type),
event.event_type);
module->pml_error_callback(&module->super,
MCA_BTL_ERROR_FLAGS_FATAL,
ompi_proc_local(), "usnic");
/* The PML error callback will likely not return (i.e., it
will likely kill the job). But in case someone
implements a non-fatal PML error callback someday, do
reasonable things just in case it does actually
return. */
fatal = true;
break;
}
/* Ack the event back to verbs */
ibv_ack_async_event(&event);
}
/* If this is fatal, invoke the upper layer error handler to abort
the job */
if (fatal) {
ompi_btl_usnic_exit();
/* Does not return */
}
}
/*
* Create a single UD queue pair.
*/
static int
init_qp(
ompi_btl_usnic_module_t* module,
struct ompi_btl_usnic_channel_t *channel)
{
struct ibv_qp_attr qp_attr;
struct ibv_qp_init_attr qp_init_attr;
/* memset to both silence valgrind warnings (since the attr struct
ends up getting written down an fd to the kernel) and actually
zero out all the fields that we don't care about / want to be
logically false. */
memset(&qp_init_attr, 0, sizeof(qp_init_attr));
qp_init_attr.send_cq = channel->cq;
qp_init_attr.recv_cq = channel->cq;
qp_init_attr.cap.max_send_wr = channel->chan_sd_num;
qp_init_attr.cap.max_recv_wr = channel->chan_rd_num;
qp_init_attr.cap.max_send_sge = 1;
qp_init_attr.cap.max_recv_sge = 1;
qp_init_attr.qp_type = IBV_QPT_UD;
/* We did math up in component_init() to know that there should be
enough QPs available. So if create_qp fails, then either the
memlock limits are too low, or something other than this MPI
job is consuming QPs. */
channel->qp = ibv_create_qp(module->pd, &qp_init_attr);
if (NULL == channel->qp) {
opal_show_help("help-mpi-btl-usnic.txt", "create ibv resource failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
"ibv_create_qp()", __FILE__, __LINE__,
"Failed to create a usNIC queue pair");
return OMPI_ERR_OUT_OF_RESOURCE;
}
/* memset to both silence valgrind warnings (since the attr struct
ends up getting written down an fd to the kernel) and actually
zero out all the fields that we don't care about / want to be
logically false. */
memset(&qp_attr, 0, sizeof(qp_attr));
qp_attr.qp_state = IBV_QPS_INIT;
qp_attr.port_num = module->port_num;
if (ibv_modify_qp(channel->qp, &qp_attr,
IBV_QP_STATE | IBV_QP_PORT)) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"ibv_modify_qp()", __FILE__, __LINE__,
"Failed to modify an existing queue pair");
return OMPI_ERR_TEMP_OUT_OF_RESOURCE;
}
/* Find the max inline size */
memset(&qp_attr, 0, sizeof(qp_attr));
memset(&qp_init_attr, 0, sizeof(qp_init_attr));
if (ibv_query_qp(channel->qp, &qp_attr, IBV_QP_CAP,
&qp_init_attr) != 0) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"ibv_query_qp()", __FILE__, __LINE__,
"Failed to query an existing queue pair");
return OMPI_ERR_TEMP_OUT_OF_RESOURCE;
}
/* only grab this once */
if (channel == &module->mod_channels[USNIC_DATA_CHANNEL]) {
module->qp_max_inline = qp_attr.cap.max_inline_data;
}
return OMPI_SUCCESS;
}
static int move_qp_to_rtr(ompi_btl_usnic_module_t *module,
struct ompi_btl_usnic_channel_t *channel)
{
struct ibv_qp_attr qp_attr;
memset(&qp_attr, 0, sizeof(qp_attr));
qp_attr.qp_state = IBV_QPS_RTR;
if (ibv_modify_qp(channel->qp, &qp_attr, IBV_QP_STATE)) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"ibv_modify_qp", __FILE__, __LINE__,
"Failed to move QP to RTR state");
return OMPI_ERROR;
}
return OMPI_SUCCESS;
}
static int move_qp_to_rts(ompi_btl_usnic_module_t *module,
struct ompi_btl_usnic_channel_t *channel)
{
struct ibv_qp_attr qp_attr;
memset(&qp_attr, 0, sizeof(qp_attr));
qp_attr.qp_state = IBV_QPS_RTS;
if (ibv_modify_qp(channel->qp, &qp_attr, IBV_QP_STATE)) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"ibv_modify_qp", __FILE__, __LINE__,
"Failed to move QP to RTS state");
return OMPI_ERROR;
}
return OMPI_SUCCESS;
}
/*
* finalize channel - release all associated resources
*/
static void
ompi_btl_usnic_channel_finalize(
ompi_btl_usnic_module_t *module,
struct ompi_btl_usnic_channel_t *channel)
{
/* gets set right after constructor called, lets us know recv_segs
* have been constructed
*/
if (channel->recv_segs.ctx == module) {
OBJ_DESTRUCT(&channel->recv_segs);
}
if (NULL != channel->qp) {
ibv_destroy_qp(channel->qp);
channel->qp = NULL;
}
/* destroy CQ if created */
if (NULL != channel->cq) {
ibv_destroy_cq(channel->cq);
channel->cq = NULL;
}
}
/*
* Initialize a channel
*/
static int
ompi_btl_usnic_channel_init(
ompi_btl_usnic_module_t *module,
struct ompi_btl_usnic_channel_t *channel,
int mtu,
int rd_num,
int sd_num)
{
struct ibv_context *ctx;
ompi_btl_usnic_recv_segment_t *rseg;
ompi_free_list_item_t* item;
struct ibv_recv_wr* bad_wr;
int i;
int rc;
ctx = module->device_context;
channel->chan_mtu = mtu;
channel->chan_rd_num = rd_num;
channel->chan_sd_num = sd_num;
channel->sd_wqe = sd_num;
channel->fastsend_wqe_thresh = sd_num - 10;
/* We did math up in component_init() to know that there should be
enough CQs available. So if create_cq fails, then either the
memlock limits are too low, or something other than this MPI
job is consuming CQs. */
channel->cq = ibv_create_cq(ctx, module->cq_num, NULL, NULL, 0);
if (NULL == channel->cq) {
opal_show_help("help-mpi-btl-usnic.txt", "create ibv resource failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
"ibv_create_cq()", __FILE__, __LINE__,
"Failed to create a usNIC completion queue");
goto error;
}
/* Set up the QP for this channel */
rc = init_qp(module, channel);
if (OMPI_SUCCESS != rc) {
goto error;
}
/* Initialize pool of receive segments */
OBJ_CONSTRUCT(&channel->recv_segs, ompi_free_list_t);
channel->recv_segs.ctx = module;
rc = ompi_free_list_init_new(&channel->recv_segs,
sizeof(ompi_btl_usnic_recv_segment_t),
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_recv_segment_t),
mtu,
opal_cache_line_size,
rd_num,
rd_num,
rd_num,
module->super.btl_mpool);
if (OMPI_SUCCESS != rc) {
goto error;
}
/* Post receive descriptors */
for (i = 0; i < rd_num; i++) {
OMPI_FREE_LIST_GET_MT(&channel->recv_segs, item);
assert(NULL != item);
rseg = (ompi_btl_usnic_recv_segment_t*)item;
if (NULL == rseg) {
opal_show_help("help-mpi-btl-usnic.txt",
"internal error during init",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"get freelist buffer()", __FILE__, __LINE__,
"Failed to get receive buffer from freelist");
abort(); /* this is impossible */
}
/* cannot find length from constructor, set it now */
rseg->rs_base.us_sg_entry[0].length = mtu;
rseg->rs_recv_desc.next = NULL;
if (ibv_post_recv(channel->qp, &rseg->rs_recv_desc, &bad_wr)) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"ibv_post_recv", __FILE__, __LINE__,
"Failed to post receive buffer");
goto error;
}
}
if (OMPI_SUCCESS != move_qp_to_rtr(module, channel)) {
goto error;
}
if (OMPI_SUCCESS != move_qp_to_rts(module, channel)) {
goto error;
}
return OMPI_SUCCESS;
error:
ompi_btl_usnic_channel_finalize(module, channel);
return OMPI_ERROR;
}
/*
* generate initial send sequence number
*/
static ompi_btl_usnic_seq_t
get_initial_seq_no(void)
{
ompi_btl_usnic_seq_t isn;
/* only utilize the bottom 62 bits to avoid hitting seq # overflow */
isn = (((uint64_t)random() & ((1LL<<30)-1)) << 32) |
((uint64_t)random() & ((1LL<<32)-1));
isn += 2; /* guarantee > 1 */
return isn;
}
/*
* Initialize the btl module by allocating a protection domain,
* memory pool, priority and data channels, and free lists
*/
int ompi_btl_usnic_module_init(ompi_btl_usnic_module_t *module)
{
int rc;
int i;
struct mca_mpool_base_resources_t mpool_resources;
struct ibv_context *ctx = module->device_context;
uint32_t ack_segment_len;
#if OPAL_HAVE_HWLOC
/* If this process is bound to a single NUMA locality, calculate
its NUMA distance from this usNIC device */
if (mca_btl_usnic_component.want_numa_device_assignment) {
ompi_btl_usnic_hwloc_distance(module);
} else {
opal_output_verbose(5, USNIC_OUT,
"btl:usnic: not sorting devices by NUMA distance (MCA btl_usnic_want_numa_device_assignment)");
}
#else
opal_output_verbose(5, USNIC_OUT,
"btl:usnic: not sorting devices by NUMA distance (topology support not included)");
#endif
/* Setup the pointer array for the procs that will be used by this
module */
OBJ_CONSTRUCT(&module->all_procs, opal_pointer_array_t);
opal_pointer_array_init(&module->all_procs, ompi_process_info.num_procs,
INT_MAX, 32);
/* Get a PD */
module->pd = ibv_alloc_pd(ctx);
if (NULL == module->pd) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"ibv_alloc_pd()", __FILE__, __LINE__,
"Failed to create a PD; is the usnic_verbs Linux kernel module loaded?");
return OMPI_ERROR;
}
/* Setup the mpool */
mpool_resources.reg_data = (void*)module;
mpool_resources.sizeof_reg = sizeof(ompi_btl_usnic_reg_t);
mpool_resources.register_mem = usnic_reg_mr;
mpool_resources.deregister_mem = usnic_dereg_mr;
asprintf(&mpool_resources.pool_name, "usnic.%" PRIu64,
module->local_addr.gid.global.interface_id);
module->super.btl_mpool =
mca_mpool_base_module_create(mca_btl_usnic_component.usnic_mpool_name,
&module->super, &mpool_resources);
if (NULL == module->super.btl_mpool) {
opal_show_help("help-mpi-btl-usnic.txt", "ibv API failed",
true,
ompi_process_info.nodename,
ibv_get_device_name(module->device),
module->port_num,
"create mpool", __FILE__, __LINE__,
"Failed to allocate registered memory; check Linux memlock limits");
goto dealloc_pd;
}
/* initialize data and priority channels */
rc = ompi_btl_usnic_channel_init(module,
&module->mod_channels[USNIC_PRIORITY_CHANNEL],
module->tiny_mtu, module->prio_rd_num, module->prio_sd_num);
if (rc != OMPI_SUCCESS) {
goto chan_destroy;
}
rc = ompi_btl_usnic_channel_init(module,
&module->mod_channels[USNIC_DATA_CHANNEL],
module->if_mtu, module->rd_num, module->sd_num);
if (rc != OMPI_SUCCESS) {
goto chan_destroy;
}
module->local_addr.isn = get_initial_seq_no();
/* Place QP number in our local address information */
module->local_addr.qp_num[USNIC_PRIORITY_CHANNEL] =
module->mod_channels[USNIC_PRIORITY_CHANNEL].qp->qp_num;
module->local_addr.qp_num[USNIC_DATA_CHANNEL] =
module->mod_channels[USNIC_DATA_CHANNEL].qp->qp_num;
/* Get the fd to receive events on this device */
opal_event_set(opal_event_base, &(module->device_async_event),
module->device_context->async_fd,
OPAL_EV_READ | OPAL_EV_PERSIST,
module_async_event_callback, module);
opal_event_add(&(module->device_async_event), NULL);
module->device_async_event_active = true;
/* No more errors anticipated - initialize everything else */
/* list of all endpoints */
OBJ_CONSTRUCT(&(module->all_endpoints), opal_list_t);
/* Pending send segs list */
OBJ_CONSTRUCT(&module->pending_resend_segs, opal_list_t);
OBJ_CONSTRUCT(&module->endpoints_that_need_acks, opal_list_t);
/* list of endpoints that are ready to send */
OBJ_CONSTRUCT(&module->endpoints_with_sends, opal_list_t);
/* Send frags freelists */
module->small_send_frags.ctx = module;
OBJ_CONSTRUCT(&module->small_send_frags, ompi_free_list_t);
rc = ompi_free_list_init_new(&module->small_send_frags,
sizeof(ompi_btl_usnic_small_send_frag_t),
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_small_send_frag_t),
module->if_mtu,
opal_cache_line_size,
module->sd_num * 4,
-1,
module->sd_num / 2,
module->super.btl_mpool);
assert(OMPI_SUCCESS == rc);
module->large_send_frags.ctx = module;
OBJ_CONSTRUCT(&module->large_send_frags, ompi_free_list_t);
rc = ompi_free_list_init_new(&module->large_send_frags,
sizeof(ompi_btl_usnic_large_send_frag_t),
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_large_send_frag_t),
0, /* payload size */
0, /* payload align */
module->sd_num / 8,
-1,
module->sd_num / 8,
NULL);
assert(OMPI_SUCCESS == rc);
module->put_dest_frags.ctx = module;
OBJ_CONSTRUCT(&module->put_dest_frags, ompi_free_list_t);
rc = ompi_free_list_init_new(&module->put_dest_frags,
sizeof(ompi_btl_usnic_put_dest_frag_t),
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_put_dest_frag_t),
0, /* payload size */
0, /* payload align */
module->sd_num / 8,
-1,
module->sd_num / 8,
NULL);
assert(OMPI_SUCCESS == rc);
/* list of segments to use for sending */
module->chunk_segs.ctx = module;
OBJ_CONSTRUCT(&module->chunk_segs, ompi_free_list_t);
rc = ompi_free_list_init_new(&module->chunk_segs,
sizeof(ompi_btl_usnic_chunk_segment_t),
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_chunk_segment_t),
module->if_mtu,
opal_cache_line_size,
module->sd_num * 4,
-1,
module->sd_num / 2,
module->super.btl_mpool);
assert(OMPI_SUCCESS == rc);
/* ACK segments freelist */
module->ack_segs.ctx = module;
ack_segment_len = sizeof(ompi_btl_usnic_btl_header_t);
OBJ_CONSTRUCT(&module->ack_segs, ompi_free_list_t);
rc = ompi_free_list_init_new(&module->ack_segs,
sizeof(ompi_btl_usnic_ack_segment_t),
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_ack_segment_t),
ack_segment_len,
opal_cache_line_size,
module->sd_num * 4,
-1,
module->sd_num / 2,
module->super.btl_mpool);
assert(OMPI_SUCCESS == rc);
/*
* Initialize pools of large recv buffers
*/
module->first_pool = 16;
module->last_pool = fls(module->super.btl_eager_limit-1);
module->module_recv_buffers = calloc(module->last_pool+1,
sizeof(ompi_free_list_t));
assert(module->module_recv_buffers != NULL);
for (i=module->first_pool; i<=module->last_pool; ++i) {
module->module_recv_buffers[i].ctx = module;
OBJ_CONSTRUCT(&module->module_recv_buffers[i], ompi_free_list_t);
rc = ompi_free_list_init_new(&module->module_recv_buffers[i],
1 << i,
opal_cache_line_size,
OBJ_CLASS(ompi_btl_usnic_large_send_frag_t),
0, /* payload size */
0, /* payload align */
8,
128,
8,
NULL);
assert(OMPI_SUCCESS == rc);
}
if (mca_btl_usnic_component.stats_enabled) {
usnic_stats_reset(module);
module->final_stats = false;
module->stats_timeout.tv_sec = mca_btl_usnic_component.stats_frequency;
module->stats_timeout.tv_usec = 0;
opal_event_set(opal_event_base, &(module->stats_timer_event),
-1, EV_TIMEOUT | EV_PERSIST, usnic_stats_callback, module);
opal_event_add(&(module->stats_timer_event),
&(module->stats_timeout));
}
return OMPI_SUCCESS;
chan_destroy:
for (i=0; i<USNIC_NUM_CHANNELS; ++i) {
ompi_btl_usnic_channel_finalize(module, &module->mod_channels[i]);
}
mca_mpool_base_module_destroy(module->super.btl_mpool);
dealloc_pd:
ibv_dealloc_pd(module->pd);
return OMPI_ERROR;
}
static int usnic_ft_event(int state)
{
return OMPI_SUCCESS;
}
ompi_btl_usnic_module_t ompi_btl_usnic_module_template = {
{
&mca_btl_usnic_component.super,
0, /* eager_limit */
0, /* min_send_size */
0, /* max_send_size */
0, /* rdma_pipeline_send_length */
0, /* rdma_pipeline_frag_size */
0, /* min_rdma_pipeline_size */
MCA_BTL_EXCLUSIVITY_DEFAULT,
0, /* latency */
0, /* bandwidth */
MCA_BTL_FLAGS_SEND |
MCA_BTL_FLAGS_PUT |
MCA_BTL_FLAGS_SEND_INPLACE,
sizeof(mca_btl_base_segment_t), /* seg size */
usnic_add_procs,
usnic_del_procs,
NULL, /*ompi_btl_usnic_register*/
usnic_finalize,
usnic_alloc,
usnic_free,
usnic_prepare_src,
usnic_prepare_dst,
usnic_send,
NULL, /* ompi_btl_usnic_sendi */
usnic_put,
NULL, /*ompi_btl_usnic_get */
mca_btl_base_dump,
NULL, /* mpool -- to be filled in later */
usnic_register_pml_err_cb,
usnic_ft_event
}
};