return ret;
 }
 
-static void mtk_mdp_ctx_lock(struct vb2_queue *vq)
-{
-       struct mtk_mdp_ctx *ctx = vb2_get_drv_priv(vq);
-
-       mutex_lock(&ctx->mdp_dev->lock);
-}
-
-static void mtk_mdp_ctx_unlock(struct vb2_queue *vq)
-{
-       struct mtk_mdp_ctx *ctx = vb2_get_drv_priv(vq);
-
-       mutex_unlock(&ctx->mdp_dev->lock);
-}
-
 static void mtk_mdp_set_frame_size(struct mtk_mdp_frame *frame, int width,
                                   int height)
 {
        .queue_setup     = mtk_mdp_m2m_queue_setup,
        .buf_prepare     = mtk_mdp_m2m_buf_prepare,
        .buf_queue       = mtk_mdp_m2m_buf_queue,
-       .wait_prepare    = mtk_mdp_ctx_unlock,
-       .wait_finish     = mtk_mdp_ctx_lock,
        .stop_streaming  = mtk_mdp_m2m_stop_streaming,
        .start_streaming = mtk_mdp_m2m_start_streaming,
+       .wait_prepare    = vb2_ops_wait_prepare,
+       .wait_finish     = vb2_ops_wait_finish,
 };
 
 static int mtk_mdp_m2m_querycap(struct file *file, void *fh,
        src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
        src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
        src_vq->dev = &ctx->mdp_dev->pdev->dev;
+       src_vq->lock = &ctx->mdp_dev->lock;
 
        ret = vb2_queue_init(src_vq);
        if (ret)
        dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
        dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
        dst_vq->dev = &ctx->mdp_dev->pdev->dev;
+       dst_vq->lock = &ctx->mdp_dev->lock;
 
        return vb2_queue_init(dst_vq);
 }