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

Build binaries

Browse Source
This commit is contained in:
Simon Layton
2016-12-07 10:54:06 -05:00
parent f3c20620ed
commit 497659ce0d
3 changed files with 56 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
cmake/Dependencies.cmake
View File

@ -91,3 +91,13 @@ find_package(NumPy REQUIRED)
include_directories(SYSTEM ${PYTHON_INCLUDE_DIRS} ${NUMPY_INCLUDE_DIRS})
list(APPEND Caffe2_LINKER_LIBS ${PYTHON_LIBRARIES})
# ---[ MPI
if(USE_MPI)
find_package(MPI)
if(MPI_CXX_FOUND)
include_directories(SYSTEM ${MPI_CXX_INCLUDE_PATH})
list(APPEND Caffe2_LINKER_LIBS ${MPI_CXX_LIBRARIES})
set(CMAKE_EXE_LINKER_FLAGS ${MPI_CXX_LINK_FLAGS})
endif()
endif()