static int __vb2_buf_mem_alloc(struct vb2_buffer *vb)
 {
        struct vb2_queue *q = vb->vb2_queue;
+       enum dma_data_direction dma_dir =
+               V4L2_TYPE_IS_OUTPUT(q->type) ? DMA_TO_DEVICE : DMA_FROM_DEVICE;
        void *mem_priv;
        int plane;
 
                unsigned long size = PAGE_ALIGN(q->plane_sizes[plane]);
 
                mem_priv = call_ptr_memop(vb, alloc, q->alloc_ctx[plane],
-                                     size, q->gfp_flags);
+                                     size, dma_dir, q->gfp_flags);
                if (IS_ERR_OR_NULL(mem_priv))
                        goto free;
 
 
        kfree(buf);
 }
 
-static void *vb2_dc_alloc(void *alloc_ctx, unsigned long size, gfp_t gfp_flags)
+static void *vb2_dc_alloc(void *alloc_ctx, unsigned long size,
+                         enum dma_data_direction dma_dir, gfp_t gfp_flags)
 {
        struct vb2_dc_conf *conf = alloc_ctx;
        struct device *dev = conf->dev;
        /* Prevent the device from being released while the buffer is used */
        buf->dev = get_device(dev);
        buf->size = size;
+       buf->dma_dir = dma_dir;
 
        buf->handler.refcount = &buf->refcount;
        buf->handler.put = vb2_dc_put;
 
        return 0;
 }
 
-static void *vb2_dma_sg_alloc(void *alloc_ctx, unsigned long size, gfp_t gfp_flags)
+static void *vb2_dma_sg_alloc(void *alloc_ctx, unsigned long size,
+                             enum dma_data_direction dma_dir, gfp_t gfp_flags)
 {
        struct vb2_dma_sg_buf *buf;
        int ret;
                return NULL;
 
        buf->vaddr = NULL;
-       buf->dma_dir = DMA_NONE;
+       buf->dma_dir = dma_dir;
        buf->offset = 0;
        buf->size = size;
        /* size is already page aligned */
 
 
 static void vb2_vmalloc_put(void *buf_priv);
 
-static void *vb2_vmalloc_alloc(void *alloc_ctx, unsigned long size, gfp_t gfp_flags)
+static void *vb2_vmalloc_alloc(void *alloc_ctx, unsigned long size,
+                              enum dma_data_direction dma_dir, gfp_t gfp_flags)
 {
        struct vb2_vmalloc_buf *buf;
 
 
        buf->size = size;
        buf->vaddr = vmalloc_user(buf->size);
+       buf->dma_dir = dma_dir;
        buf->handler.refcount = &buf->refcount;
        buf->handler.put = vb2_vmalloc_put;
        buf->handler.arg = buf;
 
  *                               unmap_dmabuf.
  */
 struct vb2_mem_ops {
-       void            *(*alloc)(void *alloc_ctx, unsigned long size, gfp_t gfp_flags);
+       void            *(*alloc)(void *alloc_ctx, unsigned long size,
+                                 enum dma_data_direction dma_dir,
+                                 gfp_t gfp_flags);
        void            (*put)(void *buf_priv);
        struct dma_buf *(*get_dmabuf)(void *buf_priv, unsigned long flags);