Skip to content

Commit f105e0f

Browse files
committed
Revert compile for Ampere
1 parent 1b25cbb commit f105e0f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -308,9 +308,9 @@ if (GGML_CUDA)
308308
# 61 == integer CUDA intrinsics
309309
# 70 == compute capability at which unrolling a loop in mul_mat_q kernels is faster
310310
if (GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
311-
set(CMAKE_CUDA_ARCHITECTURES "52;60;61;70;75;86")
311+
set(CMAKE_CUDA_ARCHITECTURES "60;61;70;75")
312312
else()
313-
set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75;86")
313+
set(CMAKE_CUDA_ARCHITECTURES "52;61;70;75")
314314
#set(CMAKE_CUDA_ARCHITECTURES "OFF") # use this to compile much faster, but only F16 models work
315315
endif()
316316
endif()

0 commit comments

Comments
 (0)