fix(core): fix unsigned noise addition for custom modulus

zama-ai · Feb 26, 2024 · 618758b · 618758b
1 parent d770a27
commit 618758b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tfhe/src/core_crypto/commons/math/random/generator.rs b/tfhe/src/core_crypto/commons/math/random/generator.rs
@@ -665,10 +665,10 @@ impl<G: ByteRandomGenerator> RandomGenerator<G> {
                 custom_modulus_as_scalar,
             );
             if let Some(elem) = s.get_mut(0) {
-                *elem = (*elem).wrapping_add(g1);
+                *elem = (*elem).wrapping_add_custom_mod(g1, custom_modulus_as_scalar);
             }
             if let Some(elem) = s.get_mut(1) {
-                *elem = (*elem).wrapping_add(g2);
+                *elem = (*elem).wrapping_add_custom_mod(g2, custom_modulus_as_scalar);
             }
         });
     }