env var

sdk/src/provers/mod.rs

@@ -7,10 +7,12 @@ use anyhow::Result;
 pub use local::LocalProver;
 pub use mock::MockProver;
 pub use network::NetworkProver;
+use sp1_core::runtime::ExecutionReport;
 use sp1_core::stark::MachineVerificationError;
 use sp1_prover::CoreSC;
 use sp1_prover::SP1CoreProofData;
 use sp1_prover::SP1Prover;
+use sp1_prover::SP1PublicValues;
 use sp1_prover::SP1ReduceProof;
 use sp1_prover::{SP1ProvingKey, SP1Stdin, SP1VerifyingKey};
 use strum_macros::EnumString;
@@ -31,6 +33,11 @@ pub trait Prover: Send + Sync {
 
     fn setup(&self, elf: &[u8]) -> (SP1ProvingKey, SP1VerifyingKey);
 
+    /// Execute the given program with the given input.
+    fn execute(&self, elf: &[u8], stdin: SP1Stdin) -> Result<(SP1PublicValues, ExecutionReport)> {
+        self.execute(elf, stdin)
+    }
+
     /// Prove the execution of a RISCV ELF with the given inputs.
     fn prove(&self, pk: &SP1ProvingKey, stdin: SP1Stdin) -> Result<SP1Proof>;
 

sdk/src/provers/network.rs

@@ -41,6 +41,20 @@ impl NetworkProver {
     ) -> Result<P> {
         let client = &self.client;
 
+        let skip_simulation = env::var("SKIP_SIMULATION")
+            .map(|val| val == "true")
+            .unwrap_or(false);
+
+        if !skip_simulation {
+            let (_, report) = self.local_prover.execute(elf, stdin.clone())?;
+            log::info!(
+                "Simulation complete, cycles: {}",
+                report.total_instruction_count()
+            );
+        } else {
+            log::info!("Skipping simulation");
+        }
+
         let proof_id = client.create_proof(elf, &stdin, mode).await?;
         log::info!("Created {}", proof_id);