1 files changed, 4 insertions, 4 deletions

diff --git a/drivers/infiniband/sw/rdmavt/mr.c b/drivers/infiniband/sw/rdmavt/mr.c
index 562cf78b6ae9..52fd15276ee6 100644
--- a/drivers/infiniband/sw/rdmavt/mr.c
+++ b/drivers/infiniband/sw/rdmavt/mr.c
@@ -793,7 +793,7 @@ int rvt_lkey_ok(struct rvt_lkey_table *rkt, struct rvt_pd *pd,
                 mr = rcu_dereference(dev->dma_mr);
                 if (!mr)
                         goto bail;
-                atomic_inc(&mr->refcount);
+                rvt_get_mr(mr);
                 rcu_read_unlock();
 
                 isge->mr = mr;
@@ -814,7 +814,7 @@ int rvt_lkey_ok(struct rvt_lkey_table *rkt, struct rvt_pd *pd,
                      off + sge->length > mr->length ||
                      (mr->access_flags & acc) != acc))
                 goto bail;
-        atomic_inc(&mr->refcount);
+        rvt_get_mr(mr);
         rcu_read_unlock();
 
         off += mr->offset;
@@ -892,7 +892,7 @@ int rvt_rkey_ok(struct rvt_qp *qp, struct rvt_sge *sge,
                 mr = rcu_dereference(rdi->dma_mr);
                 if (!mr)
                         goto bail;
-                atomic_inc(&mr->refcount);
+                rvt_get_mr(mr);
                 rcu_read_unlock();
 
                 sge->mr = mr;
@@ -913,7 +913,7 @@ int rvt_rkey_ok(struct rvt_qp *qp, struct rvt_sge *sge,
         if (unlikely(vaddr < mr->iova || off + len > mr->length ||
                      (mr->access_flags & acc) == 0))
                 goto bail;
-        atomic_inc(&mr->refcount);
+        rvt_get_mr(mr);
         rcu_read_unlock();
 
         off += mr->offset;