@@ -85,28 +85,26 @@ else()
8585 set (NLC_GGML_NATIVE ON )
8686endif ()
8787
88- if (DEFINED ENV{CI} AND ENV{CI} STREQUAL "true" )
89- if (GGML_CUDA AND NOT DEFINED CMAKE_CUDA_ARCHITECTURES AND NOT NLC_GGML_NATIVE)
90- find_package (CUDAToolkit )
91- if (CUDAToolkit_VERSION VERSION_LESS "13" )
92- list (APPEND CMAKE_CUDA_ARCHITECTURES 50-virtual 61-virtual 70-virtual)
93- list (APPEND CMAKE_CUDA_ARCHITECTURES 75-virtual 80-virtual 86-real)
94-
95- if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8" )
96- list (APPEND CMAKE_CUDA_ARCHITECTURES 89-real 90-virtual)
97- endif ()
98-
99- if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.8" )
100- list (APPEND CMAKE_CUDA_ARCHITECTURES 120a-real)
101- endif ()
102- if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.9" )
103- list (APPEND CMAKE_CUDA_ARCHITECTURES 121a-real)
104- endif ()
105-
106- # Remove `90-virtual` due to low usage and because it inflates the binary size too much
107- list (REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES 90-virtual)
108- message (STATUS "CUDAToolkit version: ${CUDAToolkit_VERSION} , setting CMAKE_CUDA_ARCHITECTURES to ${CMAKE_CUDA_ARCHITECTURES} " )
88+ if (NLC_CI_MODE AND GGML_CUDA AND NOT DEFINED CMAKE_CUDA_ARCHITECTURES AND NOT NLC_GGML_NATIVE)
89+ find_package (CUDAToolkit )
90+ if (CUDAToolkit_VERSION VERSION_LESS "13" )
91+ list (APPEND CMAKE_CUDA_ARCHITECTURES 50-virtual 61-virtual 70-virtual)
92+ list (APPEND CMAKE_CUDA_ARCHITECTURES 75-virtual 80-virtual 86-real)
93+
94+ if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8" )
95+ list (APPEND CMAKE_CUDA_ARCHITECTURES 89-real 90-virtual)
10996 endif ()
97+
98+ if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.8" )
99+ list (APPEND CMAKE_CUDA_ARCHITECTURES 120a-real)
100+ endif ()
101+ if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.9" )
102+ list (APPEND CMAKE_CUDA_ARCHITECTURES 121a-real)
103+ endif ()
104+
105+ # Remove `90-virtual` due to low usage and because it inflates the binary size too much
106+ list (REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES 90-virtual)
107+ message (STATUS "CUDAToolkit version: ${CUDAToolkit_VERSION} , setting CMAKE_CUDA_ARCHITECTURES to ${CMAKE_CUDA_ARCHITECTURES} " )
110108 endif ()
111109endif ()
112110
0 commit comments