diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 700dbd19eb44df58185d2642d1c94f08628f7fa3..b9b3f536959d4ff7102ccef38dd7a1e772471452 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -11555,6 +11555,15 @@ static int cpu_affinity_stat_show(struct seq_file *sf, void *v)
 
 	return 0;
 }
+
+static int cpu_rebuild_affinity_domain_u64(struct cgroup_subsys_state *css,
+					   struct cftype *cftype,
+					   u64 cpu)
+{
+	struct task_group *tg = css_tg(css);
+
+	return tg_rebuild_affinity_domains(cpu, tg->auto_affinity);
+}
 #endif /* CONFIG_QOS_SCHED_SMART_GRID */
 
 #ifdef CONFIG_QOS_SCHED
@@ -11735,6 +11744,10 @@ static struct cftype cpu_legacy_files[] = {
 		.name = "affinity_stat",
 		.seq_show = cpu_affinity_stat_show,
 	},
+	{
+		.name = "rebuild_affinity_domain",
+		.write_u64 = cpu_rebuild_affinity_domain_u64,
+	},
 #endif
 #ifdef CONFIG_CFS_BANDWIDTH
 	{
diff --git a/kernel/sched/fair.c b/kernel/sched/fair.c
index 240e315d8e02a56c8185c0e111a02b004b27da10..e27058ab099d5270656ae134409b26115c9ab844 100644
--- a/kernel/sched/fair.c
+++ b/kernel/sched/fair.c
@@ -7252,6 +7252,49 @@ static void destroy_auto_affinity(struct task_group *tg)
 	kfree(tg->auto_affinity);
 	tg->auto_affinity = NULL;
 }
+
+int tg_rebuild_affinity_domains(int cpu, struct auto_affinity *auto_affi)
+{
+	int ret = 0;
+	int level = 0;
+	struct sched_domain *tmp;
+
+	if (unlikely(!auto_affi))
+		return -EPERM;
+
+	mutex_lock(&smart_grid_used_mutex);
+	raw_spin_lock_irq(&auto_affi->lock);
+	/* Only build domain while auto mode disabled */
+	if (auto_affi->mode) {
+		ret = -EPERM;
+		goto unlock_all;
+	}
+
+	/* Only build on active and housekeeping cpu */
+	if (!cpu_active(cpu) || !housekeeping_cpu(cpu, HK_TYPE_DOMAIN)) {
+		ret = -EINVAL;
+		goto unlock_all;
+	}
+
+	for_each_domain(cpu, tmp) {
+		if (!auto_affi->ad.domains[level] || !auto_affi->ad.domains_orig[level])
+			continue;
+
+		/* rebuild domain[,_orig] and reset schedstat counter */
+		cpumask_copy(auto_affi->ad.domains[level], sched_domain_span(tmp));
+		cpumask_copy(auto_affi->ad.domains_orig[level], auto_affi->ad.domains[level]);
+		__schedstat_set(auto_affi->ad.stay_cnt[level], 0);
+		level++;
+	}
+
+	/* trigger to update smart grid zone */
+	sched_grid_zone_update(false);
+
+unlock_all:
+	raw_spin_unlock_irq(&auto_affi->lock);
+	mutex_unlock(&smart_grid_used_mutex);
+	return ret;
+}
 #else
 static void __maybe_unused destroy_auto_affinity(struct task_group *tg) {}
 
diff --git a/kernel/sched/sched.h b/kernel/sched/sched.h
index 9de2bac6444f0bd6b79603fafa8558699d280dc1..79b93b0a2ef0ba13b4c24f25e0bdfe32f7447344 100644
--- a/kernel/sched/sched.h
+++ b/kernel/sched/sched.h
@@ -551,6 +551,7 @@ extern void start_auto_affinity(struct auto_affinity *auto_affi);
 extern void stop_auto_affinity(struct auto_affinity *auto_affi);
 extern int init_auto_affinity(struct task_group *tg);
 extern void tg_update_affinity_domains(int cpu, int online);
+extern int tg_rebuild_affinity_domains(int cpu, struct auto_affinity *auto_affi);
 
 #else
 static inline int init_auto_affinity(struct task_group *tg)