Skip to content

Commit b143f71

Browse files
authored
Merge pull request opencv#9505 from cbalint13:cuda
Enable CUDA_NVCC_FLAGS at arch detect step.
2 parents 5a94610 + 0c4e294 commit b143f71

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cmake/OpenCVDetectCUDA.cmake

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ if(CUDA_FOUND)
7373
elseif(CUDA_GENERATION STREQUAL "Volta")
7474
set(__cuda_arch_bin "7.0")
7575
elseif(CUDA_GENERATION STREQUAL "Auto")
76-
execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run"
76+
execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" ${CUDA_NVCC_FLAGS} "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run"
7777
WORKING_DIRECTORY "${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/"
7878
RESULT_VARIABLE _nvcc_res OUTPUT_VARIABLE _nvcc_out
7979
ERROR_QUIET OUTPUT_STRIP_TRAILING_WHITESPACE)
@@ -90,7 +90,7 @@ if(CUDA_FOUND)
9090
set(__cuda_arch_bin "3.2")
9191
set(__cuda_arch_ptx "")
9292
elseif(AARCH64)
93-
execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run"
93+
execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" ${CUDA_NVCC_FLAGS} "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run"
9494
WORKING_DIRECTORY "${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/"
9595
RESULT_VARIABLE _nvcc_res OUTPUT_VARIABLE _nvcc_out
9696
ERROR_QUIET OUTPUT_STRIP_TRAILING_WHITESPACE)

0 commit comments

Comments
 (0)