1 files changed, 4 insertions, 3 deletions

diff --git a/fs/proc/task_mmu.c b/fs/proc/task_mmu.c
index 288185e24762..ef44bb474ea2 100644
--- a/fs/proc/task_mmu.c
+++ b/fs/proc/task_mmu.c
@@ -792,19 +792,20 @@ static inline void clear_soft_dirty(struct vm_area_struct *vma,
         pte_t ptent = *pte;
 
         if (pte_present(ptent)) {
+                ptent = ptep_modify_prot_start(vma->vm_mm, addr, pte);
                 ptent = pte_wrprotect(ptent);
                 ptent = pte_clear_soft_dirty(ptent);
+                ptep_modify_prot_commit(vma->vm_mm, addr, pte, ptent);
         } else if (is_swap_pte(ptent)) {
                 ptent = pte_swp_clear_soft_dirty(ptent);
+                set_pte_at(vma->vm_mm, addr, pte, ptent);
         }
-
-        set_pte_at(vma->vm_mm, addr, pte, ptent);
 }
 
 static inline void clear_soft_dirty_pmd(struct vm_area_struct *vma,
                 unsigned long addr, pmd_t *pmdp)
 {
-        pmd_t pmd = *pmdp;
+        pmd_t pmd = pmdp_huge_get_and_clear(vma->vm_mm, addr, pmdp);
 
         pmd = pmd_wrprotect(pmd);
         pmd = pmd_clear_soft_dirty(pmd);