return 0;
 }
 
-static inline bool
+static inline void
 enable_policy_freq_counters(int cpu, cpumask_var_t valid_cpus)
 {
        struct cpufreq_policy *policy = cpufreq_cpu_get(cpu);
 
        if (!policy) {
                pr_debug("CPU%d: No cpufreq policy found.\n", cpu);
-               return false;
+               return;
        }
 
        if (cpumask_subset(policy->related_cpus, valid_cpus))
                           amu_fie_cpus);
 
        cpufreq_cpu_put(policy);
-
-       return true;
 }
 
 static DEFINE_STATIC_KEY_FALSE(amu_fie_key);
 {
        bool invariance_status = topology_scale_freq_invariant();
        cpumask_var_t valid_cpus;
-       bool have_policy = false;
        int ret = 0;
        int cpu;
 
                        continue;
 
                cpumask_set_cpu(cpu, valid_cpus);
-               have_policy |= enable_policy_freq_counters(cpu, valid_cpus);
+               enable_policy_freq_counters(cpu, valid_cpus);
        }
 
-       /*
-        * If we are not restricted by cpufreq policies, we only enable
-        * the use of the AMU feature for FIE if all CPUs support AMU.
-        * Otherwise, enable_policy_freq_counters has already enabled
-        * policy cpus.
-        */
-       if (!have_policy && cpumask_equal(valid_cpus, cpu_present_mask))
-               cpumask_or(amu_fie_cpus, amu_fie_cpus, valid_cpus);
+       /* Overwrite amu_fie_cpus if all CPUs support AMU */
+       if (cpumask_equal(valid_cpus, cpu_present_mask))
+               cpumask_copy(amu_fie_cpus, cpu_present_mask);
 
        if (!cpumask_empty(amu_fie_cpus)) {
                pr_info("CPUs[%*pbl]: counters will be used for FIE.",