diff options
Diffstat (limited to 'CMakeLists.txt')
-rw-r--r-- | CMakeLists.txt | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index e5453dd..8f9b50e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -12,16 +12,13 @@ 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 EXACT) - +FIND_PACKAGE(CUDNN 5.1 EXACT QUIET) IF(NOT CUDNN_FOUND) - CUDNN_INSTALL(5.1 ${CMAKE_INSTALL_PREFIX}) + CUDNN_INSTALL(5.1 "${Torch_INSTALL_LIB}" "${Torch_INSTALL_INCLUDE}" "") + FIND_PACKAGE(CUDNN 5.1 EXACT REQUIRED) ENDIF() + FILE(GLOB luasrc *.lua) SET(src "") ADD_TORCH_PACKAGE(cudnn "${src}" "${luasrc}" "NVIDIA CuDNN Bindings") |