From fa32abe5f365a363625247748eb41f4e6dd6e572 Mon Sep 17 00:00:00 2001 From: StevenPuttemans Date: Fri, 14 Oct 2016 10:14:12 +0200 Subject: [PATCH] add 5.2 and 6.1 compute capability support --- cmake/OpenCVDetectCUDA.cmake | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/OpenCVDetectCUDA.cmake b/cmake/OpenCVDetectCUDA.cmake index 70b11eb532..72d7a09faa 100644 --- a/cmake/OpenCVDetectCUDA.cmake +++ b/cmake/OpenCVDetectCUDA.cmake @@ -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()