return 0;
 }
 
+void mmci_dma_finalize(struct mmci_host *host, struct mmc_data *data)
+{
+       if (!host->use_dma)
+               return;
+
+       if (host->ops && host->ops->dma_finalize)
+               host->ops->dma_finalize(host, data);
+}
+
 static void
 mmci_request_end(struct mmci_host *host, struct mmc_request *mrq)
 {
        mmci_dma_unmap(host, host->data);
 }
 
-static void mmci_dma_finalize(struct mmci_host *host, struct mmc_data *data)
+void mmci_dmae_finalize(struct mmci_host *host, struct mmc_data *data)
 {
        struct mmci_dmae_priv *dmae = host->dma_priv;
        u32 status;
        int i;
 
-       if (!host->use_dma || !dma_inprogress(host))
+       if (!dma_inprogress(host))
                return;
 
        /* Wait up to 1ms for the DMA to complete */
        .dma_setup = mmci_dmae_setup,
        .dma_release = mmci_dmae_release,
        .dma_start = mmci_dmae_start,
+       .dma_finalize = mmci_dmae_finalize,
 };
 
 void mmci_variant_init(struct mmci_host *host)
 }
 #else
 /* Blank functions if the DMA engine is not available */
-static inline void mmci_dma_finalize(struct mmci_host *host,
-                                    struct mmc_data *data)
-{
-}
-
 static inline void mmci_dma_data_error(struct mmci_host *host)
 {
 }
 
        int (*dma_setup)(struct mmci_host *host);
        void (*dma_release)(struct mmci_host *host);
        int (*dma_start)(struct mmci_host *host, unsigned int *datactrl);
+       void (*dma_finalize)(struct mmci_host *host, struct mmc_data *data);
 };
 
 struct mmci_host {
 int mmci_dmae_setup(struct mmci_host *host);
 void mmci_dmae_release(struct mmci_host *host);
 int mmci_dmae_start(struct mmci_host *host, unsigned int *datactrl);
-
+void mmci_dmae_finalize(struct mmci_host *host, struct mmc_data *data);
 
        .dma_setup = qcom_dma_setup,
        .dma_release = mmci_dmae_release,
        .dma_start = mmci_dmae_start,
+       .dma_finalize = mmci_dmae_finalize,
 };
 
 void qcom_variant_init(struct mmci_host *host)