#include <vppinfra/ring.h>
#include <vnet/ethernet/ethernet.h>
#include <vnet/devices/devices.h>
-
#include <rdma/rdma.h>
+#ifndef MLX5_ETH_L2_INLINE_HEADER_SIZE
+#define MLX5_ETH_L2_INLINE_HEADER_SIZE 18
+#endif
+
+#define RDMA_TX_RETRIES 5
+
+#define RDMA_TXQ_DV_DSEG_SZ(txq) (RDMA_MLX5_WQE_DS * RDMA_TXQ_DV_SQ_SZ(txq))
+#define RDMA_TXQ_DV_DSEG2WQE(d) (((d) + RDMA_MLX5_WQE_DS - 1) / RDMA_MLX5_WQE_DS)
+
+/*
+ * MLX5 direct verbs tx/free functions
+ */
+
+static_always_inline void
+rdma_device_output_free_mlx5 (vlib_main_t * vm,
+ const vlib_node_runtime_t * node,
+ rdma_txq_t * txq)
+{
+ u16 idx = txq->dv_cq_idx;
+ u32 cq_mask = pow2_mask (txq->dv_cq_log2sz);
+ u32 sq_mask = pow2_mask (txq->dv_sq_log2sz);
+ u32 mask = pow2_mask (txq->bufs_log2sz);
+ u32 buf_sz = RDMA_TXQ_BUF_SZ (txq);
+ u32 log2_cq_sz = txq->dv_cq_log2sz;
+ struct mlx5_cqe64 *cqes = txq->dv_cq_cqes, *cur = cqes + (idx & cq_mask);
+ u8 op_own, saved;
+ const rdma_mlx5_wqe_t *wqe;
+
+ for (;;)
+ {
+ op_own = *(volatile u8 *) &cur->op_own;
+ if (((idx >> log2_cq_sz) & MLX5_CQE_OWNER_MASK) !=
+ (op_own & MLX5_CQE_OWNER_MASK) || (op_own >> 4) == MLX5_CQE_INVALID)
+ break;
+ if (PREDICT_FALSE ((op_own >> 4)) != MLX5_CQE_REQ)
+ vlib_error_count (vm, node->node_index, RDMA_TX_ERROR_COMPLETION, 1);
+ idx++;
+ cur = cqes + (idx & cq_mask);
+ }
+
+ if (idx == txq->dv_cq_idx)
+ return; /* nothing to do */
+
+ cur = cqes + ((idx - 1) & cq_mask);
+ saved = cur->op_own;
+ (void) saved;
+ cur->op_own = 0xf0;
+ txq->dv_cq_idx = idx;
+
+ /* retrieve original WQE and get new tail counter */
+ wqe = txq->dv_sq_wqes + (be16toh (cur->wqe_counter) & sq_mask);
+ if (PREDICT_FALSE (wqe->ctrl.imm == RDMA_TXQ_DV_INVALID_ID))
+ return; /* can happen if CQE reports error for an intermediate WQE */
+
+ ASSERT (RDMA_TXQ_USED_SZ (txq->head, wqe->ctrl.imm) <= buf_sz &&
+ RDMA_TXQ_USED_SZ (wqe->ctrl.imm, txq->tail) < buf_sz);
+
+ /* free sent buffers and update txq head */
+ vlib_buffer_free_from_ring (vm, txq->bufs, txq->head & mask, buf_sz,
+ RDMA_TXQ_USED_SZ (txq->head, wqe->ctrl.imm));
+ txq->head = wqe->ctrl.imm;
+
+ /* ring doorbell */
+ CLIB_MEMORY_STORE_BARRIER ();
+ txq->dv_cq_dbrec[0] = htobe32 (idx);
+}
+
+static_always_inline void
+rdma_device_output_tx_mlx5_doorbell (rdma_txq_t * txq, rdma_mlx5_wqe_t * last,
+ const u16 tail, u32 sq_mask)
+{
+ last->ctrl.imm = tail; /* register item to free */
+ last->ctrl.fm_ce_se = MLX5_WQE_CTRL_CQ_UPDATE; /* generate a CQE so we can free buffers */
+
+ ASSERT (tail != txq->tail &&
+ RDMA_TXQ_AVAIL_SZ (txq, txq->head, txq->tail) >=
+ RDMA_TXQ_USED_SZ (txq->tail, tail));
+
+ CLIB_MEMORY_STORE_BARRIER ();
+ txq->dv_sq_dbrec[MLX5_SND_DBR] = htobe32 (tail);
+ CLIB_COMPILER_BARRIER ();
+ txq->dv_sq_db[0] = *(u64 *) (txq->dv_sq_wqes + (txq->tail & sq_mask));
+ txq->tail = tail;
+}
+
+static_always_inline void
+rdma_mlx5_wqe_init (rdma_mlx5_wqe_t * wqe, const void *tmpl,
+ vlib_buffer_t * b, const u16 tail)
+{
+ u16 sz = b->current_length;
+ u16 inline_sz = clib_min (sz, MLX5_ETH_L2_INLINE_HEADER_SIZE);
+
+ clib_memcpy_fast (wqe, tmpl, RDMA_MLX5_WQE_SZ);
+
+ wqe->ctrl.opmod_idx_opcode |= ((u32) htobe16 (tail)) << 8;
+ /* speculatively copy at least MLX5_ETH_L2_INLINE_HEADER_SIZE (18-bytes) */
+ const void *cur = vlib_buffer_get_current (b);
+ clib_memcpy_fast (wqe->eseg.inline_hdr_start,
+ cur, MLX5_ETH_L2_INLINE_HEADER_SIZE);
+ wqe->eseg.inline_hdr_sz = htobe16 (inline_sz);
+ wqe->dseg.byte_count = htobe32 (sz - inline_sz);
+ wqe->dseg.addr = htobe64 (pointer_to_uword (cur) + inline_sz);
+}
+
+/*
+ * specific data path for chained buffers, supporting ring wrap-around
+ * contrary to the normal path - otherwise we may fail to enqueue chained
+ * buffers because we are close to the end of the ring while we still have
+ * plenty of descriptors available
+ */
+static_always_inline u32
+rdma_device_output_tx_mlx5_chained (vlib_main_t * vm,
+ const vlib_node_runtime_t * node,
+ const rdma_device_t * rd,
+ rdma_txq_t * txq, u32 n_left_from, u32 n,
+ u32 * bi, vlib_buffer_t ** b,
+ rdma_mlx5_wqe_t * wqe, u16 tail)
+{
+ rdma_mlx5_wqe_t *last = wqe;
+ u32 wqe_n = RDMA_TXQ_AVAIL_SZ (txq, txq->head, tail);
+ u32 sq_mask = pow2_mask (txq->dv_sq_log2sz);
+ u32 mask = pow2_mask (txq->bufs_log2sz);
+ u32 dseg_mask = RDMA_TXQ_DV_DSEG_SZ (txq) - 1;
+ const u32 lkey = wqe[0].dseg.lkey;
+
+ vlib_buffer_copy_indices (txq->bufs + (txq->tail & mask), bi,
+ n_left_from - n);
+
+ while (n >= 1 && wqe_n >= 1)
+ {
+ u32 *bufs = txq->bufs + (tail & mask);
+ rdma_mlx5_wqe_t *wqe = txq->dv_sq_wqes + (tail & sq_mask);
+
+ /* setup the head WQE */
+ rdma_mlx5_wqe_init (wqe, txq->dv_wqe_tmpl, b[0], tail);
+
+ bufs[0] = bi[0];
+
+ if (b[0]->flags & VLIB_BUFFER_NEXT_PRESENT)
+ {
+ /*
+ * max number of available dseg:
+ * - 4 dseg per WQEBB available
+ * - max 32 dseg per WQE (5-bits length field in WQE ctrl)
+ */
+#define RDMA_MLX5_WQE_DS_MAX (1 << 5)
+ const u32 dseg_max =
+ clib_min (RDMA_MLX5_WQE_DS * (wqe_n - 1), RDMA_MLX5_WQE_DS_MAX);
+ vlib_buffer_t *chained_b = b[0];
+ u32 chained_n = 0;
+
+ /* there are exactly 4 dseg per WQEBB and we rely on that */
+ STATIC_ASSERT (RDMA_MLX5_WQE_DS *
+ sizeof (struct mlx5_wqe_data_seg) ==
+ MLX5_SEND_WQE_BB, "wrong size");
+
+ /*
+ * iterate over fragments, supporting ring wrap-around contrary to
+ * the normal path - otherwise we may fail to enqueue chained
+ * buffers because we are close to the end of the ring while we
+ * still have plenty of descriptors available
+ */
+ while (chained_n < dseg_max
+ && chained_b->flags & VLIB_BUFFER_NEXT_PRESENT)
+ {
+ struct mlx5_wqe_data_seg *dseg = (void *) txq->dv_sq_wqes;
+ dseg += ((tail + 1) * RDMA_MLX5_WQE_DS + chained_n) & dseg_mask;
+ if (((clib_address_t) dseg & (MLX5_SEND_WQE_BB - 1)) == 0)
+ {
+ /*
+ * start of new WQEBB
+ * head/tail are shared between buffers and descriptor
+ * In order to maintain 1:1 correspondance between
+ * buffer index and descriptor index, we build
+ * 4-fragments chains and save the head
+ */
+ chained_b->flags &= ~(VLIB_BUFFER_NEXT_PRESENT |
+ VLIB_BUFFER_TOTAL_LENGTH_VALID);
+ u32 idx = tail + 1 + RDMA_TXQ_DV_DSEG2WQE (chained_n);
+ idx &= mask;
+ txq->bufs[idx] = chained_b->next_buffer;
+ }
+
+ chained_b = vlib_get_buffer (vm, chained_b->next_buffer);
+ dseg->byte_count = htobe32 (chained_b->current_length);
+ dseg->lkey = lkey;
+ dseg->addr = htobe64 (vlib_buffer_get_current_va (chained_b));
+
+ chained_n += 1;
+ }
+
+ if (chained_b->flags & VLIB_BUFFER_NEXT_PRESENT)
+ {
+ /*
+ * no descriptors left: drop the chain including 1st WQE
+ * skip the problematic packet and continue
+ */
+ vlib_buffer_free_from_ring (vm, txq->bufs, tail & mask,
+ RDMA_TXQ_BUF_SZ (txq), 1 +
+ RDMA_TXQ_DV_DSEG2WQE (chained_n));
+ vlib_error_count (vm, node->node_index,
+ dseg_max == chained_n ?
+ RDMA_TX_ERROR_SEGMENT_SIZE_EXCEEDED :
+ RDMA_TX_ERROR_NO_FREE_SLOTS, 1);
+
+ /* fixup tail to overwrite wqe head with next packet */
+ tail -= 1;
+ }
+ else
+ {
+ /* update WQE descriptor with new dseg number */
+ ((u8 *) & wqe[0].ctrl.qpn_ds)[3] = RDMA_MLX5_WQE_DS + chained_n;
+
+ tail += RDMA_TXQ_DV_DSEG2WQE (chained_n);
+ wqe_n -= RDMA_TXQ_DV_DSEG2WQE (chained_n);
+
+ last = wqe;
+ }
+ }
+ else
+ {
+ /* not chained */
+ last = wqe;
+ }
+
+ tail += 1;
+ bi += 1;
+ b += 1;
+ wqe_n -= 1;
+ n -= 1;
+ }
+
+ if (n == n_left_from)
+ return 0; /* we fail to enqueue even a single packet */
+
+ rdma_device_output_tx_mlx5_doorbell (txq, last, tail, sq_mask);
+ return n_left_from - n;
+}
+
+static_always_inline u32
+rdma_device_output_tx_mlx5 (vlib_main_t * vm,
+ const vlib_node_runtime_t * node,
+ const rdma_device_t * rd, rdma_txq_t * txq,
+ const u32 n_left_from, u32 * bi,
+ vlib_buffer_t ** b)
+{
+ u32 sq_mask = pow2_mask (txq->dv_sq_log2sz);
+ u32 mask = pow2_mask (txq->bufs_log2sz);
+ rdma_mlx5_wqe_t *wqe = txq->dv_sq_wqes + (txq->tail & sq_mask);
+ u32 n = n_left_from;
+ u16 tail = txq->tail;
+
+ ASSERT (RDMA_TXQ_BUF_SZ (txq) <= RDMA_TXQ_DV_SQ_SZ (txq));
+
+ while (n >= 4)
+ {
+ u32 flags = b[0]->flags | b[1]->flags | b[2]->flags | b[3]->flags;
+ if (PREDICT_FALSE (flags & VLIB_BUFFER_NEXT_PRESENT))
+ return rdma_device_output_tx_mlx5_chained (vm, node, rd, txq,
+ n_left_from, n, bi, b, wqe,
+ tail);
+
+ if (PREDICT_TRUE (n >= 8))
+ {
+ vlib_prefetch_buffer_header (b + 4, LOAD);
+ vlib_prefetch_buffer_header (b + 5, LOAD);
+ vlib_prefetch_buffer_header (b + 6, LOAD);
+ vlib_prefetch_buffer_header (b + 7, LOAD);
+ clib_prefetch_load (wqe + 4);
+ }
+
+ rdma_mlx5_wqe_init (wqe + 0, txq->dv_wqe_tmpl, b[0], tail + 0);
+ rdma_mlx5_wqe_init (wqe + 1, txq->dv_wqe_tmpl, b[1], tail + 1);
+ rdma_mlx5_wqe_init (wqe + 2, txq->dv_wqe_tmpl, b[2], tail + 2);
+ rdma_mlx5_wqe_init (wqe + 3, txq->dv_wqe_tmpl, b[3], tail + 3);
+
+ b += 4;
+ tail += 4;
+ wqe += 4;
+ n -= 4;
+ }
+
+ while (n >= 1)
+ {
+ if (PREDICT_FALSE (b[0]->flags & VLIB_BUFFER_NEXT_PRESENT))
+ return rdma_device_output_tx_mlx5_chained (vm, node, rd, txq,
+ n_left_from, n, bi, b, wqe,
+ tail);
+
+ rdma_mlx5_wqe_init (wqe, txq->dv_wqe_tmpl, b[0], tail);
+
+ b += 1;
+ tail += 1;
+ wqe += 1;
+ n -= 1;
+ }
+
+ vlib_buffer_copy_indices (txq->bufs + (txq->tail & mask), bi, n_left_from);
+
+ rdma_device_output_tx_mlx5_doorbell (txq, &wqe[-1], tail, sq_mask);
+ return n_left_from;
+}
+
+/*
+ * standard ibverb tx/free functions
+ */
+
static_always_inline void
-rdma_device_output_free (vlib_main_t * vm, rdma_txq_t * txq)
+rdma_device_output_free_ibverb (vlib_main_t * vm,
+ const vlib_node_runtime_t * node,
+ rdma_txq_t * txq)
{
struct ibv_wc wc[VLIB_FRAME_SIZE];
- u32 tail, slot;
+ u32 mask = pow2_mask (txq->bufs_log2sz);
+ u16 tail;
int n;
- n = ibv_poll_cq (txq->cq, VLIB_FRAME_SIZE, wc);
+ n = ibv_poll_cq (txq->ibv_cq, VLIB_FRAME_SIZE, wc);
if (n <= 0)
- return;
+ {
+ if (PREDICT_FALSE (n < 0))
+ vlib_error_count (vm, node->node_index, RDMA_TX_ERROR_COMPLETION, 1);
+ return;
+ }
+
+ while (PREDICT_FALSE (IBV_WC_SUCCESS != wc[n - 1].status))
+ {
+ vlib_error_count (vm, node->node_index, RDMA_TX_ERROR_COMPLETION, 1);
+ n--;
+ if (0 == n)
+ return;
+ }
tail = wc[n - 1].wr_id;
- slot = txq->head & (txq->size - 1);
- vlib_buffer_free_from_ring (vm, txq->bufs, slot, txq->size,
- tail - txq->head);
+ vlib_buffer_free_from_ring (vm, txq->bufs, txq->head & mask,
+ RDMA_TXQ_BUF_SZ (txq),
+ RDMA_TXQ_USED_SZ (txq->head, tail));
txq->head = tail;
}
static_always_inline u32
-rmda_device_output_tx (vlib_main_t * vm, const rdma_device_t * rd,
- rdma_txq_t * txq, u32 n_left_from, u32 * bi)
+rdma_device_output_tx_ibverb (vlib_main_t * vm,
+ const vlib_node_runtime_t * node,
+ const rdma_device_t * rd, rdma_txq_t * txq,
+ u32 n_left_from, u32 * bi, vlib_buffer_t ** b)
{
- vlib_buffer_t *bufs[VLIB_FRAME_SIZE], **b = bufs;
struct ibv_send_wr wr[VLIB_FRAME_SIZE], *w = wr;
struct ibv_sge sge[VLIB_FRAME_SIZE], *s = sge;
- u32 n, slot = txq->tail & (txq->size - 1);
- u32 *tx = &txq->bufs[slot];
-
- /* do not enqueue more packet than ring space */
- n_left_from = clib_min (n_left_from, txq->size - (txq->tail - txq->head));
- /* avoid wrap-around logic in core loop */
- n = n_left_from = clib_min (n_left_from, txq->size - slot);
+ u32 mask = txq->bufs_log2sz;
+ u32 n = n_left_from;
- /* if ring is full, do nothing */
- if (PREDICT_FALSE (0 == n_left_from))
- return 0;
-
- vlib_get_buffers (vm, bi, bufs, n_left_from);
memset (w, 0, n_left_from * sizeof (w[0]));
while (n >= 4)
CLIB_PREFETCH (&w[4 + 3], CLIB_CACHE_LINE_BYTES, STORE);
}
- vlib_buffer_copy_indices (tx, bi, 4);
-
s[0].addr = vlib_buffer_get_current_va (b[0]);
s[0].length = b[0]->current_length;
s[0].lkey = rd->lkey;
s += 4;
w += 4;
b += 4;
- bi += 4;
- tx += 4;
n -= 4;
}
while (n >= 1)
{
- vlib_buffer_copy_indices (tx, bi, 1);
-
s[0].addr = vlib_buffer_get_current_va (b[0]);
s[0].length = b[0]->current_length;
s[0].lkey = rd->lkey;
s += 1;
w += 1;
b += 1;
- bi += 1;
- tx += 1;
n -= 1;
}
- w[-1].wr_id = txq->tail + n_left_from; /* register item to free */
+ w[-1].wr_id = txq->tail; /* register item to free */
w[-1].next = 0; /* fix next pointer in WR linked-list */
w[-1].send_flags = IBV_SEND_SIGNALED; /* generate a CQE so we can free buffers */
w = wr;
- if (PREDICT_FALSE (0 != ibv_post_send (txq->qp, w, &w)))
- n_left_from = w - wr;
+ if (PREDICT_FALSE (0 != ibv_post_send (txq->ibv_qp, w, &w)))
+ {
+ vlib_error_count (vm, node->node_index, RDMA_TX_ERROR_SUBMISSION,
+ n_left_from - (w - wr));
+ n_left_from = w - wr;
+ }
+ vlib_buffer_copy_indices (txq->bufs + (txq->tail & mask), bi, n_left_from);
txq->tail += n_left_from;
return n_left_from;
}
-VNET_DEVICE_CLASS_TX_FN (rdma_device_class) (vlib_main_t * vm,
- vlib_node_runtime_t * node,
- vlib_frame_t * frame)
+/*
+ * common tx/free functions
+ */
+
+static_always_inline void
+rdma_device_output_free (vlib_main_t * vm, const vlib_node_runtime_t * node,
+ rdma_txq_t * txq, int is_mlx5dv)
+{
+ if (is_mlx5dv)
+ rdma_device_output_free_mlx5 (vm, node, txq);
+ else
+ rdma_device_output_free_ibverb (vm, node, txq);
+}
+
+static_always_inline u32
+rdma_device_output_tx_try (vlib_main_t * vm, const vlib_node_runtime_t * node,
+ const rdma_device_t * rd, rdma_txq_t * txq,
+ u32 n_left_from, u32 * bi, int is_mlx5dv)
+{
+ vlib_buffer_t *b[VLIB_FRAME_SIZE];
+ u32 mask = pow2_mask (txq->bufs_log2sz);
+
+ /* do not enqueue more packet than ring space */
+ n_left_from = clib_min (n_left_from, RDMA_TXQ_AVAIL_SZ (txq, txq->head,
+ txq->tail));
+ /* avoid wrap-around logic in core loop */
+ n_left_from = clib_min (n_left_from, RDMA_TXQ_BUF_SZ (txq) -
+ (txq->tail & mask));
+
+ /* if ring is full, do nothing */
+ if (PREDICT_FALSE (n_left_from == 0))
+ return 0;
+
+ vlib_get_buffers (vm, bi, b, n_left_from);
+
+ return is_mlx5dv ?
+ rdma_device_output_tx_mlx5 (vm, node, rd, txq, n_left_from, bi, b) :
+ rdma_device_output_tx_ibverb (vm, node, rd, txq, n_left_from, bi, b);
+}
+
+static_always_inline uword
+rdma_device_output_tx (vlib_main_t * vm, vlib_node_runtime_t * node,
+ vlib_frame_t * frame, rdma_device_t * rd,
+ int is_mlx5dv)
{
- rdma_main_t *rm = &rdma_main;
- vnet_interface_output_runtime_t *ord = (void *) node->runtime_data;
- rdma_device_t *rd = pool_elt_at_index (rm->devices, ord->dev_instance);
u32 thread_index = vm->thread_index;
rdma_txq_t *txq =
vec_elt_at_index (rd->txqs, thread_index % vec_len (rd->txqs));
u32 n_left_from;
int i;
- ASSERT (txq->size >= VLIB_FRAME_SIZE && is_pow2 (txq->size));
- ASSERT (txq->tail - txq->head <= txq->size);
+ ASSERT (RDMA_TXQ_BUF_SZ (txq) >= VLIB_FRAME_SIZE);
from = vlib_frame_vector_args (frame);
n_left_from = frame->n_vectors;
clib_spinlock_lock_if_init (&txq->lock);
- for (i = 0; i < 5 && n_left_from > 0; i++)
+ for (i = 0; i < RDMA_TX_RETRIES && n_left_from > 0; i++)
{
u32 n_enq;
- rdma_device_output_free (vm, txq);
- n_enq = rmda_device_output_tx (vm, rd, txq, n_left_from, from);
+ rdma_device_output_free (vm, node, txq, is_mlx5dv);
+ n_enq = rdma_device_output_tx_try (vm, node, rd, txq, n_left_from, from,
+ is_mlx5dv);
+
n_left_from -= n_enq;
from += n_enq;
}
return frame->n_vectors - n_left_from;
}
+VNET_DEVICE_CLASS_TX_FN (rdma_device_class) (vlib_main_t * vm,
+ vlib_node_runtime_t * node,
+ vlib_frame_t * frame)
+{
+ rdma_main_t *rm = &rdma_main;
+ vnet_interface_output_runtime_t *ord = (void *) node->runtime_data;
+ rdma_device_t *rd = pool_elt_at_index (rm->devices, ord->dev_instance);
+
+ if (PREDICT_TRUE (rd->flags & RDMA_DEVICE_F_MLX5DV))
+ return rdma_device_output_tx (vm, node, frame, rd, 1 /* is_mlx5dv */ );
+
+ return rdma_device_output_tx (vm, node, frame, rd, 0 /* is_mlx5dv */ );
+}
+
/*
* fd.io coding-style-patch-verification: ON
*
Code Review / vpp.git / blobdiff