commit 26ef59d2d7745ef1d438f50fa565bd2700e1baea
author James Zern <jzern@google.com> Thu Apr 28 12:54:06 2022 -0700
committer James Zern <jzern@google.com> Fri May 06 18:35:17 2022 +0000
tree 7b20cae4d63cd7452baacaefe332b4e28e711216
parent 61f1e35f2db473de5b9971ee754fdc6c43a96aae

av1_rd_partition_search: add missing alloc check

Bug: aomedia:3276
Change-Id: Ibe72737de4715818094c99fcded494d5aeb3abc7

av1/encoder/partition_search.c

diff --git a/av1/encoder/partition_search.c b/av1/encoder/partition_search.c
index a0d071a..60365cf 100644
--- a/av1/encoder/partition_search.c
+++ b/av1/encoder/partition_search.c
@@ -4852,7 +4852,7 @@
     PC_TREE *const pc_tree = av1_alloc_pc_tree_node(bsize);
     num_configs = read_partition_tree(cpi, pc_tree, i);
     if (i == 0) {
-      rdcost = aom_calloc(num_configs, sizeof(*rdcost));
+      CHECK_MEM_ERROR(cm, rdcost, aom_calloc(num_configs, sizeof(*rdcost)));
     }
     if (num_configs <= 0) {
       av1_free_pc_tree_recursive(pc_tree, av1_num_planes(cm), 0, 0);