Fix cmake building with cblas_bgemm

This commit is contained in:
youcai
2025-07-23 21:51:30 +08:00
parent f4caa61e47
commit 41f9701ebc
2 changed files with 6 additions and 0 deletions

View File

@@ -14,6 +14,7 @@ foreach (GEMM_DEFINE ${GEMM_DEFINES})
endif ()
if (BUILD_BFLOAT16)
GenerateNamedObjects("gemm.c" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};BGEMM" "gemm_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
endif ()