aboutsummarylogtreecommitdiffstats
path: root/SelectCudaComputeArch.patch
blob: 3fa6c0a12d418917f883995b74c1f9d65aa2f4b0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
diff --git a/intern/cycles/cmake/external_libs.cmake b/intern/cycles/cmake/external_libs.cmake
index 5bf681792ca..b975fb5db14 100644
--- a/intern/cycles/cmake/external_libs.cmake
+++ b/intern/cycles/cmake/external_libs.cmake
@@ -41,6 +41,10 @@ if(WITH_CYCLES_CUDA_BINARIES OR NOT WITH_CUDA_DYNLOAD)
   find_package(CUDA) # Try to auto locate CUDA toolkit
   if(CUDA_FOUND)
     message(STATUS "CUDA nvcc = ${CUDA_NVCC_EXECUTABLE}")
+      CUDA_SELECT_NVCC_ARCH_FLAGS(CUDA_ARCH_FLAGS Auto)
+        set(CYCLES_CUDA_BINARIES_ARCH "${CUDA_ARCH_FLAGS_readable}")
+        message(STATUS "Enabling CUDA support (version: ${CUDA_VERSION_STRING},"
+          " archs: ${CUDA_ARCH_FLAGS_readable})")
   else()
     message(STATUS "CUDA compiler not found, disabling WITH_CYCLES_CUDA_BINARIES")
     set(WITH_CYCLES_CUDA_BINARIES OFF)