if (irq < 0)
                return irq;
 
-       host = spi_alloc_host(&pdev->dev, sizeof(*spi_engine));
+       host = devm_spi_alloc_host(&pdev->dev, sizeof(*spi_engine));
        if (!host)
                return -ENOMEM;
 
        spin_lock_init(&spi_engine->lock);
 
        spi_engine->clk = devm_clk_get_enabled(&pdev->dev, "s_axi_aclk");
-       if (IS_ERR(spi_engine->clk)) {
-               ret = PTR_ERR(spi_engine->clk);
-               goto err_put_host;
-       }
+       if (IS_ERR(spi_engine->clk))
+               return PTR_ERR(spi_engine->clk);
 
        spi_engine->ref_clk = devm_clk_get_enabled(&pdev->dev, "spi_clk");
-       if (IS_ERR(spi_engine->ref_clk)) {
-               ret = PTR_ERR(spi_engine->ref_clk);
-               goto err_put_host;
-       }
+       if (IS_ERR(spi_engine->ref_clk))
+               return PTR_ERR(spi_engine->ref_clk);
 
        spi_engine->base = devm_platform_ioremap_resource(pdev, 0);
-       if (IS_ERR(spi_engine->base)) {
-               ret = PTR_ERR(spi_engine->base);
-               goto err_put_host;
-       }
+       if (IS_ERR(spi_engine->base))
+               return PTR_ERR(spi_engine->base);
 
        version = readl(spi_engine->base + SPI_ENGINE_REG_VERSION);
        if (SPI_ENGINE_VERSION_MAJOR(version) != 1) {
                        SPI_ENGINE_VERSION_MAJOR(version),
                        SPI_ENGINE_VERSION_MINOR(version),
                        SPI_ENGINE_VERSION_PATCH(version));
-               ret = -ENODEV;
-               goto err_put_host;
+               return -ENODEV;
        }
 
        writel_relaxed(0x00, spi_engine->base + SPI_ENGINE_REG_RESET);
 
        ret = request_irq(irq, spi_engine_irq, 0, pdev->name, host);
        if (ret)
-               goto err_put_host;
+               return ret;
 
        host->dev.of_node = pdev->dev.of_node;
        host->mode_bits = SPI_CPOL | SPI_CPHA | SPI_3WIRE;
        return 0;
 err_free_irq:
        free_irq(irq, host);
-err_put_host:
-       spi_controller_put(host);
        return ret;
 }
 
 static void spi_engine_remove(struct platform_device *pdev)
 {
-       struct spi_controller *host = spi_controller_get(platform_get_drvdata(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);
 
 
        free_irq(irq, host);
 
-       spi_controller_put(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);