1 files changed, 6 insertions, 22 deletions

diff --git a/drivers/infiniband/hw/mthca/mthca_srq.c b/drivers/infiniband/hw/mthca/mthca_srq.c
index b8f05a526673..88d219e730ad 100644
--- a/drivers/infiniband/hw/mthca/mthca_srq.c
+++ b/drivers/infiniband/hw/mthca/mthca_srq.c
@@ -543,20 +543,12 @@ int mthca_tavor_post_srq_recv(struct ib_srq *ibsrq, struct ib_recv_wr *wr,
                 }
 
                 for (i = 0; i < wr->num_sge; ++i) {
-                        ((struct mthca_data_seg *) wqe)->byte_count =
-                                cpu_to_be32(wr->sg_list[i].length);
-                        ((struct mthca_data_seg *) wqe)->lkey =
-                                cpu_to_be32(wr->sg_list[i].lkey);
-                        ((struct mthca_data_seg *) wqe)->addr =
-                                cpu_to_be64(wr->sg_list[i].addr);
+                        mthca_set_data_seg(wqe, wr->sg_list + i);
                         wqe += sizeof (struct mthca_data_seg);
                 }
 
-                if (i < srq->max_gs) {
-                        ((struct mthca_data_seg *) wqe)->byte_count = 0;
-                        ((struct mthca_data_seg *) wqe)->lkey = cpu_to_be32(MTHCA_INVAL_LKEY);
-                        ((struct mthca_data_seg *) wqe)->addr = 0;
-                }
+                if (i < srq->max_gs)
+                        mthca_set_data_seg_inval(wqe);
 
                 ((struct mthca_next_seg *) prev_wqe)->nda_op =
                         cpu_to_be32((ind << srq->wqe_shift) | 1);
@@ -662,20 +654,12 @@ int mthca_arbel_post_srq_recv(struct ib_srq *ibsrq, struct ib_recv_wr *wr,
                 }
 
                 for (i = 0; i < wr->num_sge; ++i) {
-                        ((struct mthca_data_seg *) wqe)->byte_count =
-                                cpu_to_be32(wr->sg_list[i].length);
-                        ((struct mthca_data_seg *) wqe)->lkey =
-                                cpu_to_be32(wr->sg_list[i].lkey);
-                        ((struct mthca_data_seg *) wqe)->addr =
-                                cpu_to_be64(wr->sg_list[i].addr);
+                        mthca_set_data_seg(wqe, wr->sg_list + i);
                         wqe += sizeof (struct mthca_data_seg);
                 }
 
-                if (i < srq->max_gs) {
-                        ((struct mthca_data_seg *) wqe)->byte_count = 0;
-                        ((struct mthca_data_seg *) wqe)->lkey = cpu_to_be32(MTHCA_INVAL_LKEY);
-                        ((struct mthca_data_seg *) wqe)->addr = 0;
-                }
+                if (i < srq->max_gs)
+                        mthca_set_data_seg_inval(wqe);
 
                 srq->wrid[ind]  = wr->wr_id;
                 srq->first_free = next_ind;