bool compound = flags & RMAP_COMPOUND;
        bool first = true;
 
-       if (unlikely(PageKsm(page)))
-               lock_page_memcg(page);
-
        /* Is page being mapped by PTE? Is this its first map to be added? */
        if (likely(!compound)) {
                first = atomic_inc_and_test(&page->_mapcount);
        if (nr)
                __mod_lruvec_page_state(page, NR_ANON_MAPPED, nr);
 
-       if (unlikely(PageKsm(page)))
-               unlock_page_memcg(page);
-
-       /* address might be in next vma when migration races vma_adjust */
-       else if (first)
-               __page_set_anon_rmap(page, vma, address,
-                                    !!(flags & RMAP_EXCLUSIVE));
-       else
-               __page_check_anon_rmap(page, vma, address);
+       if (likely(!PageKsm(page))) {
+               /* address might be in next vma when migration races vma_adjust */
+               if (first)
+                       __page_set_anon_rmap(page, vma, address,
+                                            !!(flags & RMAP_EXCLUSIVE));
+               else
+                       __page_check_anon_rmap(page, vma, address);
+       }
 
        mlock_vma_page(page, vma, compound);
 }
        bool first;
 
        VM_BUG_ON_PAGE(compound && !PageTransHuge(page), page);
-       lock_page_memcg(page);
 
        /* Is page being mapped by PTE? Is this its first map to be added? */
        if (likely(!compound)) {
                        NR_SHMEM_PMDMAPPED : NR_FILE_PMDMAPPED, nr_pmdmapped);
        if (nr)
                __mod_lruvec_page_state(page, NR_FILE_MAPPED, nr);
-       unlock_page_memcg(page);
 
        mlock_vma_page(page, vma, compound);
 }
                return;
        }
 
-       lock_page_memcg(page);
-
        /* Is page being unmapped by PTE? Is this its last map to be removed? */
        if (likely(!compound)) {
                last = atomic_add_negative(-1, &page->_mapcount);
         * and remember that it's only reliable while mapped.
         */
 
-       unlock_page_memcg(page);
-
        munlock_vma_page(page, vma, compound);
 }