|
|
|
@ -67,9 +67,9 @@ if(CUDA_FOUND) |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Kepler") |
|
|
|
|
set(__cuda_arch_bin "3.0 3.5 3.7") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Maxwell") |
|
|
|
|
set(__cuda_arch_bin "5.0") |
|
|
|
|
set(__cuda_arch_bin "5.0 5.2") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Pascal") |
|
|
|
|
set(__cuda_arch_bin "6.0") |
|
|
|
|
set(__cuda_arch_bin "6.0 6.1") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Auto") |
|
|
|
|
execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run" |
|
|
|
|
WORKING_DIRECTORY "${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/" |
|
|
|
@ -92,9 +92,9 @@ if(CUDA_FOUND) |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
else() |
|
|
|
|
if(${CUDA_VERSION} VERSION_LESS "8.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 5.2") |
|
|
|
|
else() |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 6.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 5.2 6.0 6.1") |
|
|
|
|
endif() |
|
|
|
|
set(__cuda_arch_ptx "") |
|
|
|
|
endif() |
|
|
|
|