From: Zhao Zhili <zhiliz...@tencent.com>
---
tests/checkasm/h264dsp.c | 20 +++++++++++++++-----
1 file changed, 15 insertions(+), 5 deletions(-)
diff --git a/tests/checkasm/h264dsp.c b/tests/checkasm/h264dsp.c
index f5f9650224..a0f8fd858a 100644
--- a/tests/checkasm/h264dsp.c
+++ b/tests/checkasm/h264dsp.c
@@ -328,25 +328,35 @@ static void check_idct_multiple(void)
static void check_idct_dequant(void)
{
static const int depths[5] = { 8, 9, 10, 12, 14 };
- LOCAL_ALIGNED_16(int16_t, src, [16]);
- /* Ensure dst buffers are large enough to hold dctcoefs of all bit-depths.
*/
+ /* Ensure buffers are large enough to hold dctcoefs of all bit-depths. */
+ LOCAL_ALIGNED_16(uint8_t, src_buf, [16 * sizeof(int32_t)]);
LOCAL_ALIGNED_16(uint8_t, dst0, [16 * 16 * sizeof(int32_t)]);
LOCAL_ALIGNED_16(uint8_t, dst1, [16 * 16 * sizeof(int32_t)]);
+ int16_t *src = (int16_t *)src_buf;
int16_t *dst_ref = (int16_t *)dst0;
int16_t *dst_new = (int16_t *)dst1;
H264DSPContext h;
int bit_depth, i, qmul;
declare_func_emms(AV_CPU_FLAG_MMX | AV_CPU_FLAG_SSE2, void, int16_t
*output, int16_t *input, int qmul);
- for (int j = 0; j < 16; j++)
- src[j] = (rnd() % 512) - 256;
-
qmul = rnd() % 4096;
for (i = 0; i < FF_ARRAY_ELEMS(depths); i++) {
bit_depth = depths[i];
ff_h264dsp_init(&h, bit_depth, 1);
+ if (bit_depth == 8) {
+ for (size_t j = 0; j < 16; j++) {
+ int16_t r = (rnd() % 512) - 256;
+ AV_WN16A(&src_buf[j << 1], r);
+ }
+ } else {
+ for (size_t j = 0; j < 16; j++) {
+ int32_t r = (rnd() % (1 << (bit_depth + 1))) - (1 <<
bit_depth);
+ AV_WN32A(&src_buf[j << 2], r);
+ }
+ }
+
memset(dst0, 0, 16 * 16 * SIZEOF_COEF);
memset(dst1, 0, 16 * 16 * SIZEOF_COEF);
--
2.25.1
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".
