On Tue 09-03-21 12:32:55, Matthew Wilcox wrote:
> On Tue, Mar 09, 2021 at 10:02:00AM +0100, Michal Hocko wrote:
[...]
> > diff --git a/mm/memcontrol.c b/mm/memcontrol.c
> > index 913c2b9e5c72..d44dea2b8d22 100644
> > --- a/mm/memcontrol.c
> > +++ b/mm/memcontrol.c
> > @@ -3135,8 +3135,21 @@ int __memcg_kmem_charge_page(struct page *page, 
> > gfp_t gfp, int order)
> >     if (memcg && !mem_cgroup_is_root(memcg)) {
> >             ret = __memcg_kmem_charge(memcg, gfp, 1 << order);
> >             if (!ret) {
> > +                   int nr_pages = 1 << order;
> >                     page->memcg_data = (unsigned long)memcg |
> >                             MEMCG_DATA_KMEM;
> > +                   
> > +                   /*
> > +                    * Compound pages are normally split or freed
> > +                    * via their head pages so memcg_data in in the
> > +                    * head page should be sufficient but there
> > +                    * are exceptions to the rule (see __free_pages).
> > +                    * Non compound pages would need to copy memcg anyway.
> > +                    */
> > +                   for (i = 1; i < nr_pages; i++) {
> > +                           struct page * p = page + i;
> > +                           p->memcg_data = page->memcg_data
> > +                   }
> >                     return 0;
> 
> I would condition this loop on if (!(gfp & __GFP_COMP)), but yes, something
> along these lines.  I might phrase the comment a little differently ...
> 
>                       /*
>                        * Compound pages are treated as a single unit,
>                        * but non-compound pages can be freed individually
>                        * so each page needs to have its memcg set to get
>                        * the accounting right.
>                        */

OK, I must have misunderstood your __free_pages fix then. I thought this
was about compound pages. Btw. again I forgot about css ref counting so
here is an updated version.

diff --git a/mm/memcontrol.c b/mm/memcontrol.c
index 913c2b9e5c72..ec2c705f38fa 100644
--- a/mm/memcontrol.c
+++ b/mm/memcontrol.c
@@ -3133,10 +3133,22 @@ int __memcg_kmem_charge_page(struct page *page, gfp_t 
gfp, int order)
 
        memcg = get_mem_cgroup_from_current();
        if (memcg && !mem_cgroup_is_root(memcg)) {
-               ret = __memcg_kmem_charge(memcg, gfp, 1 << order);
+               int nr_pages = 1 << order;
+               ret = __memcg_kmem_charge(memcg, gfp, nr_pages);
                if (!ret) {
                        page->memcg_data = (unsigned long)memcg |
                                MEMCG_DATA_KMEM;
+                       if (nr_pages > 1) {
+                               /*
+                                * comment goes here
+                                */
+                               for (i = 1; i < nr_pages; i++) {
+                                       struct page * p = page + i;
+                                       p->memcg_data = page->memcg_data
+                               }
+                               /* Head page reference from 
get_mem_cgroup_from_current */
+                               css_get_many(&memcg->css, nr_pages - 1);
+                       }
                        return 0;
                }
                css_put(&memcg->css);

-- 
Michal Hocko
SUSE Labs

Reply via email to