sweep_jolt.py

import subprocess
from itertools import product
import os

filename = "benchmark"

# First we loop through simple programs for SP1 vs. JOLT since JOLT only supports simplistic programs.
trials = 1
options_program = ['sha2-chain', 'loop', 'fibonacci']
options_prover = ["jolt-zkvm", "sp1"]
options_hashfn = ['poseidon']
options_shard_size = [21]

os.environ["NO_GROTH16"] = "true"
os.environ["RECONSTRUCT_COMMITMENTS"] = "false"
# Generate the Cartesian product of the options
option_combinations = product(options_program, options_prover, options_hashfn, options_shard_size)

# Run the shell script with each combination of options
for program, prover, hashfn, shard_size in option_combinations:
    first_shard_size = options_shard_size[0]
    if prover != 'sp1' and shard_size != first_shard_size: # Only sp1 supports different shard sizes
        continue
    print(program, prover, hashfn, shard_size)
    for i in range(trials):
        subprocess.run(['bash', 'eval.sh', program, prover, hashfn, str(shard_size), filename])