Avoid copying FnBounds

Author: manh9203

crates/toolchain/instructions/src/exe.rs

@@ -1,4 +1,4 @@
-use std::collections::BTreeMap;
+use std::{collections::BTreeMap, sync::Arc};
 
 use openvm_stark_backend::p3_field::Field;
 use serde::{Deserialize, Serialize};
@@ -24,7 +24,7 @@ pub struct VmExe<F> {
     /// Initial memory image.
     pub init_memory: MemoryImage<F>,
     /// Starting + ending bounds for each function.
-    pub fn_bounds: FnBounds,
+    pub fn_bounds: Arc<FnBounds>,
 }
 
 impl<F> VmExe<F> {
@@ -33,7 +33,7 @@ impl<F> VmExe<F> {
             program,
             pc_start: 0,
             init_memory: BTreeMap::new(),
-            fn_bounds: Default::default(),
+            fn_bounds: Arc::new(Default::default()),
         }
     }
     pub fn with_pc_start(mut self, pc_start: u32) -> Self {

crates/toolchain/transpiler/src/lib.rs

@@ -1,5 +1,7 @@
 //! A transpiler from custom RISC-V ELFs to OpenVM executable binaries.
 
+use std::sync::Arc;
+
 use elf::Elf;
 use openvm_instructions::{
     exe::VmExe,
@@ -41,7 +43,7 @@ impl<F: PrimeField32> FromElf for VmExe<F> {
             program,
             pc_start: elf.pc_start,
             init_memory,
-            fn_bounds: elf.fn_bounds,
+            fn_bounds: Arc::new(elf.fn_bounds),
         })
     }
 }

crates/vm/src/arch/segment.rs

@@ -170,7 +170,7 @@ impl<F: PrimeField32, VC: VmConfig<F>> ExecutionSegment<F, VC> {
         init_streams: Streams<F>,
         initial_memory: Option<MemoryImage<F>>,
         trace_height_constraints: Vec<LinearConstraint>,
-        #[allow(unused_variables)] fn_bounds: FnBounds,
+        #[allow(unused_variables)] fn_bounds: Arc<FnBounds>,
     ) -> Self {
         let mut chip_complex = config.create_chip_complex().unwrap();
         chip_complex.set_streams(init_streams);

crates/vm/src/metrics/mod.rs

@@ -1,4 +1,4 @@
-use std::{collections::BTreeMap, mem};
+use std::{collections::BTreeMap, mem, sync::Arc};
 
 use cycle_tracker::CycleTracker;
 use metrics::counter;
@@ -23,7 +23,7 @@ pub struct VmMetrics {
     /// Metric collection tools. Only collected when `config.profiling` is true.
     pub cycle_tracker: CycleTracker,
     #[allow(dead_code)]
-    pub(crate) fn_bounds: FnBounds,
+    pub(crate) fn_bounds: Arc<FnBounds>,
     /// Cycle span by function if function start/end addresses are available
     #[allow(dead_code)]
     pub(crate) current_fn: FnBound,

crates/vm/tests/integration_test.rs

@@ -325,7 +325,7 @@ fn test_vm_initial_memory() {
         program,
         pc_start: 0,
         init_memory,
-        fn_bounds: Default::default(),
+        fn_bounds: Arc::new(Default::default()),
     };
     air_test(config, exe);
 }