tfhe-rs execution context example

tfhe-execution-context.rs

use tfhe::shortint::{gen_keys, Parameters, context::{ExecutionContext, OperationFlavor}};

let (client_key, server_key) = gen_key(Parameters::default());

let msg1 = 1;
let msg2 = 0;

let ct_1 = client_key.encrypt(msg1);
let ct_2 = client_key.encrypt(msg2);

let ct_3 = ExecutionContext::new(&server_key, OperationFlavor::Unchecked).execute(|| ct_1 + ct_2);

let output = client_key.decrypt(&ct_3);
assert_eq!(output, 1);

y-get-context-items.rs

use tfhe::shortint::{gen_keys, Parameters, context::{self, ExecutionContext, OperationFlavor}};

let (_client_key, server_key) = gen_key(Parameters::default());

ExecutionContext::new(&server_key, OperationFlavor::Unchecked)
    .execute(|| {
        assert_eq!(&server_key, context::server_key());
        assert_eq!(OperationFlavor::Unchecked, context::operation_flavor());
    });

// context::server_key(); // would probably panic outside of an `ExecutionContext`

z-example-add-impl.rs

impl Add<Ciphertext> for Ciphertext {
    type Output = Ciphertext;
  
    fn add(self, rhs: Self) -> Self::Output {
        let server_key = context::server_key();
        match context::operation_flavor() {
            OperationFlavor::Unchecked => server_key.unchecked_add(&self, &rhs),
            // would have to panic instead. caller could either `catch_unwind` or call
            // `ServerKey::checked_add` from within context manually
            OperationFlavor::Checked => server_key.checked_add(&self, &rhs)
                .expect("failed checked_add from within execution context"),
            OperationFlavor::Smart => server_key.smart_add(&self, &rhs),
        }
    }
}

could be implemented via https://docs.rs/execution-context/latest/execution_context/struct.ExecutionContext.html