1 files changed, 4 insertions, 2 deletions

diff --git a/include/media/videobuf2-core.h b/include/media/videobuf2-core.h
index 70ace7cc6087..d607871749c7 100644
--- a/include/media/videobuf2-core.h
+++ b/include/media/videobuf2-core.h
@@ -87,14 +87,16 @@ struct vb2_mem_ops {
         struct dma_buf *(*get_dmabuf)(void *buf_priv, unsigned long flags);
 
         void            *(*get_userptr)(void *alloc_ctx, unsigned long vaddr,
-                                        unsigned long size, int write);
+                                        unsigned long size,
+                                        enum dma_data_direction dma_dir);
         void            (*put_userptr)(void *buf_priv);
 
         void            (*prepare)(void *buf_priv);
         void            (*finish)(void *buf_priv);
 
         void            *(*attach_dmabuf)(void *alloc_ctx, struct dma_buf *dbuf,
-                                unsigned long size, int write);
+                                          unsigned long size,
+                                          enum dma_data_direction dma_dir);
         void            (*detach_dmabuf)(void *buf_priv);
         int             (*map_dmabuf)(void *buf_priv);
         void            (*unmap_dmabuf)(void *buf_priv);