File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -230,10 +230,12 @@ macro(SetDefaultL2)
230230if (BUILD_BFLOAT16)
231231 SetFallback (BGEMVNKERNEL ../generic/gemv_n.c )
232232 SetFallback (BGEMVTKERNEL ../generic/gemv_t.c )
233- SetFallback (SHGEMVNKERNEL ../generic/gemv_n.c )
234- SetFallback (SHGEMVTKERNEL ../generic/gemv_t.c )
235233 SetFallback (SBGEMVNKERNEL ../x86_64/sbgemv_n.c )
236234 SetFallback (SBGEMVTKERNEL ../x86_64/sbgemv_t.c )
235+ endif ()
236+ if (BUILD_HFLOAT16)
237+ SetFallback (SHGEMVNKERNEL ../generic/gemv_n.c )
238+ SetFallback (SHGEMVTKERNEL ../generic/gemv_t.c )
237239 SetFallback (SHGERKERNEL ../generic/ger.c )
238240endif ()
239241endmacro ()
Original file line number Diff line number Diff line change @@ -202,6 +202,15 @@ if (BUILD_BFLOAT16)
202202 if (USE_THREAD)
203203 GenerateNamedObjects ("sbgemv_thread.c" "" "gemv_thread_n" false "" "" false "BFLOAT16" )
204204 GenerateNamedObjects ("sbgemv_thread.c" "TRANSA" "gemv_thread_t" false "" "" false "BFLOAT16" )
205+ GenerateNamedObjects ("sbgemv_thread.c" "BGEMM;BFLOAT16" "bgemv_thread_n" false "" "" true "" )
206+ GenerateNamedObjects ("sbgemv_thread.c" "BGEMM;BFLOAT16;TRANSA" "bgemv_thread_t" false "" "" true "" )
207+ endif ()
208+ endif ()
209+
210+ if (BUILD_HFLOAT16)
211+ if (USE_THREAD)
212+ GenerateNamedObjects ("sbgemv_thread.c" "" "gemv_thread_n" false "" "" false "HFLOAT16" )
213+ GenerateNamedObjects ("sbgemv_thread.c" "TRANSA" "gemv_thread_t" false "" "" false "HFLOAT16" )
205214 endif ()
206215endif ()
207216
Original file line number Diff line number Diff line change @@ -17,6 +17,7 @@ foreach (GEMM_DEFINE ${GEMM_DEFINES})
1717 GenerateNamedObjects ("gemm.c" "${GEMM_DEFINE} ;BGEMM" "gemm_${GEMM_DEFINE_LC} " 0 "" "" false "BFLOAT16" )
1818 if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
1919 GenerateNamedObjects ("gemm.c" "${GEMM_DEFINE} ;THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC} " 0 "" "" false "BFLOAT16" )
20+ GenerateNamedObjects ("gemm.c" "${GEMM_DEFINE} ;THREADED_LEVEL3;BGEMM" "gemm_thread_${GEMM_DEFINE_LC} " 0 "" "" false "BFLOAT16" )
2021 endif ()
2122 endif ()
2223 if (BUILD_HFLOAT16)
You can’t perform that action at this time.
0 commit comments