Merge pull request #1318 from pv/potrf-smoketest

Add trivial smoketest for xpotrf

Author: martin-frbg

utest/test_potrs.c

@@ -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);
+        }
+      }
+    }
+  }
+}