|
|
|
@ -29,10 +29,42 @@ if(CUDA_FOUND) |
|
|
|
|
if(${CUDA_VERSION} VERSION_LESS "5.5") |
|
|
|
|
find_cuda_helper_libs(npp) |
|
|
|
|
else() |
|
|
|
|
find_cuda_helper_libs(nppc) |
|
|
|
|
find_cuda_helper_libs(nppi) |
|
|
|
|
find_cuda_helper_libs(npps) |
|
|
|
|
set(CUDA_npp_LIBRARY ${CUDA_nppc_LIBRARY} ${CUDA_nppi_LIBRARY} ${CUDA_npps_LIBRARY}) |
|
|
|
|
# hack for CUDA 5.5 |
|
|
|
|
if(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "arm") |
|
|
|
|
unset(CUDA_TOOLKIT_INCLUDE CACHE) |
|
|
|
|
unset(CUDA_CUDART_LIBRARY CACHE) |
|
|
|
|
unset(CUDA_cublas_LIBRARY CACHE) |
|
|
|
|
unset(CUDA_cufft_LIBRARY CACHE) |
|
|
|
|
unset(CUDA_npp_LIBRARY CACHE) |
|
|
|
|
|
|
|
|
|
if(SOFTFP) |
|
|
|
|
set(cuda_arm_path "${CUDA_TOOLKIT_ROOT_DIR}/targets/armv7-linux-gnueabi") |
|
|
|
|
else() |
|
|
|
|
set(cuda_arm_path "${CUDA_TOOLKIT_ROOT_DIR}/targets/armv7-linux-gnueabihf") |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
set(CUDA_TOOLKIT_INCLUDE "${cuda_arm_path}/include" CACHE PATH "include path") |
|
|
|
|
set(CUDA_INCLUDE_DIRS ${CUDA_TOOLKIT_INCLUDE}) |
|
|
|
|
|
|
|
|
|
set(cuda_arm_library_path "${cuda_arm_path}/lib") |
|
|
|
|
|
|
|
|
|
set(CUDA_CUDART_LIBRARY "${cuda_arm_library_path}/libcudart.so" CACHE FILEPATH "cudart library") |
|
|
|
|
set(CUDA_LIBRARIES ${CUDA_CUDART_LIBRARY}) |
|
|
|
|
set(CUDA_cublas_LIBRARY "${cuda_arm_library_path}/libcublas.so" CACHE FILEPATH "cublas library") |
|
|
|
|
set(CUDA_cufft_LIBRARY "${cuda_arm_library_path}/libcufft.so" CACHE FILEPATH "cufft library") |
|
|
|
|
set(CUDA_nppc_LIBRARY "${cuda_arm_library_path}/libnppc.so" CACHE FILEPATH "nppc library") |
|
|
|
|
set(CUDA_nppi_LIBRARY "${cuda_arm_library_path}/libnppi.so" CACHE FILEPATH "nppi library") |
|
|
|
|
set(CUDA_npps_LIBRARY "${cuda_arm_library_path}/libnpps.so" CACHE FILEPATH "npps library") |
|
|
|
|
set(CUDA_npp_LIBRARY "${CUDA_nppc_LIBRARY};${CUDA_nppi_LIBRARY};${CUDA_npps_LIBRARY}" CACHE STRING "npp library") |
|
|
|
|
else() |
|
|
|
|
unset(CUDA_npp_LIBRARY CACHE) |
|
|
|
|
|
|
|
|
|
find_cuda_helper_libs(nppc) |
|
|
|
|
find_cuda_helper_libs(nppi) |
|
|
|
|
find_cuda_helper_libs(npps) |
|
|
|
|
|
|
|
|
|
set(CUDA_npp_LIBRARY "${CUDA_nppc_LIBRARY};${CUDA_nppi_LIBRARY};${CUDA_npps_LIBRARY}" CACHE STRING "npp library") |
|
|
|
|
endif() |
|
|
|
|
endif() |
|
|
|
|
|
|
|
|
|
if(WITH_NVCUVID) |
|
|
|
|