File: all_cuda_architectures.patch

package info (click to toggle)
cccl 2.3.2-2
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 89,900 kB
  • sloc: cpp: 697,664; ansic: 26,964; python: 11,928; sh: 3,284; asm: 2,154; perl: 460; makefile: 112; xml: 13
file content (26 lines) | stat: -rw-r--r-- 825 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
@@ -61,7 +61,20 @@ if (LIBCUDACXX_TOPLEVEL_PROJECT)
 endif()
 
 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 ()
 
 set(_libcudacxx_enable_upstream_tests OFF)