fix(gpu): fix wrong number of blocks used in cast

zama-ai · Feb 21, 2025 · 219c755 · 219c755
1 parent fc4abd5
commit 219c755
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tfhe/src/integer/gpu/server_key/radix/mod.rs b/tfhe/src/integer/gpu/server_key/radix/mod.rs
@@ -692,7 +692,7 @@ impl CudaServerKey {
             ciphertext_modulus,
         );
 
-        let trimmed_ct_info = ct.as_ref().info.after_trim_radix_blocks_lsb(num_blocks);
+        let trimmed_ct_info = ct.as_ref().info.after_trim_radix_blocks_lsb(new_num_blocks);
         T::from(CudaRadixCiphertext::new(trimmed_ct_list, trimmed_ct_info))
     }
 
@@ -775,8 +775,7 @@ impl CudaServerKey {
             LweCiphertextCount(new_num_blocks),
             ciphertext_modulus,
         );
-
-        let trimmed_ct_info = ct.as_ref().info.after_trim_radix_blocks_msb(num_blocks);
+        let trimmed_ct_info = ct.as_ref().info.after_trim_radix_blocks_msb(new_num_blocks);
         T::from(CudaRadixCiphertext::new(trimmed_ct_list, trimmed_ct_info))
     }