1 files changed, 2 insertions, 1 deletions

diff --git a/mm/migrate.c b/mm/migrate.c
index 253474c22239..b1d02127e1be 100644
--- a/mm/migrate.c
+++ b/mm/migrate.c
@@ -1862,7 +1862,7 @@ fail_putback:
          */
         flush_cache_range(vma, mmun_start, mmun_end);
         page_add_anon_rmap(new_page, vma, mmun_start);
-        pmdp_clear_flush(vma, mmun_start, pmd);
+        pmdp_clear_flush_notify(vma, mmun_start, pmd);
         set_pmd_at(mm, mmun_start, pmd, entry);
         flush_tlb_range(vma, mmun_start, mmun_end);
         update_mmu_cache_pmd(vma, address, &entry);
@@ -1870,6 +1870,7 @@ fail_putback:
         if (page_count(page) != 2) {
                 set_pmd_at(mm, mmun_start, pmd, orig_entry);
                 flush_tlb_range(vma, mmun_start, mmun_end);
+                mmu_notifier_invalidate_range(mm, mmun_start, mmun_end);
                 update_mmu_cache_pmd(vma, address, &entry);
                 page_remove_rmap(new_page);
                 goto fail_putback;