diff --git a/.devops/s390x.Dockerfile b/.devops/s390x.Dockerfile index 17aca5537f..01cdef79a8 100644 --- a/.devops/s390x.Dockerfile +++ b/.devops/s390x.Dockerfile @@ -1,36 +1,91 @@ -# vim: filetype=dockerfile - ARG GCC_VERSION=15.2.0 ARG UBUNTU_VERSION=24.10 -ARG SUPPORT_ZDNN=OFF -FROM --platform=linux/s390x gcc:${GCC_VERSION} AS build -RUN apt update -y \ - && apt upgrade -y \ - && apt install -y --no-install-recommends cmake ninja-build \ - && apt install -y --no-install-recommends libopenblas-openmp-dev \ - && rm -rf /var/lib/apt/lists/* +FROM gcc:${GCC_VERSION} AS build + +ARG TARGETARCH + +RUN apt-get update && \ + apt-get install -y build-essential git cmake libcurl4-openssl-dev WORKDIR /app COPY . . -RUN --mount=type=cache,target=/root/.ccache \ - cmake -S . -B build -G Ninja \ - -DCMAKE_BUILD_TYPE=Release \ - -DBUILD_SHARED_LIBS=OFF \ - -DGGML_BLAS=ON \ - -DGGML_BLAS_VENDOR=OpenBLAS \ - -DGGML_ZDNN=$SUPPORT_ZDNN \ - && cmake --build build --config Release -j $(nproc) +RUN if [ "$TARGETARCH" = "amd64" ] || [ "$TARGETARCH" = "arm64" ]; then \ + cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DGGML_NATIVE=OFF -DLLAMA_BUILD_TESTS=OFF -DGGML_BACKEND_DL=ON -DGGML_CPU_ALL_VARIANTS=ON; \ + elif [ "${TARGETARCH}" = "s390x" ]; then \ + cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DGGML_NATIVE=OFF -DLLAMA_BUILD_TESTS=OFF -DGGML_BACKEND_DL=ON -DGGML_CPU_ALL_VARIANTS=OFF; \ + else \ + echo "Unsupported architecture"; \ + exit 1; \ + fi && \ + cmake --build build -j $(nproc) -RUN cmake --install build --prefix /opt/llama.cpp +RUN mkdir -p /app/lib && \ + find build -name "*.so" -exec cp {} /app/lib \; -FROM --platform=linux/s390x scratch AS llama-server +RUN mkdir -p /app/full \ + && cp build/bin/* /app/full \ + && cp *.py /app/full \ + && cp -r gguf-py /app/full \ + && cp -r requirements /app/full \ + && cp requirements.txt /app/full \ + && cp .devops/tools.sh /app/full/tools.sh -COPY --from=build /opt/llama.cpp/bin/llama-server / +## Base image +FROM ubuntu:$UBUNTU_VERSION AS base -WORKDIR /models -EXPOSE 8080 +RUN apt-get update \ + && apt-get install -y libgomp1 curl\ + && apt autoremove -y \ + && apt clean -y \ + && rm -rf /tmp/* /var/tmp/* \ + && find /var/cache/apt/archives /var/lib/apt/lists -not -name lock -type f -delete \ + && find /var/cache -type f -delete -ENTRYPOINT ["/llama-server"] +COPY --from=build /app/lib/ /app + +### Full +FROM base AS full + +COPY --from=build /app/full /app + +WORKDIR /app + +RUN apt-get update \ + && apt-get install -y \ + git \ + python3 \ + python3-pip \ + && pip install --upgrade pip setuptools wheel \ + && pip install -r requirements.txt \ + && apt autoremove -y \ + && apt clean -y \ + && rm -rf /tmp/* /var/tmp/* \ + && find /var/cache/apt/archives /var/lib/apt/lists -not -name lock -type f -delete \ + && find /var/cache -type f -delete + +ENTRYPOINT ["/app/tools.sh"] + +### Light, CLI only +FROM base AS light + +COPY --from=build /app/full/llama-cli /app + +WORKDIR /app + +ENTRYPOINT [ "/app/llama-cli" ] + +### Server, Server only +FROM base AS server + +ENV LLAMA_ARG_HOST=0.0.0.0 + +COPY --from=build /app/full/llama-server /app + +WORKDIR /app + +HEALTHCHECK CMD [ "curl", "-f", "http://localhost:8080/health" ] + +ENTRYPOINT [ "/app/llama-server" ]