*/
 void mem_cgroup_swapout(struct page *page, swp_entry_t entry)
 {
+       struct mem_cgroup *memcg;
        struct page_cgroup *pc;
        unsigned short oldid;
 
                return;
 
        VM_BUG_ON_PAGE(!(pc->flags & PCG_MEMSW), page);
+       memcg = pc->mem_cgroup;
 
-       oldid = swap_cgroup_record(entry, mem_cgroup_id(pc->mem_cgroup));
+       oldid = swap_cgroup_record(entry, mem_cgroup_id(memcg));
        VM_BUG_ON_PAGE(oldid, page);
+       mem_cgroup_swap_statistics(memcg, true);
+
+       pc->flags = 0;
 
-       pc->flags &= ~PCG_MEMSW;
-       css_get(&pc->mem_cgroup->css);
-       mem_cgroup_swap_statistics(pc->mem_cgroup, true);
+       if (!mem_cgroup_is_root(memcg))
+               page_counter_uncharge(&memcg->memory, 1);
+
+       /* XXX: caller holds IRQ-safe mapping->tree_lock */
+       VM_BUG_ON(!irqs_disabled());
+
+       mem_cgroup_charge_statistics(memcg, page, -1);
+       memcg_check_events(memcg, page);
 }
 
 /**