commit 680ba4fafa222313735d9938e8d232aa11c05c39
author Yunqing Wang <yunqingwang@google.com> Fri Mar 05 14:21:51 2021 -0800
committer Yunqing Wang <yunqingwang@google.com> Fri Mar 05 23:26:45 2021 +0000
tree d3713ace13acd9966f2e22f998e53b21b0e53329
parent 0ea7211e43c41b727d50c695e143ef9f8ad31bd9

Record the correct end time in tpl_setup_stats

Change-Id: I2ed2135d5fa36ebf92af5d39e15507c4360e69d0

av1/encoder/tpl_model.c

diff --git a/av1/encoder/tpl_model.c b/av1/encoder/tpl_model.c
index 62a3488..234afe8 100644
--- a/av1/encoder/tpl_model.c
+++ b/av1/encoder/tpl_model.c
@@ -1535,6 +1535,13 @@
   cm->current_frame.frame_type = frame_params->frame_type;
   cm->show_frame = frame_params->show_frame;
 
+#if CONFIG_COLLECT_COMPONENT_TIMING
+  // Record the time if the function returns.
+  if (cpi->common.tiles.large_scale || gf_group->max_layer_depth_allowed == 0 ||
+      !gop_eval)
+    end_timing(cpi, av1_tpl_setup_stats_time);
+#endif
+
   if (cpi->common.tiles.large_scale) return 0;
   if (gf_group->max_layer_depth_allowed == 0) return 1;
   if (!gop_eval) return 0;