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)
|