[RFC PATCH v7 17/29] sched/rt: Update task's RT runqueue when switching scheduling class

[Yuri Andriaccio <yurand2000@gmail.com>]

Signed-off-by: Yuri Andriaccio <yurand2000@gmail.com>
---
 kernel/sched/rt.c | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/kernel/sched/rt.c b/kernel/sched/rt.c
index cf756639d9e8..b16cc77d7015 100644
--- a/kernel/sched/rt.c
+++ b/kernel/sched/rt.c
@@ -1926,6 +1926,27 @@ void __init init_sched_rt_class(void)
 	}
 }

+#ifdef CONFIG_RT_GROUP_SCHED
+static void switching_to_rt(struct rq *rq, struct task_struct *p)
+{
+	struct task_group *tg = p->sched_task_group;
+	int cpu = rq->cpu;
+
+	if (!rt_group_sched_enabled())
+		return;
+
+	if (tg == &root_task_group) {
+		p->rt.rt_rq = root_task_group.rt_rq[cpu];
+		return;
+	}
+
+	guard(raw_spinlock_irqsave)(dl_bw_lock_of_tg(tg));
+	p->rt.rt_rq = dl_bandwidth_read(tg)->active_context->rt_rq[cpu];
+}
+#else
+static void switching_to_rt(struct rq *rq, struct task_struct *p) {}
+#endif
+
 /*
  * When switching a task to RT, we may overload the runqueue
  * with RT tasks. In this case we try to push them off to
@@ -2127,6 +2148,7 @@ DEFINE_SCHED_CLASS(rt) = {

 	.get_rr_interval	= get_rr_interval_rt,

+	.switching_to		= switching_to_rt,
 	.switched_to		= switched_to_rt,
 	.prio_changed		= prio_changed_rt,

--
2.56.0