chore(fhe_strings): limit max n to improve performance

zama-ai · Nov 6, 2024 · e882229 · e882229
1 parent bcae0f1
commit e882229
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tfhe/src/strings/server_key/pattern/replace.rs b/tfhe/src/strings/server_key/pattern/replace.rs
@@ -313,7 +313,7 @@ impl ServerKey {
 
             UIntArg::Enc(enc_n) => {
                 // As we don't know the number n we perform the maximum number of iterations
-                let max = enc_n.max().unwrap_or(u16::MAX);
+                let max = (enc_n.max().unwrap_or(u16::MAX) as usize).min(result.len() + 1) as u16;
 
                 self.replace_n_times(max, &mut result, from, to, Some(enc_n.cipher()));
             }