|
|
@@ -44,3 +44,14 @@ if(CMAKE_CUDA_COMPILER_TOOLKIT_VERSION VERSION_GREATER_EQUAL 11.4
|
|
|
AND (NOT CMAKE_CUDA_COMPILER_ID STREQUAL "Clang"))
|
|
|
list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 87)
|
|
|
endif()
|
|
|
+
|
|
|
+# only generate jit code for the newest arch for all/all-major
|
|
|
+list(POP_BACK CMAKE_CUDA_ARCHITECTURES_ALL _latest_arch)
|
|
|
+list(TRANSFORM CMAKE_CUDA_ARCHITECTURES_ALL APPEND "-real")
|
|
|
+list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL ${_latest_arch})
|
|
|
+
|
|
|
+list(POP_BACK CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR _latest_arch)
|
|
|
+list(TRANSFORM CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR APPEND "-real")
|
|
|
+list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR ${_latest_arch})
|
|
|
+
|
|
|
+unset(_latest_arch)
|