Hello, On Fri, Feb 05, 2021 at 01:28:03PM -0500, Johannes Weiner wrote: > Current users of the rstat code can source root-level statistics from > the native counters of their respective subsystem, allowing them to > forego aggregation at the root level. This optimization is currently > implemented inside the generic rstat code, which doesn't track the > root cgroup and doesn't invoke the subsystem flush callbacks on it. > > However, the memory controller cannot do this optimization, because > cgroup1 breaks out memory specifically for the local level, including > at the root level. In preparation for the memory controller switching > to rstat, move the optimization from rstat core to the controllers. > > Afterwards, rstat will always track the root cgroup for changes and > invoke the subsystem callbacks on it; and it's up to the subsystem to > special-case and skip aggregation of the root cgroup if it can source > this information through other, cheaper means. > > The extra cost of tracking the root cgroup is negligible: on stat > changes, we actually remove a branch that checks for the root. The > queueing for a flush touches only per-cpu data, and only the first > stat change since a flush requires a (per-cpu) lock. > > Signed-off-by: Johannes Weiner <hannes@xxxxxxxxxxx> Generally looks good to me. Acked-by: Tejun Heo <tj@xxxxxxxxxx> A couple nits below. > diff --git a/block/blk-cgroup.c b/block/blk-cgroup.c > index 02ce2058c14b..76725e1cad7f 100644 > --- a/block/blk-cgroup.c > +++ b/block/blk-cgroup.c > @@ -766,6 +766,10 @@ static void blkcg_rstat_flush(struct cgroup_subsys_state *css, int cpu) > struct blkcg *blkcg = css_to_blkcg(css); > struct blkcg_gq *blkg; > > + /* Root-level stats are sourced from system-wide IO stats */ > + if (!cgroup_parent(css->cgroup)) > + return; > + > rcu_read_lock(); > > hlist_for_each_entry_rcu(blkg, &blkcg->blkg_list, blkcg_node) { > @@ -789,6 +793,7 @@ static void blkcg_rstat_flush(struct cgroup_subsys_state *css, int cpu) > u64_stats_update_end(&blkg->iostat.sync); > > /* propagate global delta to parent */ > + /* XXX: could skip this if parent is root */ > if (parent) { > u64_stats_update_begin(&parent->iostat.sync); > blkg_iostat_set(&delta, &blkg->iostat.cur); Might as well update this similar to cgroup_base_stat_flush()? > @@ -58,8 +53,16 @@ void cgroup_rstat_updated(struct cgroup *cgrp, int cpu) > if (rstatc->updated_next) > break; > > + if (!parent) { Maybe useful to note that the node is being marked busy but not added to the non-existent parent. > + rstatc->updated_next = cgrp; > + break; > + } > + > + prstatc = cgroup_rstat_cpu(parent, cpu); > rstatc->updated_next = prstatc->updated_children; > prstatc->updated_children = cgrp; > + > + cgrp = parent; > } > > raw_spin_unlock_irqrestore(cpu_lock, flags); ... > static void cgroup_base_stat_flush(struct cgroup *cgrp, int cpu) > { > - struct cgroup *parent = cgroup_parent(cgrp); > struct cgroup_rstat_cpu *rstatc = cgroup_rstat_cpu(cgrp, cpu); > + struct cgroup *parent = cgroup_parent(cgrp); Is this chunk intentional? > struct cgroup_base_stat cur, delta; > unsigned seq; > > + /* Root-level stats are sourced from system-wide CPU stats */ > + if (!parent) > + return; > + > /* fetch the current per-cpu values */ > do { > seq = __u64_stats_fetch_begin(&rstatc->bsync); > @@ -326,8 +336,8 @@ static void cgroup_base_stat_flush(struct cgroup *cgrp, int cpu) > cgroup_base_stat_add(&cgrp->bstat, &delta); > cgroup_base_stat_add(&rstatc->last_bstat, &delta); > > - /* propagate global delta to parent */ > - if (parent) { > + /* propagate global delta to parent (unless that's root) */ > + if (cgroup_parent(parent)) { Yeah, this makes sense. Can you add a short while-at-it note in the patch description? Thanks. -- tejun