OpenCVConfig-CUDA.cmake.in 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. # Version Compute Capability from which OpenCV has been compiled is remembered
  2. set(OpenCV_COMPUTE_CAPABILITIES "@OpenCV_CUDA_CC@")
  3. set(OpenCV_CUDA_VERSION "@CUDA_VERSION_STRING@")
  4. set(OpenCV_USE_CUBLAS "@HAVE_CUBLAS@")
  5. set(OpenCV_USE_CUFFT "@HAVE_CUFFT@")
  6. set(OpenCV_USE_NVCUVID "@HAVE_NVCUVID@")
  7. set(OpenCV_USE_NVCUVENC "@HAVE_NVCUVENC@")
  8. set(OpenCV_CUDNN_VERSION "@CUDNN_VERSION@")
  9. set(OpenCV_USE_CUDNN "@HAVE_CUDNN@")
  10. if(NOT CUDA_FOUND)
  11. find_host_package(CUDA ${OpenCV_CUDA_VERSION} EXACT REQUIRED)
  12. else()
  13. if(NOT CUDA_VERSION_STRING VERSION_EQUAL OpenCV_CUDA_VERSION)
  14. message(FATAL_ERROR "OpenCV static library was compiled with CUDA ${OpenCV_CUDA_VERSION} support. Please, use the same version or rebuild OpenCV with CUDA ${CUDA_VERSION_STRING}")
  15. endif()
  16. endif()
  17. set(OpenCV_CUDA_LIBS_ABSPATH ${CUDA_LIBRARIES})
  18. if(CUDA_VERSION VERSION_LESS "5.5")
  19. list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_npp_LIBRARY})
  20. else()
  21. find_cuda_helper_libs(nppc)
  22. find_cuda_helper_libs(nppi)
  23. find_cuda_helper_libs(npps)
  24. list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nppc_LIBRARY} ${CUDA_nppi_LIBRARY} ${CUDA_npps_LIBRARY})
  25. endif()
  26. if(OpenCV_USE_CUBLAS)
  27. list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_CUBLAS_LIBRARIES})
  28. endif()
  29. if(OpenCV_USE_CUFFT)
  30. list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_CUFFT_LIBRARIES})
  31. endif()
  32. set(OpenCV_CUDA_LIBS_RELPATH "")
  33. foreach(l ${OpenCV_CUDA_LIBS_ABSPATH})
  34. get_filename_component(_tmp ${l} PATH)
  35. if(NOT ${_tmp} MATCHES "-Wl.*")
  36. list(APPEND OpenCV_CUDA_LIBS_RELPATH ${_tmp})
  37. endif()
  38. endforeach()
  39. list(REMOVE_DUPLICATES OpenCV_CUDA_LIBS_RELPATH)
  40. link_directories(${OpenCV_CUDA_LIBS_RELPATH})