commit a087ddc9836e607723095a5c0896bdabeaaff3eb
author Fyodor Kyslov <kyslov@google.com> Tue Mar 17 16:55:56 2020 -0700
committer Fyodor Kyslov <kyslov@google.com> Wed Mar 18 01:49:49 2020 +0000
tree 4ad8ae2a388b6ab432c72238d15f002e910ffb08
parent a816bceaa3514190d06417ffbdd64f010de4b062

AV1 RT: Dynamically allocate space for 64x64 blocks

Change-Id: I24d261c446eaede0fdc32effff235f58cc5542a9

av1/encoder/var_based_part.c

diff --git a/av1/encoder/var_based_part.c b/av1/encoder/var_based_part.c
index 2202d37..dddc21f 100644
--- a/av1/encoder/var_based_part.c
+++ b/av1/encoder/var_based_part.c
@@ -79,7 +79,7 @@
 
 typedef struct {
   partition_variance part_variances;
-  v64x64 split[4];
+  v64x64 *split;
 } v128x128;
 
 typedef struct {
@@ -642,11 +642,14 @@
   unsigned int y_sad = UINT_MAX;
   unsigned int y_sad_g = UINT_MAX;
   BLOCK_SIZE bsize = is_small_sb ? BLOCK_64X64 : BLOCK_128X128;
+  v64x64 *vt64x64 = NULL;
 
   // Ref frame used in partitioning.
   MV_REFERENCE_FRAME ref_frame_partition = LAST_FRAME;
 
+  CHECK_MEM_ERROR(cm, vt64x64, aom_malloc(sizeof(*vt64x64) * num_64x64_blocks));
   CHECK_MEM_ERROR(cm, vt, aom_malloc(sizeof(*vt)));
+  vt->split = vt64x64;
 
   int64_t thresholds[5] = { cpi->vbp_thresholds[0], cpi->vbp_thresholds[1],
                             cpi->vbp_thresholds[2], cpi->vbp_thresholds[3],
@@ -1012,6 +1015,7 @@
   }
   chroma_check(cpi, x, bsize, y_sad, is_key_frame);
 
+  if (vt64x64) aom_free(vt64x64);
   if (vt2) aom_free(vt2);
   if (vt) aom_free(vt);
   return 0;