floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add trivial smoketest for xpotrf

This commit is contained in:
Pauli Virtanen 2017-09-30 18:40:03 +02:00
parent 00c42dc815
commit 845e6d750f
1 changed files with 147 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

147
utest/test_potrs.c
View File

@ -394,3 +394,150 @@ CTEST(potrf, bug_695){
CTEST_ERR("%s:%d got NaN", __FILE__, __LINE__); CTEST_ERR("%s:%d got NaN", __FILE__, __LINE__);
} }
} }
// Check potrf factorizes a small problem correctly
CTEST(potrf, smoketest_trivial){
float A1s[4] = {2, 0.3, 0.3, 3};
double A1d[4] = {2, 0.3, 0.3, 3};
openblas_complex_float A1c[4] = {
openblas_make_complex_float(2,0),
openblas_make_complex_float(0.3,0.1),
openblas_make_complex_float(0.3,-0.1),
openblas_make_complex_float(3,0)
};
openblas_complex_double A1z[4] = {
openblas_make_complex_double(2,0),
openblas_make_complex_double(0.3,0.1),
openblas_make_complex_double(0.3,-0.1),
openblas_make_complex_double(3,0)
};
float zeros = 0, ones = 1;
double zerod = 0, oned = 1;
openblas_complex_float zeroc = openblas_make_complex_float(0, 0),
onec = openblas_make_complex_float(1, 0);
openblas_complex_double zeroz = openblas_make_complex_double(0, 0),
onez = openblas_make_complex_float(1, 0);
char uplo, trans1, trans2;
blasint nv = 4;
blasint n = 2;
blasint inc = 1;
blasint info = 0;
int i, j, cycle;
float As[4], Bs[4];
double Ad[4], Bd[4];
openblas_complex_float Ac[4], Bc[4];
openblas_complex_double Az[4], Bz[4];
for (cycle = 0; cycle < 2; ++cycle) {
if (cycle == 0) {
uplo = 'L';
}
else {
uplo = 'U';
}
BLASFUNC(scopy)(&nv, A1s, &inc, As, &inc);
BLASFUNC(dcopy)(&nv, A1d, &inc, Ad, &inc);
BLASFUNC(ccopy)(&nv, (float *)A1c, &inc, (float *)Ac, &inc);
BLASFUNC(zcopy)(&nv, (double *)A1z, &inc, (double *)Az, &inc);
BLASFUNC(spotrf)(&uplo, &n, As, &n, &info);
if (info != 0) {
CTEST_ERR("%s:%d info != 0", __FILE__, __LINE__);
}
BLASFUNC(dpotrf)(&uplo, &n, Ad, &n, &info);
if (info != 0) {
CTEST_ERR("%s:%d info != 0", __FILE__, __LINE__);
}
BLASFUNC(cpotrf)(&uplo, &n, (float *)Ac, &n, &info);
if (info != 0) {
CTEST_ERR("%s:%d info != 0", __FILE__, __LINE__);
}
BLASFUNC(zpotrf)(&uplo, &n, (double *)Az, &n, &info);
if (info != 0) {
CTEST_ERR("%s:%d info != 0", __FILE__, __LINE__);
}
/* Fill the other triangle */
if (uplo == 'L') {
for (i = 0; i < n; ++i) {
for (j = i+1; j < n; ++j) {
As[i+n*j] = 0;
Ad[i+n*j] = 0;
Ac[i+n*j] = zeroc;
Az[i+n*j] = zeroz;
}
}
}
else {
for (i = 0; i < n; ++i) {
for (j = 0; j < i; ++j) {
As[i+n*j] = 0;
Ad[i+n*j] = 0;
Ac[i+n*j] = zeroc;
Az[i+n*j] = zeroz;
}
}
}
/* B = A A^H or A^H A */
if (uplo == 'L') {
trans1 = 'N';
trans2 = 'C';
}
else {
trans1 = 'C';
trans2 = 'N';
}
BLASFUNC(sgemm)(&trans1, &trans2, &n, &n, &n, &ones, As, &n, As, &n, &zeros, Bs, &n);
BLASFUNC(dgemm)(&trans1, &trans2, &n, &n, &n, &oned, Ad, &n, Ad, &n, &zerod, Bd, &n);
BLASFUNC(cgemm)(&trans1, &trans2, &n, &n, &n, (float *)&onec,
(float *)Ac, &n, (float *)Ac, &n, (float *)&zeroc, (float *)Bc, &n);
BLASFUNC(zgemm)(&trans1, &trans2, &n, &n, &n, (double *)&onez,
(double *)Az, &n, (double *)Az, &n, (double *)&zeroz, (double *)Bz, &n);
/* Check result is close to original */
for (i = 0; i < n; ++i) {
for (j = 0; j < n; ++j) {
double err;
err = fabs(A1s[i+n*j] - Bs[i+n*j]);
if (err > 1e-5) {
CTEST_ERR("%s:%d %c s(%d,%d) difference: %g", __FILE__, __LINE__, uplo, i, j, err);
}
err = fabs(A1d[i+n*j] - Bd[i+n*j]);
if (err > 1e-12) {
CTEST_ERR("%s:%d %c d(%d,%d) difference: %g", __FILE__, __LINE__, uplo, i, j, err);
}
#ifdef OPENBLAS_COMPLEX_C99
err = cabsf(A1c[i+n*j] - Bc[i+n*j]);
#else
err = hypot(A1c[i+n*j].real - Bc[i+n*j].real,
A1c[i+n*j].imag - Bc[i+n*j].imag);
#endif
if (err > 1e-5) {
CTEST_ERR("%s:%d %c c(%d,%d) difference: %g", __FILE__, __LINE__, uplo, i, j, err);
}
#ifdef OPENBLAS_COMPLEX_C99
err = cabs(A1z[i+n*j] - Bz[i+n*j]);
#else
err = hypot(A1z[i+n*j].real - Bz[i+n*j].real,
A1z[i+n*j].imag - Bz[i+n*j].imag);
#endif
if (err > 1e-12) {
CTEST_ERR("%s:%d %c z(%d,%d) difference: %g", __FILE__, __LINE__, uplo, i, j, err);
}
}
}
}
}