* add openmp lib to dockerfiles * build only main and server in their docker images
@@ -23,7 +23,7 @@ ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
# Enable CUDA
ENV LLAMA_CUDA=1
-RUN make -j$(nproc)
+RUN make -j$(nproc) main
FROM ${BASE_CUDA_RUN_CONTAINER} as runtime
@@ -40,6 +40,6 @@ ENV LLAMA_HIPBLAS=1
ENV CC=/opt/rocm/llvm/bin/clang
ENV CXX=/opt/rocm/llvm/bin/clang++
ENTRYPOINT [ "/app/main" ]
@@ -9,7 +9,7 @@ WORKDIR /app
COPY . .
FROM ubuntu:$UBUNTU_VERSION as runtime
@@ -25,7 +25,7 @@ ENV LLAMA_CUDA=1
# Enable cURL
ENV LLAMA_CURL=1
+RUN make -j$(nproc) server
@@ -11,7 +11,7 @@ COPY . .