#define DRIVER_NAME "mmci-pl18x"
 
+#ifdef CONFIG_DMA_ENGINE
+void mmci_variant_init(struct mmci_host *host);
+#else
+static inline void mmci_variant_init(struct mmci_host *host) {}
+#endif
+
 static unsigned int fmax = 515633;
 
 static struct variant_data variant_arm = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_ROD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_arm_extended_fifo = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_ROD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_arm_extended_fifo_hwfc = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_ROD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_u300 = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_OD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_nomadik = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_OD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_ux500 = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_OD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_ux500v2 = {
        .mmcimask1              = true,
        .start_err              = MCI_STARTBITERR,
        .opendrain              = MCI_OD,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_stm32 = {
        .f_max                  = 48000000,
        .pwrreg_clkgate         = true,
        .pwrreg_nopower         = true,
+       .init                   = mmci_variant_init,
 };
 
 static struct variant_data variant_qcom = {
        mmci_write_clkreg(host, clk);
 }
 
+void mmci_dma_release(struct mmci_host *host)
+{
+       if (host->ops && host->ops->dma_release)
+               host->ops->dma_release(host);
+
+       host->use_dma = false;
+}
+
+void mmci_dma_setup(struct mmci_host *host)
+{
+       if (!host->ops || !host->ops->dma_setup)
+               return;
+
+       if (host->ops->dma_setup(host))
+               return;
+
+       host->use_dma = true;
+}
+
 static void
 mmci_request_end(struct mmci_host *host, struct mmc_request *mrq)
 {
  * no custom DMA interfaces are supported.
  */
 #ifdef CONFIG_DMA_ENGINE
-static void mmci_dma_setup(struct mmci_host *host)
+int mmci_dmae_setup(struct mmci_host *host)
 {
        const char *rxname, *txname;
 
                        host->mmc->max_seg_size = max_seg_size;
        }
 
-       if (host->ops && host->ops->dma_setup)
-               host->ops->dma_setup(host);
+       if (!host->dma_tx_channel && !host->dma_rx_channel) {
+               mmci_dmae_release(host);
+               return -EINVAL;
+       }
+
+       return 0;
 }
 
 /*
  * This is used in or so inline it
  * so it can be discarded.
  */
-static inline void mmci_dma_release(struct mmci_host *host)
+void mmci_dmae_release(struct mmci_host *host)
 {
        if (host->dma_rx_channel)
                dma_release_channel(host->dma_rx_channel);
 
 static void mmci_dma_data_error(struct mmci_host *host)
 {
-       if (!dma_inprogress(host))
+       if (!host->use_dma || !dma_inprogress(host))
                return;
 
        dev_err(mmc_dev(host->mmc), "error during DMA transfer!\n");
        u32 status;
        int i;
 
-       if (!dma_inprogress(host))
+       if (!host->use_dma || !dma_inprogress(host))
                return;
 
        /* Wait up to 1ms for the DMA to complete */
        int ret;
        struct mmc_data *data = host->data;
 
+       if (!host->use_dma)
+               return -EINVAL;
+
        ret = mmci_dma_prep_data(host, host->data);
        if (ret)
                return ret;
 {
        struct mmci_host_next *next = &host->next_data;
 
+       if (!host->use_dma)
+               return;
+
        WARN_ON(data->host_cookie && data->host_cookie != next->cookie);
        WARN_ON(!data->host_cookie && (next->dma_desc || next->dma_chan));
 
        struct mmc_data *data = mrq->data;
        struct mmci_host_next *nd = &host->next_data;
 
-       if (!data)
+       if (!host->use_dma || !data)
                return;
 
        BUG_ON(data->host_cookie);
        struct mmci_host *host = mmc_priv(mmc);
        struct mmc_data *data = mrq->data;
 
-       if (!data || !data->host_cookie)
+       if (!host->use_dma || !data || !data->host_cookie)
                return;
 
        mmci_dma_unmap(host, data);
        }
 }
 
+static struct mmci_host_ops mmci_variant_ops = {
+       .dma_setup = mmci_dmae_setup,
+       .dma_release = mmci_dmae_release,
+};
+
+void mmci_variant_init(struct mmci_host *host)
+{
+       host->ops = &mmci_variant_ops;
+}
 #else
 /* Blank functions if the DMA engine is not available */
 static void mmci_get_next_data(struct mmci_host *host, struct mmc_data *data)
 {
 }
-static inline void mmci_dma_setup(struct mmci_host *host)
-{
-}
-
-static inline void mmci_dma_release(struct mmci_host *host)
-{
-}
 
 static inline void mmci_dma_finalize(struct mmci_host *host,
                                     struct mmc_data *data)
 
 }
 
 /* Initialize the dml hardware connected to SD Card controller */
-static void qcom_dma_setup(struct mmci_host *host)
+static int qcom_dma_setup(struct mmci_host *host)
 {
        u32 config;
        void __iomem *base;
        int consumer_id, producer_id;
        struct device_node *np = host->mmc->parent->of_node;
 
+       if (mmci_dmae_setup(host))
+               return -EINVAL;
+
        consumer_id = of_get_dml_pipe_index(np, "tx");
        producer_id = of_get_dml_pipe_index(np, "rx");
 
        if (producer_id < 0 || consumer_id < 0) {
                host->variant->qcom_dml = false;
-               return;
+               mmci_dmae_release(host);
+               return -EINVAL;
        }
 
        base = host->base + DML_OFFSET;
 
        /* Make sure dml initialization is finished */
        mb();
+
+       return 0;
 }
 
 static struct mmci_host_ops qcom_variant_ops = {
        .dma_setup = qcom_dma_setup,
+       .dma_release = mmci_dmae_release,
 };
 
 void qcom_variant_init(struct mmci_host *host)