mm: hugetlbfs: use __compound_tail_refcounted in __get_page_tail too

[cascardo/linux.git] / mm / swap.c

diff --git a/mm/swap.c b/mm/swap.c
index bba4aa5..7434e36 100644 (file)
--- a/mm/swap.c
+++ b/mm/swap.c
@@ -31,7 +31,6 @@
 #include <linux/memcontrol.h>
 #include <linux/gfp.h>
 #include <linux/uio.h>
-#include <linux/hugetlb.h>
 
 #include "internal.h"
 
@@ -261,7 +260,7 @@ bool __get_page_tail(struct page *page)
        struct page *page_head = compound_trans_head(page);
 
        /* Ref to put_compound_page() comment. */
-       if (PageSlab(page_head) || PageHeadHuge(page_head)) {
+       if (!__compound_tail_refcounted(page_head)) {
                smp_rmb();
                if (likely(PageTail(page))) {
                        /*