hw/mlx5/srq.c

11 #include "srq.h"
13 static void *get_wqe(struct mlx5_ib_srq *srq, int n)
15 	return mlx5_frag_buf_get_wqe(&srq->fbc, n);
18 static void mlx5_ib_srq_event(struct mlx5_core_srq *srq, enum mlx5_event type)
21 	struct ib_srq *ibsrq = &to_mibsrq(srq)->ibsrq;
25 		event.element.srq = ibsrq;
35 				type, srq->srqn);
43 static int create_srq_user(struct ib_pd *pd, struct mlx5_ib_srq *srq,
76 	srq->wq_sig = !!(ucmd.flags & MLX5_SRQ_FLAG_SIGNATURE);
78 	srq->umem = ib_umem_get(pd->device, ucmd.buf_addr, buf_size, 0);
79 	if (IS_ERR(srq->umem)) {
81 		err = PTR_ERR(srq->umem);
84 	in->umem = srq->umem;
86 	err = mlx5_ib_db_map_user(ucontext, ucmd.db_addr, &srq->db);
100 	ib_umem_release(srq->umem);
105 static int create_srq_kernel(struct mlx5_ib_dev *dev, struct mlx5_ib_srq *srq,
112 	err = mlx5_db_alloc(dev->mdev, &srq->db);
118 	if (mlx5_frag_buf_alloc_node(dev->mdev, buf_size, &srq->buf,
125 	mlx5_init_fbc(srq->buf.frags, srq->msrq.wqe_shift, ilog2(srq->msrq.max),
126 		      &srq->fbc);
128 	srq->head    = 0;
129 	srq->tail    = srq->msrq.max - 1;
130 	srq->wqe_ctr = 0;
132 	for (i = 0; i < srq->msrq.max; i++) {
133 		next = get_wqe(srq, i);
135 			cpu_to_be16((i + 1) & (srq->msrq.max - 1));
138 	mlx5_ib_dbg(dev, "srq->buf.page_shift = %d\n", srq->buf.page_shift);
139 	in->pas = kvcalloc(srq->buf.npages, sizeof(*in->pas), GFP_KERNEL);
144 	mlx5_fill_page_frag_array(&srq->buf, in->pas);
146 	srq->wrid = kvmalloc_array(srq->msrq.max, sizeof(u64), GFP_KERNEL);
147 	if (!srq->wrid) {
151 	srq->wq_sig = 0;
153 	in->log_page_size = srq->buf.page_shift - MLX5_ADAPTER_PAGE_SHIFT;
164 	mlx5_frag_buf_free(dev->mdev, &srq->buf);
167 	mlx5_db_free(dev->mdev, &srq->db);
171 static void destroy_srq_user(struct ib_pd *pd, struct mlx5_ib_srq *srq,
179 		&srq->db);
180 	ib_umem_release(srq->umem);
184 static void destroy_srq_kernel(struct mlx5_ib_dev *dev, struct mlx5_ib_srq *srq)
186 	kvfree(srq->wrid);
187 	mlx5_frag_buf_free(dev->mdev, &srq->buf);
188 	mlx5_db_free(dev->mdev, &srq->db);
196 	struct mlx5_ib_srq *srq = to_msrq(ib_srq);
216 	mutex_init(&srq->mutex);
217 	spin_lock_init(&srq->lock);
218 	srq->msrq.max    = roundup_pow_of_two(init_attr->attr.max_wr + 1);
219 	srq->msrq.max_gs = init_attr->attr.max_sge;
222 		    srq->msrq.max_gs * sizeof(struct mlx5_wqe_data_seg);
223 	if (desc_size == 0 || srq->msrq.max_gs > desc_size)
231 	srq->msrq.max_avail_gather = (desc_size - sizeof(struct mlx5_wqe_srq_next_seg)) /
233 	srq->msrq.wqe_shift = ilog2(desc_size);
234 	buf_size = srq->msrq.max * desc_size;
241 		err = create_srq_user(ib_srq->pd, srq, &in, udata, buf_size);
243 		err = create_srq_kernel(dev, srq, &in, buf_size);
246 		mlx5_ib_warn(dev, "create srq %s failed, err %d\n",
251 	in.log_size = ilog2(srq->msrq.max);
252 	in.wqe_shift = srq->msrq.wqe_shift - 4;
253 	if (srq->wq_sig)
279 	in.db_record = srq->db.dma;
280 	err = mlx5_cmd_create_srq(dev, &srq->msrq, &in);
287 	mlx5_ib_dbg(dev, "create SRQ with srqn 0x%x\n", srq->msrq.srqn);
289 	srq->msrq.event = mlx5_ib_srq_event;
290 	srq->ibsrq.ext.xrc.srq_num = srq->msrq.srqn;
294 			.srqn = srq->msrq.srqn,
305 	init_attr->attr.max_wr = srq->msrq.max - 1;
310 	mlx5_cmd_destroy_srq(dev, &srq->msrq);
314 		destroy_srq_user(ib_srq->pd, srq, udata);
316 		destroy_srq_kernel(dev, srq);
325 	struct mlx5_ib_srq *srq = to_msrq(ibsrq);
333 		if (attr->srq_limit >= srq->msrq.max)
336 		mutex_lock(&srq->mutex);
337 		ret = mlx5_cmd_arm_srq(dev, &srq->msrq, attr->srq_limit, 1);
338 		mutex_unlock(&srq->mutex);
350 	struct mlx5_ib_srq *srq = to_msrq(ibsrq);
358 	ret = mlx5_cmd_query_srq(dev, &srq->msrq, out);
363 	srq_attr->max_wr    = srq->msrq.max - 1;
364 	srq_attr->max_sge   = srq->msrq.max_gs;
371 int mlx5_ib_destroy_srq(struct ib_srq *srq, struct ib_udata *udata)
373 	struct mlx5_ib_dev *dev = to_mdev(srq->device);
374 	struct mlx5_ib_srq *msrq = to_msrq(srq);
382 		destroy_srq_user(srq->pd, msrq, udata);
388 void mlx5_ib_free_srq_wqe(struct mlx5_ib_srq *srq, int wqe_index)
393 	spin_lock(&srq->lock);
395 	next = get_wqe(srq, srq->tail);
397 	srq->tail = wqe_index;
399 	spin_unlock(&srq->lock);
405 	struct mlx5_ib_srq *srq = to_msrq(ibsrq);
415 	spin_lock_irqsave(&srq->lock, flags);
424 		if (unlikely(wr->num_sge > srq->msrq.max_gs)) {
430 		if (unlikely(srq->head == srq->tail)) {
436 		srq->wrid[srq->head] = wr->wr_id;
438 		next      = get_wqe(srq, srq->head);
439 		srq->head = be16_to_cpu(next->next_wqe_index);
448 		if (i < srq->msrq.max_avail_gather) {
456 		srq->wqe_ctr += nreq;
463 		*srq->db.db = cpu_to_be32(srq->wqe_ctr);
466 	spin_unlock_irqrestore(&srq->lock, flags);