1 files changed, 4 insertions, 1 deletions

diff --git a/mm/huge_memory.c b/mm/huge_memory.c
index d79f7a55bf6f..ee8133794a56 100644
--- a/mm/huge_memory.c
+++ b/mm/huge_memory.c
@@ -1046,6 +1046,7 @@ int do_huge_pmd_numa_page(struct mm_struct *mm, struct vm_area_struct *vma,
          */
         split_huge_page(page);
         put_page(page);
+
         return 0;
 
 clear_pmdnuma:
@@ -1060,8 +1061,10 @@ clear_pmdnuma:
 
 out_unlock:
         spin_unlock(&mm->page_table_lock);
-        if (page)
+        if (page) {
                 put_page(page);
+                task_numa_fault(numa_node_id(), HPAGE_PMD_NR);
+        }
         return 0;
 }