commit 71ddaec1fb8d4b3108bc2d684e62458bbe487fe9
author chiyotsai <chiyotsai@google.com> Fri Apr 16 20:51:39 2021 -0700
committer Chi Yo Tsai <chiyotsai@google.com> Mon Apr 19 17:45:06 2021 +0000
tree ab4a34e63a7a2c2e4ed84feffd9c18558e5d850f
parent 94ee9bfa5736158b19b8701ad7380e392fa456a5

Fix a crash in compute_gradient_info_sb for monochrome images

BUG=aomedia:2996

Change-Id: I20d5a7891d9e3e42bb52991a79e058447e68f030

av1/encoder/intra_mode_search_utils.h

diff --git a/av1/encoder/intra_mode_search_utils.h b/av1/encoder/intra_mode_search_utils.h
index d1a9861..11c2952 100644
--- a/av1/encoder/intra_mode_search_utils.h
+++ b/av1/encoder/intra_mode_search_utils.h
@@ -328,14 +328,15 @@
       sf->part_sf.partition_search_type != SEARCH_PARTITION)
     return;
 
-  av1_setup_src_planes(x, cpi->source, mi_row, mi_col,
-                       av1_num_planes(&cpi->common), sb_size);
+  const int num_planes = av1_num_planes(&cpi->common);
+
+  av1_setup_src_planes(x, cpi->source, mi_row, mi_col, num_planes, sb_size);
 
   if (sf->intra_sf.intra_pruning_with_hog) {
     compute_gradient_info_sb(x, sb_size, PLANE_TYPE_Y);
     x->is_sb_gradient_cached[PLANE_TYPE_Y] = true;
   }
-  if (sf->intra_sf.chroma_intra_pruning_with_hog) {
+  if (sf->intra_sf.chroma_intra_pruning_with_hog && num_planes > 1) {
     compute_gradient_info_sb(x, sb_size, PLANE_TYPE_UV);
     x->is_sb_gradient_cached[PLANE_TYPE_UV] = true;
   }