amlogic/vh264: non-idr or non-I frame will set pts_valid

diff --git a/drivers/amlogic/media_modules/frame_provider/decoder/h264/vh264.c b/drivers/amlogic/media_modules/frame_provider/decoder/h264/vh264.c
index 7cbc73c..f5fe8c3 100644 (file)
--- a/drivers/amlogic/media_modules/frame_provider/decoder/h264/vh264.c
+++ b/drivers/amlogic/media_modules/frame_provider/decoder/h264/vh264.c
@@ -2342,7 +2342,8 @@ static void vh264_isr(void)
 
                                h264_pts_count++;
                        } else {
-                               if (!idr_flag)
+                               /* non-idr or non-I frame will set pts_valid */
+                               if (!idr_flag && !(slice_type == SLICE_TYPE_I))
                                        pts_valid = 0;
                        }