2 files changed, 9 insertions, 7 deletions

diff --git a/arch/powerpc/kernel/eeh.c b/arch/powerpc/kernel/eeh.c
index a4c62eb0ee48..44b480e3a5af 100644
--- a/arch/powerpc/kernel/eeh.c
+++ b/arch/powerpc/kernel/eeh.c
@@ -334,9 +334,11 @@ static inline unsigned long eeh_token_to_phys(unsigned long token)
         int hugepage_shift;
 
         /*
-         * We won't find hugepages here, iomem
+         * We won't find hugepages here(this is iomem). Hence we are not
+         * worried about _PAGE_SPLITTING/collapse. Also we will not hit
+         * page table free, because of init_mm.
          */
-        ptep = find_linux_pte_or_hugepte(init_mm.pgd, token, &hugepage_shift);
+        ptep = __find_linux_pte_or_hugepte(init_mm.pgd, token, &hugepage_shift);
         if (!ptep)
                 return token;
         WARN_ON(hugepage_shift);
diff --git a/arch/powerpc/kernel/io-workarounds.c b/arch/powerpc/kernel/io-workarounds.c
index 24b968f8e4d8..63d9cc4d7366 100644
--- a/arch/powerpc/kernel/io-workarounds.c
+++ b/arch/powerpc/kernel/io-workarounds.c
@@ -71,15 +71,15 @@ struct iowa_bus *iowa_mem_find_bus(const PCI_IO_ADDR addr)
                 vaddr = (unsigned long)PCI_FIX_ADDR(addr);
                 if (vaddr < PHB_IO_BASE || vaddr >= PHB_IO_END)
                         return NULL;
-
-                ptep = find_linux_pte_or_hugepte(init_mm.pgd, vaddr,
+                /*
+                 * We won't find huge pages here (iomem). Also can't hit
+                 * a page table free due to init_mm
+                 */
+                ptep = __find_linux_pte_or_hugepte(init_mm.pgd, vaddr,
                                                  &hugepage_shift);
                 if (ptep == NULL)
                         paddr = 0;
                 else {
-                        /*
-                         * we don't have hugepages backing iomem
-                         */
                         WARN_ON(hugepage_shift);
                         paddr = pte_pfn(*ptep) << PAGE_SHIFT;
                 }