moving scratch into less-than-variable scope

o1-labs · Dec 24, 2024 · 7751170 · 7751170
1 parent fa494c3
commit 7751170
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/o1vm/src/interpreters/riscv32im/interpreter.rs b/o1vm/src/interpreters/riscv32im/interpreter.rs
@@ -2271,10 +2271,10 @@ pub fn interpret_sbtype<Env: InterpreterEnv>(env: &mut Env, instr: SBInstruction
             let local_rs1 = env.read_register(&rs1);
             let local_rs2 = env.read_register(&rs2);
 
-            let rd_scratch = env.alloc_scratch();
-
-            let less_than =
-                unsafe { env.test_less_than_signed(&local_rs1, &local_rs2, rd_scratch) };
+            let less_than = {
+                let rd_scratch = env.alloc_scratch();
+                unsafe { env.test_less_than_signed(&local_rs1, &local_rs2, rd_scratch) }
+            };
             let offset = (less_than.clone()) * imm0_12
                 + (Env::constant(1) - less_than.clone()) * Env::constant(4);