Skip to content

Commit 9ee77fc

Browse files
authored
fix(ci): disable dynamic CPU backends for arm64 CUDA image (#1709)
1 parent 39f7962 commit 9ee77fc

3 files changed

Lines changed: 23 additions & 13 deletions

File tree

.github/workflows/build.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,7 @@ jobs:
207207
UBUNTU_VERSION=24.04
208208
CUDA_ARCHITECTURES=121
209209
GGML_CUDA_FA_ALL_QUANTS=ON
210+
GGML_CUDA_ENABLE_DYNAMIC_CPU_BACKENDS=OFF
210211
211212
env:
212213
REGISTRY: ghcr.io

Dockerfile.cuda

Lines changed: 21 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -23,18 +23,27 @@ COPY . .
2323
ARG CUDACXX=/usr/local/cuda/bin/nvcc
2424
ARG CUDA_ARCHITECTURES=""
2525
ARG GGML_CUDA_FA_ALL_QUANTS=""
26-
27-
RUN cmake . -B ./build \
28-
-DSD_CUDA=ON \
29-
-DSD_BUILD_SHARED_LIBS=ON \
30-
-DGGML_NATIVE=OFF \
31-
-DSD_BUILD_SHARED_GGML_LIB=ON \
32-
-DGGML_BACKEND_DL=ON \
33-
-DGGML_CPU_ALL_VARIANTS=ON \
34-
-DCMAKE_BUILD_WITH_INSTALL_RPATH=ON \
35-
-DCMAKE_INSTALL_RPATH='$ORIGIN' \
36-
${CUDA_ARCHITECTURES:+-DCMAKE_CUDA_ARCHITECTURES="${CUDA_ARCHITECTURES}"} \
37-
${GGML_CUDA_FA_ALL_QUANTS:+-DGGML_CUDA_FA_ALL_QUANTS=${GGML_CUDA_FA_ALL_QUANTS}}
26+
ARG GGML_CUDA_ENABLE_DYNAMIC_CPU_BACKENDS=ON
27+
28+
RUN set -- \
29+
-DSD_CUDA=ON; \
30+
if [ "${GGML_CUDA_ENABLE_DYNAMIC_CPU_BACKENDS}" = "ON" ]; then \
31+
set -- "$@" \
32+
-DSD_BUILD_SHARED_LIBS=ON \
33+
-DGGML_NATIVE=OFF \
34+
-DSD_BUILD_SHARED_GGML_LIB=ON \
35+
-DGGML_BACKEND_DL=ON \
36+
-DGGML_CPU_ALL_VARIANTS=ON \
37+
-DCMAKE_BUILD_WITH_INSTALL_RPATH=ON \
38+
"-DCMAKE_INSTALL_RPATH=\$ORIGIN"; \
39+
fi; \
40+
if [ -n "${CUDA_ARCHITECTURES}" ]; then \
41+
set -- "$@" "-DCMAKE_CUDA_ARCHITECTURES=${CUDA_ARCHITECTURES}"; \
42+
fi; \
43+
if [ -n "${GGML_CUDA_FA_ALL_QUANTS}" ]; then \
44+
set -- "$@" "-DGGML_CUDA_FA_ALL_QUANTS=${GGML_CUDA_FA_ALL_QUANTS}"; \
45+
fi; \
46+
cmake . -B ./build "$@"
3847
RUN cmake --build ./build --config Release -j$(nproc)
3948

4049
FROM nvidia/cuda:${CUDA_VERSION}-cudnn-runtime-ubuntu${UBUNTU_VERSION} AS runtime

Dockerfile.vulkan

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ RUN cmake . -B ./build \
2828
-DGGML_CPU_ALL_VARIANTS=ON \
2929
-DCMAKE_BUILD_WITH_INSTALL_RPATH=ON \
3030
-DCMAKE_INSTALL_RPATH='$ORIGIN'
31-
RUN cmake --build ./build --config Release --parallel
31+
RUN cmake --build ./build --config Release -j$(nproc)
3232

3333
FROM ubuntu:$UBUNTU_VERSION AS runtime
3434

0 commit comments

Comments
 (0)