File tree Expand file tree Collapse file tree 3 files changed +6
-1
lines changed Expand file tree Collapse file tree 3 files changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -110,6 +110,7 @@ macro(SetDefaultL1)
110
110
SetFallback (SROTMKERNEL rotm.S )
111
111
SetFallback (DROTMKERNEL rotm.S )
112
112
SetFallback (QROTMKERNEL rotm.S )
113
+ SetFallback (BSCALKERNEL ../generic/scal.c )
113
114
SetFallback (SSCALKERNEL scal.S )
114
115
SetFallback (DSCALKERNEL scal.S )
115
116
SetFallback (CSCALKERNEL zscal.S )
Original file line number Diff line number Diff line change @@ -150,12 +150,13 @@ endif ()
150
150
GenerateNamedObjects ("imax.c" "USE_MIN" "i*min" ${CBLAS_FLAG} )
151
151
152
152
if (BUILD_BFLOAT16 )
153
+ GenerateNamedObjects ("scal.c" "BGEMM" "bscal" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
153
154
GenerateNamedObjects ("bf16dot.c" "" "sbdot" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
154
155
GenerateNamedObjects ("gemm.c" "BGEMM" "bgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
155
156
GenerateNamedObjects ("gemm.c" "" "sbgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
156
157
GenerateNamedObjects ("sbgemmt.c" "" "sbgemmt" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
157
158
GenerateNamedObjects ("sbgemmt.c" "RNAME" "sbgemmtr" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
158
- GenerateNamedObjects ("bgemv .c" "BGEMV " "bgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
159
+ GenerateNamedObjects ("sbgemv .c" "BGEMM " "bgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
159
160
GenerateNamedObjects ("sbgemv.c" "" "sbgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
160
161
GenerateNamedObjects ("tobf16.c" "SINGLE_PREC" "sbstobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
161
162
GenerateNamedObjects ("tobf16.c" "DOUBLE_PREC" "sbdtobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
Original file line number Diff line number Diff line change @@ -121,6 +121,7 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
121
121
122
122
# sbdot
123
123
if (BUILD_BFLOAT16 )
124
+ GenerateNamedObjects ("${KERNELDIR} /${BSCALKERNEL} " "BGEMM" "scal_k" false "" "" false "BFLOAT16" )
124
125
GenerateNamedObjects ("${KERNELDIR} /${SBDOTKERNEL} " "SBDOT" "dot_k" false "" "" false "BFLOAT16" )
125
126
GenerateNamedObjects ("${KERNELDIR} /${BF16TOKERNEL} " "SINGLE" "f16tos_k" false "" "" false "BFLOAT16" )
126
127
GenerateNamedObjects ("${KERNELDIR} /${BF16TOKERNEL} " "DOUBLE" "bf16tod_k" false "" "" false "DOUBLE" )
@@ -222,6 +223,8 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
222
223
GenerateNamedObjects ("${KERNELDIR} /${SGEMVTKERNEL} " "TRANS" "gemv_t" false "" "" false "SINGLE" )
223
224
endif ()
224
225
if (BUILD_BFLOAT16 )
226
+ GenerateNamedObjects ("${KERNELDIR} /${BGEMVNKERNEL} " "BGEMM" "gemv_n" false "" "" false "BFLOAT16" )
227
+ GenerateNamedObjects ("${KERNELDIR} /${BGEMVTKERNEL} " "BGEMM" "gemv_t" false "" "" false "BFLOAT16" )
225
228
GenerateNamedObjects ("${KERNELDIR} /${SBGEMVNKERNEL} " "" "gemv_n" false "" "" false "BFLOAT16" )
226
229
GenerateNamedObjects ("${KERNELDIR} /${SBGEMVTKERNEL} " "" "gemv_t" false "" "" false "BFLOAT16" )
227
230
endif ()
You can’t perform that action at this time.
0 commit comments