diff options
| author | Bram Wasti <bwasti@fb.com> | 2017-01-03 15:19:58 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-01-03 15:19:58 -0800 |
| commit | 4ab03516475f827a0419bae188aac48e1522681c (patch) | |
| tree | eb010ce809cb50a30efa66bf128e7b002b87b6c4 /cmake | |
| parent | 737f507786ae82421948140e7464fd91ebbdfbfb (diff) | |
| parent | 0ce23319c25efe6bb7a736589dc2a451799fb70e (diff) | |
| download | pytorch-4ab03516475f827a0419bae188aac48e1522681c.tar.gz pytorch-4ab03516475f827a0419bae188aac48e1522681c.tar.bz2 pytorch-4ab03516475f827a0419bae188aac48e1522681c.zip | |
Merge pull request #6 from caffe2/cmake
Cmake
Diffstat (limited to 'cmake')
| -rw-r--r-- | cmake/Dependencies.cmake | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake index 106d8dd041..1f223abea1 100644 --- a/cmake/Dependencies.cmake +++ b/cmake/Dependencies.cmake @@ -9,22 +9,28 @@ find_package(Threads REQUIRED) list(APPEND Caffe2_LINKER_LIBS ${CMAKE_THREAD_LIBS_INIT}) # ---[ BLAS -set(BLAS "Atlas" CACHE STRING "Selected BLAS library") +set(BLAS "Eigen" CACHE STRING "Selected BLAS library") set_property(CACHE BLAS PROPERTY STRINGS "Atlas;OpenBLAS;MKL") -list(APPEND Caffe2_LINKER_LIBS cblas) -if(BLAS STREQUAL "Atlas") +if(BLAS STREQUAL "Eigen") + # Eigen is header-only and we do not have any dependent libraries + add_definitions(-DCAFFE2_USE_EIGEN_FOR_BLAS) +elseif(BLAS STREQUAL "Atlas") find_package(Atlas REQUIRED) include_directories(SYSTEM ${ATLAS_INCLUDE_DIRS}) list(APPEND Caffe2_LINKER_LIBS ${ATLAS_LIBRARIES}) + list(APPEND Caffe2_LINKER_LIBS cblas) elseif(BLAS STREQUAL "OpenBLAS") find_package(OpenBLAS REQUIRED) include_directories(SYSTEM ${OpenBLAS_INCLUDE_DIR}) list(APPEND Caffe2_LINKER_LIBS ${OpenBLAS_LIB}) + list(APPEND Caffe2_LINKER_LIBS cblas) elseif(BLAS STREQUAL "MKL") find_package(MKL REQUIRED) include_directories(SYSTEM ${MKL_INCLUDE_DIR}) list(APPEND Caffe2_LINKER_LIBS ${MKL_LIBRARIES}) + list(APPEND Caffe2_LINKER_LIBS cblas) + add_definitions(-DCAFFE2_USE_MKL) endif() # ---[ Google-glog |