@@ -15,7 +15,6 @@ if (CUDAToolkit_FOUND)
1515 # 80 == Ampere, asynchronous data loading, faster tensor core instructions
1616 # 86 == RTX 3000, needs CUDA v11.1
1717 # 89 == RTX 4000, needs CUDA v11.8
18- # 120 == Blackwell, compute capability 12.8.1 minimum
1918 #
2019 # XX-virtual == compile CUDA code as PTX, do JIT compilation to binary code on first run
2120 # XX-real == compile CUDA code as device code for this specific architecture
@@ -27,15 +26,15 @@ if (CUDAToolkit_FOUND)
2726 set (CMAKE_CUDA_ARCHITECTURES "native" )
2827 elseif (GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
2928 if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8" )
30- set (CMAKE_CUDA_ARCHITECTURES "60-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;89-real;120-real " )
29+ set (CMAKE_CUDA_ARCHITECTURES "60-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;89-real" )
3130 else ()
32- set (CMAKE_CUDA_ARCHITECTURES "60-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;120-real " )
31+ set (CMAKE_CUDA_ARCHITECTURES "60-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real" )
3332 endif ()
3433 else ()
3534 if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8" )
36- set (CMAKE_CUDA_ARCHITECTURES "50-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;89-real;120-real " )
35+ set (CMAKE_CUDA_ARCHITECTURES "50-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;89-real" )
3736 else ()
38- set (CMAKE_CUDA_ARCHITECTURES "50-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real;120-real " )
37+ set (CMAKE_CUDA_ARCHITECTURES "50-virtual;61-virtual;70-virtual;75-virtual;80-virtual;86-real" )
3938 endif ()
4039 endif ()
4140 endif ()
0 commit comments