ref: 39dba4cd6d2a43df68916b1fd05b2f2409b92775
dir: /tests/checkasm/msac.c/
/* * Copyright © 2019, VideoLAN and dav1d authors * Copyright © 2019, Two Orioles, LLC * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "tests/checkasm/checkasm.h" #include "src/cpu.h" #include "src/msac.h" #include <string.h> #define BUF_SIZE 8192 /* The normal code doesn't use function pointers */ typedef unsigned (*decode_symbol_adapt_fn)(MsacContext *s, uint16_t *cdf, size_t n_symbols); typedef unsigned (*decode_bool_adapt_fn)(MsacContext *s, uint16_t *cdf); typedef unsigned (*decode_bool_equi_fn)(MsacContext *s); typedef unsigned (*decode_bool_fn)(MsacContext *s, unsigned f); typedef struct { decode_symbol_adapt_fn symbol_adapt4; decode_symbol_adapt_fn symbol_adapt8; decode_symbol_adapt_fn symbol_adapt16; decode_bool_adapt_fn bool_adapt; decode_bool_equi_fn bool_equi; decode_bool_fn bool; } MsacDSPContext; static void randomize_cdf(uint16_t *const cdf, int n) { for (int i = 16; i > n; i--) cdf[i] = rnd(); /* randomize padding */ cdf[n] = cdf[n-1] = 0; while (--n > 0) cdf[n-1] = cdf[n] + rnd() % (32768 - cdf[n] - n) + 1; } /* memcmp() on structs can have weird behavior due to padding etc. */ static int msac_cmp(const MsacContext *const a, const MsacContext *const b) { return a->buf_pos != b->buf_pos || a->buf_end != b->buf_end || a->dif != b->dif || a->rng != b->rng || a->cnt != b->cnt || a->allow_update_cdf != b->allow_update_cdf; } #define CHECK_SYMBOL_ADAPT(n, n_min, n_max) do { \ if (check_func(c->symbol_adapt##n, "msac_decode_symbol_adapt%d", n)) { \ for (int cdf_update = 0; cdf_update <= 1; cdf_update++) { \ for (int ns = n_min; ns <= n_max; ns++) { \ dav1d_msac_init(&s_c, buf, BUF_SIZE, !cdf_update); \ s_a = s_c; \ randomize_cdf(cdf[0], ns); \ memcpy(cdf[1], cdf[0], sizeof(*cdf)); \ for (int i = 0; i < 64; i++) { \ unsigned c_res = call_ref(&s_c, cdf[0], ns); \ unsigned a_res = call_new(&s_a, cdf[1], ns); \ if (c_res != a_res || msac_cmp(&s_c, &s_a) || \ memcmp(cdf[0], cdf[1], sizeof(**cdf) * (ns + 1))) \ { \ fail(); \ } \ } \ if (cdf_update && ns == n) \ bench_new(&s_a, cdf[0], n); \ } \ } \ } \ } while (0) static void check_decode_symbol(MsacDSPContext *const c, uint8_t *const buf) { /* Use an aligned CDF buffer for more consistent benchmark * results, and a misaligned one for checking correctness. */ ALIGN_STK_16(uint16_t, cdf, 2, [17]); MsacContext s_c, s_a; declare_func(unsigned, MsacContext *s, uint16_t *cdf, size_t n_symbols); CHECK_SYMBOL_ADAPT( 4, 1, 5); CHECK_SYMBOL_ADAPT( 8, 1, 8); CHECK_SYMBOL_ADAPT(16, 4, 16); report("decode_symbol"); } static void check_decode_bool(MsacDSPContext *const c, uint8_t *const buf) { MsacContext s_c, s_a; if (check_func(c->bool_adapt, "msac_decode_bool_adapt")) { declare_func(unsigned, MsacContext *s, uint16_t *cdf); uint16_t cdf[2][2]; for (int cdf_update = 0; cdf_update <= 1; cdf_update++) { dav1d_msac_init(&s_c, buf, BUF_SIZE, !cdf_update); s_a = s_c; cdf[0][0] = cdf[1][0] = rnd() % 32767 + 1; cdf[0][1] = cdf[1][1] = 0; for (int i = 0; i < 64; i++) { unsigned c_res = call_ref(&s_c, cdf[0]); unsigned a_res = call_new(&s_a, cdf[1]); if (c_res != a_res || msac_cmp(&s_c, &s_a) || memcmp(cdf[0], cdf[1], sizeof(*cdf))) { fail(); } } if (cdf_update) bench_new(&s_a, cdf[0]); } } if (check_func(c->bool_equi, "msac_decode_bool_equi")) { declare_func(unsigned, MsacContext *s); dav1d_msac_init(&s_c, buf, BUF_SIZE, 1); s_a = s_c; for (int i = 0; i < 64; i++) { unsigned c_res = call_ref(&s_c); unsigned a_res = call_new(&s_a); if (c_res != a_res || msac_cmp(&s_c, &s_a)) fail(); } bench_new(&s_a); } if (check_func(c->bool, "msac_decode_bool")) { declare_func(unsigned, MsacContext *s, unsigned f); dav1d_msac_init(&s_c, buf, BUF_SIZE, 1); s_a = s_c; for (int i = 0; i < 64; i++) { const unsigned f = rnd() & 0x7fff; unsigned c_res = call_ref(&s_c, f); unsigned a_res = call_new(&s_a, f); if (c_res != a_res || msac_cmp(&s_c, &s_a)) fail(); } bench_new(&s_a, 16384); } report("decode_bool"); } void checkasm_check_msac(void) { MsacDSPContext c; c.symbol_adapt4 = dav1d_msac_decode_symbol_adapt_c; c.symbol_adapt8 = dav1d_msac_decode_symbol_adapt_c; c.symbol_adapt16 = dav1d_msac_decode_symbol_adapt_c; c.bool_adapt = dav1d_msac_decode_bool_adapt_c; c.bool_equi = dav1d_msac_decode_bool_equi_c; c.bool = dav1d_msac_decode_bool_c; #if ARCH_AARCH64 && HAVE_ASM if (dav1d_get_cpu_flags() & DAV1D_ARM_CPU_FLAG_NEON) { c.symbol_adapt4 = dav1d_msac_decode_symbol_adapt4_neon; c.symbol_adapt8 = dav1d_msac_decode_symbol_adapt8_neon; c.symbol_adapt16 = dav1d_msac_decode_symbol_adapt16_neon; c.bool_adapt = dav1d_msac_decode_bool_adapt_neon; c.bool_equi = dav1d_msac_decode_bool_equi_neon; c.bool = dav1d_msac_decode_bool_neon; } #elif ARCH_X86 && HAVE_ASM if (dav1d_get_cpu_flags() & DAV1D_X86_CPU_FLAG_SSE2) { c.symbol_adapt4 = dav1d_msac_decode_symbol_adapt4_sse2; c.symbol_adapt8 = dav1d_msac_decode_symbol_adapt8_sse2; c.symbol_adapt16 = dav1d_msac_decode_symbol_adapt16_sse2; c.bool_adapt = dav1d_msac_decode_bool_adapt_sse2; c.bool_equi = dav1d_msac_decode_bool_equi_sse2; c.bool = dav1d_msac_decode_bool_sse2; } #endif uint8_t buf[BUF_SIZE]; for (int i = 0; i < BUF_SIZE; i++) buf[i] = rnd(); check_decode_symbol(&c, buf); check_decode_bool(&c, buf); }