Fix av1_write_coeffs_mb when ext_partition is on
Use av1_foreach_transformed_block_in_plane.
Change-Id: I22f92b601f6a90e92a93dbff8492df86c717ab6f
diff --git a/av1/encoder/bitstream.c b/av1/encoder/bitstream.c
index b597ec7..1c04554 100644
--- a/av1/encoder/bitstream.c
+++ b/av1/encoder/bitstream.c
@@ -2097,7 +2097,7 @@
#endif // CONFIG_RD_DEBUG
} else {
#if CONFIG_LV_MAP
- av1_write_coeffs_mb(cm, x, w, plane);
+ av1_write_coeffs_mb(cm, x, w, plane, bsize);
#else
const TX_SIZE tx = av1_get_tx_size(plane, xd);
const int bkw = tx_size_wide_unit[tx];