Update LBB

Co-authored-by: Dimitris Mouris <[email protected]>
TrustworthyComputing · Sep 7, 2023 · 8bf09ba · 8bf09ba
1 parent c1da8f6
commit 8bf09ba
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/circuit.rs b/src/circuit.rs
@@ -621,7 +621,7 @@ impl<'a> EvalCircuit<CtxtShortInt> for LutCircuit<'a> {
         {
             // Evaluate all the gates in the level in parallel
             gates.par_iter_mut().for_each(|gate| {
-                let input_values: Vec<CtxtShortInt> = gate
+                let mut input_values: Vec<CtxtShortInt> = gate
                     .get_input_wires()
                     .iter()
                     .map(|input| {
@@ -637,7 +637,7 @@ impl<'a> EvalCircuit<CtxtShortInt> for LutCircuit<'a> {
                     .collect();
                 let output_value = {
                     if gate.get_gate_type() == GateType::Lut {
-                        gate.evaluate_encrypted_lut(&self.server_key, &input_values, cycle)
+                        gate.evaluate_encrypted_lut(&self.server_key, &mut input_values, cycle)
                     } else {
                         gate.evaluate_encrypted_dff(&input_values, cycle)
                     }

diff --git a/src/gates.rs b/src/gates.rs
@@ -278,7 +278,7 @@ impl Gate {
     pub fn evaluate_encrypted_lut(
         &mut self,
         server_key: &ServerKeyShortInt,
-        input_values: &Vec<CiphertextBase>,
+        input_values: &mut Vec<CiphertextBase>,
         cycle: usize,
     ) -> CiphertextBase {
         if let Some(encrypted_lut_output) = self.encrypted_lut_output.clone() {
@@ -744,13 +744,14 @@ fn eval_luts(x: u64, lut_table: &Vec<u64>) -> u64 {
 pub fn lut(
     sks: &ServerKeyShortInt,
     lut_const: &Vec<u64>,
-    ctxts: &Vec<CiphertextBase>,
+    ctxts: &mut Vec<CiphertextBase>,
 ) -> CiphertextBase {
     // Σ ctxts[i] * 2^i
+    let ctxts_len = (ctxts.len() - 1) as u8;
     let ct_sum = ctxts
-        .iter()
+        .iter_mut()
         .enumerate()
-        .map(|(i, ct)| sks.scalar_mul(ct, 1 << (ctxts.len() - 1 - i)))
+        .map(|(i, ct)| sks.smart_scalar_left_shift(ct, ctxts_len - i as u8))
         .fold(sks.create_trivial(0), |acc, ct| sks.add(&acc, &ct));
 
     // Generate LUT entries from lut_const