diff options
| author | Soumith Chintala <soumith@gmail.com> | 2014-09-11 01:03:02 +0400 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2014-09-11 01:03:02 +0400 |
| commit | a8bf53f8738fd319568593718280343c3ebc93e6 (patch) | |
| tree | 58f632560c26f31c9d87e81d6f14c546c3020bcd /CMakeLists.txt | |
first commit
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt new file mode 100644 index 0000000..4efe497 --- /dev/null +++ b/CMakeLists.txt @@ -0,0 +1,12 @@ +CMAKE_MINIMUM_REQUIRED(VERSION 2.6 FATAL_ERROR) +CMAKE_POLICY(VERSION 2.6) +IF(LUAROCKS_PREFIX) + MESSAGE(STATUS "Installing CuDNN through Luarocks") + STRING(REGEX REPLACE "(.*)lib/luarocks/rocks.*" "\\1" CMAKE_INSTALL_PREFIX "${LUAROCKS_PREFIX}") + MESSAGE(STATUS "Prefix inferred from Luarocks: ${CMAKE_INSTALL_PREFIX}") +ENDIF() +FIND_PACKAGE(Torch REQUIRED) + +FILE(GLOB luasrc *.lua) +SET(src "") +ADD_TORCH_PACKAGE(cudnn "${src}" "${luasrc}" "NVIDIA CuDNN Bindings")
\ No newline at end of file |