diff options
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 637af3c..1acb0bc 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -1,5 +1,9 @@ + CMAKE_MINIMUM_REQUIRED(VERSION 2.6 FATAL_ERROR) CMAKE_POLICY(VERSION 2.6) + +SET(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake ${CMAKE_MODULE_PATH}) + IF(LUAROCKS_PREFIX) MESSAGE(STATUS "Installing CuDNN through Luarocks") STRING(REGEX REPLACE "(.*)lib/luarocks/rocks.*" "\\1" CMAKE_INSTALL_PREFIX "${LUAROCKS_PREFIX}") @@ -8,6 +12,18 @@ ENDIF() FIND_PACKAGE(Torch REQUIRED) FIND_PACKAGE(CUDA 7.0 REQUIRED) +IF(NOT COMMAND CUDNN_INSTALL) + INCLUDE(${CMAKE_CURRENT_SOURCE_DIR}/cmake/FindCUDNN.cmake) +ENDIF() + +FIND_PACKAGE(CUDNN 5.1 QUIET) + +IF(NOT CUDNN_FOUND) + CUDNN_INSTALL(5.1 ${CMAKE_INSTALL_PREFIX}) +ENDIF() + FILE(GLOB luasrc *.lua) SET(src "") ADD_TORCH_PACKAGE(cudnn "${src}" "${luasrc}" "NVIDIA CuDNN Bindings") + + |