Install THCUNN into ${Torch_INSTALL_LUA_CPATH_SUBDIR}
diff --git a/CMakeLists.txt b/CMakeLists.txt
index dd58697..9956f4c 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,6 +1,13 @@
CMAKE_MINIMUM_REQUIRED(VERSION 2.8 FATAL_ERROR)
CMAKE_POLICY(VERSION 2.8)
+FIND_PACKAGE(Torch REQUIRED)
+FIND_PACKAGE(CUDA 4.0 REQUIRED)
+
+IF(NOT THCUNN_INSTALL_LIB_SUBDIR)
+ SET(THCUNN_INSTALL_LIB_SUBDIR "lib" CACHE PATH "THCUNN install library directory")
+ENDIF()
+
FILE(GLOB src-cuda *.cu)
CUDA_ADD_LIBRARY(THCUNN MODULE ${src-cuda})
@@ -8,6 +15,4 @@
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
TARGET_LINK_LIBRARIES(THCUNN THC TH)
-INSTALL(TARGETS THCUNN
- RUNTIME DESTINATION ${Torch_INSTALL_LIB_SUBDIR}
- LIBRARY DESTINATION ${Torch_INSTALL_LIB_SUBDIR})
+INSTALL(TARGETS THCUNN LIBRARY DESTINATION ${THCUNN_INSTALL_LIB_SUBDIR})
