Skip to content

Commit 49c77e6

Browse files
committed
cmake : fix cuda vars (#0)
1 parent 1ba593b commit 49c77e6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ if (GGML_CUDA)
194194
# 60 == f16 CUDA intrinsics
195195
# 61 == integer CUDA intrinsics
196196
# 70 == compute capability at which unrolling a loop in mul_mat_q kernels is faster
197-
if (LLAMA_CUDA_F16 OR LLAMA_CUDA_DMMV_F16)
197+
if (GGML_CUDA_F16 OR GGML_CUDA_DMMV_F16)
198198
set(CMAKE_CUDA_ARCHITECTURES "60;61;70") # needed for f16 CUDA intrinsics
199199
else()
200200
set(CMAKE_CUDA_ARCHITECTURES "52;61;70") # lowest CUDA 12 standard + lowest for integer intrinsics

0 commit comments

Comments
 (0)