|
|
@@ -217,6 +217,9 @@ if (LLAMA_BLAS)
|
|
|
message(STATUS "BLAS found, Includes: ${BLAS_INCLUDE_DIRS}")
|
|
|
add_compile_options(${BLAS_LINKER_FLAGS})
|
|
|
add_compile_definitions(GGML_USE_OPENBLAS)
|
|
|
+ if (${BLAS_INCLUDE_DIRS} MATCHES "mkl" AND (${LLAMA_BLAS_VENDOR} MATCHES "Generic" OR ${LLAMA_BLAS_VENDOR} MATCHES "Intel"))
|
|
|
+ add_compile_definitions(GGML_BLAS_USE_MKL)
|
|
|
+ endif()
|
|
|
set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} ${BLAS_LIBRARIES})
|
|
|
set(LLAMA_EXTRA_INCLUDES ${LLAMA_EXTRA_INCLUDES} ${BLAS_INCLUDE_DIRS})
|
|
|
|