Implement Neg for FieldElement. (#2252)

powdr-labs · Dec 18, 2024 · a3d9f38 · a3d9f38
1 parent c8dcd07
commit a3d9f38
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/executor/src/witgen/jit/compiler.rs b/executor/src/witgen/jit/compiler.rs
@@ -368,6 +368,17 @@ impl std::ops::Sub for FieldElement {{
         Self(IntType::try_from(((self.0 as DoubleIntType) + (MODULUS as DoubleIntType) - (b.0 as DoubleIntType)) % (MODULUS as DoubleIntType)).unwrap())
     }}
 }}
+impl std::ops::Neg for FieldElement {{
+    type Output = Self;
+    #[inline]
+    fn neg(self) -> Self {{
+        if self.0 == 0 {{
+            self
+        }} else {{
+            Self(MODULUS - self.0)
+        }}
+    }}
+}}
 impl std::ops::Mul<FieldElement> for FieldElement {{
     type Output = Self;
     #[inline]