2 files changed, 21 insertions, 3 deletions

diff --git a/drivers/cpufreq/armada-8k-cpufreq.c b/drivers/cpufreq/armada-8k-cpufreq.c
index 8a5ddb93fc58..b3f4bd647e9b 100644
--- a/drivers/cpufreq/armada-8k-cpufreq.c
+++ b/drivers/cpufreq/armada-8k-cpufreq.c
@@ -128,8 +128,10 @@ static int __init armada_8k_cpufreq_init(void)
         struct cpumask cpus;
 
         node = of_find_compatible_node(NULL, NULL, "marvell,ap806-cpu-clock");
-        if (!node || !of_device_is_available(node))
+        if (!node || !of_device_is_available(node)) {
+                of_node_put(node);
                 return -ENODEV;
+        }
 
         nb_cpus = num_possible_cpus();
         freq_tables = kcalloc(nb_cpus, sizeof(*freq_tables), GFP_KERNEL);
diff --git a/drivers/cpufreq/qcom-cpufreq-kryo.c b/drivers/cpufreq/qcom-cpufreq-kryo.c
index 1c8583cc06a2..dd64dcf89c74 100644
--- a/drivers/cpufreq/qcom-cpufreq-kryo.c
+++ b/drivers/cpufreq/qcom-cpufreq-kryo.c
@@ -75,7 +75,7 @@ static enum _msm8996_version qcom_cpufreq_kryo_get_msm_id(void)
 
 static int qcom_cpufreq_kryo_probe(struct platform_device *pdev)
 {
-        struct opp_table *opp_tables[NR_CPUS] = {0};
+        struct opp_table **opp_tables;
         enum _msm8996_version msm8996_version;
         struct nvmem_cell *speedbin_nvmem;
         struct device_node *np;
@@ -133,6 +133,10 @@ static int qcom_cpufreq_kryo_probe(struct platform_device *pdev)
         }
         kfree(speedbin);
 
+        opp_tables = kcalloc(num_possible_cpus(), sizeof(*opp_tables), GFP_KERNEL);
+        if (!opp_tables)
+                return -ENOMEM;
+
         for_each_possible_cpu(cpu) {
                 cpu_dev = get_cpu_device(cpu);
                 if (NULL == cpu_dev) {
@@ -151,8 +155,10 @@ static int qcom_cpufreq_kryo_probe(struct platform_device *pdev)
 
         cpufreq_dt_pdev = platform_device_register_simple("cpufreq-dt", -1,
                                                           NULL, 0);
-        if (!IS_ERR(cpufreq_dt_pdev))
+        if (!IS_ERR(cpufreq_dt_pdev)) {
+                platform_set_drvdata(pdev, opp_tables);
                 return 0;
+        }
 
         ret = PTR_ERR(cpufreq_dt_pdev);
         dev_err(cpu_dev, "Failed to register platform device\n");
@@ -163,13 +169,23 @@ free_opp:
                         break;
                 dev_pm_opp_put_supported_hw(opp_tables[cpu]);
         }
+        kfree(opp_tables);
 
         return ret;
 }
 
 static int qcom_cpufreq_kryo_remove(struct platform_device *pdev)
 {
+        struct opp_table **opp_tables = platform_get_drvdata(pdev);
+        unsigned int cpu;
+
         platform_device_unregister(cpufreq_dt_pdev);
+
+        for_each_possible_cpu(cpu)
+                dev_pm_opp_put_supported_hw(opp_tables[cpu]);
+
+        kfree(opp_tables);
+
         return 0;
 }