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

Compiling most of the project

Browse Source 
Now compiles all CPU + GPU code, tests + binaries with deps
This commit is contained in:
Simon Layton
2016-12-08 08:40:29 -05:00
parent 497659ce0d
commit 4255ee9944
2 changed files with 85 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
cmake/Dependencies.cmake
View File

@ -101,3 +101,18 @@ if(USE_MPI)
set(CMAKE_EXE_LINKER_FLAGS ${MPI_CXX_LINK_FLAGS})
endif()
endif()
# ---[ OpenMP
find_package(OpenMP)
if(OpenMP_FOUND)
# set(CMAKE_CXX_FLAGS ${OpenMP_CXX_FLAGS})
list(APPEND Caffe2_LINKER_LIBS ${OpenMP_CXX_FLAGS})
endif()
# ---[ CUB
include_directories(SYSTEM ${CMAKE_SOURCE_DIR}/third_party/cub)
# ---[ CNMEM
add_subdirectory(${CMAKE_SOURCE_DIR}/third_party/cnmem)
include_directories(SYSTEM ${CMAKE_SOURCE_DIR}/third_party/cnmem/include)
list(APPEND ${Caffe2_LINKER_LIBS} ${CMAKE_SOURCE_DIR}/third_party/cnmem/libcnmem.so)