parasol_cpu 0.10.0

This crate contains the Parasol CPU, which runs programs over a mix of encrypted and plaintext data.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use std::sync::Arc;

use parasol_concurrency::AtomicRefCell;
use parasol_runtime::{FheCircuit, FheEdge, FheOp};

use crate::{
    Ciphertext, Error, Register, Result,
    proc::{DispatchIsaOp, fhe_processor::FheProcessor, ops::make_parent_op},
    tomasulo::{registers::RobEntryRef, tomasulo_processor::RetirementInfo},
    unwrap_registers,
};

impl FheProcessor {
    /// Execute a load instruction.
    pub fn not(
        &mut self,
        retirement_info: RetirementInfo<DispatchIsaOp>,
        dst: RobEntryRef<Register>,
        src: RobEntryRef<Register>,
        _instruction_id: usize,
        _pc: u32,
    ) {
        let mut add_impl = || -> Result<()> {
            unwrap_registers!((mut dst) (src) );

            match src {
                Register::Plaintext { val, width } => {
                    let mask = (0x1 << width) - 1;

                    *dst = Register::Plaintext {
                        val: (!val) & mask,
                        width: *width,
                    };

                    FheProcessor::retire(&retirement_info, Ok(()));
                }
                Register::Ciphertext(Ciphertext::L1Glwe { data: c }) => {
                    let width = c.len();

                    let mut graph = FheCircuit::new();

                    let dst_data = (0..width)
                        .map(|_| Arc::new(AtomicRefCell::new(self.aux_data.enc.allocate_glwe_l1())))
                        .collect::<Vec<_>>();

                    for (bit, dst) in c.iter().zip(dst_data.iter()) {
                        let bit_in_node = graph.add_node(FheOp::InputGlwe1(bit.clone()));
                        let neg_node = graph.add_node(FheOp::Not);
                        graph.add_edge(bit_in_node, neg_node, FheEdge::Unary);
                        let out = graph.add_node(FheOp::OutputGlwe1(dst.clone()));
                        graph.add_edge(neg_node, out, FheEdge::Unary);
                    }

                    let parent_op = make_parent_op(&retirement_info);

                    *dst = Register::Ciphertext(Ciphertext::L1Glwe { data: dst_data });

                    self.aux_data
                        .uop_processor
                        .spawn_graph(&graph, &self.aux_data.flow, parent_op);
                }
                _ => return Err(Error::EncryptionMismatch),
            };

            Ok(())
        };

        if let Err(e) = add_impl() {
            FheProcessor::retire(&retirement_info, Err(e));
        }
    }
}