return 0;
 }
 
+static void spi_engine_release_hw(void *p)
+{
+       struct spi_engine *spi_engine = p;
+
+       writel_relaxed(0xff, spi_engine->base + SPI_ENGINE_REG_INT_PENDING);
+       writel_relaxed(0x00, spi_engine->base + SPI_ENGINE_REG_INT_ENABLE);
+       writel_relaxed(0x01, spi_engine->base + SPI_ENGINE_REG_RESET);
+}
+
 static int spi_engine_probe(struct platform_device *pdev)
 {
        struct spi_engine *spi_engine;
        writel_relaxed(0xff, spi_engine->base + SPI_ENGINE_REG_INT_PENDING);
        writel_relaxed(0x00, spi_engine->base + SPI_ENGINE_REG_INT_ENABLE);
 
+       ret = devm_add_action_or_reset(&pdev->dev, spi_engine_release_hw,
+                                      spi_engine);
+       if (ret)
+               return ret;
+
        ret = request_irq(irq, spi_engine_irq, 0, pdev->name, host);
        if (ret)
                return ret;
 static void spi_engine_remove(struct platform_device *pdev)
 {
        struct spi_controller *host = platform_get_drvdata(pdev);
-       struct spi_engine *spi_engine = spi_controller_get_devdata(host);
        int irq = platform_get_irq(pdev, 0);
 
        spi_unregister_controller(host);
 
        free_irq(irq, host);
-
-       writel_relaxed(0xff, spi_engine->base + SPI_ENGINE_REG_INT_PENDING);
-       writel_relaxed(0x00, spi_engine->base + SPI_ENGINE_REG_INT_ENABLE);
-       writel_relaxed(0x01, spi_engine->base + SPI_ENGINE_REG_RESET);
 }
 
 static const struct of_device_id spi_engine_match_table[] = {