Added more span logs to stwo_run_and_prove

Author: yuvalsw

crates/cairo-program-runner-lib/Cargo.toml

@@ -12,15 +12,16 @@ cairo-lang-runner.workspace = true
 cairo-lang-casm.workspace = true
 cairo-lang-execute-utils.workspace = true
 clap.workspace = true
+num-bigint.workspace = true
 num-traits.workspace = true
+regex.workspace = true
 serde.workspace = true
 serde_json.workspace = true
 starknet-crypto.workspace = true
 starknet-types-core.workspace = true
 thiserror.workspace = true
 thiserror-no-std.workspace = true
-regex.workspace = true
-num-bigint.workspace = true
+tracing.workspace = true
 
 [dev-dependencies]
 assert_matches = "1.5.0"

crates/cairo-program-runner-lib/src/lib.rs

@@ -4,6 +4,7 @@ use cairo_vm::types::program::Program;
 use cairo_vm::vm::errors::cairo_run_errors::CairoRunError;
 use cairo_vm::vm::runners::cairo_runner::CairoRunner;
 pub use hints::*;
+use tracing::{span, Level};
 
 pub mod hints;
 pub mod tasks;
@@ -37,6 +38,8 @@ pub fn cairo_run_program(
     program_input_contents: Option<String>,
     cairo_run_config: CairoRunConfig,
 ) -> Result<CairoRunner, CairoRunError> {
+    let _span = span!(Level::INFO, "cairo_run_program").entered();
+
     let mut hint_processor = BootloaderHintProcessor::new();
 
     let mut exec_scopes = ExecutionScopes::new();

crates/stwo_run_and_prove/src/main.rs

@@ -2,13 +2,13 @@ use anyhow::Result;
 use cairo_air::utils::ProofFormat;
 use cairo_program_runner_lib::cairo_run_program;
 use cairo_program_runner_lib::utils::{get_cairo_run_config, get_program, get_program_input};
-use cairo_vm::Felt252;
 use cairo_vm::types::errors::program_errors::ProgramError;
 use cairo_vm::types::layout_name::LayoutName;
 use cairo_vm::vm::errors::cairo_run_errors::CairoRunError;
 use cairo_vm::vm::errors::runner_errors::RunnerError;
 use cairo_vm::vm::errors::vm_errors::VirtualMachineError;
 use cairo_vm::vm::runners::cairo_runner::CairoRunner;
+use cairo_vm::Felt252;
 use clap::Parser;
 #[cfg(test)]
 use mockall::automock;
@@ -17,12 +17,12 @@ use std::env;
 use std::fs;
 use std::path::PathBuf;
 use std::process::ExitCode;
-use stwo_cairo_adapter::ProverInput;
 use stwo_cairo_adapter::adapter::adapt;
+use stwo_cairo_adapter::ProverInput;
 use stwo_cairo_prover::prover::create_and_serialize_proof;
 use stwo_cairo_utils::binary_utils::run_binary;
 use thiserror::Error;
-use tracing::{Level, error, info, span};
+use tracing::{error, info, span, Level};
 
 static PROVER_INPUT_FILE_NAME: &str = "prover_input.json";
 
@@ -147,6 +147,7 @@ fn stwo_run_and_prove(
     debug_data_dir: Option<PathBuf>,
     save_debug_data: bool,
 ) -> Result<(), StwoRunAndProveError> {
+    let _span = span!(Level::INFO, "stwo_run_and_prove").entered();
     let cairo_run_config = get_cairo_run_config(
         // we don't use dynamic layout in stwo
         &None,
@@ -476,7 +477,7 @@ mod tests {
         assert!(
             file_exists(&debug_data_tempdir.path().join(PROVER_INPUT_FILE_NAME)),
             "Prover input file was not created in the debug data directory, or was created with an
-            incorrect name, after running with a proving failure. NOTE: Changing the file name may 
+            incorrect name, after running with a proving failure. NOTE: Changing the file name may
             break external dependencies.",
         );