fix(gpu): fix noise level calculation in full propagation

zama-ai · Dec 18, 2024 · 9e8db21 · 9e8db21
1 parent 950cece
commit 9e8db21
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tfhe/src/integer/gpu/server_key/radix/mod.rs b/tfhe/src/integer/gpu/server_key/radix/mod.rs
@@ -427,11 +427,14 @@ impl CudaServerKey {
                 );
             }
         };
-        ciphertext
-            .info
-            .blocks
-            .iter_mut()
-            .for_each(|b| b.degree = Degree::new(b.message_modulus.0 - 1));
+        ciphertext.info.blocks.iter_mut().for_each(|b| {
+            b.degree = Degree::new(b.message_modulus.0 - 1);
+            b.noise_level = if b.noise_level == NoiseLevel::ZERO {
+                NoiseLevel::ZERO
+            } else {
+                NoiseLevel::NOMINAL
+            };
+        });
     }
 
     /// Prepend trivial zero LSB blocks to an existing [`CudaUnsignedRadixCiphertext`] or