File: all_cuda_architectures.patch

package info (click to toggle)
cccl 2.4.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 41,200 kB
  • sloc: cpp: 236,258; python: 6,415; sh: 2,407; perl: 460; makefile: 114; xml: 13
file content (26 lines) | stat: -rw-r--r-- 946 bytes parent folder | download
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
Author: Andreas Beckmann <anbe@debian.org>
Description: compile CUDA tests for all supported CUDA architectures

--- a/libcudacxx/CMakeLists.txt
+++ b/libcudacxx/CMakeLists.txt
@@ -51,7 +51,20 @@ set(LLVM_PATH "${libcudacxx_SOURCE_DIR}"
 # Configuration options.
 option(LIBCUDACXX_ENABLE_CUDA "Enable the CUDA language support." ON)
 if (LIBCUDACXX_ENABLE_CUDA)
+  if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+    set(CMAKE_CUDA_ARCHITECTURES "all")
+  endif()
+
   enable_language(CUDA)
+
+  if(CMAKE_CUDA_ARCHITECTURES STREQUAL "all")
+    set(CMAKE_CUDA_ARCHITECTURES "${CMAKE_CUDA_ARCHITECTURES_ALL}")
+    list(TRANSFORM CMAKE_CUDA_ARCHITECTURES REPLACE "-real" "")
+  endif()
+  # for libcudacxx
+  set(LIBCUDACXX_COMPUTE_ARCHS "${CMAKE_CUDA_ARCHITECTURES}")
+  # for cub and thrust
+  set(CMAKE_CUDA_ARCHITECTURES "${CMAKE_CUDA_ARCHITECTURES}" PARENT_SCOPE)
 endif ()
 
 option(LIBCUDACXX_ENABLE_LIBCUDACXX_TESTS "Enable libcu++ tests." ON)