Kernel/linux-next
1
0
Fork 0
mirror of https://git.kernel.org/pub/scm/linux/kernel/git/next/linux-next.git synced 2024-12-29 09:12:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

memcg: factor out mem_cgroup_stat_aggregate()

Browse Source 
Currently mem_cgroup_css_rstat_flush() is used to flush the per-CPU
statistics from a specified CPU into the global statistics of the
memcg. It processes three kinds of data in three for loops using exactly
the same method. Therefore, the for loop can be factored out and may
make the code more clean.

Link: https://lkml.kernel.org/r/20241026093407.310955-1-xiujianfeng@huaweicloud.com
Signed-off-by: Xiu Jianfeng <xiujianfeng@huawei.com>
Cc: Johannes Weiner <hannes@cmpxchg.org>
Cc: Michal Hocko <mhocko@kernel.org>
Cc: Muchun Song <muchun.song@linux.dev>
Cc: Roman Gushchin <roman.gushchin@linux.dev>
Cc: Shakeel Butt <shakeel.butt@linux.dev>
Cc: Wang Weiyang <wangweiyang2@huawei.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
This commit is contained in:
Xiu Jianfeng 2024-10-26 09:34:07 +00:00 committed by Andrew Morton
parent e8c1a296b8
commit 2b1d55498b
1 changed files with 69 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

127
mm/memcontrol.c
View File

@ -3730,68 +3730,90 @@ static void mem_cgroup_css_reset(struct cgroup_subsys_state *css)
memcg_wb_domain_size_changed(memcg);
}
static void mem_cgroup_css_rstat_flush(struct cgroup_subsys_state *css, int cpu)
struct aggregate_control {
/* pointer to the aggregated (CPU and subtree aggregated) counters */
long *aggregate;
/* pointer to the non-hierarchichal (CPU aggregated) counters */
long *local;
/* pointer to the pending child counters during tree propagation */
long *pending;
/* pointer to the parent's pending counters, could be NULL */
long *ppending;
/* pointer to the percpu counters to be aggregated */
long *cstat;
/* pointer to the percpu counters of the last aggregation*/
long *cstat_prev;
/* size of the above counters */
int size;
};
static void mem_cgroup_stat_aggregate(struct aggregate_control *ac)
{
struct mem_cgroup *memcg = mem_cgroup_from_css(css);
struct mem_cgroup *parent = parent_mem_cgroup(memcg);
struct memcg_vmstats_percpu *statc;
int i;
long delta, delta_cpu, v;
int i, nid;
statc = per_cpu_ptr(memcg->vmstats_percpu, cpu);
for (i = 0; i < MEMCG_VMSTAT_SIZE; i++) {
for (i = 0; i < ac->size; i++) {
/*
* Collect the aggregated propagation counts of groups
* below us. We're in a per-cpu loop here and this is
* a global counter, so the first cycle will get them.
*/
delta = memcg->vmstats->state_pending[i];
delta = ac->pending[i];
if (delta)
memcg->vmstats->state_pending[i] = 0;
ac->pending[i] = 0;
/* Add CPU changes on this level since the last flush */
delta_cpu = 0;
v = READ_ONCE(statc->state[i]);
if (v != statc->state_prev[i]) {
delta_cpu = v - statc->state_prev[i];
v = READ_ONCE(ac->cstat[i]);
if (v != ac->cstat_prev[i]) {
delta_cpu = v - ac->cstat_prev[i];
delta += delta_cpu;
statc->state_prev[i] = v;
ac->cstat_prev[i] = v;
}
/* Aggregate counts on this level and propagate upwards */
if (delta_cpu)
memcg->vmstats->state_local[i] += delta_cpu;
ac->local[i] += delta_cpu;
if (delta) {
memcg->vmstats->state[i] += delta;
if (parent)
parent->vmstats->state_pending[i] += delta;
ac->aggregate[i] += delta;
if (ac->ppending)
ac->ppending[i] += delta;
}
}
}
for (i = 0; i < NR_MEMCG_EVENTS; i++) {
delta = memcg->vmstats->events_pending[i];
if (delta)
memcg->vmstats->events_pending[i] = 0;
static void mem_cgroup_css_rstat_flush(struct cgroup_subsys_state *css, int cpu)
{
struct mem_cgroup *memcg = mem_cgroup_from_css(css);
struct mem_cgroup *parent = parent_mem_cgroup(memcg);
struct memcg_vmstats_percpu *statc;
struct aggregate_control ac;
int nid;
delta_cpu = 0;
v = READ_ONCE(statc->events[i]);
if (v != statc->events_prev[i]) {
delta_cpu = v - statc->events_prev[i];
delta += delta_cpu;
statc->events_prev[i] = v;
}
statc = per_cpu_ptr(memcg->vmstats_percpu, cpu);
if (delta_cpu)
memcg->vmstats->events_local[i] += delta_cpu;
ac = (struct aggregate_control) {
.aggregate = memcg->vmstats->state,
.local = memcg->vmstats->state_local,
.pending = memcg->vmstats->state_pending,
.ppending = parent ? parent->vmstats->state_pending : NULL,
.cstat = statc->state,
.cstat_prev = statc->state_prev,
.size = MEMCG_VMSTAT_SIZE,
};
mem_cgroup_stat_aggregate(&ac);
if (delta) {
memcg->vmstats->events[i] += delta;
if (parent)
parent->vmstats->events_pending[i] += delta;
}
}
ac = (struct aggregate_control) {
.aggregate = memcg->vmstats->events,
.local = memcg->vmstats->events_local,
.pending = memcg->vmstats->events_pending,
.ppending = parent ? parent->vmstats->events_pending : NULL,
.cstat = statc->events,
.cstat_prev = statc->events_prev,
.size = NR_MEMCG_EVENTS,
};
mem_cgroup_stat_aggregate(&ac);
for_each_node_state(nid, N_MEMORY) {
struct mem_cgroup_per_node *pn = memcg->nodeinfo[nid];
@ -3804,28 +3826,17 @@ static void mem_cgroup_css_rstat_flush(struct cgroup_subsys_state *css, int cpu)
lstatc = per_cpu_ptr(pn->lruvec_stats_percpu, cpu);
for (i = 0; i < NR_MEMCG_NODE_STAT_ITEMS; i++) {
delta = lstats->state_pending[i];
if (delta)
lstats->state_pending[i] = 0;
ac = (struct aggregate_control) {
.aggregate = lstats->state,
.local = lstats->state_local,
.pending = lstats->state_pending,
.ppending = plstats ? plstats->state_pending : NULL,
.cstat = lstatc->state,
.cstat_prev = lstatc->state_prev,
.size = NR_MEMCG_NODE_STAT_ITEMS,
};
mem_cgroup_stat_aggregate(&ac);
delta_cpu = 0;
v = READ_ONCE(lstatc->state[i]);
if (v != lstatc->state_prev[i]) {
delta_cpu = v - lstatc->state_prev[i];
delta += delta_cpu;
lstatc->state_prev[i] = v;
}
if (delta_cpu)
lstats->state_local[i] += delta_cpu;
if (delta) {
lstats->state[i] += delta;
if (plstats)
plstats->state_pending[i] += delta;
}
}
}
WRITE_ONCE(statc->stats_updates, 0);
/* We are in a per-cpu loop here, only do the atomic write once */