Import alloc_scratch

o1-labs · Dec 4, 2023 · 228b1b6 · 228b1b6
1 parent bb4076b
commit 228b1b6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/optimism/src/mips/interpreter.rs b/optimism/src/mips/interpreter.rs
@@ -194,6 +194,10 @@ pub enum ITypeInstruction {
 }
 
 pub trait InterpreterEnv {
+    type Position;
+
+    fn alloc_scratch(&mut self) -> Self::Position;
+
     type Variable: Clone
         + std::ops::Add<Self::Variable, Output = Self::Variable>
         + std::ops::Mul<Self::Variable, Output = Self::Variable>

diff --git a/optimism/src/mips/witness.rs b/optimism/src/mips/witness.rs
@@ -99,6 +99,14 @@ fn memory_size(total: usize) -> String {
 }
 
 impl<Fp: Field> InterpreterEnv for Env<Fp> {
+    type Position = Column;
+
+    fn alloc_scratch(&mut self) -> Self::Position {
+        let scratch_idx = self.scratch_state_idx;
+        self.scratch_state_idx += 1;
+        Column::ScratchState(scratch_idx)
+    }
+
     type Variable = u32;
 
     fn overwrite_register_checked(