[<prev] [next>] [<thread-prev] [day] [month] [year] [list]
Message-ID: <17fd8e7f-eefb-4c85-a086-fa32741c4d94@efficios.com>
Date: Thu, 3 Jul 2025 11:59:29 -0400
From: Mathieu Desnoyers <mathieu.desnoyers@...icios.com>
To: Andrew Morton <akpm@...ux-foundation.org>
Cc: linux-kernel@...r.kernel.org, "Paul E. McKenney" <paulmck@...nel.org>,
Steven Rostedt <rostedt@...dmis.org>, Masami Hiramatsu
<mhiramat@...nel.org>, Dennis Zhou <dennis@...nel.org>,
Tejun Heo <tj@...nel.org>, Christoph Lameter <cl@...ux.com>,
Martin Liu <liumartin@...gle.com>, David Rientjes <rientjes@...gle.com>,
christian.koenig@....com, Shakeel Butt <shakeel.butt@...ux.dev>,
Johannes Weiner <hannes@...xchg.org>, Sweet Tea Dorminy
<sweettea@...gle.com>, Lorenzo Stoakes <lorenzo.stoakes@...cle.com>,
"Liam R . Howlett" <Liam.Howlett@...cle.com>,
Suren Baghdasaryan <surenb@...gle.com>, Vlastimil Babka <vbabka@...e.cz>,
Christian Brauner <brauner@...nel.org>, Wei Yang
<richard.weiyang@...il.com>, David Hildenbrand <david@...hat.com>,
Miaohe Lin <linmiaohe@...wei.com>, Al Viro <viro@...iv.linux.org.uk>,
linux-mm@...ck.org, linux-trace-kernel@...r.kernel.org,
Yu Zhao <yuzhao@...gle.com>, Roman Gushchin <roman.gushchin@...ux.dev>,
Mateusz Guzik <mjguzik@...il.com>, Matthew Wilcox <willy@...radead.org>
Subject: Re: [RFC PATCH v4 2/2] mm: Fix OOM killer inaccuracy on large
many-core systems
On 2025-07-03 11:20, Mathieu Desnoyers wrote:
> Use hierarchical per-cpu counters for rss tracking to fix the per-mm RSS
> tracking which has become too inaccurate for OOM killer purposes on
> large many-core systems.
>
> The following rss tracking issues were noted by Sweet Tea Dorminy [1],
> which lead to picking wrong tasks as OOM kill target:
>
> Recently, several internal services had an RSS usage regression as part of a
> kernel upgrade. Previously, they were on a pre-6.2 kernel and were able to
> read RSS statistics in a backup watchdog process to monitor and decide if
> they'd overrun their memory budget. Now, however, a representative service
> with five threads, expected to use about a hundred MB of memory, on a 250-cpu
> machine had memory usage tens of megabytes different from the expected amount
> -- this constituted a significant percentage of inaccuracy, causing the
> watchdog to act.
>
> This was a result of f1a7941243c1 ("mm: convert mm's rss stats into
> percpu_counter") [1]. Previously, the memory error was bounded by
> 64*nr_threads pages, a very livable megabyte. Now, however, as a result of
> scheduler decisions moving the threads around the CPUs, the memory error could
> be as large as a gigabyte.
>
> This is a really tremendous inaccuracy for any few-threaded program on a
> large machine and impedes monitoring significantly. These stat counters are
> also used to make OOM killing decisions, so this additional inaccuracy could
> make a big difference in OOM situations -- either resulting in the wrong
> process being killed, or in less memory being returned from an OOM-kill than
> expected.
>
> Here is a (possibly incomplete) list of the prior approaches that were
> used or proposed, along with their downside:
>
> 1) Per-thread rss tracking: large error on many-thread processes.
>
> 2) Per-CPU counters: up to 12% slower for short-lived processes and 9%
> increased system time in make test workloads [1]. Moreover, the
> inaccuracy increases with O(n^2) with the number of CPUs.
>
> 3) Per-NUMA-node counters: requires atomics on fast-path (overhead),
> error is high with systems that have lots of NUMA nodes (32 times
> the number of NUMA nodes).
>
> The approach proposed here is to replace this by the hierarchical
> per-cpu counters, which bounds the inaccuracy based on the system
> topology with O(N*logN).
>
> Link: https://lore.kernel.org/lkml/20250331223516.7810-2-sweettea-kernel@dorminy.me/ # [1]
> Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@...icios.com>
> Cc: Andrew Morton <akpm@...ux-foundation.org>
> Cc: "Paul E. McKenney" <paulmck@...nel.org>
> Cc: Steven Rostedt <rostedt@...dmis.org>
> Cc: Masami Hiramatsu <mhiramat@...nel.org>
> Cc: Mathieu Desnoyers <mathieu.desnoyers@...icios.com>
> Cc: Dennis Zhou <dennis@...nel.org>
> Cc: Tejun Heo <tj@...nel.org>
> Cc: Christoph Lameter <cl@...ux.com>
> Cc: Martin Liu <liumartin@...gle.com>
> Cc: David Rientjes <rientjes@...gle.com>
> Cc: christian.koenig@....com
> Cc: Shakeel Butt <shakeel.butt@...ux.dev>
> Cc: Johannes Weiner <hannes@...xchg.org>
> Cc: Sweet Tea Dorminy <sweettea@...gle.com>
> Cc: Lorenzo Stoakes <lorenzo.stoakes@...cle.com>
> Cc: "Liam R . Howlett" <Liam.Howlett@...cle.com>
> Cc: Suren Baghdasaryan <surenb@...gle.com>
> Cc: Vlastimil Babka <vbabka@...e.cz>
> Cc: Christian Brauner <brauner@...nel.org>
> Cc: Wei Yang <richard.weiyang@...il.com>
> Cc: David Hildenbrand <david@...hat.com>
> Cc: Miaohe Lin <linmiaohe@...wei.com>
> Cc: Al Viro <viro@...iv.linux.org.uk>
> Cc: linux-mm@...ck.org
> Cc: linux-trace-kernel@...r.kernel.org
> Cc: Yu Zhao <yuzhao@...gle.com>
> Cc: Roman Gushchin <roman.gushchin@...ux.dev>
> Cc: Mateusz Guzik <mjguzik@...il.com>
> Cc: Matthew Wilcox <willy@...radead.org>
> ---
> include/linux/mm.h | 8 ++++----
> include/linux/mm_types.h | 4 ++--
> include/trace/events/kmem.h | 2 +-
> kernel/fork.c | 31 +++++++++++++++++++++----------
> 4 files changed, 28 insertions(+), 17 deletions(-)
>
> diff --git a/include/linux/mm.h b/include/linux/mm.h
> index e51dba8398f7..79defe89aefa 100644
> --- a/include/linux/mm.h
> +++ b/include/linux/mm.h
> @@ -2705,28 +2705,28 @@ static inline bool get_user_page_fast_only(unsigned long addr,
> */
> static inline unsigned long get_mm_counter(struct mm_struct *mm, int member)
> {
> - return percpu_counter_read_positive(&mm->rss_stat[member]);
> + return percpu_counter_tree_approximate_sum(&mm->rss_stat[member]);
This should be:
int v = percpu_counter_tree_approximate_sum(&mm->rss_stat[member]);
return v > 0 ? v : 0;
To make sure the value returned is positive, as expected by some callers.
Otherwise I get divide by 0 OOPS in task_scan_max after a while.
Thanks,
Mathieu
> }
>
> void mm_trace_rss_stat(struct mm_struct *mm, int member);
>
> static inline void add_mm_counter(struct mm_struct *mm, int member, long value)
> {
> - percpu_counter_add(&mm->rss_stat[member], value);
> + percpu_counter_tree_add(&mm->rss_stat[member], value);
>
> mm_trace_rss_stat(mm, member);
> }
>
> static inline void inc_mm_counter(struct mm_struct *mm, int member)
> {
> - percpu_counter_inc(&mm->rss_stat[member]);
> + percpu_counter_tree_add(&mm->rss_stat[member], 1);
>
> mm_trace_rss_stat(mm, member);
> }
>
> static inline void dec_mm_counter(struct mm_struct *mm, int member)
> {
> - percpu_counter_dec(&mm->rss_stat[member]);
> + percpu_counter_tree_add(&mm->rss_stat[member], -1);
>
> mm_trace_rss_stat(mm, member);
> }
> diff --git a/include/linux/mm_types.h b/include/linux/mm_types.h
> index 56d07edd01f9..85b15109106a 100644
> --- a/include/linux/mm_types.h
> +++ b/include/linux/mm_types.h
> @@ -18,7 +18,7 @@
> #include <linux/page-flags-layout.h>
> #include <linux/workqueue.h>
> #include <linux/seqlock.h>
> -#include <linux/percpu_counter.h>
> +#include <linux/percpu_counter_tree.h>
> #include <linux/types.h>
>
> #include <asm/mmu.h>
> @@ -1059,7 +1059,7 @@ struct mm_struct {
>
> unsigned long saved_auxv[AT_VECTOR_SIZE]; /* for /proc/PID/auxv */
>
> - struct percpu_counter rss_stat[NR_MM_COUNTERS];
> + struct percpu_counter_tree rss_stat[NR_MM_COUNTERS];
>
> struct linux_binfmt *binfmt;
>
> diff --git a/include/trace/events/kmem.h b/include/trace/events/kmem.h
> index f74925a6cf69..d6199b99c771 100644
> --- a/include/trace/events/kmem.h
> +++ b/include/trace/events/kmem.h
> @@ -477,7 +477,7 @@ TRACE_EVENT(rss_stat,
> __entry->mm_id = mm_ptr_to_hash(mm);
> __entry->curr = !!(current->mm == mm);
> __entry->member = member;
> - __entry->size = (percpu_counter_sum_positive(&mm->rss_stat[member])
> + __entry->size = (percpu_counter_tree_approximate_sum(&mm->rss_stat[member])
> << PAGE_SHIFT);
> ),
>
> diff --git a/kernel/fork.c b/kernel/fork.c
> index 168681fc4b25..dd458adc5543 100644
> --- a/kernel/fork.c
> +++ b/kernel/fork.c
> @@ -129,6 +129,11 @@
> */
> #define MAX_THREADS FUTEX_TID_MASK
>
> +/*
> + * Batch size of rss stat approximation
> + */
> +#define RSS_STAT_BATCH_SIZE 32
> +
> /*
> * Protected counters by write_lock_irq(&tasklist_lock)
> */
> @@ -843,11 +848,10 @@ static void check_mm(struct mm_struct *mm)
> "Please make sure 'struct resident_page_types[]' is updated as well");
>
> for (i = 0; i < NR_MM_COUNTERS; i++) {
> - long x = percpu_counter_sum(&mm->rss_stat[i]);
> -
> - if (unlikely(x))
> - pr_alert("BUG: Bad rss-counter state mm:%p type:%s val:%ld\n",
> - mm, resident_page_types[i], x);
> + if (unlikely(percpu_counter_tree_precise_compare_value(&mm->rss_stat[i], 0) != 0))
> + pr_alert("BUG: Bad rss-counter state mm:%p type:%s val:%d\n",
> + mm, resident_page_types[i],
> + percpu_counter_tree_precise_sum(&mm->rss_stat[i]));
> }
>
> if (mm_pgtables_bytes(mm))
> @@ -930,6 +934,8 @@ static void cleanup_lazy_tlbs(struct mm_struct *mm)
> */
> void __mmdrop(struct mm_struct *mm)
> {
> + int i;
> +
> BUG_ON(mm == &init_mm);
> WARN_ON_ONCE(mm == current->mm);
>
> @@ -945,8 +951,8 @@ void __mmdrop(struct mm_struct *mm)
> put_user_ns(mm->user_ns);
> mm_pasid_drop(mm);
> mm_destroy_cid(mm);
> - percpu_counter_destroy_many(mm->rss_stat, NR_MM_COUNTERS);
> -
> + for (i = 0; i < NR_MM_COUNTERS; i++)
> + percpu_counter_tree_destroy(&mm->rss_stat[i]);
> free_mm(mm);
> }
> EXPORT_SYMBOL_GPL(__mmdrop);
> @@ -1285,6 +1291,8 @@ static void mmap_init_lock(struct mm_struct *mm)
> static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p,
> struct user_namespace *user_ns)
> {
> + int i;
> +
> mt_init_flags(&mm->mm_mt, MM_MT_FLAGS);
> mt_set_external_lock(&mm->mm_mt, &mm->mmap_lock);
> atomic_set(&mm->mm_users, 1);
> @@ -1332,15 +1340,18 @@ static struct mm_struct *mm_init(struct mm_struct *mm, struct task_struct *p,
> if (mm_alloc_cid(mm, p))
> goto fail_cid;
>
> - if (percpu_counter_init_many(mm->rss_stat, 0, GFP_KERNEL_ACCOUNT,
> - NR_MM_COUNTERS))
> - goto fail_pcpu;
> + for (i = 0; i < NR_MM_COUNTERS; i++) {
> + if (percpu_counter_tree_init(&mm->rss_stat[i], RSS_STAT_BATCH_SIZE, GFP_KERNEL_ACCOUNT))
> + goto fail_pcpu;
> + }
>
> mm->user_ns = get_user_ns(user_ns);
> lru_gen_init_mm(mm);
> return mm;
>
> fail_pcpu:
> + for (i--; i >= 0; i--)
> + percpu_counter_tree_destroy(&mm->rss_stat[i]);
> mm_destroy_cid(mm);
> fail_cid:
> destroy_context(mm);
--
Mathieu Desnoyers
EfficiOS Inc.
https://www.efficios.com
Powered by blists - more mailing lists