struct clk *spi_clk;
        struct clk *ctlr_clk;
        struct clk *axi_clk;
+       struct clk *axi_s_clk;
        unsigned int spi_freq;
        bool wbuf_en;
        bool has_irq;
        clk_disable_unprepare(sp->spi_clk);
        clk_disable_unprepare(sp->ctlr_clk);
        clk_disable_unprepare(sp->axi_clk);
+       clk_disable_unprepare(sp->axi_s_clk);
 }
 
 static int mtk_nor_enable_clk(struct mtk_nor *sp)
                return ret;
        }
 
+       ret = clk_prepare_enable(sp->axi_s_clk);
+       if (ret) {
+               clk_disable_unprepare(sp->spi_clk);
+               clk_disable_unprepare(sp->ctlr_clk);
+               clk_disable_unprepare(sp->axi_clk);
+               return ret;
+       }
+
        return 0;
 }
 
        struct mtk_nor *sp;
        struct mtk_nor_caps *caps;
        void __iomem *base;
-       struct clk *spi_clk, *ctlr_clk, *axi_clk;
+       struct clk *spi_clk, *ctlr_clk, *axi_clk, *axi_s_clk;
        int ret, irq;
 
        base = devm_platform_ioremap_resource(pdev, 0);
        if (IS_ERR(axi_clk))
                return PTR_ERR(axi_clk);
 
+       axi_s_clk = devm_clk_get_optional(&pdev->dev, "axi_s");
+       if (IS_ERR(axi_s_clk))
+               return PTR_ERR(axi_s_clk);
+
        caps = (struct mtk_nor_caps *)of_device_get_match_data(&pdev->dev);
 
        ret = dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(caps->dma_bits));
        sp->spi_clk = spi_clk;
        sp->ctlr_clk = ctlr_clk;
        sp->axi_clk = axi_clk;
+       sp->axi_s_clk = axi_s_clk;
        sp->caps = caps;
        sp->high_dma = caps->dma_bits > 32;
        sp->buffer = dmam_alloc_coherent(&pdev->dev,