xref: /freebsd/sys/dev/irdma/irdma_uk.c (revision d9a42747950146bf03cda7f6e25d219253f8a57a)
1 /*-
2  * SPDX-License-Identifier: GPL-2.0 or Linux-OpenIB
3  *
4  * Copyright (c) 2015 - 2022 Intel Corporation
5  *
6  * This software is available to you under a choice of one of two
7  * licenses.  You may choose to be licensed under the terms of the GNU
8  * General Public License (GPL) Version 2, available from the file
9  * COPYING in the main directory of this source tree, or the
10  * OpenFabrics.org BSD license below:
11  *
12  *   Redistribution and use in source and binary forms, with or
13  *   without modification, are permitted provided that the following
14  *   conditions are met:
15  *
16  *    - Redistributions of source code must retain the above
17  *	copyright notice, this list of conditions and the following
18  *	disclaimer.
19  *
20  *    - Redistributions in binary form must reproduce the above
21  *	copyright notice, this list of conditions and the following
22  *	disclaimer in the documentation and/or other materials
23  *	provided with the distribution.
24  *
25  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
26  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
27  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
28  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
29  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
30  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
31  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
32  * SOFTWARE.
33  */
34 /*$FreeBSD$*/
35 
36 #include "osdep.h"
37 #include "irdma_defs.h"
38 #include "irdma_user.h"
39 #include "irdma.h"
40 
41 /**
42  * irdma_set_fragment - set fragment in wqe
43  * @wqe: wqe for setting fragment
44  * @offset: offset value
45  * @sge: sge length and stag
46  * @valid: The wqe valid
47  */
48 static void
49 irdma_set_fragment(__le64 * wqe, u32 offset, struct irdma_sge *sge,
50 		   u8 valid)
51 {
52 	if (sge) {
53 		set_64bit_val(wqe, offset,
54 			      FIELD_PREP(IRDMAQPSQ_FRAG_TO, sge->tag_off));
55 		set_64bit_val(wqe, offset + IRDMA_BYTE_8,
56 			      FIELD_PREP(IRDMAQPSQ_VALID, valid) |
57 			      FIELD_PREP(IRDMAQPSQ_FRAG_LEN, sge->len) |
58 			      FIELD_PREP(IRDMAQPSQ_FRAG_STAG, sge->stag));
59 	} else {
60 		set_64bit_val(wqe, offset, 0);
61 		set_64bit_val(wqe, offset + IRDMA_BYTE_8,
62 			      FIELD_PREP(IRDMAQPSQ_VALID, valid));
63 	}
64 }
65 
66 /**
67  * irdma_set_fragment_gen_1 - set fragment in wqe
68  * @wqe: wqe for setting fragment
69  * @offset: offset value
70  * @sge: sge length and stag
71  * @valid: wqe valid flag
72  */
73 static void
74 irdma_set_fragment_gen_1(__le64 * wqe, u32 offset,
75 			 struct irdma_sge *sge, u8 valid)
76 {
77 	if (sge) {
78 		set_64bit_val(wqe, offset,
79 			      FIELD_PREP(IRDMAQPSQ_FRAG_TO, sge->tag_off));
80 		set_64bit_val(wqe, offset + IRDMA_BYTE_8,
81 			      FIELD_PREP(IRDMAQPSQ_GEN1_FRAG_LEN, sge->len) |
82 			      FIELD_PREP(IRDMAQPSQ_GEN1_FRAG_STAG, sge->stag));
83 	} else {
84 		set_64bit_val(wqe, offset, 0);
85 		set_64bit_val(wqe, offset + IRDMA_BYTE_8, 0);
86 	}
87 }
88 
89 /**
90  * irdma_nop_hdr - Format header section of noop WQE
91  * @qp: hw qp ptr
92  */
93 static inline u64 irdma_nop_hdr(struct irdma_qp_uk *qp){
94 	return FIELD_PREP(IRDMAQPSQ_OPCODE, IRDMAQP_OP_NOP) |
95 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, false) |
96 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
97 }
98 
99 /**
100  * irdma_nop_1 - insert a NOP wqe
101  * @qp: hw qp ptr
102  */
103 static int
104 irdma_nop_1(struct irdma_qp_uk *qp)
105 {
106 	__le64 *wqe;
107 	u32 wqe_idx;
108 
109 	if (!qp->sq_ring.head)
110 		return -EINVAL;
111 
112 	wqe_idx = IRDMA_RING_CURRENT_HEAD(qp->sq_ring);
113 	wqe = qp->sq_base[wqe_idx].elem;
114 
115 	qp->sq_wrtrk_array[wqe_idx].quanta = IRDMA_QP_WQE_MIN_QUANTA;
116 
117 	set_64bit_val(wqe, IRDMA_BYTE_0, 0);
118 	set_64bit_val(wqe, IRDMA_BYTE_8, 0);
119 	set_64bit_val(wqe, IRDMA_BYTE_16, 0);
120 
121 	/* make sure WQE is written before valid bit is set */
122 	irdma_wmb();
123 
124 	set_64bit_val(wqe, IRDMA_BYTE_24, irdma_nop_hdr(qp));
125 
126 	return 0;
127 }
128 
129 /**
130  * irdma_clr_wqes - clear next 128 sq entries
131  * @qp: hw qp ptr
132  * @qp_wqe_idx: wqe_idx
133  */
134 void
135 irdma_clr_wqes(struct irdma_qp_uk *qp, u32 qp_wqe_idx)
136 {
137 	__le64 *wqe;
138 	u32 wqe_idx;
139 
140 	if (!(qp_wqe_idx & 0x7F)) {
141 		wqe_idx = (qp_wqe_idx + 128) % qp->sq_ring.size;
142 		wqe = qp->sq_base[wqe_idx].elem;
143 		if (wqe_idx)
144 			memset(wqe, qp->swqe_polarity ? 0 : 0xFF, 0x1000);
145 		else
146 			memset(wqe, qp->swqe_polarity ? 0xFF : 0, 0x1000);
147 	}
148 }
149 
150 /**
151  * irdma_uk_qp_post_wr - ring doorbell
152  * @qp: hw qp ptr
153  */
154 void
155 irdma_uk_qp_post_wr(struct irdma_qp_uk *qp)
156 {
157 	u64 temp;
158 	u32 hw_sq_tail;
159 	u32 sw_sq_head;
160 
161 	/* valid bit is written and loads completed before reading shadow */
162 	irdma_mb();
163 
164 	/* read the doorbell shadow area */
165 	get_64bit_val(qp->shadow_area, IRDMA_BYTE_0, &temp);
166 
167 	hw_sq_tail = (u32)FIELD_GET(IRDMA_QP_DBSA_HW_SQ_TAIL, temp);
168 	sw_sq_head = IRDMA_RING_CURRENT_HEAD(qp->sq_ring);
169 	if (sw_sq_head != qp->initial_ring.head) {
170 		if (qp->push_dropped) {
171 			db_wr32(qp->qp_id, qp->wqe_alloc_db);
172 			qp->push_dropped = false;
173 		} else if (sw_sq_head != hw_sq_tail) {
174 			if (sw_sq_head > qp->initial_ring.head) {
175 				if (hw_sq_tail >= qp->initial_ring.head &&
176 				    hw_sq_tail < sw_sq_head)
177 					db_wr32(qp->qp_id, qp->wqe_alloc_db);
178 			} else {
179 				if (hw_sq_tail >= qp->initial_ring.head ||
180 				    hw_sq_tail < sw_sq_head)
181 					db_wr32(qp->qp_id, qp->wqe_alloc_db);
182 			}
183 		}
184 	}
185 
186 	qp->initial_ring.head = qp->sq_ring.head;
187 }
188 
189 /**
190  * irdma_qp_ring_push_db -  ring qp doorbell
191  * @qp: hw qp ptr
192  * @wqe_idx: wqe index
193  */
194 static void
195 irdma_qp_ring_push_db(struct irdma_qp_uk *qp, u32 wqe_idx)
196 {
197 	set_32bit_val(qp->push_db, 0,
198 		      FIELD_PREP(IRDMA_WQEALLOC_WQE_DESC_INDEX, wqe_idx >> 3) | qp->qp_id);
199 	qp->initial_ring.head = qp->sq_ring.head;
200 	qp->push_mode = true;
201 	qp->push_dropped = false;
202 }
203 
204 void
205 irdma_qp_push_wqe(struct irdma_qp_uk *qp, __le64 * wqe, u16 quanta,
206 		  u32 wqe_idx, bool post_sq)
207 {
208 	__le64 *push;
209 
210 	if (IRDMA_RING_CURRENT_HEAD(qp->initial_ring) !=
211 	    IRDMA_RING_CURRENT_TAIL(qp->sq_ring) &&
212 	    !qp->push_mode) {
213 		if (post_sq)
214 			irdma_uk_qp_post_wr(qp);
215 	} else {
216 		push = (__le64 *) ((uintptr_t)qp->push_wqe +
217 				   (wqe_idx & 0x7) * 0x20);
218 		irdma_memcpy(push, wqe, quanta * IRDMA_QP_WQE_MIN_SIZE);
219 		irdma_qp_ring_push_db(qp, wqe_idx);
220 	}
221 }
222 
223 /**
224  * irdma_qp_get_next_send_wqe - pad with NOP if needed, return where next WR should go
225  * @qp: hw qp ptr
226  * @wqe_idx: return wqe index
227  * @quanta: (in/out) ptr to size of WR in quanta. Modified in case pad is needed
228  * @total_size: size of WR in bytes
229  * @info: info on WR
230  */
231 __le64 *
232 irdma_qp_get_next_send_wqe(struct irdma_qp_uk *qp, u32 *wqe_idx,
233 			   u16 *quanta, u32 total_size,
234 			   struct irdma_post_sq_info *info)
235 {
236 	__le64 *wqe;
237 	__le64 *wqe_0 = NULL;
238 	u32 nop_wqe_idx;
239 	u16 avail_quanta, wqe_quanta = *quanta;
240 	u16 i;
241 
242 	avail_quanta = qp->uk_attrs->max_hw_sq_chunk -
243 	    (IRDMA_RING_CURRENT_HEAD(qp->sq_ring) %
244 	     qp->uk_attrs->max_hw_sq_chunk);
245 
246 	if (*quanta <= avail_quanta) {
247 		/* WR fits in current chunk */
248 		if (*quanta > IRDMA_SQ_RING_FREE_QUANTA(qp->sq_ring))
249 			return NULL;
250 	} else {
251 		/* Need to pad with NOP */
252 		if (*quanta + avail_quanta >
253 		    IRDMA_SQ_RING_FREE_QUANTA(qp->sq_ring))
254 			return NULL;
255 
256 		nop_wqe_idx = IRDMA_RING_CURRENT_HEAD(qp->sq_ring);
257 		for (i = 0; i < avail_quanta; i++) {
258 			irdma_nop_1(qp);
259 			IRDMA_RING_MOVE_HEAD_NOCHECK(qp->sq_ring);
260 		}
261 		if (qp->push_db && info->push_wqe)
262 			irdma_qp_push_wqe(qp, qp->sq_base[nop_wqe_idx].elem,
263 					  avail_quanta, nop_wqe_idx, true);
264 	}
265 
266 	*wqe_idx = IRDMA_RING_CURRENT_HEAD(qp->sq_ring);
267 	if (!*wqe_idx)
268 		qp->swqe_polarity = !qp->swqe_polarity;
269 
270 	IRDMA_RING_MOVE_HEAD_BY_COUNT_NOCHECK(qp->sq_ring, *quanta);
271 
272 	irdma_clr_wqes(qp, *wqe_idx);
273 
274 	wqe = qp->sq_base[*wqe_idx].elem;
275 	if (qp->uk_attrs->hw_rev == IRDMA_GEN_1 && wqe_quanta == 1 &&
276 	    (IRDMA_RING_CURRENT_HEAD(qp->sq_ring) & 1)) {
277 		wqe_0 = qp->sq_base[IRDMA_RING_CURRENT_HEAD(qp->sq_ring)].elem;
278 		wqe_0[3] = cpu_to_le64(FIELD_PREP(IRDMAQPSQ_VALID, !qp->swqe_polarity));
279 	}
280 	qp->sq_wrtrk_array[*wqe_idx].wrid = info->wr_id;
281 	qp->sq_wrtrk_array[*wqe_idx].wr_len = total_size;
282 	qp->sq_wrtrk_array[*wqe_idx].quanta = wqe_quanta;
283 	qp->sq_wrtrk_array[*wqe_idx].signaled = info->signaled;
284 
285 	return wqe;
286 }
287 
288 /**
289  * irdma_qp_get_next_recv_wqe - get next qp's rcv wqe
290  * @qp: hw qp ptr
291  * @wqe_idx: return wqe index
292  */
293 __le64 *
294 irdma_qp_get_next_recv_wqe(struct irdma_qp_uk *qp, u32 *wqe_idx)
295 {
296 	__le64 *wqe;
297 	int ret_code;
298 
299 	if (IRDMA_RING_FULL_ERR(qp->rq_ring))
300 		return NULL;
301 
302 	IRDMA_ATOMIC_RING_MOVE_HEAD(qp->rq_ring, *wqe_idx, ret_code);
303 	if (ret_code)
304 		return NULL;
305 
306 	if (!*wqe_idx)
307 		qp->rwqe_polarity = !qp->rwqe_polarity;
308 	/* rq_wqe_size_multiplier is no of 32 byte quanta in one rq wqe */
309 	wqe = qp->rq_base[*wqe_idx * qp->rq_wqe_size_multiplier].elem;
310 
311 	return wqe;
312 }
313 
314 /**
315  * irdma_uk_rdma_write - rdma write operation
316  * @qp: hw qp ptr
317  * @info: post sq information
318  * @post_sq: flag to post sq
319  */
320 int
321 irdma_uk_rdma_write(struct irdma_qp_uk *qp, struct irdma_post_sq_info *info,
322 		    bool post_sq)
323 {
324 	u64 hdr;
325 	__le64 *wqe;
326 	struct irdma_rdma_write *op_info;
327 	u32 i, wqe_idx;
328 	u32 total_size = 0, byte_off;
329 	int ret_code;
330 	u32 frag_cnt, addl_frag_cnt;
331 	bool read_fence = false;
332 	u16 quanta;
333 
334 	info->push_wqe = qp->push_db ? true : false;
335 
336 	op_info = &info->op.rdma_write;
337 	if (op_info->num_lo_sges > qp->max_sq_frag_cnt)
338 		return -EINVAL;
339 
340 	for (i = 0; i < op_info->num_lo_sges; i++)
341 		total_size += op_info->lo_sg_list[i].len;
342 
343 	read_fence |= info->read_fence;
344 
345 	if (info->imm_data_valid)
346 		frag_cnt = op_info->num_lo_sges + 1;
347 	else
348 		frag_cnt = op_info->num_lo_sges;
349 	addl_frag_cnt = frag_cnt > 1 ? (frag_cnt - 1) : 0;
350 	ret_code = irdma_fragcnt_to_quanta_sq(frag_cnt, &quanta);
351 	if (ret_code)
352 		return ret_code;
353 
354 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, total_size, info);
355 	if (!wqe)
356 		return -ENOSPC;
357 
358 	qp->sq_wrtrk_array[wqe_idx].signaled = info->signaled;
359 	set_64bit_val(wqe, IRDMA_BYTE_16,
360 		      FIELD_PREP(IRDMAQPSQ_FRAG_TO, op_info->rem_addr.tag_off));
361 
362 	if (info->imm_data_valid) {
363 		set_64bit_val(wqe, IRDMA_BYTE_0,
364 			      FIELD_PREP(IRDMAQPSQ_IMMDATA, info->imm_data));
365 		i = 0;
366 	} else {
367 		qp->wqe_ops.iw_set_fragment(wqe, IRDMA_BYTE_0,
368 					    op_info->lo_sg_list,
369 					    qp->swqe_polarity);
370 		i = 1;
371 	}
372 
373 	for (byte_off = IRDMA_BYTE_32; i < op_info->num_lo_sges; i++) {
374 		qp->wqe_ops.iw_set_fragment(wqe, byte_off,
375 					    &op_info->lo_sg_list[i],
376 					    qp->swqe_polarity);
377 		byte_off += 16;
378 	}
379 
380 	/* if not an odd number set valid bit in next fragment */
381 	if (qp->uk_attrs->hw_rev >= IRDMA_GEN_2 && !(frag_cnt & 0x01) &&
382 	    frag_cnt) {
383 		qp->wqe_ops.iw_set_fragment(wqe, byte_off, NULL,
384 					    qp->swqe_polarity);
385 		if (qp->uk_attrs->hw_rev == IRDMA_GEN_2)
386 			++addl_frag_cnt;
387 	}
388 
389 	hdr = FIELD_PREP(IRDMAQPSQ_REMSTAG, op_info->rem_addr.stag) |
390 	    FIELD_PREP(IRDMAQPSQ_OPCODE, info->op_type) |
391 	    FIELD_PREP(IRDMAQPSQ_IMMDATAFLAG, info->imm_data_valid) |
392 	    FIELD_PREP(IRDMAQPSQ_REPORTRTT, info->report_rtt) |
393 	    FIELD_PREP(IRDMAQPSQ_ADDFRAGCNT, addl_frag_cnt) |
394 	    FIELD_PREP(IRDMAQPSQ_PUSHWQE, info->push_wqe) |
395 	    FIELD_PREP(IRDMAQPSQ_READFENCE, read_fence) |
396 	    FIELD_PREP(IRDMAQPSQ_LOCALFENCE, info->local_fence) |
397 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, info->signaled) |
398 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
399 
400 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
401 
402 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
403 	if (info->push_wqe)
404 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
405 	else if (post_sq)
406 		irdma_uk_qp_post_wr(qp);
407 
408 	return 0;
409 }
410 
411 /**
412  * irdma_uk_rdma_read - rdma read command
413  * @qp: hw qp ptr
414  * @info: post sq information
415  * @inv_stag: flag for inv_stag
416  * @post_sq: flag to post sq
417  */
418 int
419 irdma_uk_rdma_read(struct irdma_qp_uk *qp, struct irdma_post_sq_info *info,
420 		   bool inv_stag, bool post_sq)
421 {
422 	struct irdma_rdma_read *op_info;
423 	int ret_code;
424 	u32 i, byte_off, total_size = 0;
425 	bool local_fence = false;
426 	bool ord_fence = false;
427 	u32 addl_frag_cnt;
428 	__le64 *wqe;
429 	u32 wqe_idx;
430 	u16 quanta;
431 	u64 hdr;
432 
433 	info->push_wqe = qp->push_db ? true : false;
434 
435 	op_info = &info->op.rdma_read;
436 	if (qp->max_sq_frag_cnt < op_info->num_lo_sges)
437 		return -EINVAL;
438 
439 	for (i = 0; i < op_info->num_lo_sges; i++)
440 		total_size += op_info->lo_sg_list[i].len;
441 
442 	ret_code = irdma_fragcnt_to_quanta_sq(op_info->num_lo_sges, &quanta);
443 	if (ret_code)
444 		return ret_code;
445 
446 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, total_size, info);
447 	if (!wqe)
448 		return -ENOSPC;
449 
450 	if (qp->rd_fence_rate && (qp->ord_cnt++ == qp->rd_fence_rate)) {
451 		ord_fence = true;
452 		qp->ord_cnt = 0;
453 	}
454 
455 	qp->sq_wrtrk_array[wqe_idx].signaled = info->signaled;
456 	addl_frag_cnt = op_info->num_lo_sges > 1 ?
457 	    (op_info->num_lo_sges - 1) : 0;
458 	local_fence |= info->local_fence;
459 
460 	qp->wqe_ops.iw_set_fragment(wqe, IRDMA_BYTE_0, op_info->lo_sg_list,
461 				    qp->swqe_polarity);
462 	for (i = 1, byte_off = IRDMA_BYTE_32; i < op_info->num_lo_sges; ++i) {
463 		qp->wqe_ops.iw_set_fragment(wqe, byte_off,
464 					    &op_info->lo_sg_list[i],
465 					    qp->swqe_polarity);
466 		byte_off += IRDMA_BYTE_16;
467 	}
468 
469 	/* if not an odd number set valid bit in next fragment */
470 	if (qp->uk_attrs->hw_rev >= IRDMA_GEN_2 &&
471 	    !(op_info->num_lo_sges & 0x01) && op_info->num_lo_sges) {
472 		qp->wqe_ops.iw_set_fragment(wqe, byte_off, NULL,
473 					    qp->swqe_polarity);
474 		if (qp->uk_attrs->hw_rev == IRDMA_GEN_2)
475 			++addl_frag_cnt;
476 	}
477 	set_64bit_val(wqe, IRDMA_BYTE_16,
478 		      FIELD_PREP(IRDMAQPSQ_FRAG_TO, op_info->rem_addr.tag_off));
479 	hdr = FIELD_PREP(IRDMAQPSQ_REMSTAG, op_info->rem_addr.stag) |
480 	    FIELD_PREP(IRDMAQPSQ_REPORTRTT, (info->report_rtt ? 1 : 0)) |
481 	    FIELD_PREP(IRDMAQPSQ_ADDFRAGCNT, addl_frag_cnt) |
482 	    FIELD_PREP(IRDMAQPSQ_OPCODE,
483 		       (inv_stag ? IRDMAQP_OP_RDMA_READ_LOC_INV : IRDMAQP_OP_RDMA_READ)) |
484 	    FIELD_PREP(IRDMAQPSQ_PUSHWQE, info->push_wqe) |
485 	    FIELD_PREP(IRDMAQPSQ_READFENCE,
486 		       info->read_fence || ord_fence ? 1 : 0) |
487 	    FIELD_PREP(IRDMAQPSQ_LOCALFENCE, local_fence) |
488 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, info->signaled) |
489 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
490 
491 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
492 
493 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
494 	if (info->push_wqe)
495 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
496 	else if (post_sq)
497 		irdma_uk_qp_post_wr(qp);
498 
499 	return 0;
500 }
501 
502 /**
503  * irdma_uk_send - rdma send command
504  * @qp: hw qp ptr
505  * @info: post sq information
506  * @post_sq: flag to post sq
507  */
508 int
509 irdma_uk_send(struct irdma_qp_uk *qp, struct irdma_post_sq_info *info,
510 	      bool post_sq)
511 {
512 	__le64 *wqe;
513 	struct irdma_post_send *op_info;
514 	u64 hdr;
515 	u32 i, wqe_idx, total_size = 0, byte_off;
516 	int ret_code;
517 	u32 frag_cnt, addl_frag_cnt;
518 	bool read_fence = false;
519 	u16 quanta;
520 
521 	info->push_wqe = qp->push_db ? true : false;
522 
523 	op_info = &info->op.send;
524 	if (qp->max_sq_frag_cnt < op_info->num_sges)
525 		return -EINVAL;
526 
527 	for (i = 0; i < op_info->num_sges; i++)
528 		total_size += op_info->sg_list[i].len;
529 
530 	if (info->imm_data_valid)
531 		frag_cnt = op_info->num_sges + 1;
532 	else
533 		frag_cnt = op_info->num_sges;
534 	ret_code = irdma_fragcnt_to_quanta_sq(frag_cnt, &quanta);
535 	if (ret_code)
536 		return ret_code;
537 
538 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, total_size, info);
539 	if (!wqe)
540 		return -ENOSPC;
541 
542 	read_fence |= info->read_fence;
543 	addl_frag_cnt = frag_cnt > 1 ? (frag_cnt - 1) : 0;
544 	if (info->imm_data_valid) {
545 		set_64bit_val(wqe, IRDMA_BYTE_0,
546 			      FIELD_PREP(IRDMAQPSQ_IMMDATA, info->imm_data));
547 		i = 0;
548 	} else {
549 		qp->wqe_ops.iw_set_fragment(wqe, IRDMA_BYTE_0,
550 					    frag_cnt ? op_info->sg_list : NULL,
551 					    qp->swqe_polarity);
552 		i = 1;
553 	}
554 
555 	for (byte_off = IRDMA_BYTE_32; i < op_info->num_sges; i++) {
556 		qp->wqe_ops.iw_set_fragment(wqe, byte_off, &op_info->sg_list[i],
557 					    qp->swqe_polarity);
558 		byte_off += IRDMA_BYTE_16;
559 	}
560 
561 	/* if not an odd number set valid bit in next fragment */
562 	if (qp->uk_attrs->hw_rev >= IRDMA_GEN_2 && !(frag_cnt & 0x01) &&
563 	    frag_cnt) {
564 		qp->wqe_ops.iw_set_fragment(wqe, byte_off, NULL,
565 					    qp->swqe_polarity);
566 		if (qp->uk_attrs->hw_rev == IRDMA_GEN_2)
567 			++addl_frag_cnt;
568 	}
569 
570 	set_64bit_val(wqe, IRDMA_BYTE_16,
571 		      FIELD_PREP(IRDMAQPSQ_DESTQKEY, op_info->qkey) |
572 		      FIELD_PREP(IRDMAQPSQ_DESTQPN, op_info->dest_qp));
573 	hdr = FIELD_PREP(IRDMAQPSQ_REMSTAG, info->stag_to_inv) |
574 	    FIELD_PREP(IRDMAQPSQ_AHID, op_info->ah_id) |
575 	    FIELD_PREP(IRDMAQPSQ_IMMDATAFLAG,
576 		       (info->imm_data_valid ? 1 : 0)) |
577 	    FIELD_PREP(IRDMAQPSQ_REPORTRTT, (info->report_rtt ? 1 : 0)) |
578 	    FIELD_PREP(IRDMAQPSQ_OPCODE, info->op_type) |
579 	    FIELD_PREP(IRDMAQPSQ_ADDFRAGCNT, addl_frag_cnt) |
580 	    FIELD_PREP(IRDMAQPSQ_PUSHWQE, info->push_wqe) |
581 	    FIELD_PREP(IRDMAQPSQ_READFENCE, read_fence) |
582 	    FIELD_PREP(IRDMAQPSQ_LOCALFENCE, info->local_fence) |
583 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, info->signaled) |
584 	    FIELD_PREP(IRDMAQPSQ_UDPHEADER, info->udp_hdr) |
585 	    FIELD_PREP(IRDMAQPSQ_L4LEN, info->l4len) |
586 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
587 
588 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
589 
590 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
591 	if (info->push_wqe)
592 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
593 	else if (post_sq)
594 		irdma_uk_qp_post_wr(qp);
595 
596 	return 0;
597 }
598 
599 /**
600  * irdma_set_mw_bind_wqe_gen_1 - set mw bind wqe
601  * @wqe: wqe for setting fragment
602  * @op_info: info for setting bind wqe values
603  */
604 static void
605 irdma_set_mw_bind_wqe_gen_1(__le64 * wqe,
606 			    struct irdma_bind_window *op_info)
607 {
608 	set_64bit_val(wqe, IRDMA_BYTE_0, (uintptr_t)op_info->va);
609 	set_64bit_val(wqe, IRDMA_BYTE_8,
610 		      FIELD_PREP(IRDMAQPSQ_PARENTMRSTAG, op_info->mw_stag) |
611 		      FIELD_PREP(IRDMAQPSQ_MWSTAG, op_info->mr_stag));
612 	set_64bit_val(wqe, IRDMA_BYTE_16, op_info->bind_len);
613 }
614 
615 /**
616  * irdma_copy_inline_data_gen_1 - Copy inline data to wqe
617  * @wqe: pointer to wqe
618  * @sge_list: table of pointers to inline data
619  * @num_sges: Total inline data length
620  * @polarity: compatibility parameter
621  */
622 static void
623 irdma_copy_inline_data_gen_1(u8 *wqe, struct irdma_sge *sge_list,
624 			     u32 num_sges, u8 polarity)
625 {
626 	u32 quanta_bytes_remaining = 16;
627 	u32 i;
628 
629 	for (i = 0; i < num_sges; i++) {
630 		u8 *cur_sge = (u8 *)(uintptr_t)sge_list[i].tag_off;
631 		u32 sge_len = sge_list[i].len;
632 
633 		while (sge_len) {
634 			u32 bytes_copied;
635 
636 			bytes_copied = min(sge_len, quanta_bytes_remaining);
637 			irdma_memcpy(wqe, cur_sge, bytes_copied);
638 			wqe += bytes_copied;
639 			cur_sge += bytes_copied;
640 			quanta_bytes_remaining -= bytes_copied;
641 			sge_len -= bytes_copied;
642 
643 			if (!quanta_bytes_remaining) {
644 				/* Remaining inline bytes reside after hdr */
645 				wqe += 16;
646 				quanta_bytes_remaining = 32;
647 			}
648 		}
649 	}
650 }
651 
652 /**
653  * irdma_inline_data_size_to_quanta_gen_1 - based on inline data, quanta
654  * @data_size: data size for inline
655  *
656  * Gets the quanta based on inline and immediate data.
657  */
658 static inline u16 irdma_inline_data_size_to_quanta_gen_1(u32 data_size) {
659 	return data_size <= 16 ? IRDMA_QP_WQE_MIN_QUANTA : 2;
660 }
661 
662 /**
663  * irdma_set_mw_bind_wqe - set mw bind in wqe
664  * @wqe: wqe for setting mw bind
665  * @op_info: info for setting wqe values
666  */
667 static void
668 irdma_set_mw_bind_wqe(__le64 * wqe,
669 		      struct irdma_bind_window *op_info)
670 {
671 	set_64bit_val(wqe, IRDMA_BYTE_0, (uintptr_t)op_info->va);
672 	set_64bit_val(wqe, IRDMA_BYTE_8,
673 		      FIELD_PREP(IRDMAQPSQ_PARENTMRSTAG, op_info->mr_stag) |
674 		      FIELD_PREP(IRDMAQPSQ_MWSTAG, op_info->mw_stag));
675 	set_64bit_val(wqe, IRDMA_BYTE_16, op_info->bind_len);
676 }
677 
678 /**
679  * irdma_copy_inline_data - Copy inline data to wqe
680  * @wqe: pointer to wqe
681  * @sge_list: table of pointers to inline data
682  * @num_sges: number of SGE's
683  * @polarity: polarity of wqe valid bit
684  */
685 static void
686 irdma_copy_inline_data(u8 *wqe, struct irdma_sge *sge_list,
687 		       u32 num_sges, u8 polarity)
688 {
689 	u8 inline_valid = polarity << IRDMA_INLINE_VALID_S;
690 	u32 quanta_bytes_remaining = 8;
691 	u32 i;
692 	bool first_quanta = true;
693 
694 	wqe += 8;
695 
696 	for (i = 0; i < num_sges; i++) {
697 		u8 *cur_sge = (u8 *)(uintptr_t)sge_list[i].tag_off;
698 		u32 sge_len = sge_list[i].len;
699 
700 		while (sge_len) {
701 			u32 bytes_copied;
702 
703 			bytes_copied = min(sge_len, quanta_bytes_remaining);
704 			irdma_memcpy(wqe, cur_sge, bytes_copied);
705 			wqe += bytes_copied;
706 			cur_sge += bytes_copied;
707 			quanta_bytes_remaining -= bytes_copied;
708 			sge_len -= bytes_copied;
709 
710 			if (!quanta_bytes_remaining) {
711 				quanta_bytes_remaining = 31;
712 
713 				/* Remaining inline bytes reside after hdr */
714 				if (first_quanta) {
715 					first_quanta = false;
716 					wqe += 16;
717 				} else {
718 					*wqe = inline_valid;
719 					wqe++;
720 				}
721 			}
722 		}
723 	}
724 	if (!first_quanta && quanta_bytes_remaining < 31)
725 		*(wqe + quanta_bytes_remaining) = inline_valid;
726 }
727 
728 /**
729  * irdma_inline_data_size_to_quanta - based on inline data, quanta
730  * @data_size: data size for inline
731  *
732  * Gets the quanta based on inline and immediate data.
733  */
734 static u16 irdma_inline_data_size_to_quanta(u32 data_size) {
735 	if (data_size <= 8)
736 		return IRDMA_QP_WQE_MIN_QUANTA;
737 	else if (data_size <= 39)
738 		return 2;
739 	else if (data_size <= 70)
740 		return 3;
741 	else if (data_size <= 101)
742 		return 4;
743 	else if (data_size <= 132)
744 		return 5;
745 	else if (data_size <= 163)
746 		return 6;
747 	else if (data_size <= 194)
748 		return 7;
749 	else
750 		return 8;
751 }
752 
753 /**
754  * irdma_uk_inline_rdma_write - inline rdma write operation
755  * @qp: hw qp ptr
756  * @info: post sq information
757  * @post_sq: flag to post sq
758  */
759 int
760 irdma_uk_inline_rdma_write(struct irdma_qp_uk *qp,
761 			   struct irdma_post_sq_info *info, bool post_sq)
762 {
763 	__le64 *wqe;
764 	struct irdma_rdma_write *op_info;
765 	u64 hdr = 0;
766 	u32 wqe_idx;
767 	bool read_fence = false;
768 	u16 quanta;
769 	u32 i, total_size = 0;
770 
771 	info->push_wqe = qp->push_db ? true : false;
772 	op_info = &info->op.rdma_write;
773 
774 	if (unlikely(qp->max_sq_frag_cnt < op_info->num_lo_sges))
775 		return -EINVAL;
776 
777 	for (i = 0; i < op_info->num_lo_sges; i++)
778 		total_size += op_info->lo_sg_list[i].len;
779 
780 	if (unlikely(total_size > qp->max_inline_data))
781 		return -EINVAL;
782 
783 	quanta = qp->wqe_ops.iw_inline_data_size_to_quanta(total_size);
784 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, total_size, info);
785 	if (!wqe)
786 		return -ENOSPC;
787 
788 	qp->sq_wrtrk_array[wqe_idx].signaled = info->signaled;
789 	read_fence |= info->read_fence;
790 	set_64bit_val(wqe, IRDMA_BYTE_16,
791 		      FIELD_PREP(IRDMAQPSQ_FRAG_TO, op_info->rem_addr.tag_off));
792 
793 	hdr = FIELD_PREP(IRDMAQPSQ_REMSTAG, op_info->rem_addr.stag) |
794 	    FIELD_PREP(IRDMAQPSQ_OPCODE, info->op_type) |
795 	    FIELD_PREP(IRDMAQPSQ_INLINEDATALEN, total_size) |
796 	    FIELD_PREP(IRDMAQPSQ_REPORTRTT, info->report_rtt ? 1 : 0) |
797 	    FIELD_PREP(IRDMAQPSQ_INLINEDATAFLAG, 1) |
798 	    FIELD_PREP(IRDMAQPSQ_IMMDATAFLAG, info->imm_data_valid ? 1 : 0) |
799 	    FIELD_PREP(IRDMAQPSQ_PUSHWQE, info->push_wqe ? 1 : 0) |
800 	    FIELD_PREP(IRDMAQPSQ_READFENCE, read_fence) |
801 	    FIELD_PREP(IRDMAQPSQ_LOCALFENCE, info->local_fence) |
802 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, info->signaled) |
803 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
804 
805 	if (info->imm_data_valid)
806 		set_64bit_val(wqe, IRDMA_BYTE_0,
807 			      FIELD_PREP(IRDMAQPSQ_IMMDATA, info->imm_data));
808 
809 	qp->wqe_ops.iw_copy_inline_data((u8 *)wqe, op_info->lo_sg_list,
810 					op_info->num_lo_sges, qp->swqe_polarity);
811 
812 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
813 
814 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
815 
816 	if (info->push_wqe)
817 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
818 	else if (post_sq)
819 		irdma_uk_qp_post_wr(qp);
820 
821 	return 0;
822 }
823 
824 /**
825  * irdma_uk_inline_send - inline send operation
826  * @qp: hw qp ptr
827  * @info: post sq information
828  * @post_sq: flag to post sq
829  */
830 int
831 irdma_uk_inline_send(struct irdma_qp_uk *qp,
832 		     struct irdma_post_sq_info *info, bool post_sq)
833 {
834 	__le64 *wqe;
835 	struct irdma_post_send *op_info;
836 	u64 hdr;
837 	u32 wqe_idx;
838 	bool read_fence = false;
839 	u16 quanta;
840 	u32 i, total_size = 0;
841 
842 	info->push_wqe = qp->push_db ? true : false;
843 	op_info = &info->op.send;
844 
845 	if (unlikely(qp->max_sq_frag_cnt < op_info->num_sges))
846 		return -EINVAL;
847 
848 	for (i = 0; i < op_info->num_sges; i++)
849 		total_size += op_info->sg_list[i].len;
850 
851 	if (unlikely(total_size > qp->max_inline_data))
852 		return -EINVAL;
853 
854 	quanta = qp->wqe_ops.iw_inline_data_size_to_quanta(total_size);
855 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, total_size, info);
856 	if (!wqe)
857 		return -ENOSPC;
858 
859 	set_64bit_val(wqe, IRDMA_BYTE_16,
860 		      FIELD_PREP(IRDMAQPSQ_DESTQKEY, op_info->qkey) |
861 		      FIELD_PREP(IRDMAQPSQ_DESTQPN, op_info->dest_qp));
862 
863 	read_fence |= info->read_fence;
864 	hdr = FIELD_PREP(IRDMAQPSQ_REMSTAG, info->stag_to_inv) |
865 	    FIELD_PREP(IRDMAQPSQ_AHID, op_info->ah_id) |
866 	    FIELD_PREP(IRDMAQPSQ_OPCODE, info->op_type) |
867 	    FIELD_PREP(IRDMAQPSQ_INLINEDATALEN, total_size) |
868 	    FIELD_PREP(IRDMAQPSQ_IMMDATAFLAG,
869 		       (info->imm_data_valid ? 1 : 0)) |
870 	    FIELD_PREP(IRDMAQPSQ_REPORTRTT, (info->report_rtt ? 1 : 0)) |
871 	    FIELD_PREP(IRDMAQPSQ_INLINEDATAFLAG, 1) |
872 	    FIELD_PREP(IRDMAQPSQ_PUSHWQE, info->push_wqe) |
873 	    FIELD_PREP(IRDMAQPSQ_READFENCE, read_fence) |
874 	    FIELD_PREP(IRDMAQPSQ_LOCALFENCE, info->local_fence) |
875 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, info->signaled) |
876 	    FIELD_PREP(IRDMAQPSQ_UDPHEADER, info->udp_hdr) |
877 	    FIELD_PREP(IRDMAQPSQ_L4LEN, info->l4len) |
878 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
879 
880 	if (info->imm_data_valid)
881 		set_64bit_val(wqe, IRDMA_BYTE_0,
882 			      FIELD_PREP(IRDMAQPSQ_IMMDATA, info->imm_data));
883 	qp->wqe_ops.iw_copy_inline_data((u8 *)wqe, op_info->sg_list,
884 					op_info->num_sges, qp->swqe_polarity);
885 
886 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
887 
888 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
889 
890 	if (info->push_wqe)
891 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
892 	else if (post_sq)
893 		irdma_uk_qp_post_wr(qp);
894 
895 	return 0;
896 }
897 
898 /**
899  * irdma_uk_stag_local_invalidate - stag invalidate operation
900  * @qp: hw qp ptr
901  * @info: post sq information
902  * @post_sq: flag to post sq
903  */
904 int
905 irdma_uk_stag_local_invalidate(struct irdma_qp_uk *qp,
906 			       struct irdma_post_sq_info *info,
907 			       bool post_sq)
908 {
909 	__le64 *wqe;
910 	struct irdma_inv_local_stag *op_info;
911 	u64 hdr;
912 	u32 wqe_idx;
913 	bool local_fence = false;
914 	struct irdma_sge sge = {0};
915 	u16 quanta = IRDMA_QP_WQE_MIN_QUANTA;
916 
917 	info->push_wqe = qp->push_db ? true : false;
918 	op_info = &info->op.inv_local_stag;
919 	local_fence = info->local_fence;
920 
921 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, 0, info);
922 	if (!wqe)
923 		return -ENOSPC;
924 
925 	sge.stag = op_info->target_stag;
926 	qp->wqe_ops.iw_set_fragment(wqe, IRDMA_BYTE_0, &sge, 0);
927 
928 	set_64bit_val(wqe, IRDMA_BYTE_16, 0);
929 
930 	hdr = FIELD_PREP(IRDMAQPSQ_OPCODE, IRDMA_OP_TYPE_INV_STAG) |
931 	    FIELD_PREP(IRDMAQPSQ_PUSHWQE, info->push_wqe) |
932 	    FIELD_PREP(IRDMAQPSQ_READFENCE, info->read_fence) |
933 	    FIELD_PREP(IRDMAQPSQ_LOCALFENCE, local_fence) |
934 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, info->signaled) |
935 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
936 
937 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
938 
939 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
940 
941 	if (info->push_wqe)
942 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
943 	else if (post_sq)
944 		irdma_uk_qp_post_wr(qp);
945 
946 	return 0;
947 }
948 
949 /**
950  * irdma_uk_post_receive - post receive wqe
951  * @qp: hw qp ptr
952  * @info: post rq information
953  */
954 int
955 irdma_uk_post_receive(struct irdma_qp_uk *qp,
956 		      struct irdma_post_rq_info *info)
957 {
958 	u32 wqe_idx, i, byte_off;
959 	u32 addl_frag_cnt;
960 	__le64 *wqe;
961 	u64 hdr;
962 
963 	if (qp->max_rq_frag_cnt < info->num_sges)
964 		return -EINVAL;
965 
966 	wqe = irdma_qp_get_next_recv_wqe(qp, &wqe_idx);
967 	if (!wqe)
968 		return -ENOSPC;
969 
970 	qp->rq_wrid_array[wqe_idx] = info->wr_id;
971 	addl_frag_cnt = info->num_sges > 1 ? (info->num_sges - 1) : 0;
972 	qp->wqe_ops.iw_set_fragment(wqe, IRDMA_BYTE_0, info->sg_list,
973 				    qp->rwqe_polarity);
974 
975 	for (i = 1, byte_off = IRDMA_BYTE_32; i < info->num_sges; i++) {
976 		qp->wqe_ops.iw_set_fragment(wqe, byte_off, &info->sg_list[i],
977 					    qp->rwqe_polarity);
978 		byte_off += 16;
979 	}
980 
981 	/* if not an odd number set valid bit in next fragment */
982 	if (qp->uk_attrs->hw_rev >= IRDMA_GEN_2 && !(info->num_sges & 0x01) &&
983 	    info->num_sges) {
984 		qp->wqe_ops.iw_set_fragment(wqe, byte_off, NULL,
985 					    qp->rwqe_polarity);
986 		if (qp->uk_attrs->hw_rev == IRDMA_GEN_2)
987 			++addl_frag_cnt;
988 	}
989 
990 	set_64bit_val(wqe, IRDMA_BYTE_16, 0);
991 	hdr = FIELD_PREP(IRDMAQPSQ_ADDFRAGCNT, addl_frag_cnt) |
992 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->rwqe_polarity);
993 
994 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
995 
996 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
997 
998 	return 0;
999 }
1000 
1001 /**
1002  * irdma_uk_cq_resize - reset the cq buffer info
1003  * @cq: cq to resize
1004  * @cq_base: new cq buffer addr
1005  * @cq_size: number of cqes
1006  */
1007 void
1008 irdma_uk_cq_resize(struct irdma_cq_uk *cq, void *cq_base, int cq_size)
1009 {
1010 	cq->cq_base = cq_base;
1011 	cq->cq_size = cq_size;
1012 	IRDMA_RING_INIT(cq->cq_ring, cq->cq_size);
1013 	cq->polarity = 1;
1014 }
1015 
1016 /**
1017  * irdma_uk_cq_set_resized_cnt - record the count of the resized buffers
1018  * @cq: cq to resize
1019  * @cq_cnt: the count of the resized cq buffers
1020  */
1021 void
1022 irdma_uk_cq_set_resized_cnt(struct irdma_cq_uk *cq, u16 cq_cnt)
1023 {
1024 	u64 temp_val;
1025 	u16 sw_cq_sel;
1026 	u8 arm_next_se;
1027 	u8 arm_next;
1028 	u8 arm_seq_num;
1029 
1030 	get_64bit_val(cq->shadow_area, 32, &temp_val);
1031 
1032 	sw_cq_sel = (u16)FIELD_GET(IRDMA_CQ_DBSA_SW_CQ_SELECT, temp_val);
1033 	sw_cq_sel += cq_cnt;
1034 
1035 	arm_seq_num = (u8)FIELD_GET(IRDMA_CQ_DBSA_ARM_SEQ_NUM, temp_val);
1036 	arm_next_se = (u8)FIELD_GET(IRDMA_CQ_DBSA_ARM_NEXT_SE, temp_val);
1037 	arm_next = (u8)FIELD_GET(IRDMA_CQ_DBSA_ARM_NEXT, temp_val);
1038 
1039 	temp_val = FIELD_PREP(IRDMA_CQ_DBSA_ARM_SEQ_NUM, arm_seq_num) |
1040 	    FIELD_PREP(IRDMA_CQ_DBSA_SW_CQ_SELECT, sw_cq_sel) |
1041 	    FIELD_PREP(IRDMA_CQ_DBSA_ARM_NEXT_SE, arm_next_se) |
1042 	    FIELD_PREP(IRDMA_CQ_DBSA_ARM_NEXT, arm_next);
1043 
1044 	set_64bit_val(cq->shadow_area, 32, temp_val);
1045 }
1046 
1047 /**
1048  * irdma_uk_cq_request_notification - cq notification request (door bell)
1049  * @cq: hw cq
1050  * @cq_notify: notification type
1051  */
1052 void
1053 irdma_uk_cq_request_notification(struct irdma_cq_uk *cq,
1054 				 enum irdma_cmpl_notify cq_notify)
1055 {
1056 	u64 temp_val;
1057 	u16 sw_cq_sel;
1058 	u8 arm_next_se = 0;
1059 	u8 arm_next = 0;
1060 	u8 arm_seq_num;
1061 
1062 	get_64bit_val(cq->shadow_area, IRDMA_BYTE_32, &temp_val);
1063 	arm_seq_num = (u8)FIELD_GET(IRDMA_CQ_DBSA_ARM_SEQ_NUM, temp_val);
1064 	arm_seq_num++;
1065 	sw_cq_sel = (u16)FIELD_GET(IRDMA_CQ_DBSA_SW_CQ_SELECT, temp_val);
1066 	arm_next_se = (u8)FIELD_GET(IRDMA_CQ_DBSA_ARM_NEXT_SE, temp_val);
1067 	arm_next_se |= 1;
1068 	if (cq_notify == IRDMA_CQ_COMPL_EVENT)
1069 		arm_next = 1;
1070 	temp_val = FIELD_PREP(IRDMA_CQ_DBSA_ARM_SEQ_NUM, arm_seq_num) |
1071 	    FIELD_PREP(IRDMA_CQ_DBSA_SW_CQ_SELECT, sw_cq_sel) |
1072 	    FIELD_PREP(IRDMA_CQ_DBSA_ARM_NEXT_SE, arm_next_se) |
1073 	    FIELD_PREP(IRDMA_CQ_DBSA_ARM_NEXT, arm_next);
1074 
1075 	set_64bit_val(cq->shadow_area, IRDMA_BYTE_32, temp_val);
1076 
1077 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
1078 
1079 	db_wr32(cq->cq_id, cq->cqe_alloc_db);
1080 }
1081 
1082 static int
1083 irdma_check_rq_cqe(struct irdma_qp_uk *qp, u32 *array_idx)
1084 {
1085 	u32 exp_idx = (qp->last_rx_cmpl_idx + 1) % qp->rq_size;
1086 
1087 	if (*array_idx != exp_idx) {
1088 
1089 		*array_idx = exp_idx;
1090 		qp->last_rx_cmpl_idx = exp_idx;
1091 
1092 		return -1;
1093 	}
1094 
1095 	qp->last_rx_cmpl_idx = *array_idx;
1096 
1097 	return 0;
1098 }
1099 
1100 /**
1101  * irdma_skip_duplicate_flush_cmpl - check last cmpl and update wqe if needed
1102  *
1103  * @ring: sq/rq ring
1104  * @flush_seen: information if flush for specific ring was already seen
1105  * @comp_status: completion status
1106  * @wqe_idx: new value of WQE index returned if there is more work on ring
1107  */
1108 static inline int
1109 irdma_skip_duplicate_flush_cmpl(struct irdma_ring ring, u8 flush_seen,
1110 				enum irdma_cmpl_status comp_status,
1111 				u32 *wqe_idx)
1112 {
1113 	if (flush_seen) {
1114 		if (IRDMA_RING_MORE_WORK(ring))
1115 			*wqe_idx = ring.tail;
1116 		else
1117 			return -ENOENT;
1118 	}
1119 
1120 	return 0;
1121 }
1122 
1123 /**
1124  * irdma_detect_unsignaled_cmpls - check if unsignaled cmpl is to be reported
1125  * @cq: hw cq
1126  * @qp: hw qp
1127  * @info: cq poll information collected
1128  * @wge_idx: index of the WR in SQ ring
1129  */
1130 static int
1131 irdma_detect_unsignaled_cmpls(struct irdma_cq_uk *cq,
1132 			      struct irdma_qp_uk *qp,
1133 			      struct irdma_cq_poll_info *info,
1134 			      u32 wqe_idx)
1135 {
1136 	u64 qword0, qword1, qword2, qword3;
1137 	__le64 *cqe, *wqe;
1138 	int i;
1139 	u32 widx;
1140 
1141 	if (qp->sq_wrtrk_array[wqe_idx].signaled == 0) {
1142 		cqe = IRDMA_GET_CURRENT_CQ_ELEM(cq);
1143 		irdma_pr_err("%p %d %d\n", cqe, cq->cq_ring.head, wqe_idx);
1144 		for (i = -10; i <= 10; i++) {
1145 			IRDMA_GET_CQ_ELEM_AT_OFFSET(cq, i + cq->cq_ring.size, cqe);
1146 			get_64bit_val(cqe, IRDMA_BYTE_0, &qword0);
1147 			get_64bit_val(cqe, IRDMA_BYTE_8, &qword1);
1148 			get_64bit_val(cqe, IRDMA_BYTE_16, &qword2);
1149 			get_64bit_val(cqe, IRDMA_BYTE_24, &qword3);
1150 			widx = (u32)FIELD_GET(IRDMA_CQ_WQEIDX, qword3);
1151 			irdma_pr_err("%d %04x %p %016lx %016lx %016lx %016lx ",
1152 				     i, widx, cqe, qword0, qword1, qword2, qword3);
1153 			if ((u8)FIELD_GET(IRDMA_CQ_SQ, qword3)) {
1154 				irdma_pr_err("%lx %x %x %x ",
1155 					     qp->sq_wrtrk_array[widx].wrid, qp->sq_wrtrk_array[widx].wr_len,
1156 					     qp->sq_wrtrk_array[widx].quanta, qp->sq_wrtrk_array[widx].signaled);
1157 				wqe = qp->sq_base[widx].elem;
1158 				get_64bit_val(wqe, IRDMA_BYTE_0, &qword0);
1159 				get_64bit_val(wqe, IRDMA_BYTE_8, &qword1);
1160 				get_64bit_val(wqe, IRDMA_BYTE_16, &qword2);
1161 				get_64bit_val(wqe, IRDMA_BYTE_24, &qword3);
1162 
1163 				irdma_pr_err("%016lx %016lx %016lx %016lx \n",
1164 					     qword0, qword1, qword2, qword3);
1165 			} else {
1166 				irdma_pr_err("\n");
1167 			}
1168 		}
1169 		return -ENOENT;
1170 	}
1171 
1172 	return 0;
1173 }
1174 
1175 /**
1176  * irdma_uk_cq_poll_cmpl - get cq completion info
1177  * @cq: hw cq
1178  * @info: cq poll information returned
1179  */
1180 int
1181 irdma_uk_cq_poll_cmpl(struct irdma_cq_uk *cq,
1182 		      struct irdma_cq_poll_info *info)
1183 {
1184 	u64 comp_ctx, qword0, qword2, qword3;
1185 	__le64 *cqe;
1186 	struct irdma_qp_uk *qp;
1187 	struct irdma_ring *pring = NULL;
1188 	u32 wqe_idx;
1189 	int ret_code;
1190 	bool move_cq_head = true;
1191 	u8 polarity;
1192 	bool ext_valid;
1193 	__le64 *ext_cqe;
1194 
1195 	if (cq->avoid_mem_cflct)
1196 		cqe = IRDMA_GET_CURRENT_EXTENDED_CQ_ELEM(cq);
1197 	else
1198 		cqe = IRDMA_GET_CURRENT_CQ_ELEM(cq);
1199 
1200 	get_64bit_val(cqe, IRDMA_BYTE_24, &qword3);
1201 	polarity = (u8)FIELD_GET(IRDMA_CQ_VALID, qword3);
1202 	if (polarity != cq->polarity)
1203 		return -ENOENT;
1204 
1205 	/* Ensure CQE contents are read after valid bit is checked */
1206 	rmb();
1207 
1208 	ext_valid = (bool)FIELD_GET(IRDMA_CQ_EXTCQE, qword3);
1209 	if (ext_valid) {
1210 		u64 qword6, qword7;
1211 		u32 peek_head;
1212 
1213 		if (cq->avoid_mem_cflct) {
1214 			ext_cqe = (__le64 *) ((u8 *)cqe + 32);
1215 			get_64bit_val(ext_cqe, IRDMA_BYTE_24, &qword7);
1216 			polarity = (u8)FIELD_GET(IRDMA_CQ_VALID, qword7);
1217 		} else {
1218 			peek_head = (cq->cq_ring.head + 1) % cq->cq_ring.size;
1219 			ext_cqe = cq->cq_base[peek_head].buf;
1220 			get_64bit_val(ext_cqe, IRDMA_BYTE_24, &qword7);
1221 			polarity = (u8)FIELD_GET(IRDMA_CQ_VALID, qword7);
1222 			if (!peek_head)
1223 				polarity ^= 1;
1224 		}
1225 		if (polarity != cq->polarity)
1226 			return -ENOENT;
1227 
1228 		/* Ensure ext CQE contents are read after ext valid bit is checked */
1229 		rmb();
1230 
1231 		info->imm_valid = (bool)FIELD_GET(IRDMA_CQ_IMMVALID, qword7);
1232 		if (info->imm_valid) {
1233 			u64 qword4;
1234 
1235 			get_64bit_val(ext_cqe, IRDMA_BYTE_0, &qword4);
1236 			info->imm_data = (u32)FIELD_GET(IRDMA_CQ_IMMDATALOW32, qword4);
1237 		}
1238 		info->ud_smac_valid = (bool)FIELD_GET(IRDMA_CQ_UDSMACVALID, qword7);
1239 		info->ud_vlan_valid = (bool)FIELD_GET(IRDMA_CQ_UDVLANVALID, qword7);
1240 		if (info->ud_smac_valid || info->ud_vlan_valid) {
1241 			get_64bit_val(ext_cqe, IRDMA_BYTE_16, &qword6);
1242 			if (info->ud_vlan_valid)
1243 				info->ud_vlan = (u16)FIELD_GET(IRDMA_CQ_UDVLAN, qword6);
1244 			if (info->ud_smac_valid) {
1245 				info->ud_smac[5] = qword6 & 0xFF;
1246 				info->ud_smac[4] = (qword6 >> 8) & 0xFF;
1247 				info->ud_smac[3] = (qword6 >> 16) & 0xFF;
1248 				info->ud_smac[2] = (qword6 >> 24) & 0xFF;
1249 				info->ud_smac[1] = (qword6 >> 32) & 0xFF;
1250 				info->ud_smac[0] = (qword6 >> 40) & 0xFF;
1251 			}
1252 		}
1253 	} else {
1254 		info->imm_valid = false;
1255 		info->ud_smac_valid = false;
1256 		info->ud_vlan_valid = false;
1257 	}
1258 
1259 	info->q_type = (u8)FIELD_GET(IRDMA_CQ_SQ, qword3);
1260 	info->error = (bool)FIELD_GET(IRDMA_CQ_ERROR, qword3);
1261 	info->push_dropped = (bool)FIELD_GET(IRDMACQ_PSHDROP, qword3);
1262 	info->ipv4 = (bool)FIELD_GET(IRDMACQ_IPV4, qword3);
1263 	get_64bit_val(cqe, IRDMA_BYTE_8, &comp_ctx);
1264 	qp = (struct irdma_qp_uk *)(irdma_uintptr) comp_ctx;
1265 	if (info->error) {
1266 		info->major_err = FIELD_GET(IRDMA_CQ_MAJERR, qword3);
1267 		info->minor_err = FIELD_GET(IRDMA_CQ_MINERR, qword3);
1268 		switch (info->major_err) {
1269 		case IRDMA_FLUSH_MAJOR_ERR:
1270 			/* Set the min error to standard flush error code for remaining cqes */
1271 			if (info->minor_err != FLUSH_GENERAL_ERR) {
1272 				qword3 &= ~IRDMA_CQ_MINERR;
1273 				qword3 |= FIELD_PREP(IRDMA_CQ_MINERR, FLUSH_GENERAL_ERR);
1274 				set_64bit_val(cqe, IRDMA_BYTE_24, qword3);
1275 			}
1276 			info->comp_status = IRDMA_COMPL_STATUS_FLUSHED;
1277 			break;
1278 		default:
1279 			info->comp_status = IRDMA_COMPL_STATUS_UNKNOWN;
1280 			break;
1281 		}
1282 	} else {
1283 		info->comp_status = IRDMA_COMPL_STATUS_SUCCESS;
1284 	}
1285 
1286 	get_64bit_val(cqe, IRDMA_BYTE_0, &qword0);
1287 	get_64bit_val(cqe, IRDMA_BYTE_16, &qword2);
1288 
1289 	info->stat.raw = (u32)FIELD_GET(IRDMACQ_TCPSQN_ROCEPSN_RTT_TS, qword0);
1290 	info->qp_id = (u32)FIELD_GET(IRDMACQ_QPID, qword2);
1291 	info->ud_src_qpn = (u32)FIELD_GET(IRDMACQ_UDSRCQPN, qword2);
1292 
1293 	info->solicited_event = (bool)FIELD_GET(IRDMACQ_SOEVENT, qword3);
1294 	if (!qp || qp->destroy_pending) {
1295 		ret_code = -EFAULT;
1296 		goto exit;
1297 	}
1298 	wqe_idx = (u32)FIELD_GET(IRDMA_CQ_WQEIDX, qword3);
1299 	info->qp_handle = (irdma_qp_handle) (irdma_uintptr) qp;
1300 	info->op_type = (u8)FIELD_GET(IRDMACQ_OP, qword3);
1301 
1302 	if (info->q_type == IRDMA_CQE_QTYPE_RQ) {
1303 		u32 array_idx;
1304 
1305 		ret_code = irdma_skip_duplicate_flush_cmpl(qp->rq_ring,
1306 							   qp->rq_flush_seen,
1307 							   info->comp_status,
1308 							   &wqe_idx);
1309 		if (ret_code != 0)
1310 			goto exit;
1311 
1312 		array_idx = wqe_idx / qp->rq_wqe_size_multiplier;
1313 
1314 		if (info->comp_status == IRDMA_COMPL_STATUS_FLUSHED ||
1315 		    info->comp_status == IRDMA_COMPL_STATUS_UNKNOWN) {
1316 			if (!IRDMA_RING_MORE_WORK(qp->rq_ring)) {
1317 				ret_code = -ENOENT;
1318 				goto exit;
1319 			}
1320 
1321 			info->wr_id = qp->rq_wrid_array[qp->rq_ring.tail];
1322 			info->signaled = 1;
1323 			array_idx = qp->rq_ring.tail;
1324 		} else {
1325 			info->wr_id = qp->rq_wrid_array[array_idx];
1326 			info->signaled = 1;
1327 			if (irdma_check_rq_cqe(qp, &array_idx)) {
1328 				info->wr_id = qp->rq_wrid_array[array_idx];
1329 				info->comp_status = IRDMA_COMPL_STATUS_UNKNOWN;
1330 				IRDMA_RING_SET_TAIL(qp->rq_ring, array_idx + 1);
1331 				return 0;
1332 			}
1333 		}
1334 
1335 		info->bytes_xfered = (u32)FIELD_GET(IRDMACQ_PAYLDLEN, qword0);
1336 
1337 		if (qword3 & IRDMACQ_STAG) {
1338 			info->stag_invalid_set = true;
1339 			info->inv_stag = (u32)FIELD_GET(IRDMACQ_INVSTAG, qword2);
1340 		} else {
1341 			info->stag_invalid_set = false;
1342 		}
1343 		IRDMA_RING_SET_TAIL(qp->rq_ring, array_idx + 1);
1344 		if (info->comp_status == IRDMA_COMPL_STATUS_FLUSHED) {
1345 			qp->rq_flush_seen = true;
1346 			if (!IRDMA_RING_MORE_WORK(qp->rq_ring))
1347 				qp->rq_flush_complete = true;
1348 			else
1349 				move_cq_head = false;
1350 		}
1351 		pring = &qp->rq_ring;
1352 	} else {		/* q_type is IRDMA_CQE_QTYPE_SQ */
1353 		if (qp->first_sq_wq) {
1354 			if (wqe_idx + 1 >= qp->conn_wqes)
1355 				qp->first_sq_wq = false;
1356 
1357 			if (wqe_idx < qp->conn_wqes && qp->sq_ring.head == qp->sq_ring.tail) {
1358 				IRDMA_RING_MOVE_HEAD_NOCHECK(cq->cq_ring);
1359 				IRDMA_RING_MOVE_TAIL(cq->cq_ring);
1360 				set_64bit_val(cq->shadow_area, IRDMA_BYTE_0,
1361 					      IRDMA_RING_CURRENT_HEAD(cq->cq_ring));
1362 				memset(info, 0,
1363 				       sizeof(struct irdma_cq_poll_info));
1364 				return irdma_uk_cq_poll_cmpl(cq, info);
1365 			}
1366 		}
1367 		/* cease posting push mode on push drop */
1368 		if (info->push_dropped) {
1369 			qp->push_mode = false;
1370 			qp->push_dropped = true;
1371 		}
1372 		ret_code = irdma_skip_duplicate_flush_cmpl(qp->sq_ring,
1373 							   qp->sq_flush_seen,
1374 							   info->comp_status,
1375 							   &wqe_idx);
1376 		if (ret_code != 0)
1377 			goto exit;
1378 		if (info->comp_status != IRDMA_COMPL_STATUS_FLUSHED) {
1379 			info->wr_id = qp->sq_wrtrk_array[wqe_idx].wrid;
1380 			info->signaled = qp->sq_wrtrk_array[wqe_idx].signaled;
1381 			if (!info->comp_status)
1382 				info->bytes_xfered = qp->sq_wrtrk_array[wqe_idx].wr_len;
1383 			ret_code = irdma_detect_unsignaled_cmpls(cq, qp, info, wqe_idx);
1384 			if (ret_code != 0)
1385 				goto exit;
1386 			info->op_type = (u8)FIELD_GET(IRDMACQ_OP, qword3);
1387 			IRDMA_RING_SET_TAIL(qp->sq_ring,
1388 					    wqe_idx + qp->sq_wrtrk_array[wqe_idx].quanta);
1389 		} else {
1390 			unsigned long flags;
1391 
1392 			spin_lock_irqsave(qp->lock, flags);
1393 			if (!IRDMA_RING_MORE_WORK(qp->sq_ring)) {
1394 				spin_unlock_irqrestore(qp->lock, flags);
1395 				ret_code = -ENOENT;
1396 				goto exit;
1397 			}
1398 
1399 			do {
1400 				__le64 *sw_wqe;
1401 				u64 wqe_qword;
1402 				u32 tail;
1403 
1404 				tail = qp->sq_ring.tail;
1405 				sw_wqe = qp->sq_base[tail].elem;
1406 				get_64bit_val(sw_wqe, IRDMA_BYTE_24,
1407 					      &wqe_qword);
1408 				info->op_type = (u8)FIELD_GET(IRDMAQPSQ_OPCODE,
1409 							      wqe_qword);
1410 				IRDMA_RING_SET_TAIL(qp->sq_ring,
1411 						    tail + qp->sq_wrtrk_array[tail].quanta);
1412 				if (info->op_type != IRDMAQP_OP_NOP) {
1413 					info->wr_id = qp->sq_wrtrk_array[tail].wrid;
1414 					info->signaled = qp->sq_wrtrk_array[tail].signaled;
1415 					info->bytes_xfered = qp->sq_wrtrk_array[tail].wr_len;
1416 					break;
1417 				}
1418 			} while (1);
1419 
1420 			if (info->op_type == IRDMA_OP_TYPE_BIND_MW &&
1421 			    info->minor_err == FLUSH_PROT_ERR)
1422 				info->minor_err = FLUSH_MW_BIND_ERR;
1423 			qp->sq_flush_seen = true;
1424 			if (!IRDMA_RING_MORE_WORK(qp->sq_ring))
1425 				qp->sq_flush_complete = true;
1426 			spin_unlock_irqrestore(qp->lock, flags);
1427 		}
1428 		pring = &qp->sq_ring;
1429 	}
1430 
1431 	ret_code = 0;
1432 
1433 exit:
1434 	if (!ret_code && info->comp_status == IRDMA_COMPL_STATUS_FLUSHED) {
1435 		if (pring && IRDMA_RING_MORE_WORK(*pring))
1436 			move_cq_head = false;
1437 	}
1438 
1439 	if (move_cq_head) {
1440 		IRDMA_RING_MOVE_HEAD_NOCHECK(cq->cq_ring);
1441 		if (!IRDMA_RING_CURRENT_HEAD(cq->cq_ring))
1442 			cq->polarity ^= 1;
1443 
1444 		if (ext_valid && !cq->avoid_mem_cflct) {
1445 			IRDMA_RING_MOVE_HEAD_NOCHECK(cq->cq_ring);
1446 			if (!IRDMA_RING_CURRENT_HEAD(cq->cq_ring))
1447 				cq->polarity ^= 1;
1448 		}
1449 
1450 		IRDMA_RING_MOVE_TAIL(cq->cq_ring);
1451 		if (!cq->avoid_mem_cflct && ext_valid)
1452 			IRDMA_RING_MOVE_TAIL(cq->cq_ring);
1453 		set_64bit_val(cq->shadow_area, IRDMA_BYTE_0,
1454 			      IRDMA_RING_CURRENT_HEAD(cq->cq_ring));
1455 	} else {
1456 		qword3 &= ~IRDMA_CQ_WQEIDX;
1457 		qword3 |= FIELD_PREP(IRDMA_CQ_WQEIDX, pring->tail);
1458 		set_64bit_val(cqe, IRDMA_BYTE_24, qword3);
1459 	}
1460 
1461 	return ret_code;
1462 }
1463 
1464 /**
1465  * irdma_round_up_wq - return round up qp wq depth
1466  * @wqdepth: wq depth in quanta to round up
1467  */
1468 static int
1469 irdma_round_up_wq(u32 wqdepth)
1470 {
1471 	int scount = 1;
1472 
1473 	for (wqdepth--; scount <= 16; scount *= 2)
1474 		wqdepth |= wqdepth >> scount;
1475 
1476 	return ++wqdepth;
1477 }
1478 
1479 /**
1480  * irdma_get_wqe_shift - get shift count for maximum wqe size
1481  * @uk_attrs: qp HW attributes
1482  * @sge: Maximum Scatter Gather Elements wqe
1483  * @inline_data: Maximum inline data size
1484  * @shift: Returns the shift needed based on sge
1485  *
1486  * Shift can be used to left shift the wqe size based on number of SGEs and inlind data size.
1487  * For 1 SGE or inline data <= 8, shift = 0 (wqe size of 32
1488  * bytes). For 2 or 3 SGEs or inline data <= 39, shift = 1 (wqe
1489  * size of 64 bytes).
1490  * For 4-7 SGE's and inline <= 101 Shift of 2 otherwise (wqe
1491  * size of 256 bytes).
1492  */
1493 void
1494 irdma_get_wqe_shift(struct irdma_uk_attrs *uk_attrs, u32 sge,
1495 		    u32 inline_data, u8 *shift)
1496 {
1497 	*shift = 0;
1498 	if (uk_attrs->hw_rev >= IRDMA_GEN_2) {
1499 		if (sge > 1 || inline_data > 8) {
1500 			if (sge < 4 && inline_data <= 39)
1501 				*shift = 1;
1502 			else if (sge < 8 && inline_data <= 101)
1503 				*shift = 2;
1504 			else
1505 				*shift = 3;
1506 		}
1507 	} else if (sge > 1 || inline_data > 16) {
1508 		*shift = (sge < 4 && inline_data <= 48) ? 1 : 2;
1509 	}
1510 }
1511 
1512 /*
1513  * irdma_get_sqdepth - get SQ depth (quanta) @uk_attrs: qp HW attributes @sq_size: SQ size @shift: shift which
1514  * determines size of WQE @sqdepth: depth of SQ
1515  */
1516 int
1517 irdma_get_sqdepth(struct irdma_uk_attrs *uk_attrs, u32 sq_size, u8 shift, u32 *sqdepth)
1518 {
1519 	*sqdepth = irdma_round_up_wq((sq_size << shift) + IRDMA_SQ_RSVD);
1520 
1521 	if (*sqdepth < ((u32)uk_attrs->min_hw_wq_size << shift))
1522 		*sqdepth = uk_attrs->min_hw_wq_size << shift;
1523 	else if (*sqdepth > uk_attrs->max_hw_wq_quanta)
1524 		return -EINVAL;
1525 
1526 	return 0;
1527 }
1528 
1529 /*
1530  * irdma_get_rqdepth - get RQ depth (quanta) @uk_attrs: qp HW attributes @rq_size: SRQ size @shift: shift which
1531  * determines size of WQE @rqdepth: depth of RQ/SRQ
1532  */
1533 int
1534 irdma_get_rqdepth(struct irdma_uk_attrs *uk_attrs, u32 rq_size, u8 shift, u32 *rqdepth)
1535 {
1536 	*rqdepth = irdma_round_up_wq((rq_size << shift) + IRDMA_RQ_RSVD);
1537 
1538 	if (*rqdepth < ((u32)uk_attrs->min_hw_wq_size << shift))
1539 		*rqdepth = uk_attrs->min_hw_wq_size << shift;
1540 	else if (*rqdepth > uk_attrs->max_hw_rq_quanta)
1541 		return -EINVAL;
1542 
1543 	return 0;
1544 }
1545 
1546 static const struct irdma_wqe_uk_ops iw_wqe_uk_ops = {
1547 	.iw_copy_inline_data = irdma_copy_inline_data,
1548 	.iw_inline_data_size_to_quanta = irdma_inline_data_size_to_quanta,
1549 	.iw_set_fragment = irdma_set_fragment,
1550 	.iw_set_mw_bind_wqe = irdma_set_mw_bind_wqe,
1551 };
1552 
1553 static const struct irdma_wqe_uk_ops iw_wqe_uk_ops_gen_1 = {
1554 	.iw_copy_inline_data = irdma_copy_inline_data_gen_1,
1555 	.iw_inline_data_size_to_quanta = irdma_inline_data_size_to_quanta_gen_1,
1556 	.iw_set_fragment = irdma_set_fragment_gen_1,
1557 	.iw_set_mw_bind_wqe = irdma_set_mw_bind_wqe_gen_1,
1558 };
1559 
1560 /**
1561  * irdma_setup_connection_wqes - setup WQEs necessary to complete
1562  * connection.
1563  * @qp: hw qp (user and kernel)
1564  * @info: qp initialization info
1565  */
1566 static void
1567 irdma_setup_connection_wqes(struct irdma_qp_uk *qp,
1568 			    struct irdma_qp_uk_init_info *info)
1569 {
1570 	u16 move_cnt = 1;
1571 
1572 	if (qp->uk_attrs->feature_flags & IRDMA_FEATURE_RTS_AE)
1573 		move_cnt = 3;
1574 
1575 	qp->conn_wqes = move_cnt;
1576 	IRDMA_RING_MOVE_HEAD_BY_COUNT_NOCHECK(qp->sq_ring, move_cnt);
1577 	IRDMA_RING_MOVE_TAIL_BY_COUNT(qp->sq_ring, move_cnt);
1578 	IRDMA_RING_MOVE_HEAD_BY_COUNT_NOCHECK(qp->initial_ring, move_cnt);
1579 }
1580 
1581 /**
1582  * irdma_uk_calc_shift_wq - calculate WQE shift for both SQ and RQ
1583  * @ukinfo: qp initialization info
1584  * @sq_shift: Returns shift of SQ
1585  * @rq_shift: Returns shift of RQ
1586  */
1587 void
1588 irdma_uk_calc_shift_wq(struct irdma_qp_uk_init_info *ukinfo, u8 *sq_shift,
1589 		       u8 *rq_shift)
1590 {
1591 	bool imm_support = ukinfo->uk_attrs->hw_rev >= IRDMA_GEN_2 ? true : false;
1592 
1593 	irdma_get_wqe_shift(ukinfo->uk_attrs,
1594 			    imm_support ? ukinfo->max_sq_frag_cnt + 1 :
1595 			    ukinfo->max_sq_frag_cnt,
1596 			    ukinfo->max_inline_data, sq_shift);
1597 
1598 	irdma_get_wqe_shift(ukinfo->uk_attrs, ukinfo->max_rq_frag_cnt, 0,
1599 			    rq_shift);
1600 
1601 	if (ukinfo->uk_attrs->hw_rev == IRDMA_GEN_1) {
1602 		if (ukinfo->abi_ver > 4)
1603 			*rq_shift = IRDMA_MAX_RQ_WQE_SHIFT_GEN1;
1604 	}
1605 }
1606 
1607 /**
1608  * irdma_uk_calc_depth_shift_sq - calculate depth and shift for SQ size.
1609  * @ukinfo: qp initialization info
1610  * @sq_depth: Returns depth of SQ
1611  * @sq_shift: Returns shift of SQ
1612  */
1613 int
1614 irdma_uk_calc_depth_shift_sq(struct irdma_qp_uk_init_info *ukinfo,
1615 			     u32 *sq_depth, u8 *sq_shift)
1616 {
1617 	bool imm_support = ukinfo->uk_attrs->hw_rev >= IRDMA_GEN_2 ? true : false;
1618 	int status;
1619 	irdma_get_wqe_shift(ukinfo->uk_attrs,
1620 			    imm_support ? ukinfo->max_sq_frag_cnt + 1 :
1621 			    ukinfo->max_sq_frag_cnt,
1622 			    ukinfo->max_inline_data, sq_shift);
1623 	status = irdma_get_sqdepth(ukinfo->uk_attrs, ukinfo->sq_size,
1624 				   *sq_shift, sq_depth);
1625 
1626 	return status;
1627 }
1628 
1629 /**
1630  * irdma_uk_calc_depth_shift_rq - calculate depth and shift for RQ size.
1631  * @ukinfo: qp initialization info
1632  * @rq_depth: Returns depth of RQ
1633  * @rq_shift: Returns shift of RQ
1634  */
1635 int
1636 irdma_uk_calc_depth_shift_rq(struct irdma_qp_uk_init_info *ukinfo,
1637 			     u32 *rq_depth, u8 *rq_shift)
1638 {
1639 	int status;
1640 
1641 	irdma_get_wqe_shift(ukinfo->uk_attrs, ukinfo->max_rq_frag_cnt, 0,
1642 			    rq_shift);
1643 
1644 	if (ukinfo->uk_attrs->hw_rev == IRDMA_GEN_1) {
1645 		if (ukinfo->abi_ver > 4)
1646 			*rq_shift = IRDMA_MAX_RQ_WQE_SHIFT_GEN1;
1647 	}
1648 
1649 	status = irdma_get_rqdepth(ukinfo->uk_attrs, ukinfo->rq_size,
1650 				   *rq_shift, rq_depth);
1651 
1652 	return status;
1653 }
1654 
1655 /**
1656  * irdma_uk_qp_init - initialize shared qp
1657  * @qp: hw qp (user and kernel)
1658  * @info: qp initialization info
1659  *
1660  * initializes the vars used in both user and kernel mode.
1661  * size of the wqe depends on numbers of max. fragements
1662  * allowed. Then size of wqe * the number of wqes should be the
1663  * amount of memory allocated for sq and rq.
1664  */
1665 int
1666 irdma_uk_qp_init(struct irdma_qp_uk *qp, struct irdma_qp_uk_init_info *info)
1667 {
1668 	int ret_code = 0;
1669 	u32 sq_ring_size;
1670 
1671 	qp->uk_attrs = info->uk_attrs;
1672 	if (info->max_sq_frag_cnt > qp->uk_attrs->max_hw_wq_frags ||
1673 	    info->max_rq_frag_cnt > qp->uk_attrs->max_hw_wq_frags)
1674 		return -EINVAL;
1675 
1676 	qp->qp_caps = info->qp_caps;
1677 	qp->sq_base = info->sq;
1678 	qp->rq_base = info->rq;
1679 	qp->qp_type = info->type ? info->type : IRDMA_QP_TYPE_IWARP;
1680 	qp->shadow_area = info->shadow_area;
1681 	qp->sq_wrtrk_array = info->sq_wrtrk_array;
1682 
1683 	qp->rq_wrid_array = info->rq_wrid_array;
1684 	qp->wqe_alloc_db = info->wqe_alloc_db;
1685 	qp->last_rx_cmpl_idx = 0xffffffff;
1686 	qp->rd_fence_rate = info->rd_fence_rate;
1687 	qp->qp_id = info->qp_id;
1688 	qp->sq_size = info->sq_size;
1689 	qp->push_mode = false;
1690 	qp->max_sq_frag_cnt = info->max_sq_frag_cnt;
1691 	sq_ring_size = qp->sq_size << info->sq_shift;
1692 	IRDMA_RING_INIT(qp->sq_ring, sq_ring_size);
1693 	IRDMA_RING_INIT(qp->initial_ring, sq_ring_size);
1694 	if (info->first_sq_wq) {
1695 		irdma_setup_connection_wqes(qp, info);
1696 		qp->swqe_polarity = 1;
1697 		qp->first_sq_wq = true;
1698 	} else {
1699 		qp->swqe_polarity = 0;
1700 	}
1701 	qp->swqe_polarity_deferred = 1;
1702 	qp->rwqe_polarity = 0;
1703 	qp->rq_size = info->rq_size;
1704 	qp->max_rq_frag_cnt = info->max_rq_frag_cnt;
1705 	qp->max_inline_data = info->max_inline_data;
1706 	qp->rq_wqe_size = info->rq_shift;
1707 	IRDMA_RING_INIT(qp->rq_ring, qp->rq_size);
1708 	qp->rq_wqe_size_multiplier = 1 << info->rq_shift;
1709 	if (qp->uk_attrs->hw_rev == IRDMA_GEN_1)
1710 		qp->wqe_ops = iw_wqe_uk_ops_gen_1;
1711 	else
1712 		qp->wqe_ops = iw_wqe_uk_ops;
1713 	return ret_code;
1714 }
1715 
1716 /**
1717  * irdma_uk_cq_init - initialize shared cq (user and kernel)
1718  * @cq: hw cq
1719  * @info: hw cq initialization info
1720  */
1721 int
1722 irdma_uk_cq_init(struct irdma_cq_uk *cq, struct irdma_cq_uk_init_info *info)
1723 {
1724 	cq->cq_base = info->cq_base;
1725 	cq->cq_id = info->cq_id;
1726 	cq->cq_size = info->cq_size;
1727 	cq->cqe_alloc_db = info->cqe_alloc_db;
1728 	cq->cq_ack_db = info->cq_ack_db;
1729 	cq->shadow_area = info->shadow_area;
1730 	cq->avoid_mem_cflct = info->avoid_mem_cflct;
1731 	IRDMA_RING_INIT(cq->cq_ring, cq->cq_size);
1732 	cq->polarity = 1;
1733 
1734 	return 0;
1735 }
1736 
1737 /**
1738  * irdma_uk_clean_cq - clean cq entries
1739  * @q: completion context
1740  * @cq: cq to clean
1741  */
1742 int
1743 irdma_uk_clean_cq(void *q, struct irdma_cq_uk *cq)
1744 {
1745 	__le64 *cqe;
1746 	u64 qword3, comp_ctx;
1747 	u32 cq_head;
1748 	u8 polarity, temp;
1749 
1750 	cq_head = cq->cq_ring.head;
1751 	temp = cq->polarity;
1752 	do {
1753 		if (cq->avoid_mem_cflct)
1754 			cqe = ((struct irdma_extended_cqe *)(cq->cq_base))[cq_head].buf;
1755 		else
1756 			cqe = cq->cq_base[cq_head].buf;
1757 		get_64bit_val(cqe, IRDMA_BYTE_24, &qword3);
1758 		polarity = (u8)FIELD_GET(IRDMA_CQ_VALID, qword3);
1759 
1760 		if (polarity != temp)
1761 			break;
1762 
1763 		get_64bit_val(cqe, IRDMA_BYTE_8, &comp_ctx);
1764 		if ((void *)(irdma_uintptr) comp_ctx == q)
1765 			set_64bit_val(cqe, IRDMA_BYTE_8, 0);
1766 
1767 		cq_head = (cq_head + 1) % cq->cq_ring.size;
1768 		if (!cq_head)
1769 			temp ^= 1;
1770 	} while (true);
1771 	return 0;
1772 }
1773 
1774 /**
1775  * irdma_nop - post a nop
1776  * @qp: hw qp ptr
1777  * @wr_id: work request id
1778  * @signaled: signaled for completion
1779  * @post_sq: ring doorbell
1780  */
1781 int
1782 irdma_nop(struct irdma_qp_uk *qp, u64 wr_id, bool signaled, bool post_sq)
1783 {
1784 	__le64 *wqe;
1785 	u64 hdr;
1786 	u32 wqe_idx;
1787 	struct irdma_post_sq_info info = {0};
1788 	u16 quanta = IRDMA_QP_WQE_MIN_QUANTA;
1789 
1790 	info.push_wqe = qp->push_db ? true : false;
1791 	info.wr_id = wr_id;
1792 	wqe = irdma_qp_get_next_send_wqe(qp, &wqe_idx, &quanta, 0, &info);
1793 	if (!wqe)
1794 		return -ENOSPC;
1795 
1796 	set_64bit_val(wqe, IRDMA_BYTE_0, 0);
1797 	set_64bit_val(wqe, IRDMA_BYTE_8, 0);
1798 	set_64bit_val(wqe, IRDMA_BYTE_16, 0);
1799 
1800 	hdr = FIELD_PREP(IRDMAQPSQ_OPCODE, IRDMAQP_OP_NOP) |
1801 	    FIELD_PREP(IRDMAQPSQ_SIGCOMPL, signaled) |
1802 	    FIELD_PREP(IRDMAQPSQ_VALID, qp->swqe_polarity);
1803 
1804 	irdma_wmb();		/* make sure WQE is populated before valid bit is set */
1805 
1806 	set_64bit_val(wqe, IRDMA_BYTE_24, hdr);
1807 
1808 	if (info.push_wqe)
1809 		irdma_qp_push_wqe(qp, wqe, quanta, wqe_idx, post_sq);
1810 	else if (post_sq)
1811 		irdma_uk_qp_post_wr(qp);
1812 
1813 	return 0;
1814 }
1815 
1816 /**
1817  * irdma_fragcnt_to_quanta_sq - calculate quanta based on fragment count for SQ
1818  * @frag_cnt: number of fragments
1819  * @quanta: quanta for frag_cnt
1820  */
1821 int
1822 irdma_fragcnt_to_quanta_sq(u32 frag_cnt, u16 *quanta)
1823 {
1824 	switch (frag_cnt) {
1825 	case 0:
1826 	case 1:
1827 		*quanta = IRDMA_QP_WQE_MIN_QUANTA;
1828 		break;
1829 	case 2:
1830 	case 3:
1831 		*quanta = 2;
1832 		break;
1833 	case 4:
1834 	case 5:
1835 		*quanta = 3;
1836 		break;
1837 	case 6:
1838 	case 7:
1839 		*quanta = 4;
1840 		break;
1841 	case 8:
1842 	case 9:
1843 		*quanta = 5;
1844 		break;
1845 	case 10:
1846 	case 11:
1847 		*quanta = 6;
1848 		break;
1849 	case 12:
1850 	case 13:
1851 		*quanta = 7;
1852 		break;
1853 	case 14:
1854 	case 15:		/* when immediate data is present */
1855 		*quanta = 8;
1856 		break;
1857 	default:
1858 		return -EINVAL;
1859 	}
1860 
1861 	return 0;
1862 }
1863 
1864 /**
1865  * irdma_fragcnt_to_wqesize_rq - calculate wqe size based on fragment count for RQ
1866  * @frag_cnt: number of fragments
1867  * @wqe_size: size in bytes given frag_cnt
1868  */
1869 int
1870 irdma_fragcnt_to_wqesize_rq(u32 frag_cnt, u16 *wqe_size)
1871 {
1872 	switch (frag_cnt) {
1873 	case 0:
1874 	case 1:
1875 		*wqe_size = 32;
1876 		break;
1877 	case 2:
1878 	case 3:
1879 		*wqe_size = 64;
1880 		break;
1881 	case 4:
1882 	case 5:
1883 	case 6:
1884 	case 7:
1885 		*wqe_size = 128;
1886 		break;
1887 	case 8:
1888 	case 9:
1889 	case 10:
1890 	case 11:
1891 	case 12:
1892 	case 13:
1893 	case 14:
1894 		*wqe_size = 256;
1895 		break;
1896 	default:
1897 		return -EINVAL;
1898 	}
1899 
1900 	return 0;
1901 }
1902