|
|
|
@ -81,6 +81,8 @@ if(CUDA_FOUND) |
|
|
|
|
set(__cuda_arch_bin "6.0 6.1") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Volta") |
|
|
|
|
set(__cuda_arch_bin "7.0") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Turing") |
|
|
|
|
set(__cuda_arch_bin "7.5") |
|
|
|
|
elseif(CUDA_GENERATION STREQUAL "Auto") |
|
|
|
|
execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" ${CUDA_NVCC_FLAGS} "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run" |
|
|
|
|
WORKING_DIRECTORY "${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/" |
|
|
|
@ -105,7 +107,7 @@ if(CUDA_FOUND) |
|
|
|
|
ERROR_QUIET OUTPUT_STRIP_TRAILING_WHITESPACE) |
|
|
|
|
if(NOT _nvcc_res EQUAL 0) |
|
|
|
|
message(STATUS "Automatic detection of CUDA generation failed. Going to build for all known architectures.") |
|
|
|
|
set(__cuda_arch_bin "5.3 6.2 7.0") |
|
|
|
|
set(__cuda_arch_bin "5.3 6.2 7.0 7.5") |
|
|
|
|
else() |
|
|
|
|
set(__cuda_arch_bin "${_nvcc_out}") |
|
|
|
|
string(REPLACE "2.1" "2.1(2.0)" __cuda_arch_bin "${__cuda_arch_bin}") |
|
|
|
@ -114,8 +116,10 @@ if(CUDA_FOUND) |
|
|
|
|
else() |
|
|
|
|
if(${CUDA_VERSION} VERSION_LESS "9.0") |
|
|
|
|
set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 5.2 6.0 6.1") |
|
|
|
|
else() |
|
|
|
|
elseif(${CUDA_VERSION} VERSION_LESS "10.0") |
|
|
|
|
set(__cuda_arch_bin "3.0 3.5 3.7 5.0 5.2 6.0 6.1 7.0") |
|
|
|
|
else() |
|
|
|
|
set(__cuda_arch_bin "3.0 3.5 3.7 5.0 5.2 6.0 6.1 7.0 7.5") |
|
|
|
|
endif() |
|
|
|
|
endif() |
|
|
|
|
endif() |
|
|
|
|