RISCV: Support pmd_leaf() in the function kernel_page_present()

Update kernel_page_present() function to support pmd_leaf().
Function kernel_page_present() will be invoked when hibernation is
started.

Signed-off-by: Sia Jee Heng <jeeheng.sia@starfivetech.com>

diff --git a/arch/riscv/mm/pageattr.c b/arch/riscv/mm/pageattr.c
index 5e49e4b..58c61df 100644 (file)
--- a/arch/riscv/mm/pageattr.c
+++ b/arch/riscv/mm/pageattr.c
@@ -229,6 +229,8 @@ bool kernel_page_present(struct page *page)
        pmd = pmd_offset(pud, addr);
        if (!pmd_present(*pmd))
                return false;
+       if (pmd_leaf(*pmd))
+               return true;
 
        pte = pte_offset_kernel(pmd, addr);
        return pte_present(*pte);