Skip to content

Commit 85154c2

Browse files
authored
Change "HALF" and "sh" to "BFLOAT16" and "sb"
1 parent ae1ab5b commit 85154c2

File tree

4 files changed

+14
-14
lines changed

4 files changed

+14
-14
lines changed

driver/others/blas_server.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ static void legacy_exec(void *func, int mode, blas_arg_t *args, void *sb){
227227
args -> a, args -> lda,
228228
args -> b, args -> ldb,
229229
args -> c, args -> ldc, sb);
230-
#ifdef BUILD_HALF
230+
#ifdef BUILD_BFLOAT16
231231
} else if ((mode & BLAS_PREC) == BLAS_BFLOAT16){
232232
/* REAL / BFLOAT16 */
233233
void (*afunc)(BLASLONG, BLASLONG, BLASLONG, bfloat16,

driver/others/blas_server_omp.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ static void legacy_exec(void *func, int mode, blas_arg_t *args, void *sb){
192192
args -> a, args -> lda,
193193
args -> b, args -> ldb,
194194
args -> c, args -> ldc, sb);
195-
#ifdef BUILD_HALF
195+
#ifdef BUILD_BFLOAT16
196196
} else if ((mode & BLAS_PREC) == BLAS_BFLOAT16){
197197
/* REAL / BFLOAT16 */
198198
void (*afunc)(BLASLONG, BLASLONG, BLASLONG, bfloat16,

driver/others/blas_server_win32.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ static void legacy_exec(void *func, int mode, blas_arg_t *args, void *sb){
112112
args -> a, args -> lda,
113113
args -> b, args -> ldb,
114114
args -> c, args -> ldc, sb);
115-
#ifdef BUILD_HALF
115+
#ifdef BUILD_BFLOAT16
116116
} else if ((mode & BLAS_PREC) == BLAS_BFLOAT16){
117117
/* REAL / BFLOAT16 */
118118
void (*afunc)(BLASLONG, BLASLONG, BLASLONG, bfloat16,

driver/others/parameter.c

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,10 @@ BLASLONG gemm_offset_b = DEFAULT_GEMM_OFFSET_B;
6262
BLASLONG gemm_offset_b = GEMM_OFFSET_B;
6363
#endif
6464

65-
#if SHGEMM_P == shgemm_p
66-
BLASLONG shgemm_p = DEFAULT_GEMM_P;
65+
#if SBGEMM_P == sbgemm_p
66+
BLASLONG sbgemm_p = DEFAULT_GEMM_P;
6767
#else
68-
BLASLONG shgemm_p = SHGEMM_P;
68+
BLASLONG sbgemm_p = SBGEMM_P;
6969
#endif
7070
#if SGEMM_P == sgemm_p
7171
BLASLONG sgemm_p = DEFAULT_GEMM_P;
@@ -88,10 +88,10 @@ BLASLONG zgemm_p = DEFAULT_GEMM_P;
8888
BLASLONG zgemm_p = ZGEMM_P;
8989
#endif
9090

91-
#if SHGEMM_Q == shgemm_q
92-
BLASLONG shgemm_q = DEFAULT_GEMM_Q;
91+
#if SBGEMM_Q == sbgemm_q
92+
BLASLONG sbgemm_q = DEFAULT_GEMM_Q;
9393
#else
94-
BLASLONG shgemm_q = SHGEMM_Q;
94+
BLASLONG sbgemm_q = SBGEMM_Q;
9595
#endif
9696
#if SGEMM_Q == sgemm_q
9797
BLASLONG sgemm_q = DEFAULT_GEMM_Q;
@@ -114,10 +114,10 @@ BLASLONG zgemm_q = DEFAULT_GEMM_Q;
114114
BLASLONG zgemm_q = ZGEMM_Q;
115115
#endif
116116

117-
#if SHGEMM_R == shgemm_r
118-
BLASLONG shgemm_r = DEFAULT_GEMM_R;
117+
#if SBGEMM_R == sbgemm_r
118+
BLASLONG sbgemm_r = DEFAULT_GEMM_R;
119119
#else
120-
BLASLONG shgemm_r = SHGEMM_R;
120+
BLASLONG sbgemm_r = SBGEMM_R;
121121
#endif
122122
#if SGEMM_R == sgemm_r
123123
BLASLONG sgemm_r = DEFAULT_GEMM_R;
@@ -615,7 +615,7 @@ void blas_set_parameter(void){
615615

616616
size = BITMASK(cpuid3, 16, 0xff);
617617

618-
shgemm_p = 192 * (size + 1);
618+
sbgemm_p = 192 * (size + 1);
619619
sgemm_p = 192 * (size + 1);
620620
dgemm_p = 96 * (size + 1);
621621
cgemm_p = 96 * (size + 1);
@@ -629,7 +629,7 @@ void blas_set_parameter(void){
629629
xgemm_p = 16 * (size + 1);
630630
#endif
631631

632-
shgemm_r = (((BUFFER_SIZE - ((SHGEMM_P * SHGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN) & ~GEMM_ALIGN)) / (SHGEMM_Q * 4)) - 15) & ~15;
632+
sbgemm_r = (((BUFFER_SIZE - ((SBGEMM_P * SBGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN) & ~GEMM_ALIGN)) / (SBGEMM_Q * 4)) - 15) & ~15;
633633
sgemm_r = (((BUFFER_SIZE - ((SGEMM_P * SGEMM_Q * 4 + GEMM_OFFSET_A + GEMM_ALIGN) & ~GEMM_ALIGN)) / (SGEMM_Q * 4)) - 15) & ~15;
634634
dgemm_r = (((BUFFER_SIZE - ((DGEMM_P * DGEMM_Q * 8 + GEMM_OFFSET_A + GEMM_ALIGN) & ~GEMM_ALIGN)) / (DGEMM_Q * 8)) - 15) & ~15;
635635
cgemm_r = (((BUFFER_SIZE - ((CGEMM_P * CGEMM_Q * 8 + GEMM_OFFSET_A + GEMM_ALIGN) & ~GEMM_ALIGN)) / (CGEMM_Q * 8)) - 15) & ~15;

0 commit comments

Comments
 (0)