1 files changed, 4 insertions, 2 deletions

diff --git a/mm/rmap.c b/mm/rmap.c
index dbc2ca2057a5..57ad276900c9 100644
--- a/mm/rmap.c
+++ b/mm/rmap.c
@@ -36,7 +36,6 @@
  *                 mapping->tree_lock (widely used, in set_page_dirty,
  *                           in arch-dependent flush_dcache_mmap_lock,
  *                           within inode_lock in __sync_single_inode)
- *                   zone->lock (within radix tree node alloc)
  */
 
 #include <linux/mm.h>
@@ -284,7 +283,10 @@ static int page_referenced_one(struct page *page,
         if (!pte)
                 goto out;
 
-        if (ptep_clear_flush_young(vma, address, pte))
+        if (vma->vm_flags & VM_LOCKED) {
+                referenced++;
+                *mapcount = 1;  /* break early from loop */
+        } else if (ptep_clear_flush_young(vma, address, pte))
                 referenced++;
 
         /* Pretend the page is referenced if the task has the