Browse Source

Cleanup of SBGEMM unit test.

tags/v0.3.29
Chip Kerchner 1 year ago
parent
commit
31226740d6
1 changed files with 13 additions and 9 deletions
  1. +13
    -9
      test/compare_sgemm_sbgemm.c

+ 13
- 9
test/compare_sgemm_sbgemm.c View File

@@ -113,13 +113,19 @@ main (int argc, char *argv[])


for (j = 0; j < m; j++) for (j = 0; j < m; j++)
{ {
for (i = 0; i < n; i++)
for (i = 0; i < k; i++)
{ {
A[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5; A[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
B[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
sbstobf16_(&one, &A[j*k+i], &one, &atmp, &one); sbstobf16_(&one, &A[j*k+i], &one, &atmp, &one);
sbstobf16_(&one, &B[j*k+i], &one, &btmp, &one);
AA[j * k + i].v = atmp; AA[j * k + i].v = atmp;
}
}
for (j = 0; j < n; j++)
{
for (i = 0; i < k; i++)
{
B[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
sbstobf16_(&one, &B[j*k+i], &one, &btmp, &one);
BB[j * k + i].v = btmp; BB[j * k + i].v = btmp;
} }
} }
@@ -147,10 +153,7 @@ main (int argc, char *argv[])


for (i = 0; i < n; i++) for (i = 0; i < n; i++)
for (j = 0; j < m; j++) for (j = 0; j < m; j++)
if (fabs (CC[i * m + j] - C[i * m + j]) > 1.0)
ret++;
for (i = 0; i < n; i++)
for (j = 0; j < m; j++)
{
for (l = 0; l < k; l++) for (l = 0; l < k; l++)
if (transA == 'N' && transB == 'N') if (transA == 'N' && transB == 'N')
{ {
@@ -169,10 +172,11 @@ main (int argc, char *argv[])
DD[i * m + j] += DD[i * m + j] +=
float16to32 (AA[k * j + l]) * float16to32 (BB[i + l * n]); float16to32 (AA[k * j + l]) * float16to32 (BB[i + l * n]);
} }
for (i = 0; i < n; i++)
for (j = 0; j < m; j++)
if (fabs (CC[i * m + j] - C[i * m + j]) > 1.0)
ret++;
if (fabs (CC[i * m + j] - DD[i * m + j]) > 1.0) if (fabs (CC[i * m + j] - DD[i * m + j]) > 1.0)
ret++; ret++;
}
} }
free(A); free(A);
free(B); free(B);


Loading…
Cancel
Save