Merge branch 'sai/branch-lt' into sai/branch-gte

o1-labs · Dec 23, 2024 · 0b721aa · 0b721aa
2 parents 16f66ac + badda6b
commit 0b721aa
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
@@ -2210,10 +2210,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);