Fix MKL build issue by correctly finding and linking MKL libraries

This commit is contained in:
Greener 2024-07-01 16:55:52 +08:00
parent c118733a29
commit dac938391d
1 changed files with 9 additions and 1 deletions

View File

@ -28,6 +28,12 @@ if (WHISPER_OPENVINO)
find_package(OpenVINO REQUIRED COMPONENTS Runtime)
endif()
if (WHISPER_MKL)
find_package(MKL REQUIRED)
set(MKL_INCLUDE_DIRS "${MKLROOT}/include")
set(MKL_LIBRARIES "${MKLROOT}/lib/intel64")
endif()
#
# libraries
#
@ -134,7 +140,9 @@ if (WHISPER_OPENVINO)
endif()
if (WHISPER_MKL)
target_link_libraries(whisper PRIVATE MKL::MKL)
include_directories(${MKL_INCLUDE_DIRS})
link_directories(${MKL_LIBRARIES})
target_link_libraries(whisper PRIVATE mkl_rt)
endif()
if (BUILD_SHARED_LIBS)