Skip to content

Commit d9894f4

Browse files
authored
Define sbgemm_r to fix DYNAMIC_ARCH builds
1 parent 522f809 commit d9894f4

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

kernel/setparam-ref.c

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1824,6 +1824,13 @@ static void init_parameter(void) {
18241824
fprintf(stderr, "L2 = %8d DGEMM_P .. %d\n", l2, TABLE_NAME.dgemm_p);
18251825
#endif
18261826

1827+
#if BUILD_BFLOAT16==1
1828+
TABLE_NAME.sbgemm_r = (((BUFFER_SIZE -
1829+
((TABLE_NAME.sbgemm_p * TABLE_NAME.sbgemm_q * 4 + TABLE_NAME.offsetA
1830+
+ TABLE_NAME.align) & ~TABLE_NAME.align)
1831+
) / (TABLE_NAME.sbgemm_q * 4) - 15) & ~15);
1832+
#endif
1833+
18271834
#if BUILD_SINGLE==1
18281835
TABLE_NAME.sgemm_r = (((BUFFER_SIZE -
18291836
((TABLE_NAME.sgemm_p * TABLE_NAME.sgemm_q * 4 + TABLE_NAME.offsetA

0 commit comments

Comments
 (0)