diff --git a/drivers/media/platform/rockchip/isp/isp_rockit.c b/drivers/media/platform/rockchip/isp/isp_rockit.c
index 2902bece0298..c069c6f16538 100644
--- a/drivers/media/platform/rockchip/isp/isp_rockit.c
+++ b/drivers/media/platform/rockchip/isp/isp_rockit.c
@@ -290,7 +290,8 @@ int rkisp_rockit_pause_stream(struct rockit_cfg *input_rockit_cfg)
 }
 EXPORT_SYMBOL(rkisp_rockit_pause_stream);
 
-int rkisp_rockit_config_stream(struct rockit_cfg *input_rockit_cfg, int width, int height)
+int rkisp_rockit_config_stream(struct rockit_cfg *input_rockit_cfg,
+				int width, int height, int wrap_line)
 {
 	struct rkisp_stream *stream = NULL;
 	struct rkisp_buffer *isp_buf;
@@ -302,7 +303,7 @@ int rkisp_rockit_config_stream(struct rockit_cfg *input_rockit_cfg, int width, i
 		pr_err("the stream is NULL");
 		return -EINVAL;
 	}
-
+	stream->ispdev->cap_dev.wrap_line = wrap_line;
 	stream->out_fmt.width = width;
 	stream->out_fmt.height = height;
 	stream->out_fmt.plane_fmt[0].bytesperline = 0;
diff --git a/include/soc/rockchip/rockchip_rockit.h b/include/soc/rockchip/rockchip_rockit.h
index 0a897e276843..207298e5d5ca 100644
--- a/include/soc/rockchip/rockchip_rockit.h
+++ b/include/soc/rockchip/rockchip_rockit.h
@@ -80,7 +80,8 @@ int rkisp_rockit_get_ispdev(char **name);
 int rkisp_rockit_buf_queue(struct rockit_cfg *input_rockit_cfg);
 int rkisp_rockit_pause_stream(struct rockit_cfg *input_rockit_cfg);
 int rkisp_rockit_resume_stream(struct rockit_cfg *input_rockit_cfg);
-int rkisp_rockit_config_stream(struct rockit_cfg *input_rockit_cfg, int width, int height);
+int rkisp_rockit_config_stream(struct rockit_cfg *input_rockit_cfg,
+				int width, int height, int wrap_line);
 
 #else
 
@@ -99,7 +100,7 @@ static inline int rkisp_rockit_resume_stream(struct rockit_cfg *input_rockit_cfg
 	return -EINVAL;
 }
 static inline int rkisp_rockit_config_stream(struct rockit_cfg *input_rockit_cfg,
-					     int width, int height)
+					     int width, int height, int wrap_line)
 {
 	return -EINVAL;
 }