checkasm: add linear least square tests

Signed-off-by: Paul B Mahol <onemda@gmail.com>
This commit is contained in:
Rémi Denis-Courmont 2024-05-28 22:56:34 +03:00 committed by Paul B Mahol
parent e0d08f3e6e
commit 98525299e0
5 changed files with 105 additions and 0 deletions

View File

@ -71,6 +71,7 @@ CHECKASMOBJS-$(CONFIG_SWSCALE) += $(SWSCALEOBJS)
AVUTILOBJS += av_tx.o
AVUTILOBJS += fixed_dsp.o
AVUTILOBJS += float_dsp.o
AVUTILOBJS += lls.o
CHECKASMOBJS-$(CONFIG_AVUTIL) += $(AVUTILOBJS)

View File

@ -254,6 +254,7 @@ static const struct {
#if CONFIG_AVUTIL
{ "fixed_dsp", checkasm_check_fixed_dsp },
{ "float_dsp", checkasm_check_float_dsp },
{ "lls", checkasm_check_lls },
{ "av_tx", checkasm_check_av_tx },
#endif
{ NULL }

View File

@ -105,6 +105,7 @@ void checkasm_check_huffyuvdsp(void);
void checkasm_check_idctdsp(void);
void checkasm_check_jpeg2000dsp(void);
void checkasm_check_llauddsp(void);
void checkasm_check_lls(void);
void checkasm_check_llviddsp(void);
void checkasm_check_llviddspenc(void);
void checkasm_check_lpc(void);

101
tests/checkasm/lls.c Normal file
View File

@ -0,0 +1,101 @@
/*
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* FFmpeg is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with FFmpeg; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <float.h>
#include "libavutil/lls.h"
#include "checkasm.h"
#define randomize_buffer(buf) \
do { \
double bmg[2], stddev = 10.0; \
\
for (size_t i = 0; i < MAX_VARS; i += 2) { \
av_bmg_get(&checkasm_lfg, bmg); \
buf[i] = bmg[0] * stddev; \
buf[i + 1] = bmg[1] * stddev; \
} \
} while(0);
static void test_update(LLSModel *lls, const double *var)
{
double refcovar[MAX_VARS][MAX_VARS];
declare_func(void, LLSModel *, const double *);
call_ref(lls, var);
for (size_t i = 0; i < MAX_VARS; i++)
for (size_t j = 0; j < MAX_VARS; j++)
refcovar[i][j] = lls->covariance[i][j];
memset(lls->covariance, 0, sizeof (lls->covariance));
call_new(lls, var);
for (size_t i = 0; i < lls->indep_count; i++)
for (size_t j = i; j < lls->indep_count; j++)
if (!double_near_abs_eps(refcovar[i][j], lls->covariance[i][j],
2 * DBL_EPSILON)) {
fprintf(stderr, "%zu, %zu: %- .12f - %- .12f = % .12g\n", i, j,
refcovar[i][j], lls->covariance[i][j],
refcovar[i][j] - lls->covariance[i][j]);
fail();
}
bench_new(lls, var);
}
#define EPS 0.2
static void test_evaluate(LLSModel *lls, const double *param, int order)
{
double refprod, newprod;
declare_func_float(double, LLSModel *, const double *, int);
refprod = call_ref(lls, param, order);
newprod = call_new(lls, param, order);
if (!double_near_abs_eps(refprod, newprod, EPS)) {
fprintf(stderr, "%- .12f - %- .12f = % .12g\n",
refprod, newprod, refprod - newprod);
fail();
}
if (order == lls->indep_count)
bench_new(lls, param, order);
}
void checkasm_check_lls(void)
{
static const unsigned char counts[] = { 8, 12, MAX_VARS, };
for (size_t i = 0; i < FF_ARRAY_ELEMS(counts); i++) {
LOCAL_ALIGNED_32(double, var, [MAX_VARS]);
LOCAL_ALIGNED_32(double, param, [MAX_VARS]);
LLSModel lls;
avpriv_init_lls(&lls, counts[i]);
randomize_buffer(var);
randomize_buffer(param);
if (check_func(lls.update_lls, "update_lls_%d", counts[i]))
test_update(&lls, var);
for (size_t j = 0; j <= i; j++)
if (check_func(lls.evaluate_lls, "evaluate_lls_%d_%d", counts[i],
counts[j]))
test_evaluate(&lls, param, counts[j]);
}
report("lls");
}

View File

@ -28,6 +28,7 @@ FATE_CHECKASM = fate-checkasm-aacencdsp \
fate-checkasm-idctdsp \
fate-checkasm-jpeg2000dsp \
fate-checkasm-llauddsp \
fate-checkasm-lls \
fate-checkasm-llviddsp \
fate-checkasm-llviddspenc \
fate-checkasm-lpc \