diff options
| author | Kamen Mladenov <kamen@syndamia.com> | 2025-02-27 11:35:32 +0200 |
|---|---|---|
| committer | Kamen Mladenov <kamen@syndamia.com> | 2025-02-27 15:08:11 +0200 |
| commit | 5b82ac769339938570c0d1f7230afc53732993b4 (patch) | |
| tree | b510f409458097a3c473ab80791a7717961ebd5b /zkvms/zkm/host | |
| parent | 39ec264c0d90555237debff519cc890b755a0646 (diff) | |
| download | zkVMs-benchmarks-5b82ac769339938570c0d1f7230afc53732993b4.tar zkVMs-benchmarks-5b82ac769339938570c0d1f7230afc53732993b4.tar.gz zkVMs-benchmarks-5b82ac769339938570c0d1f7230afc53732993b4.zip | |
chore: Rust fmt
Diffstat (limited to 'zkvms/zkm/host')
| -rw-r--r-- | zkvms/zkm/host/src/main.rs | 56 |
1 files changed, 19 insertions, 37 deletions
diff --git a/zkvms/zkm/host/src/main.rs b/zkvms/zkm/host/src/main.rs index f71dcc5..db1a5cb 100644 --- a/zkvms/zkm/host/src/main.rs +++ b/zkvms/zkm/host/src/main.rs @@ -1,8 +1,15 @@ use anyhow::{bail, Result}; use std::{env, fs::read, time::Instant}; -use zkm_sdk::{prover::ClientCfg, prover::{ProverInput, ProverResult} , ProverClient}; +use zkm_sdk::{ + prover::ClientCfg, + prover::{ProverInput, ProverResult}, + ProverClient, +}; -use zkvms_host_io::{read_args, benchmarkable, RunType::{ Execute, Prove, Verify }}; +use zkvms_host_io::{ + benchmarkable, read_args, + RunType::{Execute, Prove, Verify}, +}; async fn setup( prover_client: &mut ProverClient, @@ -26,16 +33,12 @@ async fn get_proof( if let Ok(Some(prover_result)) = proving_result { prover_result - } - else { + } else { panic!("Failed to generate proof!"); } } -async fn execute( - prover_client: &mut ProverClient, - prover_input: &mut ProverInput, -) { +async fn execute(prover_client: &mut ProverClient, prover_input: &mut ProverInput) { let prover_result = get_proof(prover_client, prover_input).await; prover_client @@ -52,12 +55,7 @@ async fn prove( let prover_result = get_proof(prover_client, prover_input).await; prover_client - .process_proof_results( - &prover_result, - &prover_input, - &proof_results_path, - "local", - ) + .process_proof_results(&prover_result, &prover_input, &proof_results_path, "local") .expect("process proof results error"); } @@ -68,30 +66,14 @@ async fn main() -> Result<()> { panic!("Off-chain verification is not supported!"); } - let seg_size: u32 = run_info - .env_then_or( - "SEG_SIZE", - |seg| seg.parse::<u32>().ok(), - 65536, - ); + let seg_size: u32 = run_info.env_then_or("SEG_SIZE", |seg| seg.parse::<u32>().ok(), 65536); let elf_path = env::var("ELF_PATH").expect("ELF PATH is missing"); - let proof_results_path = run_info - .env_or( - "PROOF_RESULTS_PATH", - "/tmp/contracts", - ); - let vk_path = run_info - .env_or( - "VERIFYING_KEY_PATH", - "/tmp/input", - ); - - let mut client_config = ClientCfg::new( - "local".to_string(), - vk_path.to_owned(), - ); + let proof_results_path = run_info.env_or("PROOF_RESULTS_PATH", "/tmp/contracts"); + let vk_path = run_info.env_or("VERIFYING_KEY_PATH", "/tmp/input"); + + let mut client_config = ClientCfg::new("local".to_string(), vk_path.to_owned()); let mut prover_client = ProverClient::new(&client_config).await; @@ -118,11 +100,11 @@ async fn main() -> Result<()> { match run_info.run_type { // only excute the guest program without generating the proof. - Execute => benchmarkable!{ + Execute => benchmarkable! { execute(&mut prover_client, &mut prover_input).await; }, // excute the guest program and generate the proof - Prove => benchmarkable!{ + Prove => benchmarkable! { prove(&mut prover_client, &mut prover_input, &vk_path, &proof_results_path).await; }, Verify => unreachable!(), |
