struct vb2_dc_conf {
        struct device           *dev;
+       struct dma_attrs        attrs;
 };
 
 struct vb2_dc_buf {
        struct device                   *dev;
        void                            *vaddr;
        unsigned long                   size;
+       void                            *cookie;
        dma_addr_t                      dma_addr;
+       struct dma_attrs                attrs;
        enum dma_data_direction         dma_dir;
        struct sg_table                 *dma_sgt;
        struct frame_vector             *vec;
                sg_free_table(buf->sgt_base);
                kfree(buf->sgt_base);
        }
-       dma_free_coherent(buf->dev, buf->size, buf->vaddr, buf->dma_addr);
+       dma_free_attrs(buf->dev, buf->size, buf->cookie, buf->dma_addr,
+                       &buf->attrs);
        put_device(buf->dev);
        kfree(buf);
 }
        if (!buf)
                return ERR_PTR(-ENOMEM);
 
-       buf->vaddr = dma_alloc_coherent(dev, size, &buf->dma_addr,
-                                               GFP_KERNEL | gfp_flags);
-       if (!buf->vaddr) {
+       buf->attrs = conf->attrs;
+       buf->cookie = dma_alloc_attrs(dev, size, &buf->dma_addr,
+                                       GFP_KERNEL | gfp_flags, &buf->attrs);
+       if (!buf->cookie) {
                dev_err(dev, "dma_alloc_coherent of size %ld failed\n", size);
                kfree(buf);
                return ERR_PTR(-ENOMEM);
        }
 
+       if (!dma_get_attr(DMA_ATTR_NO_KERNEL_MAPPING, &buf->attrs))
+               buf->vaddr = buf->cookie;
+
        /* Prevent the device from being released while the buffer is used */
        buf->dev = get_device(dev);
        buf->size = size;
         */
        vma->vm_pgoff = 0;
 
-       ret = dma_mmap_coherent(buf->dev, vma, buf->vaddr,
-               buf->dma_addr, buf->size);
+       ret = dma_mmap_attrs(buf->dev, vma, buf->cookie,
+               buf->dma_addr, buf->size, &buf->attrs);
 
        if (ret) {
                pr_err("Remapping memory failed, error: %d\n", ret);
 {
        struct vb2_dc_buf *buf = dbuf->priv;
 
-       return buf->vaddr + pgnum * PAGE_SIZE;
+       return buf->vaddr ? buf->vaddr + pgnum * PAGE_SIZE : NULL;
 }
 
 static void *vb2_dc_dmabuf_ops_vmap(struct dma_buf *dbuf)
                return NULL;
        }
 
-       ret = dma_get_sgtable(buf->dev, sgt, buf->vaddr, buf->dma_addr,
-               buf->size);
+       ret = dma_get_sgtable_attrs(buf->dev, sgt, buf->cookie, buf->dma_addr,
+               buf->size, &buf->attrs);
        if (ret < 0) {
                dev_err(buf->dev, "failed to get scatterlist from DMA API\n");
                kfree(sgt);
 };
 EXPORT_SYMBOL_GPL(vb2_dma_contig_memops);
 
-void *vb2_dma_contig_init_ctx(struct device *dev)
+void *vb2_dma_contig_init_ctx_attrs(struct device *dev,
+                                   struct dma_attrs *attrs)
 {
        struct vb2_dc_conf *conf;
 
                return ERR_PTR(-ENOMEM);
 
        conf->dev = dev;
+       if (attrs)
+               conf->attrs = *attrs;
 
        return conf;
 }
-EXPORT_SYMBOL_GPL(vb2_dma_contig_init_ctx);
+EXPORT_SYMBOL_GPL(vb2_dma_contig_init_ctx_attrs);
 
 void vb2_dma_contig_cleanup_ctx(void *alloc_ctx)
 {
 
 #include <media/videobuf2-v4l2.h>
 #include <linux/dma-mapping.h>
 
+struct dma_attrs;
+
 static inline dma_addr_t
 vb2_dma_contig_plane_dma_addr(struct vb2_buffer *vb, unsigned int plane_no)
 {
        return *addr;
 }
 
-void *vb2_dma_contig_init_ctx(struct device *dev);
+void *vb2_dma_contig_init_ctx_attrs(struct device *dev,
+                                   struct dma_attrs *attrs);
+
+static inline void *vb2_dma_contig_init_ctx(struct device *dev)
+{
+       return vb2_dma_contig_init_ctx_attrs(dev, NULL);
+}
+
 void vb2_dma_contig_cleanup_ctx(void *alloc_ctx);
 
 extern const struct vb2_mem_ops vb2_dma_contig_memops;