Merge branch 'master' into rotation

[hercules2020/kcf.git] / src / CMakeLists.txt

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 901f0dcae82048da05644889a0e8a338a58e1152..56ff3008229adae7d2e67bc35c2b1e456cbd003c 100644 (file)
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -71,7 +71,13 @@ IF(use_cuda)
   include_directories ( ${CUDA_INCLUDE_DIRS} )
   set(CUDA_SEPARABLE_COMPILATION ON)
   set(CUDA_PROPAGATE_HOST_FLAGS OFF)
-  list( APPEND CUDA_NVCC_FLAGS -O3 --gpu-architecture sm_62 -std=c++11)
+
+  if(NOT COMMAND CUDA_SELECT_NVCC_ARCH_FLAGS)
+    include(${CMAKE_SOURCE_DIR}/cmake/FindCUDA/select_compute_arch.cmake)
+  endif()
+
+  CUDA_SELECT_NVCC_ARCH_FLAGS(ARCH_FLAGS Auto)
+  list( APPEND CUDA_NVCC_FLAGS -O3 -std=c++11 ${ARCH_FLAGS}) # --gpu-architecture sm_62 )
   find_cuda_helper_libs(cufftw)
   IF(FFT STREQUAL "cuFFT")
   add_subdirectory(cuda)