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>

1 files changed, 2 insertions, 0 deletions

diff --git a/arch/riscv/mm/pageattr.c b/arch/riscv/mm/pageattr.c
index 5e49e4b4a4cc..58c61df872bd 100644
--- 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);