update dockerfile

tlsnotary · Oct 28, 2024 · 295fd8f · 295fd8f
1 parent b4e9c52
commit 295fd8f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/crates/benches/binary/benches.Dockerfile b/crates/benches/binary/benches.Dockerfile
@@ -36,11 +36,20 @@ RUN \
 
 RUN apt clean && rm -rf /var/lib/apt/lists/*
 
-COPY --from=builder ["/usr/src/tlsn/target/release/bench", "/usr/src/tlsn/target/release/prover", "/usr/src/tlsn/target/release/verifier", "/usr/src/tlsn/target/release/plot", "/usr/local/bin/"]
+COPY --from=builder \
+  ["/usr/src/tlsn/target/release/bench", \
+  "/usr/src/tlsn/target/release/prover", \
+  "/usr/src/tlsn/target/release/prover-memory", \
+  "/usr/src/tlsn/target/release/verifier", \
+  "/usr/src/tlsn/target/release/verifier-memory", \
+  "/usr/src/tlsn/target/release/plot", \
+  "/usr/local/bin/"]
 
 ENV PROVER_PATH="/usr/local/bin/prover"
 ENV VERIFIER_PATH="/usr/local/bin/verifier"
+ENV PROVER_MEMORY_PATH="/usr/local/bin/prover-memory"
+ENV VERIFIER_MEMORY_PATH="/usr/local/bin/verifier-memory"
 
 VOLUME [ "/benches" ]
 WORKDIR "/benches"
-CMD ["/bin/bash", "-c", "bench && plot /benches/metrics.csv && cat /benches/metrics.csv"]
+CMD ["/bin/bash", "-c", "bench && bench --memory-profiling && plot /benches/metrics.csv && cat /benches/metrics.csv"]