fix: rotations of 1 blocks of 4_4

zama-ai · Feb 14, 2025 · 42e9b60 · 42e9b60
1 parent 1b7f7a5
commit 42e9b60
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tfhe/src/integer/server_key/radix_parallel/shift.rs b/tfhe/src/integer/server_key/radix_parallel/shift.rs
@@ -391,12 +391,13 @@ impl ServerKey {
                         }
                         BarrelShifterOperation::LeftRotate => {
                             let shifted = (input << shift_within_block) % self.message_modulus().0;
-                            let wrapped = input >> (shift_within_block);
+                            let wrapped = input >> (message_bits_per_block - shift_within_block);
                             shifted | wrapped
                         }
                         BarrelShifterOperation::RightRotate => {
                             let shifted = input >> shift_within_block;
-                            let wrapped = (input << shift_within_block) % self.message_modulus().0;
+                            let wrapped = (input << (message_bits_per_block - shift_within_block))
+                                % self.message_modulus().0;
                             wrapped | shifted
                         }
                         BarrelShifterOperation::RightShift => {