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)
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