Skip to content

Commit 72046e5

Browse files
authored
build: fix CI CUDA build (#619)
1 parent 4b1797f commit 72046e5

3 files changed

Lines changed: 22 additions & 22 deletions

File tree

docs/blog/v3.19-gemma-4.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
---
22
title: Gemma 4 is here!
3-
date: 2026-06-29T17:00:00Z
3+
date: 2026-06-29T18:00:00Z
44
lastUpdated: false
55
author:
66
name: Gilad S.

llama/CMakeLists.txt

Lines changed: 19 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -85,28 +85,26 @@ else()
8585
set(NLC_GGML_NATIVE ON)
8686
endif()
8787

88-
if (DEFINED ENV{CI} AND ENV{CI} STREQUAL "true")
89-
if (GGML_CUDA AND NOT DEFINED CMAKE_CUDA_ARCHITECTURES AND NOT NLC_GGML_NATIVE)
90-
find_package(CUDAToolkit)
91-
if (CUDAToolkit_VERSION VERSION_LESS "13")
92-
list(APPEND CMAKE_CUDA_ARCHITECTURES 50-virtual 61-virtual 70-virtual)
93-
list(APPEND CMAKE_CUDA_ARCHITECTURES 75-virtual 80-virtual 86-real)
94-
95-
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8")
96-
list(APPEND CMAKE_CUDA_ARCHITECTURES 89-real 90-virtual)
97-
endif()
98-
99-
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.8")
100-
list(APPEND CMAKE_CUDA_ARCHITECTURES 120a-real)
101-
endif()
102-
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.9")
103-
list(APPEND CMAKE_CUDA_ARCHITECTURES 121a-real)
104-
endif()
105-
106-
# Remove `90-virtual` due to low usage and because it inflates the binary size too much
107-
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES 90-virtual)
108-
message(STATUS "CUDAToolkit version: ${CUDAToolkit_VERSION}, setting CMAKE_CUDA_ARCHITECTURES to ${CMAKE_CUDA_ARCHITECTURES}")
88+
if (NLC_CI_MODE AND GGML_CUDA AND NOT DEFINED CMAKE_CUDA_ARCHITECTURES AND NOT NLC_GGML_NATIVE)
89+
find_package(CUDAToolkit)
90+
if (CUDAToolkit_VERSION VERSION_LESS "13")
91+
list(APPEND CMAKE_CUDA_ARCHITECTURES 50-virtual 61-virtual 70-virtual)
92+
list(APPEND CMAKE_CUDA_ARCHITECTURES 75-virtual 80-virtual 86-real)
93+
94+
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8")
95+
list(APPEND CMAKE_CUDA_ARCHITECTURES 89-real 90-virtual)
10996
endif()
97+
98+
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.8")
99+
list(APPEND CMAKE_CUDA_ARCHITECTURES 120a-real)
100+
endif()
101+
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.9")
102+
list(APPEND CMAKE_CUDA_ARCHITECTURES 121a-real)
103+
endif()
104+
105+
# Remove `90-virtual` due to low usage and because it inflates the binary size too much
106+
list(REMOVE_ITEM CMAKE_CUDA_ARCHITECTURES 90-virtual)
107+
message(STATUS "CUDAToolkit version: ${CUDAToolkit_VERSION}, setting CMAKE_CUDA_ARCHITECTURES to ${CMAKE_CUDA_ARCHITECTURES}")
110108
endif()
111109
endif()
112110

src/bindings/utils/compileLLamaCpp.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,8 @@ export async function compileLlamaCpp(buildOptions: BuildOptions, compileOptions
154154
cmakeCustomOptions.set("GGML_OPENMP", "OFF");
155155

156156
if (ciMode) {
157+
cmakeCustomOptions.set("NLC_CI_MODE", "ON");
158+
157159
if (!cmakeCustomOptions.has("CMAKE_OSX_DEPLOYMENT_TARGET"))
158160
cmakeCustomOptions.set("CMAKE_OSX_DEPLOYMENT_TARGET", "14.0");
159161

0 commit comments

Comments
 (0)