1 files changed, 15 insertions, 8 deletions

diff --git a/mm/memcontrol.c b/mm/memcontrol.c
index e7f47a38fe4f..c1f9b79817d7 100644
--- a/mm/memcontrol.c
+++ b/mm/memcontrol.c
@@ -4654,8 +4654,8 @@ static enum mc_target_type get_mctgt_type(struct vm_area_struct *vma,
 
 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
 /*
- * We don't consider swapping or file mapped pages because THP does not
- * support them for now.
+ * We don't consider PMD mapped swapping or file mapped pages because THP does
+ * not support them for now.
  * Caller should make sure that pmd_trans_huge(pmd) is true.
  */
 static enum mc_target_type get_mctgt_type_thp(struct vm_area_struct *vma,
@@ -5913,6 +5913,7 @@ static struct mem_cgroup *mem_cgroup_id_get_online(struct mem_cgroup *memcg)
 void mem_cgroup_swapout(struct page *page, swp_entry_t entry)
 {
         struct mem_cgroup *memcg, *swap_memcg;
+        unsigned int nr_entries;
         unsigned short oldid;
 
         VM_BUG_ON_PAGE(PageLRU(page), page);
@@ -5933,19 +5934,24 @@ void mem_cgroup_swapout(struct page *page, swp_entry_t entry)
          * ancestor for the swap instead and transfer the memory+swap charge.
          */
         swap_memcg = mem_cgroup_id_get_online(memcg);
-        oldid = swap_cgroup_record(entry, mem_cgroup_id(swap_memcg), 1);
+        nr_entries = hpage_nr_pages(page);
+        /* Get references for the tail pages, too */
+        if (nr_entries > 1)
+                mem_cgroup_id_get_many(swap_memcg, nr_entries - 1);
+        oldid = swap_cgroup_record(entry, mem_cgroup_id(swap_memcg),
+                                   nr_entries);
         VM_BUG_ON_PAGE(oldid, page);
-        mem_cgroup_swap_statistics(swap_memcg, 1);
+        mem_cgroup_swap_statistics(swap_memcg, nr_entries);
 
         page->mem_cgroup = NULL;
 
         if (!mem_cgroup_is_root(memcg))
-                page_counter_uncharge(&memcg->memory, 1);
+                page_counter_uncharge(&memcg->memory, nr_entries);
 
         if (memcg != swap_memcg) {
                 if (!mem_cgroup_is_root(swap_memcg))
-                        page_counter_charge(&swap_memcg->memsw, 1);
-                page_counter_uncharge(&memcg->memsw, 1);
+                        page_counter_charge(&swap_memcg->memsw, nr_entries);
+                page_counter_uncharge(&memcg->memsw, nr_entries);
         }
 
         /*
@@ -5955,7 +5961,8 @@ void mem_cgroup_swapout(struct page *page, swp_entry_t entry)
          * only synchronisation we have for udpating the per-CPU variables.
          */
         VM_BUG_ON(!irqs_disabled());
-        mem_cgroup_charge_statistics(memcg, page, false, -1);
+        mem_cgroup_charge_statistics(memcg, page, PageTransHuge(page),
+                                     -nr_entries);
         memcg_check_events(memcg, page);
 
         if (!mem_cgroup_is_root(memcg))