1 files changed, 2 insertions, 1 deletions

diff --git a/mm/huge_memory.c b/mm/huge_memory.c
index 964fc5a2edd2..5a595554bd8c 100644
--- a/mm/huge_memory.c
+++ b/mm/huge_memory.c
@@ -1026,7 +1026,7 @@ out:
 }
 
 int zap_huge_pmd(struct mmu_gather *tlb, struct vm_area_struct *vma,
-                 pmd_t *pmd)
+                 pmd_t *pmd, unsigned long addr)
 {
         int ret = 0;
 
@@ -1042,6 +1042,7 @@ int zap_huge_pmd(struct mmu_gather *tlb, struct vm_area_struct *vma,
                         pgtable = get_pmd_huge_pte(tlb->mm);
                         page = pmd_page(*pmd);
                         pmd_clear(pmd);
+                        tlb_remove_pmd_tlb_entry(tlb, pmd, addr);
                         page_remove_rmap(page);
                         VM_BUG_ON(page_mapcount(page) < 0);
                         add_mm_counter(tlb->mm, MM_ANONPAGES, -HPAGE_PMD_NR);