frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #6 from caffe2/cmake

Browse Source 
Cmake
This commit is contained in:
Bram Wasti
2017-01-03 15:19:58 -08:00
committed by GitHub
parent 737f507786 0ce23319c2
commit 4ab0351647
2 changed files with 39 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmake/Dependencies.cmake
View File

@ -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