// SPDX-License-Identifier: GPL-2.0-only
 // Copyright (c) 2018-2020, The Linux Foundation. All rights reserved.
 
+#include <linux/cleanup.h>
 #include <linux/module.h>
 #include <linux/init.h>
 #include <linux/io.h>
 static int wsa_macro_probe(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
-       struct reg_default *reg_defaults;
-       struct regmap_config *reg_config;
        struct wsa_macro *wsa;
        kernel_ulong_t flags;
        void __iomem *base;
                return PTR_ERR(base);
 
        wsa->codec_version = lpass_macro_get_codec_version();
+       struct reg_default *reg_defaults __free(kfree) = NULL;
+
        switch (wsa->codec_version) {
        case LPASS_CODEC_VERSION_1_0:
        case LPASS_CODEC_VERSION_1_1:
        case LPASS_CODEC_VERSION_2_1:
                wsa->reg_layout = &wsa_codec_v2_1;
                def_count = ARRAY_SIZE(wsa_defaults) + ARRAY_SIZE(wsa_defaults_v2_1);
-               reg_defaults = devm_kmalloc_array(dev, def_count,
-                                                 sizeof(*reg_defaults),
-                                                 GFP_KERNEL);
+               reg_defaults = kmalloc_array(def_count, sizeof(*reg_defaults),
+                                            GFP_KERNEL);
                if (!reg_defaults)
                        return -ENOMEM;
                memcpy(®_defaults[0], wsa_defaults, sizeof(wsa_defaults));
        case LPASS_CODEC_VERSION_2_8:
                wsa->reg_layout = &wsa_codec_v2_5;
                def_count = ARRAY_SIZE(wsa_defaults) + ARRAY_SIZE(wsa_defaults_v2_5);
-               reg_defaults = devm_kmalloc_array(dev, def_count,
-                                                 sizeof(*reg_defaults),
-                                                 GFP_KERNEL);
+               reg_defaults = kmalloc_array(def_count, sizeof(*reg_defaults),
+                                            GFP_KERNEL);
                if (!reg_defaults)
                        return -ENOMEM;
                memcpy(®_defaults[0], wsa_defaults, sizeof(wsa_defaults));
                return -EINVAL;
        }
 
-       reg_config = devm_kmemdup(dev, &wsa_regmap_config,
-                                 sizeof(*reg_config), GFP_KERNEL);
+       struct regmap_config *reg_config __free(kfree) = kmemdup(&wsa_regmap_config,
+                                                                sizeof(*reg_config),
+                                                                GFP_KERNEL);
        if (!reg_config)
                return -ENOMEM;
 
        if (IS_ERR(wsa->regmap))
                return PTR_ERR(wsa->regmap);
 
-       devm_kfree(dev, reg_config);
-       devm_kfree(dev, reg_defaults);
        dev_set_drvdata(dev, wsa);
 
        wsa->dev = dev;