|
|
|
@ -219,3 +219,42 @@ else() |
|
|
|
|
unset(CUDA_ARCH_BIN CACHE) |
|
|
|
|
unset(CUDA_ARCH_PTX CACHE) |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
if(HAVE_CUDA) |
|
|
|
|
set(CUDA_LIBS_PATH "") |
|
|
|
|
foreach(p ${CUDA_LIBRARIES} ${CUDA_npp_LIBRARY}) |
|
|
|
|
get_filename_component(_tmp ${p} PATH) |
|
|
|
|
list(APPEND CUDA_LIBS_PATH ${_tmp}) |
|
|
|
|
endforeach() |
|
|
|
|
|
|
|
|
|
if(HAVE_CUBLAS) |
|
|
|
|
foreach(p ${CUDA_cublas_LIBRARY}) |
|
|
|
|
get_filename_component(_tmp ${p} PATH) |
|
|
|
|
list(APPEND CUDA_LIBS_PATH ${_tmp}) |
|
|
|
|
endforeach() |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
if(HAVE_CUFFT) |
|
|
|
|
foreach(p ${CUDA_cufft_LIBRARY}) |
|
|
|
|
get_filename_component(_tmp ${p} PATH) |
|
|
|
|
list(APPEND CUDA_LIBS_PATH ${_tmp}) |
|
|
|
|
endforeach() |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
list(REMOVE_DUPLICATES CUDA_LIBS_PATH) |
|
|
|
|
link_directories(${CUDA_LIBS_PATH}) |
|
|
|
|
|
|
|
|
|
set(CUDA_LIBRARIES_ABS ${CUDA_LIBRARIES}) |
|
|
|
|
ocv_convert_to_lib_name(CUDA_LIBRARIES ${CUDA_LIBRARIES}) |
|
|
|
|
set(CUDA_npp_LIBRARY_ABS ${CUDA_npp_LIBRARY}) |
|
|
|
|
ocv_convert_to_lib_name(CUDA_npp_LIBRARY ${CUDA_npp_LIBRARY}) |
|
|
|
|
if(HAVE_CUBLAS) |
|
|
|
|
set(CUDA_cublas_LIBRARY_ABS ${CUDA_cublas_LIBRARY}) |
|
|
|
|
ocv_convert_to_lib_name(CUDA_cublas_LIBRARY ${CUDA_cublas_LIBRARY}) |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
if(HAVE_CUFFT) |
|
|
|
|
set(CUDA_cufft_LIBRARY_ABS ${CUDA_cufft_LIBRARY}) |
|
|
|
|
ocv_convert_to_lib_name(CUDA_cufft_LIBRARY ${CUDA_cufft_LIBRARY}) |
|
|
|
|
endif() |
|
|
|
|
endif() |