Skip to content

Commit 2f8220d

Browse files
authored
Add sbgemm
1 parent 5f6a609 commit 2f8220d

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

driver/level3/CMakeLists.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,12 @@ foreach (GEMM_DEFINE ${GEMM_DEFINES})
1212
if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
1313
GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0)
1414
endif ()
15+
if (BUILD_BFLOAT16)
16+
GenerateNamedObjects("gemm.c" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
17+
if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
18+
GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
19+
endif ()
20+
endif ()
1521
endforeach ()
1622

1723
if ( BUILD_COMPLEX16 AND NOT BUILD_DOUBLE)

0 commit comments

Comments
 (0)