Merge pull request #2805 from o1-labs/dw/riscv32/implement-mod_

o1vm/riscv32: implement mod_

o1vm/src/interpreters/riscv32im/constraints.rs

@@ -371,6 +371,15 @@ impl<Fp: Field> InterpreterEnv for Env<Fp> {
         self.variable(position)
     }
 
+    unsafe fn mod_unsigned(
+        &mut self,
+        _x: &Self::Variable,
+        _y: &Self::Variable,
+        position: Self::Position,
+    ) -> Self::Variable {
+        self.variable(position)
+    }
+
     unsafe fn mul_lo(
         &mut self,
         _x: &Self::Variable,

o1vm/src/interpreters/riscv32im/interpreter.rs

@@ -1299,6 +1299,20 @@ pub trait InterpreterEnv {
         position: Self::Position,
     ) -> Self::Variable;
 
+    /// Returns `x % y`, storing the results in `position`.
+    ///
+    /// # Safety
+    ///
+    /// There are no constraints on the returned values; callers must manually add constraints to
+    /// ensure that the pair of returned values correspond to the given values `x` and `y`, and
+    /// that they fall within the desired range.
+    unsafe fn mod_unsigned(
+        &mut self,
+        x: &Self::Variable,
+        y: &Self::Variable,
+        position: Self::Position,
+    ) -> Self::Variable;
+
     /// Returns `(x / y, x % y)`, storing the results in `position_quotient` and
     /// `position_remainder` respectively.
     ///

o1vm/src/interpreters/riscv32im/witness.rs

@@ -543,6 +543,20 @@ impl<Fp: Field> InterpreterEnv for Env<Fp> {
         res
     }
 
+    unsafe fn mod_unsigned(
+        &mut self,
+        x: &Self::Variable,
+        y: &Self::Variable,
+        position: Self::Position,
+    ) -> Self::Variable {
+        let x: u32 = (*x).try_into().unwrap();
+        let y: u32 = (*y).try_into().unwrap();
+        let res = x % y;
+        let res = res as u64;
+        self.write_column(position, res);
+        res
+    }
+
     unsafe fn divmod(
         &mut self,
         x: &Self::Variable,