Browse Source

Fix cmake building with cblas_bgemm

pull/5397/head
youcai 2 months ago
parent
commit
41f9701ebc
2 changed files with 6 additions and 0 deletions
  1. +1
    -0
      driver/level3/CMakeLists.txt
  2. +5
    -0
      interface/CMakeLists.txt

+ 1
- 0
driver/level3/CMakeLists.txt View File

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


+ 5
- 0
interface/CMakeLists.txt View File

@@ -283,6 +283,11 @@ if ( BUILD_COMPLEX16 AND NOT BUILD_DOUBLE)
GenerateNamedObjects("imax.c" "USE_ABS" "i*amax" 0 "" "" false "DOUBLE") GenerateNamedObjects("imax.c" "USE_ABS" "i*amax" 0 "" "" false "DOUBLE")
endif () endif ()


if ( BUILD_BFLOAT16 AND NO_FBLAS )
GenerateNamedObjects("tobf16.c" "SINGLE_PREC" "sbstobf16" 0 "" "" true "BFLOAT16")
GenerateNamedObjects("bf16to.c" "SINGLE_PREC" "sbf16tos" 0 "" "" true "BFLOAT16")
endif()

add_library(interface OBJECT ${OPENBLAS_SRC}) add_library(interface OBJECT ${OPENBLAS_SRC})


if (USE_OPENMP) if (USE_OPENMP)


Loading…
Cancel
Save