File tree Expand file tree Collapse file tree 3 files changed +14
-2
lines changed Expand file tree Collapse file tree 3 files changed +14
-2
lines changed 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