42 static void *get_wqe(struct mlx4_ib_srq *srq, int n)  in get_wqe()  argument
44 	return mlx4_buf_offset(&srq->buf, n << srq->msrq.wqe_shift);  in get_wqe()
47 static void mlx4_ib_srq_event(struct mlx4_srq *srq, enum mlx4_event type)  in mlx4_ib_srq_event()  argument
50 	struct ib_srq *ibsrq = &to_mibsrq(srq)->ibsrq;  in mlx4_ib_srq_event()
54 		event.element.srq = ibsrq;  in mlx4_ib_srq_event()
64 			       "on SRQ %06x\n", type, srq->srqn);  in mlx4_ib_srq_event()
77 	struct mlx4_ib_srq *srq;  in mlx4_ib_create_srq()  local
92 	srq = kmalloc(sizeof *srq, GFP_KERNEL);  in mlx4_ib_create_srq()
93 	if (!srq)  in mlx4_ib_create_srq()
96 	mutex_init(&srq->mutex);  in mlx4_ib_create_srq()
97 	spin_lock_init(&srq->lock);  in mlx4_ib_create_srq()
98 	srq->msrq.max    = roundup_pow_of_two(init_attr->attr.max_wr + 1);  in mlx4_ib_create_srq()
99 	srq->msrq.max_gs = init_attr->attr.max_sge;  in mlx4_ib_create_srq()
103 					   srq->msrq.max_gs *  in mlx4_ib_create_srq()
105 	srq->msrq.wqe_shift = ilog2(desc_size);  in mlx4_ib_create_srq()
107 	buf_size = srq->msrq.max * desc_size;  in mlx4_ib_create_srq()
117 		srq->umem = ib_umem_get(pd->uobject->context, ucmd.buf_addr,  in mlx4_ib_create_srq()
119 		if (IS_ERR(srq->umem)) {  in mlx4_ib_create_srq()
120 			err = PTR_ERR(srq->umem);  in mlx4_ib_create_srq()
124 		err = mlx4_mtt_init(dev->dev, ib_umem_page_count(srq->umem),  in mlx4_ib_create_srq()
125 				    ilog2(srq->umem->page_size), &srq->mtt);  in mlx4_ib_create_srq()
129 		err = mlx4_ib_umem_write_mtt(dev, &srq->mtt, srq->umem);  in mlx4_ib_create_srq()
134 					  ucmd.db_addr, &srq->db);  in mlx4_ib_create_srq()
138 		err = mlx4_db_alloc(dev->dev, &srq->db, 0, GFP_KERNEL);  in mlx4_ib_create_srq()
142 		*srq->db.db = 0;  in mlx4_ib_create_srq()
144 		if (mlx4_buf_alloc(dev->dev, buf_size, PAGE_SIZE * 2, &srq->buf,  in mlx4_ib_create_srq()
150 		srq->head    = 0;  in mlx4_ib_create_srq()
151 		srq->tail    = srq->msrq.max - 1;  in mlx4_ib_create_srq()
152 		srq->wqe_ctr = 0;  in mlx4_ib_create_srq()
154 		for (i = 0; i < srq->msrq.max; ++i) {  in mlx4_ib_create_srq()
155 			next = get_wqe(srq, i);  in mlx4_ib_create_srq()
157 				cpu_to_be16((i + 1) & (srq->msrq.max - 1));  in mlx4_ib_create_srq()
165 		err = mlx4_mtt_init(dev->dev, srq->buf.npages, srq->buf.page_shift,  in mlx4_ib_create_srq()
166 				    &srq->mtt);  in mlx4_ib_create_srq()
170 		err = mlx4_buf_write_mtt(dev->dev, &srq->mtt, &srq->buf, GFP_KERNEL);  in mlx4_ib_create_srq()
174 		srq->wrid = kmalloc(srq->msrq.max * sizeof (u64), GFP_KERNEL);  in mlx4_ib_create_srq()
175 		if (!srq->wrid) {  in mlx4_ib_create_srq()
176 			srq->wrid = __vmalloc(srq->msrq.max * sizeof(u64),  in mlx4_ib_create_srq()
178 			if (!srq->wrid) {  in mlx4_ib_create_srq()
190 	err = mlx4_srq_alloc(dev->dev, to_mpd(pd)->pdn, cqn, xrcdn, &srq->mtt,  in mlx4_ib_create_srq()
191 			     srq->db.dma, &srq->msrq);  in mlx4_ib_create_srq()
195 	srq->msrq.event = mlx4_ib_srq_event;  in mlx4_ib_create_srq()
196 	srq->ibsrq.ext.xrc.srq_num = srq->msrq.srqn;  in mlx4_ib_create_srq()
199 		if (ib_copy_to_udata(udata, &srq->msrq.srqn, sizeof (__u32))) {  in mlx4_ib_create_srq()
204 	init_attr->attr.max_wr = srq->msrq.max - 1;  in mlx4_ib_create_srq()
206 	return &srq->ibsrq;  in mlx4_ib_create_srq()
210 		mlx4_ib_db_unmap_user(to_mucontext(pd->uobject->context), &srq->db);  in mlx4_ib_create_srq()
212 		kvfree(srq->wrid);  in mlx4_ib_create_srq()
215 	mlx4_mtt_cleanup(dev->dev, &srq->mtt);  in mlx4_ib_create_srq()
219 		ib_umem_release(srq->umem);  in mlx4_ib_create_srq()
221 		mlx4_buf_free(dev->dev, buf_size, &srq->buf);  in mlx4_ib_create_srq()
225 		mlx4_db_free(dev->dev, &srq->db);  in mlx4_ib_create_srq()
228 	kfree(srq);  in mlx4_ib_create_srq()
237 	struct mlx4_ib_srq *srq = to_msrq(ibsrq);  in mlx4_ib_modify_srq()  local
245 		if (attr->srq_limit >= srq->msrq.max)  in mlx4_ib_modify_srq()
248 		mutex_lock(&srq->mutex);  in mlx4_ib_modify_srq()
249 		ret = mlx4_srq_arm(dev->dev, &srq->msrq, attr->srq_limit);  in mlx4_ib_modify_srq()
250 		mutex_unlock(&srq->mutex);  in mlx4_ib_modify_srq()
262 	struct mlx4_ib_srq *srq = to_msrq(ibsrq);  in mlx4_ib_query_srq()  local
266 	ret = mlx4_srq_query(dev->dev, &srq->msrq, &limit_watermark);  in mlx4_ib_query_srq()
271 	srq_attr->max_wr    = srq->msrq.max - 1;  in mlx4_ib_query_srq()
272 	srq_attr->max_sge   = srq->msrq.max_gs;  in mlx4_ib_query_srq()
277 int mlx4_ib_destroy_srq(struct ib_srq *srq)  in mlx4_ib_destroy_srq()  argument
279 	struct mlx4_ib_dev *dev = to_mdev(srq->device);  in mlx4_ib_destroy_srq()
280 	struct mlx4_ib_srq *msrq = to_msrq(srq);  in mlx4_ib_destroy_srq()
285 	if (srq->uobject) {  in mlx4_ib_destroy_srq()
286 		mlx4_ib_db_unmap_user(to_mucontext(srq->uobject->context), &msrq->db);  in mlx4_ib_destroy_srq()
300 void mlx4_ib_free_srq_wqe(struct mlx4_ib_srq *srq, int wqe_index)  in mlx4_ib_free_srq_wqe()  argument
305 	spin_lock(&srq->lock);  in mlx4_ib_free_srq_wqe()
307 	next = get_wqe(srq, srq->tail);  in mlx4_ib_free_srq_wqe()
309 	srq->tail = wqe_index;  in mlx4_ib_free_srq_wqe()
311 	spin_unlock(&srq->lock);  in mlx4_ib_free_srq_wqe()
317 	struct mlx4_ib_srq *srq = to_msrq(ibsrq);  in mlx4_ib_post_srq_recv()  local
326 	spin_lock_irqsave(&srq->lock, flags);  in mlx4_ib_post_srq_recv()
335 		if (unlikely(wr->num_sge > srq->msrq.max_gs)) {  in mlx4_ib_post_srq_recv()
341 		if (unlikely(srq->head == srq->tail)) {  in mlx4_ib_post_srq_recv()
347 		srq->wrid[srq->head] = wr->wr_id;  in mlx4_ib_post_srq_recv()
349 		next      = get_wqe(srq, srq->head);  in mlx4_ib_post_srq_recv()
350 		srq->head = be16_to_cpu(next->next_wqe_index);  in mlx4_ib_post_srq_recv()
359 		if (i < srq->msrq.max_gs) {  in mlx4_ib_post_srq_recv()
367 		srq->wqe_ctr += nreq;  in mlx4_ib_post_srq_recv()
375 		*srq->db.db = cpu_to_be32(srq->wqe_ctr);  in mlx4_ib_post_srq_recv()
379 	spin_unlock_irqrestore(&srq->lock, flags);  in mlx4_ib_post_srq_recv()