1 files changed, 5 insertions, 1 deletions

diff --git a/kernel/sched/topology.c b/kernel/sched/topology.c
index 9318acf1d1fe..49b835f1305f 100644
--- a/kernel/sched/topology.c
+++ b/kernel/sched/topology.c
@@ -2029,7 +2029,7 @@ build_sched_domains(const struct cpumask *cpu_map, struct sched_domain_attr *att
         rcu_read_unlock();
 
         if (has_asym)
-                static_branch_enable_cpuslocked(&sched_asym_cpucapacity);
+                static_branch_inc_cpuslocked(&sched_asym_cpucapacity);
 
         if (rq && sched_debug_enabled) {
                 pr_info("root domain span: %*pbl (max cpu_capacity = %lu)\n",
@@ -2124,8 +2124,12 @@ int sched_init_domains(const struct cpumask *cpu_map)
  */
 static void detach_destroy_domains(const struct cpumask *cpu_map)
 {
+        unsigned int cpu = cpumask_any(cpu_map);
         int i;
 
+        if (rcu_access_pointer(per_cpu(sd_asym_cpucapacity, cpu)))
+                static_branch_dec_cpuslocked(&sched_asym_cpucapacity);
+
         rcu_read_lock();
         for_each_cpu(i, cpu_map)
                 cpu_attach_domain(NULL, &def_root_domain, i);