File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -110,6 +110,7 @@ macro(SetDefaultL1)
110110 SetFallback (SROTMKERNEL rotm.S )
111111 SetFallback (DROTMKERNEL rotm.S )
112112 SetFallback (QROTMKERNEL rotm.S )
113+ SetFallback (BSCALKERNEL ../generic/scal.c )
113114 SetFallback (SSCALKERNEL scal.S )
114115 SetFallback (DSCALKERNEL scal.S )
115116 SetFallback (CSCALKERNEL zscal.S )
Original file line number Diff line number Diff line change @@ -150,12 +150,13 @@ endif ()
150150 GenerateNamedObjects ("imax.c" "USE_MIN" "i*min" ${CBLAS_FLAG} )
151151
152152if (BUILD_BFLOAT16)
153+ GenerateNamedObjects ("scal.c" "BGEMM" "bscal" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
153154 GenerateNamedObjects ("bf16dot.c" "" "sbdot" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
154155 GenerateNamedObjects ("gemm.c" "BGEMM" "bgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
155156 GenerateNamedObjects ("gemm.c" "" "sbgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
156157 GenerateNamedObjects ("sbgemmt.c" "" "sbgemmt" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
157158 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" )
159160 GenerateNamedObjects ("sbgemv.c" "" "sbgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
160161 GenerateNamedObjects ("tobf16.c" "SINGLE_PREC" "sbstobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16" )
161162 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)
121121
122122 # sbdot
123123 if (BUILD_BFLOAT16)
124+ GenerateNamedObjects ("${KERNELDIR} /${BSCALKERNEL} " "BGEMM" "scal_k" false "" "" false "BFLOAT16" )
124125 GenerateNamedObjects ("${KERNELDIR} /${SBDOTKERNEL} " "SBDOT" "dot_k" false "" "" false "BFLOAT16" )
125126 GenerateNamedObjects ("${KERNELDIR} /${BF16TOKERNEL} " "SINGLE" "f16tos_k" false "" "" false "BFLOAT16" )
126127 GenerateNamedObjects ("${KERNELDIR} /${BF16TOKERNEL} " "DOUBLE" "bf16tod_k" false "" "" false "DOUBLE" )
@@ -222,6 +223,8 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS)
222223 GenerateNamedObjects ("${KERNELDIR} /${SGEMVTKERNEL} " "TRANS" "gemv_t" false "" "" false "SINGLE" )
223224 endif ()
224225 if (BUILD_BFLOAT16)
226+ GenerateNamedObjects ("${KERNELDIR} /${BGEMVNKERNEL} " "BGEMM" "gemv_n" false "" "" false "BFLOAT16" )
227+ GenerateNamedObjects ("${KERNELDIR} /${BGEMVTKERNEL} " "BGEMM" "gemv_t" false "" "" false "BFLOAT16" )
225228 GenerateNamedObjects ("${KERNELDIR} /${SBGEMVNKERNEL} " "" "gemv_n" false "" "" false "BFLOAT16" )
226229 GenerateNamedObjects ("${KERNELDIR} /${SBGEMVTKERNEL} " "" "gemv_t" false "" "" false "BFLOAT16" )
227230 endif ()
You can’t perform that action at this time.
0 commit comments