Skip to content

Commit 343830c

Browse files
martin-frbgmartin-frbg
authored
Add BGEMM parameter tables
1 parent b37516a commit 343830c

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

‎kernel/setparam-ref.c

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -926,6 +926,7 @@ gotoblas_t TABLE_NAME = {
926926
static void init_parameter(void) {
927927
#if (BUILD_BFLOAT16)
928928
TABLE_NAME.sbgemm_p = SBGEMM_DEFAULT_P;
929+
TABLE_NAME.bgemm_p = BGEMM_DEFAULT_P;
929930
#endif
930931
#if (BUILD_SINGLE==1) || (BUILD_COMPLEX==1)
931932
TABLE_NAME.sgemm_p = SGEMM_DEFAULT_P;
@@ -942,6 +943,7 @@ static void init_parameter(void) {
942943

943944
#if (BUILD_BFLOAT16)
944945
TABLE_NAME.sbgemm_q = SBGEMM_DEFAULT_Q;
946+
TABLE_NAME.bgemm_q = BGEMM_DEFAULT_Q;
945947
#endif
946948
#if BUILD_SINGLE == 1 || (BUILD_COMPLEX==1)
947949
TABLE_NAME.sgemm_q = SGEMM_DEFAULT_Q;
@@ -958,6 +960,7 @@ static void init_parameter(void) {
958960

959961
#if (BUILD_BFLOAT16)
960962
TABLE_NAME.sbgemm_r = SBGEMM_DEFAULT_R;
963+
TABLE_NAME.bgemm_r = BGEMM_DEFAULT_R;
961964
#endif
962965
#if BUILD_SINGLE == 1 || (BUILD_COMPLEX==1)
963966
TABLE_NAME.sgemm_r = SGEMM_DEFAULT_R;
@@ -1113,10 +1116,12 @@ static void init_parameter(void) {
11131116

11141117
#ifdef BUILD_BFLOAT16
11151118
TABLE_NAME.sbgemm_p = SBGEMM_DEFAULT_P;
1119+
TABLE_NAME.bgemm_p = BGEMM_DEFAULT_P;
11161120
#endif
11171121

11181122
#ifdef BUILD_BFLOAT16
11191123
TABLE_NAME.sbgemm_r = SBGEMM_DEFAULT_R;
1124+
TABLE_NAME.bgemm_r = BGEMM_DEFAULT_R;
11201125
#endif
11211126

11221127
#if defined(LA464)
@@ -1215,6 +1220,7 @@ static void init_parameter(void) {
12151220

12161221
#ifdef BUILD_BFLOAT16
12171222
TABLE_NAME.sbgemm_q = SBGEMM_DEFAULT_Q;
1223+
TABLE_NAME.bgemm_q = BGEMM_DEFAULT_Q;
12181224
#endif
12191225
}
12201226
#else // (ARCH_LOONGARCH64)
@@ -1223,6 +1229,7 @@ static void init_parameter(void) {
12231229

12241230
#ifdef BUILD_BFLOAT16
12251231
TABLE_NAME.sbgemm_p = SBGEMM_DEFAULT_P;
1232+
TABLE_NAME.bgemm_p = BGEMM_DEFAULT_P;
12261233
#endif
12271234
TABLE_NAME.sgemm_p = SGEMM_DEFAULT_P;
12281235
TABLE_NAME.dgemm_p = DGEMM_DEFAULT_P;
@@ -1231,6 +1238,7 @@ static void init_parameter(void) {
12311238

12321239
#ifdef BUILD_BFLOAT16
12331240
TABLE_NAME.sbgemm_r = SBGEMM_DEFAULT_R;
1241+
TABLE_NAME.bgemm_r = BGEMM_DEFAULT_R;
12341242
#endif
12351243
TABLE_NAME.sgemm_r = SGEMM_DEFAULT_R;
12361244
TABLE_NAME.dgemm_r = DGEMM_DEFAULT_R;
@@ -1240,6 +1248,7 @@ static void init_parameter(void) {
12401248

12411249
#ifdef BUILD_BFLOAT16
12421250
TABLE_NAME.sbgemm_q = SBGEMM_DEFAULT_Q;
1251+
TABLE_NAME.bgemm_q = BGEMM_DEFAULT_Q;
12431252
#endif
12441253
TABLE_NAME.sgemm_q = SGEMM_DEFAULT_Q;
12451254
TABLE_NAME.dgemm_q = DGEMM_DEFAULT_Q;
@@ -1252,6 +1261,7 @@ static void init_parameter(void) {
12521261
static void init_parameter(void) {
12531262
#ifdef BUILD_BFLOAT16
12541263
TABLE_NAME.sbgemm_p = SBGEMM_DEFAULT_P;
1264+
TABLE_NAME.bgemm_p = BGEMM_DEFAULT_P;
12551265
#endif
12561266
TABLE_NAME.sgemm_p = SGEMM_DEFAULT_P;
12571267
TABLE_NAME.dgemm_p = DGEMM_DEFAULT_P;
@@ -1260,6 +1270,7 @@ static void init_parameter(void) {
12601270

12611271
#ifdef BUILD_BFLOAT16
12621272
TABLE_NAME.sbgemm_r = SBGEMM_DEFAULT_R;
1273+
TABLE_NAME.bgemm_r = BGEMM_DEFAULT_R;
12631274
#endif
12641275
TABLE_NAME.sgemm_r = SGEMM_DEFAULT_R;
12651276
TABLE_NAME.dgemm_r = DGEMM_DEFAULT_R;
@@ -1269,6 +1280,7 @@ static void init_parameter(void) {
12691280

12701281
#ifdef BUILD_BFLOAT16
12711282
TABLE_NAME.sbgemm_q = SBGEMM_DEFAULT_Q;
1283+
TABLE_NAME.bgemm_q = BGEMM_DEFAULT_Q;
12721284
#endif
12731285
TABLE_NAME.sgemm_q = SGEMM_DEFAULT_Q;
12741286
TABLE_NAME.dgemm_q = DGEMM_DEFAULT_Q;
@@ -1282,6 +1294,7 @@ static void init_parameter(void) {
12821294

12831295
#ifdef BUILD_BFLOAT16
12841296
TABLE_NAME.sbgemm_p = SBGEMM_DEFAULT_P;
1297+
TABLE_NAME.bgemm_p = BGEMM_DEFAULT_P;
12851298
#endif
12861299
#ifdef BUILD_HFLOAT16
12871300
TABLE_NAME.shgemm_p = SHGEMM_DEFAULT_P;
@@ -1293,6 +1306,7 @@ static void init_parameter(void) {
12931306

12941307
#ifdef BUILD_BFLOAT16
12951308
TABLE_NAME.sbgemm_r = SBGEMM_DEFAULT_R;
1309+
TABLE_NAME.bgemm_r = BGEMM_DEFAULT_R;
12961310
#endif
12971311
#ifdef BUILD_HFLOAT16
12981312
TABLE_NAME.shgemm_r = SHGEMM_DEFAULT_R;
@@ -1305,6 +1319,7 @@ static void init_parameter(void) {
13051319

13061320
#ifdef BUILD_BFLOAT16
13071321
TABLE_NAME.sbgemm_q = SBGEMM_DEFAULT_Q;
1322+
TABLE_NAME.bgemm_q = BGEMM_DEFAULT_Q;
13081323
#endif
13091324
#ifdef BUILD_HFLOAT16
13101325
TABLE_NAME.shgemm_q = SHGEMM_DEFAULT_Q;
@@ -1455,6 +1470,8 @@ static void init_parameter(void) {
14551470
#ifdef BUILD_BFLOAT16
14561471
TABLE_NAME.sbgemm_p = SBGEMM_DEFAULT_P;
14571472
TABLE_NAME.sbgemm_q = SBGEMM_DEFAULT_Q;
1473+
TABLE_NAME.bgemm_p = BGEMM_DEFAULT_P;
1474+
TABLE_NAME.bgemm_q = BGEMM_DEFAULT_Q;
14581475
#endif
14591476
#ifdef BUILD_HFLOAT16
14601477
TABLE_NAME.shgemm_p = SHGEMM_DEFAULT_P;
@@ -2053,6 +2070,10 @@ static void init_parameter(void) {
20532070
((TABLE_NAME.sbgemm_p * TABLE_NAME.sbgemm_q * 4 + TABLE_NAME.offsetA
20542071
+ TABLE_NAME.align) & ~TABLE_NAME.align)
20552072
) / (TABLE_NAME.sbgemm_q * 4) - 15) & ~15);
2073+
TABLE_NAME.bgemm_r = (((BUFFER_SIZE -
2074+
((TABLE_NAME.bgemm_p * TABLE_NAME.bgemm_q * 4 + TABLE_NAME.offsetA
2075+
+ TABLE_NAME.align) & ~TABLE_NAME.align)
2076+
) / (TABLE_NAME.bgemm_q * 4) - 15) & ~15);
20562077
#endif
20572078

20582079
#if BUILD_HFLOAT16==1

0 commit comments

Comments
 (0)