return sg_count;
 }
 
-static void cqhci_set_tran_desc(u8 *desc, dma_addr_t addr, int len, bool end,
-                               bool dma64)
+void cqhci_set_tran_desc(u8 *desc, dma_addr_t addr, int len, bool end,
+                        bool dma64)
 {
        __le32 *attr = (__le32 __force *)desc;
 
                dataddr[0] = cpu_to_le32(addr);
        }
 }
+EXPORT_SYMBOL(cqhci_set_tran_desc);
 
 static int cqhci_prep_tran_desc(struct mmc_request *mrq,
                               struct cqhci_host *cq_host, int tag)
 
                if ((i+1) == sg_count)
                        end = true;
-               cqhci_set_tran_desc(desc, addr, len, end, dma64);
+               if (cq_host->ops->set_tran_desc)
+                       cq_host->ops->set_tran_desc(cq_host, &desc, addr, len, end, dma64);
+               else
+                       cqhci_set_tran_desc(desc, addr, len, end, dma64);
+
                desc += cq_host->trans_desc_len;
        }
 
 
        int (*program_key)(struct cqhci_host *cq_host,
                           const union cqhci_crypto_cfg_entry *cfg, int slot);
 #endif
+       void (*set_tran_desc)(struct cqhci_host *cq_host, u8 **desc,
+                             dma_addr_t addr, int len, bool end, bool dma64);
+
 };
 
 static inline void cqhci_writel(struct cqhci_host *host, u32 val, int reg)
 int cqhci_init(struct cqhci_host *cq_host, struct mmc_host *mmc, bool dma64);
 struct cqhci_host *cqhci_pltfm_init(struct platform_device *pdev);
 int cqhci_deactivate(struct mmc_host *mmc);
+void cqhci_set_tran_desc(u8 *desc, dma_addr_t addr, int len, bool end, bool dma64);
 static inline int cqhci_suspend(struct mmc_host *mmc)
 {
        return cqhci_deactivate(mmc);