*   have the page locked
         */
        pc->mem_cgroup = memcg;
-       pc->flags = PCG_USED | PCG_MEM | (do_swap_account ? PCG_MEMSW : 0);
+       pc->flags = PCG_USED | PCG_MEM;
 
        if (lrucare)
                unlock_page_lru(page, isolated);
        if (!PageCgroupUsed(pc))
                return;
 
-       VM_BUG_ON_PAGE(!(pc->flags & PCG_MEMSW), page);
        memcg = pc->mem_cgroup;
 
        oldid = swap_cgroup_record(entry, mem_cgroup_id(memcg));
 }
 
 static void uncharge_batch(struct mem_cgroup *memcg, unsigned long pgpgout,
-                          unsigned long nr_mem, unsigned long nr_memsw,
                           unsigned long nr_anon, unsigned long nr_file,
                           unsigned long nr_huge, struct page *dummy_page)
 {
+       unsigned long nr_pages = nr_anon + nr_file;
        unsigned long flags;
 
        if (!mem_cgroup_is_root(memcg)) {
-               if (nr_mem)
-                       page_counter_uncharge(&memcg->memory, nr_mem);
-               if (nr_memsw)
-                       page_counter_uncharge(&memcg->memsw, nr_memsw);
+               page_counter_uncharge(&memcg->memory, nr_pages);
+               if (do_swap_account)
+                       page_counter_uncharge(&memcg->memsw, nr_pages);
                memcg_oom_recover(memcg);
        }
 
        __this_cpu_sub(memcg->stat->count[MEM_CGROUP_STAT_CACHE], nr_file);
        __this_cpu_sub(memcg->stat->count[MEM_CGROUP_STAT_RSS_HUGE], nr_huge);
        __this_cpu_add(memcg->stat->events[MEM_CGROUP_EVENTS_PGPGOUT], pgpgout);
-       __this_cpu_add(memcg->stat->nr_page_events, nr_anon + nr_file);
+       __this_cpu_add(memcg->stat->nr_page_events, nr_pages);
        memcg_check_events(memcg, dummy_page);
        local_irq_restore(flags);
 
        if (!mem_cgroup_is_root(memcg))
-               css_put_many(&memcg->css, max(nr_mem, nr_memsw));
+               css_put_many(&memcg->css, nr_pages);
 }
 
 static void uncharge_list(struct list_head *page_list)
 {
        struct mem_cgroup *memcg = NULL;
-       unsigned long nr_memsw = 0;
        unsigned long nr_anon = 0;
        unsigned long nr_file = 0;
        unsigned long nr_huge = 0;
        unsigned long pgpgout = 0;
-       unsigned long nr_mem = 0;
        struct list_head *next;
        struct page *page;
 
 
                if (memcg != pc->mem_cgroup) {
                        if (memcg) {
-                               uncharge_batch(memcg, pgpgout, nr_mem, nr_memsw,
-                                              nr_anon, nr_file, nr_huge, page);
-                               pgpgout = nr_mem = nr_memsw = 0;
-                               nr_anon = nr_file = nr_huge = 0;
+                               uncharge_batch(memcg, pgpgout, nr_anon, nr_file,
+                                              nr_huge, page);
+                               pgpgout = nr_anon = nr_file = nr_huge = 0;
                        }
                        memcg = pc->mem_cgroup;
                }
                else
                        nr_file += nr_pages;
 
-               if (pc->flags & PCG_MEM)
-                       nr_mem += nr_pages;
-               if (pc->flags & PCG_MEMSW)
-                       nr_memsw += nr_pages;
                pc->flags = 0;
 
                pgpgout++;
        } while (next != page_list);
 
        if (memcg)
-               uncharge_batch(memcg, pgpgout, nr_mem, nr_memsw,
-                              nr_anon, nr_file, nr_huge, page);
+               uncharge_batch(memcg, pgpgout, nr_anon, nr_file,
+                              nr_huge, page);
 }
 
 /**
                return;
 
        VM_BUG_ON_PAGE(!(pc->flags & PCG_MEM), oldpage);
-       VM_BUG_ON_PAGE(do_swap_account && !(pc->flags & PCG_MEMSW), oldpage);
 
        if (lrucare)
                lock_page_lru(oldpage, &isolated);