// 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 rx_macro_probe(struct platform_device *pdev)
 {
-       struct reg_default *reg_defaults;
        struct device *dev = &pdev->dev;
        kernel_ulong_t flags;
        struct rx_macro *rx;
                return PTR_ERR(base);
 
        rx->codec_version = lpass_macro_get_codec_version();
+       struct reg_default *reg_defaults __free(kfree) = NULL;
+
        switch (rx->codec_version) {
        case LPASS_CODEC_VERSION_1_0:
        case LPASS_CODEC_VERSION_1_1:
        rx_regmap_config.num_reg_defaults = def_count;
 
        rx->regmap = devm_regmap_init_mmio(dev, base, &rx_regmap_config);
-       if (IS_ERR(rx->regmap)) {
-               ret = PTR_ERR(rx->regmap);
-               goto err_ver;
-       }
+       if (IS_ERR(rx->regmap))
+               return PTR_ERR(rx->regmap);
 
        dev_set_drvdata(dev, rx);
 
 
        ret = clk_prepare_enable(rx->macro);
        if (ret)
-               goto err_ver;
+               return ret;
 
        ret = clk_prepare_enable(rx->dcodec);
        if (ret)
        if (ret)
                goto err_clkout;
 
-       kfree(reg_defaults);
        return 0;
 
 err_clkout:
        clk_disable_unprepare(rx->dcodec);
 err_dcodec:
        clk_disable_unprepare(rx->macro);
-err_ver:
-       kfree(reg_defaults);
 
        return ret;
 }