diff options
Diffstat (limited to 'lib/TH/CMakeLists.txt')
-rw-r--r-- | lib/TH/CMakeLists.txt | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/lib/TH/CMakeLists.txt b/lib/TH/CMakeLists.txt index bfc17cc..22cdf33 100644 --- a/lib/TH/CMakeLists.txt +++ b/lib/TH/CMakeLists.txt @@ -304,6 +304,9 @@ FIND_PACKAGE(BLAS) IF(BLAS_FOUND) SET(USE_BLAS 1) TARGET_LINK_LIBRARIES(TH ${BLAS_LIBRARIES}) + IF(BLAS_INFO STREQUAL "mkl") + ADD_DEFINITIONS(-DTH_BLAS_MKL) + ENDIF() ENDIF(BLAS_FOUND) FIND_PACKAGE(LAPACK) |