From 6e264a905bec9e4c0111eb4c91379c88accef7c6 Mon Sep 17 00:00:00 2001 From: Diego Devesa Date: Sat, 25 Jan 2025 17:22:41 +0100 Subject: [PATCH] docker : add GGML_CPU_ARM_ARCH arg to select ARM architecture to build for (#11419) --- .devops/cpu.Dockerfile | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/.devops/cpu.Dockerfile b/.devops/cpu.Dockerfile index ab0e951bc..522ee8147 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)