From nobody Thu Dec 18 01:37:13 2025 Received: from us-smtp-delivery-124.mimecast.com (us-smtp-delivery-124.mimecast.com [170.10.129.124]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id BA24623E251 for ; Mon, 10 Feb 2025 15:33:28 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=170.10.129.124 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1739201610; cv=none; b=fWaES8KL6QeVyRuLKkvmtc5Jrb8GsDVA+mDrwu4T+8DIzGfhXgFF2h6fWSubVLP3zZR52I88qMN1Ett7XBLrSdVB55Oj3wNFzjldpGXGffYwNIFZQ1DLu+xCvBUC03GKwm5n57v2P6P29vtls+0hDGKW1Un40F2CL7gBL41+A08= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1739201610; c=relaxed/simple; bh=9Rwb0Zy0TFYVzR/n/xQHWwS/ZMTa05qSikjypC3FsLY=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=kYwg8XuwAi1A3GTZ/AevDy/vba3wHqOVoExIzi5d8Gcpli60csh5UHPA5Mum/tX/XxrXKn3aiPsXGsg6N8gymZ0BxGa/dgn1Az9I5t2oOTerHEZ3jS5J1QNbjqEbRlpwSVCsvmWObK5rPxNggadfYr2iD8QtdlhjzudHXWYXTM8= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=redhat.com; spf=pass smtp.mailfrom=redhat.com; dkim=pass (1024-bit key) header.d=redhat.com header.i=@redhat.com header.b=W2O2eBD3; arc=none smtp.client-ip=170.10.129.124 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=redhat.com Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=redhat.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (1024-bit key) header.d=redhat.com header.i=@redhat.com header.b="W2O2eBD3" DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=redhat.com; s=mimecast20190719; t=1739201607; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:cc:mime-version:mime-version: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=191fmzOplUZF3eh9uENAPy4J7EnijRk6+X9V7IdEjDw=; b=W2O2eBD3E8pjhmYSE8GEGr8/XD7u8OHOXC5HitoXJTMk1iv6Pb+nl9AjgNmUS30iAQUha6 AxCRxpHn8IWnxjUpaF46unkVeNr7ofs3LmpZaFx2qI/BoPs9NWu1LNq9ObaVht2FbjkN8U Uv200T5SdStclJn8GT1NG6B2+R2+aOM= Received: from mx-prod-mc-04.mail-002.prod.us-west-2.aws.redhat.com (ec2-54-186-198-63.us-west-2.compute.amazonaws.com [54.186.198.63]) by relay.mimecast.com with ESMTP with STARTTLS (version=TLSv1.3, cipher=TLS_AES_256_GCM_SHA384) id us-mta-209-x8Sp5xoAOn6BdzEqQLm_KQ-1; Mon, 10 Feb 2025 10:33:21 -0500 X-MC-Unique: x8Sp5xoAOn6BdzEqQLm_KQ-1 X-Mimecast-MFC-AGG-ID: x8Sp5xoAOn6BdzEqQLm_KQ Received: from mx-prod-int-08.mail-002.prod.us-west-2.aws.redhat.com (mx-prod-int-08.mail-002.prod.us-west-2.aws.redhat.com [10.30.177.111]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits) server-digest SHA256) (No client certificate requested) by mx-prod-mc-04.mail-002.prod.us-west-2.aws.redhat.com (Postfix) with ESMTPS id 910BD1955DDD; Mon, 10 Feb 2025 15:33:19 +0000 (UTC) Received: from gmonaco-thinkpadt14gen3.rmtit.com (unknown [10.44.34.65]) by mx-prod-int-08.mail-002.prod.us-west-2.aws.redhat.com (Postfix) with ESMTP id ACA0118004A7; Mon, 10 Feb 2025 15:33:15 +0000 (UTC) From: Gabriele Monaco To: linux-kernel@vger.kernel.org, Andrew Morton , Ingo Molnar , Peter Zijlstra , linux-mm@kvack.org Cc: Gabriele Monaco , Mathieu Desnoyers , Ingo Molnar , "Paul E. McKenney" , Shuah Khan Subject: [PATCH v6 2/3] sched: Move task_mm_cid_work to mm delayed work Date: Mon, 10 Feb 2025 16:32:51 +0100 Message-ID: <20250210153253.460471-3-gmonaco@redhat.com> In-Reply-To: <20250210153253.460471-1-gmonaco@redhat.com> References: <20250210153253.460471-1-gmonaco@redhat.com> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable X-Scanned-By: MIMEDefang 3.4.1 on 10.30.177.111 Content-Type: text/plain; charset="utf-8" Currently, the task_mm_cid_work function is called in a task work triggered by a scheduler tick to frequently compact the mm_cids of each process. This can delay the execution of the corresponding thread for the entire duration of the function, negatively affecting the response in case of real time tasks. In practice, we observe task_mm_cid_work increasing the latency of 30-35us on a 128 cores system, this order of magnitude is meaningful under PREEMPT_RT. Run the task_mm_cid_work in a new delayed work connected to the mm_struct rather than in the task context before returning to userspace. This delayed work is initialised while allocating the mm and disabled before freeing it, its execution is no longer triggered by scheduler ticks but run periodically based on the defined MM_CID_SCAN_DELAY. The main advantage of this change is that the function can be offloaded to a different CPU and even preempted by RT tasks. Moreover, this new behaviour could be more predictable with periodic tasks with short runtime, which may rarely run during a scheduler tick. Now, the work is always scheduled with the same periodicity for each mm (though the periodicity is not guaranteed due to interference from other tasks, but mm_cid compaction is mostly best effort). To avoid excessively increased runtime, we quickly return from the function if we have no work to be done (i.e. no mm_cid is allocated). This is helpful for tasks that sleep for a long time, but also for terminated task. We are no longer following the process' state, hence the function continues to run after a process terminates but before its mm is freed. Fixes: 223baf9d17f2 ("sched: Fix performance regression introduced by mm_ci= d") Reviewed-by: Mathieu Desnoyers Signed-off-by: Gabriele Monaco --- include/linux/mm_types.h | 16 ++++++---- include/linux/sched.h | 1 - kernel/sched/core.c | 66 +++++----------------------------------- kernel/sched/sched.h | 7 ----- 4 files changed, 18 insertions(+), 72 deletions(-) diff --git a/include/linux/mm_types.h b/include/linux/mm_types.h index 0234f14f2aa6b..3aeadb519cac5 100644 --- a/include/linux/mm_types.h +++ b/include/linux/mm_types.h @@ -861,12 +861,6 @@ struct mm_struct { * runqueue locks. */ struct mm_cid __percpu *pcpu_cid; - /* - * @mm_cid_next_scan: Next mm_cid scan (in jiffies). - * - * When the next mm_cid scan is due (in jiffies). - */ - unsigned long mm_cid_next_scan; /** * @nr_cpus_allowed: Number of CPUs allowed for mm. * @@ -889,6 +883,7 @@ struct mm_struct { * mm nr_cpus_allowed updates. */ raw_spinlock_t cpus_allowed_lock; + struct delayed_work mm_cid_work; #endif #ifdef CONFIG_MMU atomic_long_t pgtables_bytes; /* size of all page tables */ @@ -1180,11 +1175,16 @@ static inline void vma_iter_init(struct vma_iterato= r *vmi, =20 #ifdef CONFIG_SCHED_MM_CID =20 +#define SCHED_MM_CID_PERIOD_NS (100ULL * 1000000) /* 100ms */ +#define MM_CID_SCAN_DELAY 100 /* 100ms */ + enum mm_cid_state { MM_CID_UNSET =3D -1U, /* Unset state has lazy_put flag set. */ MM_CID_LAZY_PUT =3D (1U << 31), }; =20 +extern void task_mm_cid_work(struct work_struct *work); + static inline bool mm_cid_is_unset(int cid) { return cid =3D=3D MM_CID_UNSET; @@ -1257,12 +1257,16 @@ static inline int mm_alloc_cid_noprof(struct mm_str= uct *mm, struct task_struct * if (!mm->pcpu_cid) return -ENOMEM; mm_init_cid(mm, p); + INIT_DELAYED_WORK(&mm->mm_cid_work, task_mm_cid_work); + schedule_delayed_work(&mm->mm_cid_work, + msecs_to_jiffies(MM_CID_SCAN_DELAY)); return 0; } #define mm_alloc_cid(...) alloc_hooks(mm_alloc_cid_noprof(__VA_ARGS__)) =20 static inline void mm_destroy_cid(struct mm_struct *mm) { + disable_delayed_work_sync(&mm->mm_cid_work); free_percpu(mm->pcpu_cid); mm->pcpu_cid =3D NULL; } diff --git a/include/linux/sched.h b/include/linux/sched.h index 9632e3318e0d6..515b15f946cac 100644 --- a/include/linux/sched.h +++ b/include/linux/sched.h @@ -1397,7 +1397,6 @@ struct task_struct { int last_mm_cid; /* Most recent cid in mm */ int migrate_from_cpu; int mm_cid_active; /* Whether cid bitmap is active */ - struct callback_head cid_work; #endif =20 struct tlbflush_unmap_batch tlb_ubc; diff --git a/kernel/sched/core.c b/kernel/sched/core.c index 165c90ba64ea9..c65003ab8c55b 100644 --- a/kernel/sched/core.c +++ b/kernel/sched/core.c @@ -4524,7 +4524,6 @@ static void __sched_fork(unsigned long clone_flags, s= truct task_struct *p) p->wake_entry.u_flags =3D CSD_TYPE_TTWU; p->migration_pending =3D NULL; #endif - init_sched_mm_cid(p); } =20 DEFINE_STATIC_KEY_FALSE(sched_numa_balancing); @@ -5662,7 +5661,6 @@ void sched_tick(void) resched_latency =3D cpu_resched_latency(rq); calc_global_load_tick(rq); sched_core_tick(rq); - task_tick_mm_cid(rq, donor); scx_tick(rq); =20 rq_unlock(rq, &rf); @@ -10528,38 +10526,17 @@ static void sched_mm_cid_remote_clear_weight(stru= ct mm_struct *mm, int cpu, sched_mm_cid_remote_clear(mm, pcpu_cid, cpu); } =20 -static void task_mm_cid_work(struct callback_head *work) +void task_mm_cid_work(struct work_struct *work) { - unsigned long now =3D jiffies, old_scan, next_scan; - struct task_struct *t =3D current; struct cpumask *cidmask; - struct mm_struct *mm; + struct delayed_work *delayed_work =3D container_of(work, struct delayed_w= ork, work); + struct mm_struct *mm =3D container_of(delayed_work, struct mm_struct, mm_= cid_work); int weight, cpu; =20 - SCHED_WARN_ON(t !=3D container_of(work, struct task_struct, cid_work)); - - work->next =3D work; /* Prevent double-add */ - if (t->flags & PF_EXITING) - return; - mm =3D t->mm; - if (!mm) - return; - old_scan =3D READ_ONCE(mm->mm_cid_next_scan); - next_scan =3D now + msecs_to_jiffies(MM_CID_SCAN_DELAY); - if (!old_scan) { - unsigned long res; - - res =3D cmpxchg(&mm->mm_cid_next_scan, old_scan, next_scan); - if (res !=3D old_scan) - old_scan =3D res; - else - old_scan =3D next_scan; - } - if (time_before(now, old_scan)) - return; - if (!try_cmpxchg(&mm->mm_cid_next_scan, &old_scan, next_scan)) - return; cidmask =3D mm_cidmask(mm); + /* Nothing to clear for now */ + if (cpumask_empty(cidmask)) + goto out; /* Clear cids that were not recently used. */ for_each_possible_cpu(cpu) sched_mm_cid_remote_clear_old(mm, cpu); @@ -10570,35 +10547,8 @@ static void task_mm_cid_work(struct callback_head = *work) */ for_each_possible_cpu(cpu) sched_mm_cid_remote_clear_weight(mm, cpu, weight); -} - -void init_sched_mm_cid(struct task_struct *t) -{ - struct mm_struct *mm =3D t->mm; - int mm_users =3D 0; - - if (mm) { - mm_users =3D atomic_read(&mm->mm_users); - if (mm_users =3D=3D 1) - mm->mm_cid_next_scan =3D jiffies + msecs_to_jiffies(MM_CID_SCAN_DELAY); - } - t->cid_work.next =3D &t->cid_work; /* Protect against double add */ - init_task_work(&t->cid_work, task_mm_cid_work); -} - -void task_tick_mm_cid(struct rq *rq, struct task_struct *curr) -{ - struct callback_head *work =3D &curr->cid_work; - unsigned long now =3D jiffies; - - if (!curr->mm || (curr->flags & (PF_EXITING | PF_KTHREAD)) || - work->next !=3D work) - return; - if (time_before(now, READ_ONCE(curr->mm->mm_cid_next_scan))) - return; - - /* No page allocation under rq lock */ - task_work_add(curr, work, TWA_RESUME); +out: + schedule_delayed_work(delayed_work, msecs_to_jiffies(MM_CID_SCAN_DELAY)); } =20 void sched_mm_cid_exit_signals(struct task_struct *t) diff --git a/kernel/sched/sched.h b/kernel/sched/sched.h index 606c96b74ebfa..fc613d9090bed 100644 --- a/kernel/sched/sched.h +++ b/kernel/sched/sched.h @@ -3622,16 +3622,11 @@ extern void sched_dynamic_update(int mode); =20 #ifdef CONFIG_SCHED_MM_CID =20 -#define SCHED_MM_CID_PERIOD_NS (100ULL * 1000000) /* 100ms */ -#define MM_CID_SCAN_DELAY 100 /* 100ms */ - extern raw_spinlock_t cid_lock; extern int use_cid_lock; =20 extern void sched_mm_cid_migrate_from(struct task_struct *t); extern void sched_mm_cid_migrate_to(struct rq *dst_rq, struct task_struct = *t); -extern void task_tick_mm_cid(struct rq *rq, struct task_struct *curr); -extern void init_sched_mm_cid(struct task_struct *t); =20 static inline void __mm_cid_put(struct mm_struct *mm, int cid) { @@ -3899,8 +3894,6 @@ static inline void switch_mm_cid(struct rq *rq, static inline void switch_mm_cid(struct rq *rq, struct task_struct *prev, = struct task_struct *next) { } static inline void sched_mm_cid_migrate_from(struct task_struct *t) { } static inline void sched_mm_cid_migrate_to(struct rq *dst_rq, struct task_= struct *t) { } -static inline void task_tick_mm_cid(struct rq *rq, struct task_struct *cur= r) { } -static inline void init_sched_mm_cid(struct task_struct *t) { } #endif /* !CONFIG_SCHED_MM_CID */ =20 extern u64 avg_vruntime(struct cfs_rq *cfs_rq); --=20 2.48.1