1 files changed, 12 insertions, 3 deletions

diff --git a/arch/sparc64/mm/fault.c b/arch/sparc64/mm/fault.c
index 63b6cc0cd5d5..d21ff3230c02 100644
--- a/arch/sparc64/mm/fault.c
+++ b/arch/sparc64/mm/fault.c
@@ -410,9 +410,18 @@ good_area:
         up_read(&mm->mmap_sem);
 
         mm_rss = get_mm_rss(mm);
-        if (unlikely(mm_rss >= mm->context.tsb_rss_limit))
-                tsb_grow(mm, mm_rss);
-
+#ifdef CONFIG_HUGETLB_PAGE
+        mm_rss -= (mm->context.huge_pte_count * (HPAGE_SIZE / PAGE_SIZE));
+#endif
+        if (unlikely(mm_rss >=
+                     mm->context.tsb_block[MM_TSB_BASE].tsb_rss_limit))
+                tsb_grow(mm, MM_TSB_BASE, mm_rss);
+#ifdef CONFIG_HUGETLB_PAGE
+        mm_rss = mm->context.huge_pte_count;
+        if (unlikely(mm_rss >=
+                     mm->context.tsb_block[MM_TSB_HUGE].tsb_rss_limit))
+                tsb_grow(mm, MM_TSB_HUGE, mm_rss);
+#endif
         return;
 
         /*