Blame view

3rdparty/opencv-4.5.4/cmake/templates/OpenCVConfig-CUDA.cmake.in 1.76 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
  # Version Compute Capability from which OpenCV has been compiled is remembered
  set(OpenCV_COMPUTE_CAPABILITIES "@OpenCV_CUDA_CC@")
  
  set(OpenCV_CUDA_VERSION "@CUDA_VERSION_STRING@")
  set(OpenCV_USE_CUBLAS   "@HAVE_CUBLAS@")
  set(OpenCV_USE_CUFFT    "@HAVE_CUFFT@")
  set(OpenCV_USE_NVCUVID  "@HAVE_NVCUVID@")
  
  set(OpenCV_CUDNN_VERSION    "@CUDNN_VERSION@")
  set(OpenCV_USE_CUDNN        "@HAVE_CUDNN@")
  
  if(NOT CUDA_FOUND)
    find_host_package(CUDA ${OpenCV_CUDA_VERSION} EXACT REQUIRED)
  else()
    if(NOT CUDA_VERSION_STRING VERSION_EQUAL OpenCV_CUDA_VERSION)
      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}")
    endif()
  endif()
  
  set(OpenCV_CUDA_LIBS_ABSPATH ${CUDA_LIBRARIES})
  
  if(CUDA_VERSION VERSION_LESS "5.5")
    list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_npp_LIBRARY})
  else()
    find_cuda_helper_libs(nppc)
    find_cuda_helper_libs(nppi)
    find_cuda_helper_libs(npps)
    list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nppc_LIBRARY} ${CUDA_nppi_LIBRARY} ${CUDA_npps_LIBRARY})
  endif()
  
  if(OpenCV_USE_CUBLAS)
    list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_CUBLAS_LIBRARIES})
  endif()
  
  if(OpenCV_USE_CUFFT)
    list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_CUFFT_LIBRARIES})
  endif()
  
  if(OpenCV_USE_NVCUVID)
    list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nvcuvid_LIBRARIES})
  endif()
  
  if(WIN32)
    list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nvcuvenc_LIBRARIES})
  endif()
  
  set(OpenCV_CUDA_LIBS_RELPATH "")
  foreach(l ${OpenCV_CUDA_LIBS_ABSPATH})
    get_filename_component(_tmp ${l} PATH)
    if(NOT ${_tmp} MATCHES "-Wl.*")
      list(APPEND OpenCV_CUDA_LIBS_RELPATH ${_tmp})
    endif()
  endforeach()
  
  list(REMOVE_DUPLICATES OpenCV_CUDA_LIBS_RELPATH)
  link_directories(${OpenCV_CUDA_LIBS_RELPATH})