User get width and height are 64 align when set format. Need to make
sure all is 64 align when use width and height to calculate buffer size.

Signed-off-by: Yunfei Dong <[email protected]>
---
 drivers/media/platform/mtk-vcodec/vdec/vdec_h264_req_if.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/drivers/media/platform/mtk-vcodec/vdec/vdec_h264_req_if.c 
b/drivers/media/platform/mtk-vcodec/vdec/vdec_h264_req_if.c
index 946c23088308..28c17204f9a1 100644
--- a/drivers/media/platform/mtk-vcodec/vdec/vdec_h264_req_if.c
+++ b/drivers/media/platform/mtk-vcodec/vdec/vdec_h264_req_if.c
@@ -562,8 +562,8 @@ static void get_pic_info(struct vdec_h264_slice_inst *inst,
 {
        struct mtk_vcodec_ctx *ctx = inst->ctx;
 
-       ctx->picinfo.buf_w = (ctx->picinfo.pic_w + 15) & 0xFFFFFFF0;
-       ctx->picinfo.buf_h = (ctx->picinfo.pic_h + 31) & 0xFFFFFFE0;
+       ctx->picinfo.buf_w = ALIGN(ctx->picinfo.pic_w, 64);
+       ctx->picinfo.buf_h = ALIGN(ctx->picinfo.pic_h, 64);
        ctx->picinfo.fb_sz[0] = ctx->picinfo.buf_w * ctx->picinfo.buf_h;
        ctx->picinfo.fb_sz[1] = ctx->picinfo.fb_sz[0] >> 1;
        inst->vsi_ctx.dec.cap_num_planes =
-- 
2.25.1

Reply via email to