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

Cleanup of SBGEMM unit test.

This commit is contained in:
Chip Kerchner 2024-08-14 08:10:25 -05:00
parent b1737698db
commit 31226740d6
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
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++) if (fabs (CC[i * m + j] - C[i * m + j]) > 1.0)
for (j = 0; j < m; j++) 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);