Merge pull request 0xPolygonZero#59 from 0xPolygonZero/fast_witness

Use faster witness-only testing method

Cargo.toml

@@ -17,17 +17,19 @@ categories = ["cryptography"]
 anyhow = { version = "1.0.71", features = ["backtrace"] }
 clap = {version = "4.2.7", features = ["derive"] }
 ethereum-types = "0.14.1"
-mpt_trie = "0.1.0"
 flexi_logger = { version = "0.25.4", features = ["async"] }
 futures = "0.3.28"
 keccak-hash = "0.10.0"
 log = "0.4.17"
-plonky2 = "0.2.0"
-evm_arithmetization = "0.1.0"
 serde = "1.0.163"
 serde_cbor = "0.11.2"
 tokio = { version = "1.28.1" }
 
+# zk-evm dependencies
+plonky2 = "0.2.0"
+mpt_trie = "0.1.1"
+evm_arithmetization = "0.1.1"
+
 [profile.release]
 opt-level = 3
 lto = "fat"

evm_test_runner/src/plonky2_runner.rs

@@ -9,7 +9,9 @@ use std::{
 use common::types::TestVariantRunInfo;
 use ethereum_types::U256;
 use evm_arithmetization::{
-    generation::generate_traces, prover::prove, verifier::verify_proof, AllStark, StarkConfig,
+    prover::{prove, testing::simulate_execution},
+    verifier::verify_proof,
+    AllStark, StarkConfig,
 };
 use futures::executor::block_on;
 use indicatif::{ProgressBar, ProgressStyle};
@@ -260,12 +262,7 @@ fn run_test_and_get_test_result(test: TestVariantRunInfo, witness_only: bool) ->
 
     match witness_only {
         true => {
-            let res = generate_traces::<GoldilocksField, 2>(
-                &AllStark::default(),
-                test.gen_inputs,
-                &StarkConfig::standard_fast_config(),
-                &mut TimingTree::default(),
-            );
+            let res = simulate_execution::<GoldilocksField>(test.gen_inputs);
 
             if let Err(evm_err) = res {
                 return handle_evm_err(evm_err, false, "witness generation");

evm_test_runner/src/report_generation.rs

@@ -153,7 +153,7 @@ impl PassedInfo {
     }
 }
 
-///
+/// Print the test report to the terminal.
 pub(crate) fn output_test_report_for_terminal(
     res: &[TestGroupRunResults],
     test_filter_str: Option<String>,