1 files changed, 6 insertions, 0 deletions

diff --git a/drivers/regulator/axp20x-regulator.c b/drivers/regulator/axp20x-regulator.c
index e2608fe770b9..f18b36dd57dd 100644
--- a/drivers/regulator/axp20x-regulator.c
+++ b/drivers/regulator/axp20x-regulator.c
@@ -691,6 +691,9 @@ static int axp20x_regulator_probe(struct platform_device *pdev)
                     (regulators == axp809_regulators && i == AXP809_DC1SW)) {
                         new_desc = devm_kzalloc(&pdev->dev, sizeof(*desc),
                                                 GFP_KERNEL);
+                        if (!new_desc)
+                                return -ENOMEM;
+
                         *new_desc = regulators[i];
                         new_desc->supply_name = dcdc1_name;
                         desc = new_desc;
@@ -700,6 +703,9 @@ static int axp20x_regulator_probe(struct platform_device *pdev)
                     (regulators == axp809_regulators && i == AXP809_DC5LDO)) {
                         new_desc = devm_kzalloc(&pdev->dev, sizeof(*desc),
                                                 GFP_KERNEL);
+                        if (!new_desc)
+                                return -ENOMEM;
+
                         *new_desc = regulators[i];
                         new_desc->supply_name = dcdc5_name;
                         desc = new_desc;