github/ere
1
1
Fork 0
mirror of https://github.com/eth-act/ere.git synced 2026-02-19 11:54:42 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: update dockerfiles to use the env var that the zkVM recognize, and allow multi compute caps

Browse Source
This commit is contained in:
han0110
2026-02-10 12:29:41 +00:00
parent 8357931eda
commit f28fa268c9
4 changed files with 7 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docker/airbender/Dockerfile.base
View File

@@ -8,11 +8,8 @@ RUN rustup default nightly-2025-07-25
# Whether to enable CUDA feature or not.
ARG CUDA
# Default to build for RTX 50 series
ARG CUDA_ARCH=sm_120
# Env read by Airbender crate `gpu_prover`, only taking the numeric part
ARG CUDAARCHS=${CUDA_ARCH#sm_}
# Env read by Airbender crate `gpu_prover`, numeric cuda arch IDs (e.g. "120" or "89;120")
ARG CUDAARCHS=120
# Copy the Airbender SDK installer script from the workspace context
COPY --chmod=755 scripts/sdk_installers/install_airbender_sdk.sh /tmp/install_airbender_sdk.sh

9
docker/openvm/Dockerfile.base
View File

@@ -8,15 +8,6 @@ FROM $BASE_IMAGE
# Set default toolchain to nightly
RUN rustup default nightly
# Whether to enable CUDA feature or not.
ARG CUDA
# Default to build for RTX 50 series
ARG CUDA_ARCH=sm_120
# Env variable read by OpenVM crate `cuda-builder`, need to persist it for building `ere-openvm`.
ENV CUDA_ARCH=${CUDA_ARCH#sm_}
# Copy the OpenVM SDK installer script from the workspace context
COPY --chmod=755 scripts/sdk_installers/install_openvm_sdk.sh /tmp/install_openvm_sdk.sh

3
docker/openvm/Dockerfile.server
View File

@@ -14,6 +14,9 @@ WORKDIR /ere
ARG CUDA
ARG RUSTFLAGS
# Env variable read by OpenVM crate `openvm-cuda-builder`, comma-separated numeric arch IDs (e.g. "120" or "89,120")
ARG CUDA_ARCH=120
RUN cargo build --release --package ere-server --bin ere-server --features openvm${CUDA:+,cuda} \
&& mkdir bin && mv target/release/ere-server bin/ere-server \
&& cargo clean && rm -rf $CARGO_HOME/registry/

6
docker/risc0/Dockerfile.base
View File

@@ -13,10 +13,8 @@ RUN [ -n "$CUDA" ] && \
(curl -o protoc.zip -L https://github.com/protocolbuffers/protobuf/releases/download/v31.1/protoc-31.1-linux-x86_64.zip \
&& unzip protoc.zip -d /usr/local) || true
# Default to build for RTX 50 series
ARG CUDA_ARCH=sm_120
ARG NVCC_APPEND_FLAGS="--generate-code arch=compute_${CUDA_ARCH#sm_},code=sm_${CUDA_ARCH#sm_}"
ENV NVCC_APPEND_FLAGS=${NVCC_APPEND_FLAGS}
# Fully formed NVCC flags for CUDA arch targeting (e.g. "--generate-code arch=compute_120,code=sm_120")
ARG NVCC_APPEND_FLAGS="--generate-code arch=compute_120,code=sm_120"
# Copy and run the Risc0 SDK installer script
COPY --chmod=755 scripts/sdk_installers/install_risc0_sdk.sh /tmp/install_risc0_sdk.sh