Skip to content

Commit 203c652

Browse files
committed
build: fix CUDA 13.0 architectures
1 parent fcb0ebe commit 203c652

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

llama/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,10 @@ execute_process(COMMAND node -p "require('node-addon-api').include.slice(1,-1)"
7676
OUTPUT_STRIP_TRAILING_WHITESPACE)
7777
include_directories(${NODE_ADDON_API_DIR} ${CMAKE_JS_INC})
7878

79+
if (GGML_CUDA AND NOT DEFINED CMAKE_CUDA_ARCHITECTURES AND NOT GGML_NATIVE AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "13.0")
80+
set(CMAKE_CUDA_ARCHITECTURES "70-virtual;75-virtual;80-virtual;86-real;89-real;90-real")
81+
endif()
82+
7983
add_subdirectory("llama.cpp")
8084
include_directories("llama.cpp")
8185
include_directories("./llama.cpp/common")

0 commit comments

Comments
 (0)