Skip to content

Commit b608728

Browse files
committed
move gpu override, update scripts
1 parent 0382099 commit b608728

File tree

2 files changed

+21
-10
lines changed

2 files changed

+21
-10
lines changed

zkvm-prover/Makefile

Lines changed: 21 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ endif
2121
ZKVM_COMMIT=$(shell echo ${ZKVM_VERSION} | cut -d " " -f2)
2222
$(info ZKVM_COMMIT is ${ZKVM_COMMIT})
2323

24-
PLONKY3_GPU_VERSION=$(shell ./print_plonky3gpu_version.sh | sed -n '2p')
24+
#PLONKY3_GPU_VERSION=$(shell ./print_plonky3gpu_version.sh | sed -n '2p')
2525

2626
GIT_REV=$(shell git rev-parse --short HEAD)
2727
GO_TAG=$(shell grep "var tag = " ../common/version/version.go | cut -d "\"" -f2)
@@ -32,16 +32,27 @@ else
3232
$(info GO_TAG is ${GO_TAG})
3333
endif
3434

35-
ifeq (${PLONKY3_GPU_VERSION},)
36-
# use plonky3 with CPU
37-
ZK_VERSION=${ZKVM_COMMIT}-${PLONKY3_VERSION}
38-
else
39-
# use halo2_gpu
40-
ZK_VERSION=${ZKVM_COMMIT}-${PLONKY3_GPU_VERSION}
41-
endif
35+
ZK_VERSION=${ZKVM_COMMIT}-${PLONKY3_VERSION}
36+
#ifeq (${PLONKY3_GPU_VERSION},)
37+
# # use plonky3 with CPU
38+
# ZK_VERSION=${ZKVM_COMMIT}-${PLONKY3_VERSION}
39+
#else
40+
# # use gpu
41+
# ZK_VERSION=${ZKVM_COMMIT}-${PLONKY3_GPU_VERSION}
42+
#endif
43+
44+
prover_gpu:
45+
cd ../crates/gpu_override && cargo tree >/dev/null
46+
$(eval PLONKY3_GPU_VERSION:=$(shell ./print_plonky3gpu_version.sh | sed -n '2p'))
47+
$(eval ZK_VERSION:=${ZKVM_COMMIT}-${PLONKY3_GPU_VERSION})
48+
@echo "Updated ZK_VERSION to ${ZK_VERSION} after prover_gpu"
49+
50+
prover: prover_gpu
51+
GO_TAG=${GO_TAG} GIT_REV=${GIT_REV} ZK_VERSION=${ZK_VERSION} cd ../crates/gpu_override && cargo build --release -p prover
52+
53+
prover_cpu:
54+
GO_TAG=${GO_TAG} GIT_REV=${GIT_REV} ZK_VERSION=${ZK_VERSION} cargo build --release -p prover
4255

43-
prover:
44-
GO_TAG=${GO_TAG} GIT_REV=${GIT_REV} ZK_VERSION=${ZK_VERSION} cd ../crates/prover-bin && cargo build --release
4556

4657
tests_binary:
4758
cargo clean && cargo test --release --no-run

0 commit comments

Comments
 (0)