Skip to content

Commit c7fc98f

Browse files
committed
Revert cmake hackery
1 parent 865d0b1 commit c7fc98f

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

ggml/src/ggml-cuda/CMakeLists.txt

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)