Signed-off-by: Leo Liu <leo....@amd.com>
---
 src/gallium/drivers/radeon/radeon_vcn_dec.c | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/src/gallium/drivers/radeon/radeon_vcn_dec.c 
b/src/gallium/drivers/radeon/radeon_vcn_dec.c
index 170cc3fa41..b4cfba1713 100644
--- a/src/gallium/drivers/radeon/radeon_vcn_dec.c
+++ b/src/gallium/drivers/radeon/radeon_vcn_dec.c
@@ -936,6 +936,32 @@ static struct pb_buffer *rvcn_dec_message_decode(struct 
radeon_decoder *dec,
 
                memcpy(codec, (void*)&vp9, sizeof(rvcn_dec_message_vp9_t));
                index->message_id = RDECODE_MESSAGE_VP9;
+
+               if (dec->ctx.res == NULL) {
+                       unsigned ctx_size;
+                       uint8_t *ptr;
+
+                       /* default probability + probability data */
+                       ctx_size = 2304 * 5;
+
+                       /* SRE collocated context data */
+                       ctx_size += 32 * 2 * 64 * 64;
+
+                       /* SMP collocated context data */
+                       ctx_size += 9 * 64 * 2 * 64 * 64;
+
+                       /* SDB left tile pixel */
+                       ctx_size += 8 * 2 * 4096;
+
+                       if (!si_vid_create_buffer(dec->screen, &dec->ctx, 
ctx_size, PIPE_USAGE_DEFAULT))
+                               RVID_ERR("Can't allocated context buffer.\n");
+                       si_vid_clear_buffer(dec->base.context, &dec->ctx);
+
+                       /* ctx needs probs table */
+                       ptr = dec->ws->buffer_map(dec->ctx.res->buf, dec->cs, 
PIPE_TRANSFER_WRITE);
+                       fill_probs_table(ptr);
+                       dec->ws->buffer_unmap(dec->ctx.res->buf);
+               }
                break;
        }
        default:
-- 
2.14.1

_______________________________________________
mesa-dev mailing list
mesa-dev@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/mesa-dev

Reply via email to