summarylogtreecommitdiffstats
path: root/0001-SelectCudaComputeArch.patch
blob: 6d7ac561b9443867145fa41c117e4d9133e7b8b8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
From 4b48470fa6378882b5b5b5cf04bebcb3840c8a89 Mon Sep 17 00:00:00 2001
From: howetuft <howetuft@gmail.com>
Date: Tue, 11 Jan 2022 21:59:38 +0100
Subject: [PATCH 1/4] SelectCudaComputeArch

---
 src/cmake/external_libs.cmake | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/cmake/external_libs.cmake b/src/cmake/external_libs.cmake
index 6aeee1a9..2e4c26b7 100644
--- a/src/cmake/external_libs.cmake
+++ b/src/cmake/external_libs.cmake
@@ -129,7 +129,12 @@ endif()
 if(WITH_CYCLES_CUDA_BINARIES OR NOT WITH_CUDA_DYNLOAD)
   find_package(CUDA) # Try to auto locate CUDA toolkit
   if(CUDA_FOUND)
-    message(STATUS "Found CUDA ${CUDA_NVCC_EXECUTABLE} (${CUDA_VERSION})")
+    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)
-- 
2.35.1