diff --git a/kernel/sched/walt/input-boost.c b/kernel/sched/walt/input-boost.c
index 5372d899bd45..382e5208507e 100644
--- a/kernel/sched/walt/input-boost.c
+++ b/kernel/sched/walt/input-boost.c
@@ -128,7 +128,7 @@ static void do_input_boost_rem(struct work_struct *work)
 
 static void do_input_boost(struct work_struct *work)
 {
-	unsigned int i, ret;
+	unsigned int cpu, ret;
 	struct cpu_sync *i_sync_info;
 
 	cancel_delayed_work_sync(&input_boost_rem);
@@ -139,9 +139,9 @@ static void do_input_boost(struct work_struct *work)
 
 	/* Set the input_boost_min for all CPUs in the system */
 	pr_debug("Setting input boost min for all CPUs\n");
-	for (i = 0; i < 8; i++) {
-		i_sync_info = &per_cpu(sync_info, i);
-		i_sync_info->input_boost_min = sysctl_input_boost_freq[i];
+	for_each_possible_cpu(cpu) {
+		i_sync_info = &per_cpu(sync_info, cpu);
+		i_sync_info->input_boost_min = sysctl_input_boost_freq[cpu];
 	}
 
 	/* Update policies for all online CPUs */