fix(frontend-python): bit extract avoid extra reinterpret_precision

zama-ai · Jun 28, 2024 · 4d37487 · 4d37487
1 parent 2b94018
commit 4d37487
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/frontends/concrete-python/concrete/fhe/mlir/context.py b/frontends/concrete-python/concrete/fhe/mlir/context.py
@@ -2507,9 +2507,6 @@ def tlu_cell_input_value(i):
 
             assert lsb is not None
             bit_value = self.to_signedness(lsb, of=resulting_type)
-            bit_value = self.reinterpret(
-                bit_value, bit_width=max(resulting_type.bit_width, max_bit)
-            )
 
             delta_precision = position - actual_position
             assert actual_position < 0 or 0 <= delta_precision < resulting_type.bit_width, (
@@ -2518,7 +2515,10 @@ def tlu_cell_input_value(i):
                 resulting_type.bit_width,
             )
             if delta_precision:
-                bit_value = self.shift_left_at_constant_precision(bit_value, delta_precision)
+                bit_value = self.reinterpret(
+                    bit_value, bit_width=max(resulting_type.bit_width, max_bit)
+                )
+                bit_value = self.shift_left(bit_value, delta_precision)
 
             bit_value = self.reinterpret(bit_value, bit_width=resulting_type.bit_width)