unsigned long           util;
        unsigned long           bw_dl;
-       unsigned long           max;
 
        /* The field below is for single-CPU policies only: */
 #ifdef CONFIG_NO_HZ_COMMON
 {
        struct rq *rq = cpu_rq(sg_cpu->cpu);
 
-       sg_cpu->max = arch_scale_cpu_capacity(sg_cpu->cpu);
        sg_cpu->bw_dl = cpu_bw_dl(rq);
        sg_cpu->util = effective_cpu_util(sg_cpu->cpu, cpu_util_cfs(sg_cpu->cpu),
                                          FREQUENCY_UTIL, NULL);
  * sugov_iowait_apply() - Apply the IO boost to a CPU.
  * @sg_cpu: the sugov data for the cpu to boost
  * @time: the update time from the caller
+ * @max_cap: the max CPU capacity
  *
  * A CPU running a task which woken up after an IO operation can have its
  * utilization boosted to speed up the completion of those IO operations.
  * This mechanism is designed to boost high frequently IO waiting tasks, while
  * being more conservative on tasks which does sporadic IO operations.
  */
-static void sugov_iowait_apply(struct sugov_cpu *sg_cpu, u64 time)
+static void sugov_iowait_apply(struct sugov_cpu *sg_cpu, u64 time,
+                              unsigned long max_cap)
 {
        unsigned long boost;
 
         * sg_cpu->util is already in capacity scale; convert iowait_boost
         * into the same scale so we can compare.
         */
-       boost = (sg_cpu->iowait_boost * sg_cpu->max) >> SCHED_CAPACITY_SHIFT;
+       boost = (sg_cpu->iowait_boost * max_cap) >> SCHED_CAPACITY_SHIFT;
        boost = uclamp_rq_util_with(cpu_rq(sg_cpu->cpu), boost, NULL);
        if (sg_cpu->util < boost)
                sg_cpu->util = boost;
 }
 
 static inline bool sugov_update_single_common(struct sugov_cpu *sg_cpu,
-                                             u64 time, unsigned int flags)
+                                             u64 time, unsigned long max_cap,
+                                             unsigned int flags)
 {
        sugov_iowait_boost(sg_cpu, time, flags);
        sg_cpu->last_update = time;
                return false;
 
        sugov_get_util(sg_cpu);
-       sugov_iowait_apply(sg_cpu, time);
+       sugov_iowait_apply(sg_cpu, time, max_cap);
 
        return true;
 }
        struct sugov_cpu *sg_cpu = container_of(hook, struct sugov_cpu, update_util);
        struct sugov_policy *sg_policy = sg_cpu->sg_policy;
        unsigned int cached_freq = sg_policy->cached_raw_freq;
+       unsigned long max_cap;
        unsigned int next_f;
 
-       if (!sugov_update_single_common(sg_cpu, time, flags))
+       max_cap = arch_scale_cpu_capacity(sg_cpu->cpu);
+
+       if (!sugov_update_single_common(sg_cpu, time, max_cap, flags))
                return;
 
-       next_f = get_next_freq(sg_policy, sg_cpu->util, sg_cpu->max);
+       next_f = get_next_freq(sg_policy, sg_cpu->util, max_cap);
        /*
         * Do not reduce the frequency if the CPU has not been idle
         * recently, as the reduction is likely to be premature then.
 {
        struct sugov_cpu *sg_cpu = container_of(hook, struct sugov_cpu, update_util);
        unsigned long prev_util = sg_cpu->util;
+       unsigned long max_cap;
 
        /*
         * Fall back to the "frequency" path if frequency invariance is not
                return;
        }
 
-       if (!sugov_update_single_common(sg_cpu, time, flags))
+       max_cap = arch_scale_cpu_capacity(sg_cpu->cpu);
+
+       if (!sugov_update_single_common(sg_cpu, time, max_cap, flags))
                return;
 
        /*
                sg_cpu->util = prev_util;
 
        cpufreq_driver_adjust_perf(sg_cpu->cpu, map_util_perf(sg_cpu->bw_dl),
-                                  map_util_perf(sg_cpu->util), sg_cpu->max);
+                                  map_util_perf(sg_cpu->util), max_cap);
 
        sg_cpu->sg_policy->last_freq_update_time = time;
 }
 {
        struct sugov_policy *sg_policy = sg_cpu->sg_policy;
        struct cpufreq_policy *policy = sg_policy->policy;
-       unsigned long util = 0, max = 1;
+       unsigned long util = 0, max_cap;
        unsigned int j;
 
+       max_cap = arch_scale_cpu_capacity(sg_cpu->cpu);
+
        for_each_cpu(j, policy->cpus) {
                struct sugov_cpu *j_sg_cpu = &per_cpu(sugov_cpu, j);
-               unsigned long j_util, j_max;
 
                sugov_get_util(j_sg_cpu);
-               sugov_iowait_apply(j_sg_cpu, time);
-               j_util = j_sg_cpu->util;
-               j_max = j_sg_cpu->max;
+               sugov_iowait_apply(j_sg_cpu, time, max_cap);
 
-               if (j_util * max > j_max * util) {
-                       util = j_util;
-                       max = j_max;
-               }
+               util = max(j_sg_cpu->util, util);
        }
 
-       return get_next_freq(sg_policy, util, max);
+       return get_next_freq(sg_policy, util, max_cap);
 }
 
 static void