cmake: fix linker deps for CUDA

09f0e177 · Alexander Alekhin · bfdbc9ce · 09f0e177 · 09f0e177
Commit 09f0e177 authored Oct 16, 2013 by Alexander Alekhin
Hide whitespace changes
Inline Side-by-side

Showing with 14 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +12 -0

CMakeLists.txt apps/CMakeLists.txt +2 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -456,6 +456,18 @@ if(WITH_OPENCL)
  include(cmake/OpenCVDetectOpenCL.cmake)
 endif()

+# ----------------------------------------------------------------------------
+# Add CUDA libraries (needed for apps/tools, samples)
+# ----------------------------------------------------------------------------
+if(HAVE_CUDA)
+  set(OPENCV_LINKER_LIBS ${OPENCV_LINKER_LIBS} ${CUDA_LIBRARIES} ${CUDA_npp_LIBRARY})
+  if(HAVE_CUBLAS)
+    set(OPENCV_LINKER_LIBS ${OPENCV_LINKER_LIBS} ${CUDA_cublas_LIBRARY})
+  endif()
+  if(HAVE_CUFFT)
+    set(OPENCV_LINKER_LIBS ${OPENCV_LINKER_LIBS} ${CUDA_cufft_LIBRARY})
+  endif()
+endif()
 # ----------------------------------------------------------------------------
 # Solution folders:
 # ----------------------------------------------------------------------------

--- a/apps/CMakeLists.txt
+++ b/apps/CMakeLists.txt
+link_libraries(${OPENCV_LINKER_LIBS})
+
 add_subdirectory(haartraining)
 add_subdirectory(traincascade)