summarylogtreecommitdiffstats
path: root/SelectCudaComputeArch.patch
blob: 385f532ac91c2aa9af7473bcd755d8db0c375b50 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
diff --git a/intern/cycles/cmake/external_libs.cmake b/intern/cycles/cmake/external_libs.cmake
index d0f473a2939..90fcb40107a 100644
--- a/blender-2.79b/intern/cycles/cmake/external_libs.cmake
+++ b/blender-2.79b/intern/cycles/cmake/external_libs.cmake
@@ -41,6 +41,11 @@ 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)
+				string(REGEX REPLACE " compute_[0-9]+" "" CYCLES_CUDA_BINARIES_ARCH "${CUDA_ARCH_FLAGS_readable}")
+				string(REGEX REPLACE " " ";" CYCLES_CUDA_BINARIES_ARCH "${CYCLES_CUDA_BINARIES_ARCH}")
+				message(STATUS "Enabling CUDA support (version: ${CUDA_VERSION_STRING},"
+					" archs: ${CYCLES_CUDA_BINARIES_ARCH})")
 	else()
 		message(STATUS "CUDA compiler not found, disabling WITH_CYCLES_CUDA_BINARIES")
 		set(WITH_CYCLES_CUDA_BINARIES OFF)