1 files changed, 2 insertions, 10 deletions

diff --git a/arch/x86/mm/gup.c b/arch/x86/mm/gup.c
index dbe34b93137..dd74e46828c 100644
--- a/arch/x86/mm/gup.c
+++ b/arch/x86/mm/gup.c
@@ -108,16 +108,6 @@ static inline void get_head_page_multiple(struct page *page, int nr)
         SetPageReferenced(page);
 }
 
-static inline void get_huge_page_tail(struct page *page)
-{
-        /*
-         * __split_huge_page_refcount() cannot run
-         * from under us.
-         */
-        VM_BUG_ON(atomic_read(&page->_count) < 0);
-        atomic_inc(&page->_count);
-}
-
 static noinline int gup_huge_pmd(pmd_t pmd, unsigned long addr,
                 unsigned long end, int write, struct page **pages, int *nr)
 {
@@ -211,6 +201,8 @@ static noinline int gup_huge_pud(pud_t pud, unsigned long addr,
         do {
                 VM_BUG_ON(compound_head(page) != head);
                 pages[*nr] = page;
+                if (PageTail(page))
+                        get_huge_page_tail(page);
                 (*nr)++;
                 page++;
                 refs++;