Skip to content

Commit 3122674

Browse files
author
Chip Kerchner
committed
Cleanup of SBGEMM unit test.
1 parent b173769 commit 3122674

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

test/compare_sgemm_sbgemm.c

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -113,13 +113,19 @@ main (int argc, char *argv[])
113113

114114
for (j = 0; j < m; j++)
115115
{
116-
for (i = 0; i < n; i++)
116+
for (i = 0; i < k; i++)
117117
{
118118
A[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
119-
B[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
120119
sbstobf16_(&one, &A[j*k+i], &one, &atmp, &one);
121-
sbstobf16_(&one, &B[j*k+i], &one, &btmp, &one);
122120
AA[j * k + i].v = atmp;
121+
}
122+
}
123+
for (j = 0; j < n; j++)
124+
{
125+
for (i = 0; i < k; i++)
126+
{
127+
B[j * k + i] = ((FLOAT) rand () / (FLOAT) RAND_MAX) + 0.5;
128+
sbstobf16_(&one, &B[j*k+i], &one, &btmp, &one);
123129
BB[j * k + i].v = btmp;
124130
}
125131
}
@@ -147,10 +153,7 @@ main (int argc, char *argv[])
147153

148154
for (i = 0; i < n; i++)
149155
for (j = 0; j < m; j++)
150-
if (fabs (CC[i * m + j] - C[i * m + j]) > 1.0)
151-
ret++;
152-
for (i = 0; i < n; i++)
153-
for (j = 0; j < m; j++)
156+
{
154157
for (l = 0; l < k; l++)
155158
if (transA == 'N' && transB == 'N')
156159
{
@@ -169,10 +172,11 @@ main (int argc, char *argv[])
169172
DD[i * m + j] +=
170173
float16to32 (AA[k * j + l]) * float16to32 (BB[i + l * n]);
171174
}
172-
for (i = 0; i < n; i++)
173-
for (j = 0; j < m; j++)
175+
if (fabs (CC[i * m + j] - C[i * m + j]) > 1.0)
176+
ret++;
174177
if (fabs (CC[i * m + j] - DD[i * m + j]) > 1.0)
175178
ret++;
179+
}
176180
}
177181
free(A);
178182
free(B);

0 commit comments

Comments
 (0)