Revise cmake install for warp-ctc

xhappy · Dec 27, 2016 · 77b03e0 · 77b03e0
1 parent d46a257
commit 77b03e0
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -77,9 +77,11 @@ IF (WITH_GPU)
     TARGET_LINK_LIBRARIES(test_gpu warpctc ${CUDA_curand_LIBRARY})
 
     INSTALL(TARGETS warpctc
-            RUNTIME DESTINATION "${CMAKE_BINARY_DIR}"
-            LIBRARY DESTINATION "${CMAKE_BINARY_DIR}"
-            ARCHIVE DESTINATION "${CMAKE_BINARY_DIR}")
+            RUNTIME DESTINATION "${CMAKE_INSTALL_PREFIX}/bin"
+            LIBRARY DESTINATION "${CMAKE_INSTALL_PREFIX}/lib"
+            ARCHIVE DESTINATION "${CMAKE_INSTALL_PREFIX}/lib")
+
+    INSTALL(DIRECTORY include/ DESTINATION "${CMAKE_INSTALL_PREFIX}/include")
 
     IF (Torch_FOUND)
         MESSAGE(STATUS "Building Torch Bindings with GPU support")
@@ -119,9 +121,11 @@ ELSE()
     SET_TARGET_PROPERTIES(test_cpu PROPERTIES COMPILE_FLAGS "${CMAKE_CXX_FLAGS} --std=c++11")
 
     INSTALL(TARGETS warpctc
-            RUNTIME DESTINATION "${CMAKE_BINARY_DIR}"
-            LIBRARY DESTINATION "${CMAKE_BINARY_DIR}"
-            ARCHIVE DESTINATION "${CMAKE_BINARY_DIR}")
+            RUNTIME DESTINATION "${CMAKE_INSTALL_PREFIX}/bin"
+            LIBRARY DESTINATION "${CMAKE_INSTALL_PREFIX}/lib"
+            ARCHIVE DESTINATION "${CMAKE_INSTALL_PREFIX}/lib")
+
+    INSTALL(DIRECTORY include/ DESTINATION "${CMAKE_INSTALL_PREFIX}/include")
 
     IF (Torch_FOUND)
         MESSAGE(STATUS "Building Torch Bindings with no GPU support")