summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--decoder/ihevcd_utils.c5
1 files changed, 4 insertions, 1 deletions
diff --git a/decoder/ihevcd_utils.c b/decoder/ihevcd_utils.c
index 1cdf3be..4587694 100644
--- a/decoder/ihevcd_utils.c
+++ b/decoder/ihevcd_utils.c
@@ -271,6 +271,9 @@ WORD32 ihevcd_get_pic_mv_bank_size(WORD32 num_luma_samples)
/* Size for storing pu_t for each PU */
mv_bank_size += num_pu * sizeof(pu_t);
+ /* Size for storing slice_idx for each CTB */
+ mv_bank_size += ALIGN4(num_ctb * sizeof(UWORD16));
+
size = mv_bank_size;
return size;
}
@@ -635,7 +638,7 @@ IHEVCD_ERROR_T ihevcd_mv_buf_mgr_add_bufs(codec_t *ps_codec)
pu1_buf += num_pu;
ps_mv_buf->pu1_pic_slice_map = (UWORD16 *)pu1_buf;
- pu1_buf += num_ctb * sizeof(UWORD16);
+ pu1_buf += ALIGN4(num_ctb * sizeof(UWORD16));
ps_mv_buf->ps_pic_pu = (pu_t *)pu1_buf;