record instret during prelight execution

Author: kunxian-xia

crates/circuits/mod-builder/src/core_chip.rs

@@ -425,6 +425,7 @@ where
             &mut adapter_record,
         );
 
+        *state.instret += 1;
         *state.pc = state.pc.wrapping_add(DEFAULT_PC_STEP);
         Ok(())
     }

crates/vm/src/arch/execution.rs

@@ -216,6 +216,7 @@ pub trait PreflightExecutor<F, RA = MatrixRecordArena<F>> {
 #[derive(derive_new::new)]
 pub struct VmStateMut<'a, F, MEM, RA> {
     pub pc: &'a mut u32,
+    pub instret: &'a mut u64,
     pub memory: &'a mut MEM,
     pub streams: &'a mut Streams<F>,
     pub rng: &'a mut StdRng,

crates/vm/src/arch/state.rs

@@ -67,6 +67,7 @@ impl<F: Clone, MEM> VmState<F, MEM> {
     pub fn into_mut<'a, RA>(&'a mut self, ctx: &'a mut RA) -> VmStateMut<'a, F, MEM, RA> {
         VmStateMut {
             pc: &mut self.pc,
+            instret: &mut self.instret,
             memory: &mut self.memory,
             streams: &mut self.streams,
             rng: &mut self.rng,

crates/vm/src/arch/testing/cpu.rs

@@ -98,8 +98,10 @@ where
         tracing::debug!("initial_timestamp={}", self.memory.memory.timestamp());
 
         let mut pc = initial_pc;
+        let mut instret = 0;
         let state_mut = VmStateMut {
             pc: &mut pc,
+            instret: &mut instret,
             memory: &mut self.memory.memory,
             streams: &mut self.streams,
             rng: &mut self.rng,

crates/vm/src/arch/testing/cuda.rs

@@ -132,8 +132,10 @@ impl TestBuilder<F> for GpuChipTestBuilder {
         tracing::debug!("initial_timestamp={}", initial_state.timestamp);
 
         let mut pc = initial_pc;
+        let mut instret = 0;
         let state_mut = VmStateMut::new(
             &mut pc,
+            &mut instret,
             &mut self.memory.memory,
             &mut self.streams,
             &mut self.rng,

crates/vm/src/system/phantom/mod.rs

@@ -157,14 +157,18 @@ where
                 SysPhantom::CtStart => {
                     let metrics = state.metrics;
                     if let Some(info) = metrics.debug_infos.get(pc) {
-                        metrics.cycle_tracker.start(info.dsl_instruction.clone());
+                        metrics
+                            .cycle_tracker
+                            .start(info.dsl_instruction.clone(), *state.instret as usize);
                     }
                 }
                 #[cfg(feature = "perf-metrics")]
                 SysPhantom::CtEnd => {
                     let metrics = state.metrics;
                     if let Some(info) = metrics.debug_infos.get(pc) {
-                        metrics.cycle_tracker.end(info.dsl_instruction.clone());
+                        metrics
+                            .cycle_tracker
+                            .end(info.dsl_instruction.clone(), *state.instret as usize);
                     }
                 }
                 _ => {}
@@ -187,6 +191,7 @@ where
                     inner: err,
                 })?;
         }
+        *state.instret += 1;
         *state.pc += DEFAULT_PC_STEP;
         state.memory.increment_timestamp();
 

crates/vm/src/system/public_values/core.rs

@@ -202,6 +202,7 @@ where
             }
         }
 
+        *state.instret += 1;
         *state.pc = state.pc.wrapping_add(DEFAULT_PC_STEP);
 
         Ok(())

extensions/keccak256/circuit/src/trace.rs

@@ -235,6 +235,7 @@ where
         }
 
         // Due to the AIR constraints, the final memory timestamp should be the following:
+        *state.instret += 1;
         state.memory.timestamp = record.inner.timestamp
             + (len + KECCAK_REGISTER_READS + KECCAK_ABSORB_READS + KECCAK_DIGEST_WRITES) as u32;
         *state.pc = state.pc.wrapping_add(DEFAULT_PC_STEP);

extensions/native/circuit/src/branch_eq/core.rs

@@ -76,6 +76,8 @@ where
             *state.pc = state.pc.wrapping_add(self.pc_step);
         }
 
+        *state.instret += 1;
+
         Ok(())
     }
 }

extensions/native/circuit/src/castf/core.rs

@@ -157,6 +157,7 @@ where
         self.adapter
             .write(state.memory, instruction, x, &mut adapter_record);
 
+        *state.instret += 1;
         *state.pc = state.pc.wrapping_add(DEFAULT_PC_STEP);
 
         Ok(())