return hugepd_page(hpd) + idx;
 }
 
+static inline void hugepd_populate(hugepd_t *hpdp, pte_t *new, unsigned int pshift)
+{
+       *hpdp = __hugepd(__pa(new) | HUGEPD_VAL_BITS | (shift_to_mmu_psize(pshift) << 2));
+}
+
 void flush_hugetlb_page(struct vm_area_struct *vma, unsigned long vmaddr);
 
 #endif
 
 #ifndef _ASM_POWERPC_NOHASH_32_HUGETLB_8XX_H
 #define _ASM_POWERPC_NOHASH_32_HUGETLB_8XX_H
 
+#define PAGE_SHIFT_8M          23
+
 static inline pte_t *hugepd_page(hugepd_t hpd)
 {
        BUG_ON(!hugepd_ok(hpd));
        flush_tlb_page(vma, vmaddr);
 }
 
+static inline void hugepd_populate(hugepd_t *hpdp, pte_t *new, unsigned int pshift)
+{
+       *hpdp = __hugepd(__pa(new) | _PMD_USER | _PMD_PRESENT |
+                        (pshift == PAGE_SHIFT_8M ? _PMD_PAGE_8M : _PMD_PAGE_512K));
+}
+
 #endif /* _ASM_POWERPC_NOHASH_32_HUGETLB_8XX_H */
 
 
 void flush_hugetlb_page(struct vm_area_struct *vma, unsigned long vmaddr);
 
+static inline void hugepd_populate(hugepd_t *hpdp, pte_t *new, unsigned int pshift)
+{
+       /* We use the old format for PPC_FSL_BOOK3E */
+       *hpdp = __hugepd(((unsigned long)new & ~PD_HUGE) | pshift);
+}
+
 #endif /* _ASM_POWERPC_NOHASH_HUGETLB_BOOK3E_H */
 
 #include <asm/hugetlb.h>
 #include <asm/pte-walk.h>
 
-#define PAGE_SHIFT_64K 16
-#define PAGE_SHIFT_512K        19
-#define PAGE_SHIFT_8M  23
-#define PAGE_SHIFT_16M 24
-#define PAGE_SHIFT_16G 34
-
 bool hugetlb_disabled = false;
 
 unsigned int HPAGE_SHIFT;
        for (i = 0; i < num_hugepd; i++, hpdp++) {
                if (unlikely(!hugepd_none(*hpdp)))
                        break;
-               else {
-#ifdef CONFIG_PPC_BOOK3S_64
-                       *hpdp = __hugepd(__pa(new) | HUGEPD_VAL_BITS |
-                                        (shift_to_mmu_psize(pshift) << 2));
-#elif defined(CONFIG_PPC_8xx)
-                       *hpdp = __hugepd(__pa(new) | _PMD_USER |
-                                        (pshift == PAGE_SHIFT_8M ? _PMD_PAGE_8M :
-                                         _PMD_PAGE_512K) | _PMD_PRESENT);
-#else
-                       /* We use the old format for PPC_FSL_BOOK3E */
-                       *hpdp = __hugepd(((unsigned long)new & ~PD_HUGE) | pshift);
-#endif
-               }
+               hugepd_populate(hpdp, new, pshift);
        }
        /* If we bailed from the for loop early, an error occurred, clean up */
        if (i < num_hugepd) {