Skip to content

Commit 481c554

Browse files
committed
build: fix CI
1 parent 70da4ab commit 481c554

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

llama/CMakeLists.txt

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,19 @@ 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;72-virtual;75-virtual;80-virtual;86-real;89-real;90-real")
79+
if (DEFINED GGML_NATIVE)
80+
set(NLC_GGML_NATIVE ${GGML_NATIVE})
81+
elseif(CMAKE_CROSSCOMPILING OR DEFINED ENV{SOURCE_DATE_EPOCH})
82+
set(NLC_GGML_NATIVE OFF)
83+
else()
84+
set(NLC_GGML_NATIVE ON)
85+
endif()
86+
87+
if (GGML_CUDA AND NOT DEFINED CMAKE_CUDA_ARCHITECTURES AND NOT NLC_GGML_NATIVE)
88+
find_package(CUDAToolkit)
89+
if (CUDAToolkit_FOUND AND CUDAToolkit_VERSION VERSION_GREATER_EQUAL "13.0")
90+
set(CMAKE_CUDA_ARCHITECTURES "70-virtual;72-virtual;75-virtual;80-virtual;86-real;89-real;90-real")
91+
endif()
8192
endif()
8293

8394
add_subdirectory("llama.cpp")

0 commit comments

Comments
 (0)