print cairo_runner and prover_input size.

Author: ormarx-starkware

.github/workflows/upload_artifacts_workflow.yml

@@ -3,7 +3,7 @@ name: Upload-Artifacts
 on:
   push:
     branches:
-      - main
+      - or/print_cairo_runner_prover_input_size
 
 jobs:
   artifacts-push:

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

@@ -294,6 +294,7 @@ fn relocate_cairo_pie_memory(
 
         vm.insert_value(relocated_address, relocated_value)?;
     }
+    tracing::info!("cairo pie memory relocated memory: {} bytes", std::mem::size_of_val(&vm.segments.memory));
     Ok(())
 }
 

crates/vm_runner/src/main.rs

@@ -62,6 +62,16 @@ enum Error {
     Anyhow(#[from] anyhow::Error),
 }
 
+/// Calculate approximate heap memory usage for a Vec
+fn vec_heap_size<T>(vec: &Vec<T>) -> usize {
+    vec.capacity() * std::mem::size_of::<T>()
+}
+
+/// Calculate approximate heap memory usage for an Option<Vec<T>>
+fn option_vec_heap_size<T>(opt_vec: &Option<Vec<T>>) -> usize {
+    opt_vec.as_ref().map(|v| vec_heap_size(v)).unwrap_or(0)
+}
+
 fn main() -> ExitCode {
     run_binary(run, "stwo_vm_runner")
 }
@@ -91,8 +101,33 @@ fn run() -> Result<ProverInput, Error> {
     };
 
     let cairo_runner = cairo_run_program(&program, program_input_contents, cairo_run_config)?;
+
+    // Log stack sizes (struct sizes).
+    log::info!("=== Stack Sizes (struct metadata only) ===");
+    log::info!("cairo_runner stack: {} bytes", std::mem::size_of_val(&cairo_runner));
+    log::info!("cairo_runner.vm stack: {} bytes", std::mem::size_of_val(&cairo_runner.vm));
+    log::info!("cairo_runner.get_program() stack: {} bytes", std::mem::size_of_val(cairo_runner.get_program()));
+    log::info!("cairo_runner.exec_scopes stack: {} bytes", std::mem::size_of_val(&cairo_runner.exec_scopes));
+    log::info!("cairo_runner.relocated_trace stack: {} bytes", std::mem::size_of_val(&cairo_runner.relocated_trace));
+
+    // Log heap sizes (actual allocated data).
+    log::info!("=== Heap Sizes (actual memory consumption) ===");
+    log::info!("cairo_runner.relocated_trace heap: {} bytes ({} MB)",
+        option_vec_heap_size(&cairo_runner.relocated_trace),
+        option_vec_heap_size(&cairo_runner.relocated_trace) / 1_048_576);
+
+    // VM segments memory is complex, log the HashMap size.
+    log::info!("cairo_runner.vm.segments.public_memory_offsets entries: {}",
+        cairo_runner.vm.segments.public_memory_offsets.len());
+
     let prover_input = adapt(&cairo_runner)?;
 
+    log::info!("=== After Adaptation ===");
+    log::info!("prover_input stack: {} bytes", std::mem::size_of_val(&prover_input));
+    log::info!("prover_input.public_memory_addresses heap: {} bytes ({} MB)",
+        vec_heap_size(&prover_input.public_memory_addresses),
+        vec_heap_size(&prover_input.public_memory_addresses) / 1_048_576);
+
     if let Some(prover_input_path) = args.output_prover_input_path {
         std::fs::write(prover_input_path, serde_json::to_string(&prover_input)?)?;
     }