sched: Fix memory leak on error branch

hulk inclusion
category: bugfix
bugzilla: https://gitee.com/openeuler/kernel/issues/I7EBNA
CVE: NA

-------------------------------

Fix memory leak on error branch for smart grid.

Fixes: 713cfd26 ("sched: Introduce smart grid scheduling strategy for cfs")
Signed-off-by: NHui Tang <tanghui20@huawei.com>
Reviewed-by: NZhang Qiao <zhangqiao22@huawei.com>
Signed-off-by: NZhang Changzhong <zhangchangzhong@huawei.com>

kernel/sched/fair.c

@@ -5628,7 +5628,6 @@ static int init_affinity_domains(struct affinity_domain *ad)
 	}
 
 	if (!sd) {
-		ad->dcount = 0;
 		rcu_read_unlock();
 		return -EINVAL;
 	}
@@ -5636,9 +5635,11 @@ static int init_affinity_domains(struct affinity_domain *ad)
 
 	for (i = 0; i < dcount; i++) {
 		ad->domains[i] = kmalloc(sizeof(cpumask_t), GFP_KERNEL);
-		if (!ad->domains[i])
+		if (!ad->domains[i]) {
+			ad->dcount = i;
 			goto err;
 		}
+	}
 
 	rcu_read_lock();
 	idlest = sd_find_idlest_group(sd);