1 files changed, 5 insertions, 4 deletions

diff --git a/arch/sparc/mm/tlb.c b/arch/sparc/mm/tlb.c
index 97d1e56e9863..f1bd83019e71 100644
--- a/arch/sparc/mm/tlb.c
+++ b/arch/sparc/mm/tlb.c
@@ -161,8 +161,8 @@ void set_pmd_at(struct mm_struct *mm, unsigned long addr,
         if (mm == &init_mm)
                 return;
 
-        if ((pmd_val(pmd) ^ pmd_val(orig)) & PMD_ISHUGE) {
-                if (pmd_val(pmd) & PMD_ISHUGE)
+        if ((pmd_val(pmd) ^ pmd_val(orig)) & _PAGE_PMD_HUGE) {
+                if (pmd_val(pmd) & _PAGE_PMD_HUGE)
                         mm->context.huge_pte_count++;
                 else
                         mm->context.huge_pte_count--;
@@ -178,10 +178,11 @@ void set_pmd_at(struct mm_struct *mm, unsigned long addr,
         }
 
         if (!pmd_none(orig)) {
-                bool exec = ((pmd_val(orig) & PMD_HUGE_EXEC) != 0);
+                pte_t orig_pte = __pte(pmd_val(orig));
+                bool exec = pte_exec(orig_pte);
 
                 addr &= HPAGE_MASK;
-                if (pmd_val(orig) & PMD_ISHUGE) {
+                if (pmd_trans_huge(orig)) {
                         tlb_batch_add_one(mm, addr, exec);
                         tlb_batch_add_one(mm, addr + REAL_HPAGE_SIZE, exec);
                 } else {