1 files changed, 4 insertions, 5 deletions

diff --git a/drivers/infiniband/core/umem.c b/drivers/infiniband/core/umem.c
index a8411232207..b645e558876 100644
--- a/drivers/infiniband/core/umem.c
+++ b/drivers/infiniband/core/umem.c
@@ -35,7 +35,6 @@
 #include <linux/mm.h>
 #include <linux/dma-mapping.h>
 #include <linux/sched.h>
-#include <linux/export.h>
 #include <linux/hugetlb.h>
 #include <linux/dma-attrs.h>
 #include <linux/slab.h>
@@ -137,7 +136,7 @@ struct ib_umem *ib_umem_get(struct ib_ucontext *context, unsigned long addr,
 
         down_write(&current->mm->mmap_sem);
 
-        locked     = npages + current->mm->pinned_vm;
+        locked     = npages + current->mm->locked_vm;
         lock_limit = rlimit(RLIMIT_MEMLOCK) >> PAGE_SHIFT;
 
         if ((locked > lock_limit) && !capable(CAP_IPC_LOCK)) {
@@ -207,7 +206,7 @@ out:
                 __ib_umem_release(context->device, umem, 0);
                 kfree(umem);
         } else
-                current->mm->pinned_vm = locked;
+                current->mm->locked_vm = locked;
 
         up_write(&current->mm->mmap_sem);
         if (vma_list)
@@ -223,7 +222,7 @@ static void ib_umem_account(struct work_struct *work)
         struct ib_umem *umem = container_of(work, struct ib_umem, work);
 
         down_write(&umem->mm->mmap_sem);
-        umem->mm->pinned_vm -= umem->diff;
+        umem->mm->locked_vm -= umem->diff;
         up_write(&umem->mm->mmap_sem);
         mmput(umem->mm);
         kfree(umem);
@@ -269,7 +268,7 @@ void ib_umem_release(struct ib_umem *umem)
         } else
                 down_write(&mm->mmap_sem);
 
-        current->mm->pinned_vm -= diff;
+        current->mm->locked_vm -= diff;
         up_write(&mm->mmap_sem);
         mmput(mm);
         kfree(umem);