Setup trace dump with jit

lambdaclass · Jan 30, 2025 · 1432029 · 1432029
1 parent 11b1922
commit 1432029
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/bin/cairo-native-run.rs b/src/bin/cairo-native-run.rs
@@ -6,7 +6,10 @@ use cairo_lang_runner::short_string::as_cairo_short_string;
 use cairo_native::{
     context::NativeContext,
     executor::{AotNativeExecutor, JitNativeExecutor},
-    metadata::gas::{GasMetadata, MetadataComputationConfig},
+    metadata::{
+        gas::{GasMetadata, MetadataComputationConfig},
+        trace_dump,
+    },
     starknet_stub::StubSyscallHandler,
 };
 use clap::{Parser, ValueEnum};
@@ -96,6 +99,16 @@ fn main() -> anyhow::Result<()> {
         RunMode::Jit => {
             let executor =
                 JitNativeExecutor::from_native_module(native_module, args.opt_level.into())?;
+
+            #[cfg(feature = "with-trace-dump")]
+            {
+                use cairo_native::metadata::trace_dump::TraceBinding;
+                if let Some(trace_id) = executor.find_symbol_ptr(TraceBinding::TraceId.symbol()) {
+                    let trace_id = trace_id.cast::<u64>();
+                    unsafe { *trace_id = 0 };
+                }
+            }
+
             Box::new(move |function_id, args, gas, syscall_handler| {
                 executor.invoke_dynamic_with_syscall_handler(
                     function_id,

diff --git a/src/executor/jit.rs b/src/executor/jit.rs
@@ -65,6 +65,9 @@ impl<'m> JitNativeExecutor<'m> {
 
         setup_runtime(|name| executor.find_symbol_ptr(name));
 
+        #[cfg(feature = "with-trace-dump")]
+        crate::metadata::trace_dump::setup_runtime(|name| executor.find_symbol_ptr(name));
+
         Ok(executor)
     }