Skip to content

Commit 098a8d5

Browse files
authored
Merge branch 'OpenMathLib:develop' into issue5497
2 parents ee6aa89 + f3cecbe commit 098a8d5

File tree

3 files changed

+14
-2
lines changed

3 files changed

+14
-2
lines changed

cmake/kernel.cmake

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -230,10 +230,12 @@ macro(SetDefaultL2)
230230
if (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)
238240
endif ()
239241
endmacro ()

driver/level2/CMakeLists.txt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff 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 ()
206215
endif ()
207216

driver/level3/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)