From: Diego Devesa <redacted>
Date: Sat, 25 Jan 2025 16:22:41 +0000 (+0100)
Subject: docker : add GGML_CPU_ARM_ARCH arg to select ARM architecture to build for (#11419)
X-Git-Tag: upstream/0.0.4631~78
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=6e264a905bec9e4c0111eb4c91379c88accef7c6;p=pkg%2Fggml%2Fsources%2Fllama.cpp

docker : add GGML_CPU_ARM_ARCH arg to select ARM architecture to build for (#11419)
---

diff --git a/.devops/cpu.Dockerfile b/.devops/cpu.Dockerfile
index ab0e951b..522ee814 100644
--- a/.devops/cpu.Dockerfile
+++ b/.devops/cpu.Dockerfile
@@ -4,6 +4,8 @@ FROM ubuntu:$UBUNTU_VERSION AS build
 
 ARG TARGETARCH
 
+ARG GGML_CPU_ARM_ARCH=armv8-a
+
 RUN apt-get update && \
     apt-get install -y build-essential git cmake libcurl4-openssl-dev
 
@@ -12,9 +14,12 @@ WORKDIR /app
 COPY . .
 
 RUN if [ "$TARGETARCH" = "amd64" ]; then \
-        cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DLLAMA_CURL=ON -DGGML_BACKEND_DL=ON -DGGML_NATIVE=OFF -DGGML_CPU_ALL_VARIANTS=ON; \
+        cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DLLAMA_CURL=ON -DGGML_NATIVE=OFF -DGGML_BACKEND_DL=ON -DGGML_CPU_ALL_VARIANTS=ON; \
+    elif [ "$TARGETARCH" = "arm64" ]; then \
+        cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DLLAMA_CURL=ON -DGGML_NATIVE=OFF -DGGML_CPU_ARM_ARCH=${GGML_CPU_ARM_ARCH}; \
     else \
-        cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DLLAMA_CURL=ON; \
+        echo "Unsupported architecture"; \
+        exit 1; \
     fi && \
     cmake --build build -j $(nproc)