diff options
Diffstat (limited to 'zkvms/zkwasm')
| -rw-r--r-- | zkvms/zkwasm/host/src/main.rs | 9 | ||||
| -rw-r--r-- | zkvms/zkwasm/wrapper_macro/src/lib.rs | 17 |
2 files changed, 19 insertions, 7 deletions
diff --git a/zkvms/zkwasm/host/src/main.rs b/zkvms/zkwasm/host/src/main.rs index 6a23504..ea8dcf5 100644 --- a/zkvms/zkwasm/host/src/main.rs +++ b/zkvms/zkwasm/host/src/main.rs @@ -1,11 +1,10 @@ use std::io::{self, Write}; use std::process::{Command, Stdio}; use zkvms_host_io::{ - benchmarkable, foreach_private_input_field, foreach_public_input_field, read_args, - PrivateInput, PublicInput, + benchmarkable, foreach_private_input_field, foreach_public_input_field, output_proof_size_raw, + read_args, PrivateInput, PublicInput, RunType::{Execute, Prove, Verify}, RunWith, - output_proof_size_raw, }; static PUBLIC_INPUT_PATH: &str = "public_input.bin"; @@ -108,7 +107,9 @@ fn main() { .arg("--output") .arg(output.clone())); - let proofSize = std::fs::metadata(output.clone() + "/prog.0.transcript.data").unwrap().len(); + let proofSize = std::fs::metadata(output.clone() + "/prog.0.transcript.data") + .unwrap() + .len(); output_proof_size_raw(proofSize as usize); benchmarkable! { diff --git a/zkvms/zkwasm/wrapper_macro/src/lib.rs b/zkvms/zkwasm/wrapper_macro/src/lib.rs index 68a78ae..690cf5e 100644 --- a/zkvms/zkwasm/wrapper_macro/src/lib.rs +++ b/zkvms/zkwasm/wrapper_macro/src/lib.rs @@ -61,8 +61,18 @@ pub fn make_wrapper(item: TokenStream) -> TokenStream { let mut out = TokenStream::new(); - insert_reads(&mut out, fd.public_patterns(), fd.public_types(), "read_public"); - insert_reads(&mut out, fd.private_patterns(), fd.private_types(), "read_private"); + insert_reads( + &mut out, + fd.public_patterns(), + fd.public_types(), + "read_public", + ); + insert_reads( + &mut out, + fd.private_patterns(), + fd.private_types(), + "read_private", + ); out.extend( format!( @@ -73,7 +83,8 @@ pub fn make_wrapper(item: TokenStream) -> TokenStream { write(val); }} ", - fd.name, fd.grouped_patterns() + fd.name, + fd.grouped_patterns() ) .parse::<TokenStream>(), ); |