Add trivial smoketest for xpotrf
This commit is contained in:
parent
00c42dc815
commit
845e6d750f
|
@ -394,3 +394,150 @@ CTEST(potrf, bug_695){
|
|||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue