test_kernel_regress.c

#include"openblas_utest.h"
#include<stdio.h>
#include<stdlib.h>
#include<cblas.h>

#defineLAPACK_ROW_MAJOR 101
blasintLAPACKE_dgesvd( blasintmatrix_layout, charjobu, charjobvt,
blasintm, blasintn, double*a,
blasintlda, double*s, double*u, blasintldu,
double*vt, blasintldvt, double*superb );


#defineDATASIZE 100

doubles[DATASIZE];
doubleu[DATASIZE*DATASIZE];
doublevt[DATASIZE*DATASIZE];
doubleX[DATASIZE*DATASIZE];
doublesuperb[DATASIZE];
doubletmp[DATASIZE*DATASIZE];
doublem[DATASIZE*DATASIZE];

CTEST(kernel_regress,skx_avx)
{
#ifdefBUILD_DOUBLE
doublenorm;
inti, j, info;
srand(0);
for (i=0; i<DATASIZE*DATASIZE; i++) {
m[i] = (rand()+0.0)/RAND_MAX*10;
tmp[i] =m[i];
 }

info=LAPACKE_dgesvd( LAPACK_ROW_MAJOR, 'A', 'A', DATASIZE, DATASIZE, m, DATASIZE,
s, u, DATASIZE, vt, DATASIZE, superb);

for (i=0; i<DATASIZE; i++) {
for (j=0; j<DATASIZE; j++) {
u[i*DATASIZE+j] =u[i*DATASIZE+j]*s[j];
 }
 }
cblas_dgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, 
DATASIZE, DATASIZE, DATASIZE, 1, u, DATASIZE, vt, DATASIZE, 0, X, DATASIZE);

for (i=0; i<DATASIZE*DATASIZE; i++) {
X[i] =X[i] -tmp[i];
 }

norm=cblas_dnrm2(DATASIZE*DATASIZE, X, 1);
ASSERT_DBL_NEAR_TOL(0.0, norm, 1e-10);
#endif
}