sched/core: Introduce sched_set_rq_on/offline() helper

Introduce sched_set_rq_on/offline() helper, so it can be called
in normal or error path simply. No functional changed.

Cc: stable@kernel.org
Signed-off-by: Yang Yingliang <yangyingliang@huawei.com>
Signed-off-by: Peter Zijlstra (Intel) <peterz@infradead.org>
Link: https://lore.kernel.org/r/20240703031610.587047-4-yangyingliang@huaweicloud.com

diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 949473e414f976cf848b67fa90050c4b87ecb405..4d119e930beb1bbb1372f764d1ed66df20418c65 100644 (file)
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -7845,6 +7845,30 @@ void set_rq_offline(struct rq *rq)
        }
 }
 
+static inline void sched_set_rq_online(struct rq *rq, int cpu)
+{
+       struct rq_flags rf;
+
+       rq_lock_irqsave(rq, &rf);
+       if (rq->rd) {
+               BUG_ON(!cpumask_test_cpu(cpu, rq->rd->span));
+               set_rq_online(rq);
+       }
+       rq_unlock_irqrestore(rq, &rf);
+}
+
+static inline void sched_set_rq_offline(struct rq *rq, int cpu)
+{
+       struct rq_flags rf;
+
+       rq_lock_irqsave(rq, &rf);
+       if (rq->rd) {
+               BUG_ON(!cpumask_test_cpu(cpu, rq->rd->span));
+               set_rq_offline(rq);
+       }
+       rq_unlock_irqrestore(rq, &rf);
+}
+
 /*
  * used to mark begin/end of suspend/resume:
  */
@@ -7914,7 +7938,6 @@ static inline void sched_smt_present_dec(int cpu)
 int sched_cpu_activate(unsigned int cpu)
 {
        struct rq *rq = cpu_rq(cpu);
-       struct rq_flags rf;
 
        /*
         * Clear the balance_push callback and prepare to schedule
@@ -7943,12 +7966,7 @@ int sched_cpu_activate(unsigned int cpu)
         * 2) At runtime, if cpuset_cpu_active() fails to rebuild the
         *    domains.
         */
-       rq_lock_irqsave(rq, &rf);
-       if (rq->rd) {
-               BUG_ON(!cpumask_test_cpu(cpu, rq->rd->span));
-               set_rq_online(rq);
-       }
-       rq_unlock_irqrestore(rq, &rf);
+       sched_set_rq_online(rq, cpu);
 
        return 0;
 }
@@ -7956,7 +7974,6 @@ int sched_cpu_activate(unsigned int cpu)
 int sched_cpu_deactivate(unsigned int cpu)
 {
        struct rq *rq = cpu_rq(cpu);
-       struct rq_flags rf;
        int ret;
 
        /*
@@ -7987,12 +8004,7 @@ int sched_cpu_deactivate(unsigned int cpu)
         */
        synchronize_rcu();
 
-       rq_lock_irqsave(rq, &rf);
-       if (rq->rd) {
-               BUG_ON(!cpumask_test_cpu(cpu, rq->rd->span));
-               set_rq_offline(rq);
-       }
-       rq_unlock_irqrestore(rq, &rf);
+       sched_set_rq_offline(rq, cpu);
 
        /*
         * When going down, decrement the number of cores with SMT present.