Add tests varying alpha and beta

Author: martin-frbg

test/compare_sgemm_shgemm.c

@@ -41,9 +41,11 @@ main (int argc, char *argv[])
   int i, j, l;
   blasint x, y;
   int ret = 0;
+  int rret = 0;
   int loop = SHGEMM_LARGEST;
   char transA = 'N', transB = 'N';
   float alpha = 1.0, beta = 0.0;
+  int xvals[6]={3,24,55,71,SHGEMM_LARGEST/2,SHGEMM_LARGEST};
 
   for (x = 0; x <= loop; x++)
   {
@@ -52,8 +54,8 @@ main (int argc, char *argv[])
     float *A = (float *)malloc_safe(m * k * sizeof(FLOAT));
     float *B = (float *)malloc_safe(k * n * sizeof(FLOAT));
     float *C = (float *)malloc_safe(m * n * sizeof(FLOAT));
-    hfloat16 *AA = (hfloat16 *)malloc_safe(m * k * sizeof(hfloat16));
-    hfloat16 *BB = (hfloat16 *)malloc_safe(k * n * sizeof(hfloat16));
+    _Float16 *AA = (_Float16 *)malloc_safe(m * k * sizeof(_Float16));
+    _Float16 *BB = (_Float16 *)malloc_safe(k * n * sizeof(_Float16));
     float *DD = (float *)malloc_safe(m * n * sizeof(FLOAT));
     float *CC = (float *)malloc_safe(m * n * sizeof(FLOAT));
     if ((A == NULL) || (B == NULL) || (C == NULL) || (AA == NULL) || (BB == NULL) ||
@@ -65,15 +67,15 @@ main (int argc, char *argv[])
       for (i = 0; i < k; i++)
       {
         A[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
-        AA[j * k + i] = (hfloat16) A[j * k + i];
+        AA[j * k + i] = (_Float16) A[j * k + i];
       }
     }
     for (j = 0; j < n; j++)
     {
       for (i = 0; i < k; i++)
       {
         B[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
-        BB[j * k + i] = (hfloat16) B[j * k + i];
+        BB[j * k + i] = (_Float16) B[j * k + i];
       }
     }
     for (y = 0; y < 4; y++)
@@ -95,8 +97,8 @@ main (int argc, char *argv[])
 
       SGEMM (&transA, &transB, &m, &n, &k, &alpha, A,
         &m, B, &k, &beta, C, &m);
-      SHGEMM (&transA, &transB, &m, &n, &k, &alpha,  AA,
-        &m, BB, &k, &beta, CC, &m);
+      SHGEMM (&transA, &transB, &m, &n, &k, &alpha, (_Float16*) AA,
+        &m, (_Float16*)BB, &k, &beta, CC, &m);
 
       for (i = 0; i < n; i++)
         for (j = 0; j < m; j++)
@@ -120,9 +122,11 @@ main (int argc, char *argv[])
                 (float)AA[k * j + l] * (float)BB[i + l * n];
             }
           if (!is_close(CC[i * m + j], C[i * m + j], 0.01, 0.001)) {
+		  fprintf(stderr,"CC %f C %f \n",(float)CC[i*m+j],C[i*m+j]);
             ret++;
           }
           if (!is_close(CC[i * m + j], DD[i * m + j], 0.001, 0.0001)) {
+		  fprintf(stderr,"CC %f DD  %f \n",(float)CC[i*m+j],(float)DD[i*m+j]);
             ret++;
           }
         }
@@ -135,11 +139,96 @@ main (int argc, char *argv[])
     free(DD);
     free(CC);
   }
-
   if (ret != 0) {
-    fprintf(stderr, "SHGEMM FAILURES: %d\n", ret);
+    fprintf(stderr, "SHGEMM FAILURES: %d!!!\n", ret);
     return 1;
   }
 
-  return ret;
+ 
+  for (loop = 0; loop<6; loop++) {
+  x=xvals[loop];
+  for (alpha=0.;alpha<=1.;alpha+=0.5)  
+  {
+   for (beta = 0.0; beta <=1.; beta+=0.5) {
+   
+    m = k = n = x;
+    float *A = (float *)malloc_safe(m * k * sizeof(FLOAT));
+    float *B = (float *)malloc_safe(k * n * sizeof(FLOAT));
+    float *C = (float *)malloc_safe(m * n * sizeof(FLOAT));
+    _Float16 *AA = (_Float16 *)malloc_safe(m * k * sizeof(_Float16));
+    _Float16 *BB = (_Float16 *)malloc_safe(k * n * sizeof(_Float16));
+    float *CC = (float *)malloc_safe(m * n * sizeof(FLOAT));
+    if ((A == NULL) || (B == NULL) || (C == NULL) || (AA == NULL) || (BB == NULL) ||
+       (CC == NULL))
+      return 1;
+
+    for (j = 0; j < m; j++)
+    {
+      for (i = 0; i < k; i++)
+      {
+        A[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
+        AA[j * k + i] = (_Float16) A[j * k + i];
+      }
+    }
+    for (j = 0; j < n; j++)
+    {
+      for (i = 0; i < k; i++)
+      {
+        B[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
+        BB[j * k + i] = (_Float16) B[j * k + i];
+      }
+    }
+
+    for (y = 0; y < 4; y++)
+    {
+      if ((y == 0) || (y == 2)) {
+        transA = 'N';
+      } else {
+        transA = 'T';
+      }
+      if ((y == 0) || (y == 1)) {
+        transB = 'N';
+      } else {
+        transB = 'T';
+      }
+
+      memset(CC, 0, m * n * sizeof(FLOAT));
+      memset(C, 0, m * n * sizeof(FLOAT));
+
+      SGEMM (&transA, &transB, &m, &n, &k, &alpha, A,
+        &m, B, &k, &beta, C, &m);
+      SHGEMM (&transA, &transB, &m, &n, &k, &alpha, (_Float16*) AA,
+        &m, (_Float16*)BB, &k, &beta, CC, &m);
+
+      for (i = 0; i < n; i++)
+        for (j = 0; j < m; j++)
+        {
+          if (!is_close(CC[i * m + j], C[i * m + j], 0.01, 0.001)) {
+            ret++;
+          }
+        }
+    }
+    free(A);
+    free(B);
+    free(C);
+    free(AA);
+    free(BB);
+    free(CC);
+
+    if (ret != 0) {
+/*
+ * fprintf(stderr, "SHGEMM FAILURES FOR n=%d, alpha=%f beta=%f : %d\n", x, alpha, beta, ret);
+   */
+      rret++;
+      ret=0;
+/*    }  else {
+      fprintf(stderr, "SHGEMM SUCCEEDED FOR n=%d, alpha=%f beta=%f : %d\n", x, alpha, beta, ret);
+*/
+    }
+  }
+  
+  }
+  }
+  if (rret > 0) return(1);
+  return(0);
 }