arnaucube/poulpy
1
0
Fork 0
mirror of https://github.com/arnaucube/poulpy.git synced 2026-02-10 05:06:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed rlwe package

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-04-24 21:53:06 +02:00
parent e25ded5430
commit f64d786819
9 changed files with 29 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rlwe/benches/gadget_product.rs
View File

@@ -1,6 +1,4 @@
use base2k::{ use base2k::{BACKEND, Module, Sampling, SvpPPolOps, VecZnx, VecZnxDft, VecZnxDftOps, VecZnxOps, VmpPMat, alloc_aligned_u8};
BACKEND, Infos, Module, Sampling, SvpPPolOps, VecZnx, VecZnxDft, VecZnxDftOps, VecZnxOps, VmpPMat, alloc_aligned_u8,
};
use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main}; use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};
use rlwe::{ use rlwe::{
ciphertext::{Ciphertext, new_gadget_ciphertext}, ciphertext::{Ciphertext, new_gadget_ciphertext},
@@ -109,7 +107,7 @@ fn bench_gadget_product_inplace(c: &mut Criterion) {
let mut res_dft_0: VecZnxDft = params.module().new_vec_znx_dft(gadget_ct.cols()); let mut res_dft_0: VecZnxDft = params.module().new_vec_znx_dft(gadget_ct.cols());
let mut res_dft_1: VecZnxDft = params.module().new_vec_znx_dft(gadget_ct.cols()); let mut res_dft_1: VecZnxDft = params.module().new_vec_znx_dft(gadget_ct.cols());
let mut a: VecZnx = params.module().new_vec_znx(params.cols_q()); let mut a: VecZnx = params.module().new_vec_znx(0, params.cols_q());
params params
.module() .module()
.fill_uniform(params.log_base2k(), &mut a, params.cols_q(), &mut source_xa); .fill_uniform(params.log_base2k(), &mut a, params.cols_q(), &mut source_xa);

8
rlwe/examples/encryption.rs
View File

@@ -39,11 +39,11 @@ fn main() {
let log_k: usize = params.log_q() - 20; let log_k: usize = params.log_q() - 20;
pt.0.value[0].encode_vec_i64(log_base2k, log_k, &want, 32); pt.0.value[0].encode_vec_i64(0, log_base2k, log_k, &want, 32);
pt.0.value[0].normalize(log_base2k, &mut tmp_bytes); pt.0.value[0].normalize(log_base2k, &mut tmp_bytes);
println!("log_k: {}", log_k); println!("log_k: {}", log_k);
pt.0.value[0].print(pt.cols(), 16); pt.0.value[0].print(0, pt.cols(), 16);
println!(); println!();
let mut ct: Ciphertext<VecZnx> = params.new_ciphertext(params.log_q()); let mut ct: Ciphertext<VecZnx> = params.new_ciphertext(params.log_q());
@@ -64,12 +64,12 @@ fn main() {
); );
params.decrypt_rlwe(&mut pt, &ct, &sk_svp_ppol, &mut tmp_bytes); params.decrypt_rlwe(&mut pt, &ct, &sk_svp_ppol, &mut tmp_bytes);
pt.0.value[0].print(pt.cols(), 16); pt.0.value[0].print(0, pt.cols(), 16);
let mut have = vec![i64::default(); params.n()]; let mut have = vec![i64::default(); params.n()];
println!("pt: {}", log_k); println!("pt: {}", log_k);
pt.0.value[0].decode_vec_i64(pt.log_base2k(), log_k, &mut have); pt.0.value[0].decode_vec_i64(0, pt.log_base2k(), log_k, &mut have);
println!("want: {:?}", &want[..16]); println!("want: {:?}", &want[..16]);
println!("have: {:?}", &have[..16]); println!("have: {:?}", &have[..16]);

6
rlwe/src/automorphism.rs
View File

@@ -11,7 +11,7 @@ use base2k::{
VmpPMatOps, assert_alignement, VmpPMatOps, assert_alignement,
}; };
use sampling::source::Source; use sampling::source::Source;
use std::{cmp::min, collections::HashMap}; use std::collections::HashMap;
/// Stores DFT([-A*AUTO(s, -p) + 2^{-K*i}*s + E, A]) where AUTO(X, p): X^{i} -> X^{i*p} /// Stores DFT([-A*AUTO(s, -p) + 2^{-K*i}*s + E, A]) where AUTO(X, p): X^{i} -> X^{i*p}
pub struct AutomorphismKey { pub struct AutomorphismKey {
@@ -295,7 +295,7 @@ mod test {
let mut pt: Plaintext = params.new_plaintext(log_q); let mut pt: Plaintext = params.new_plaintext(log_q);
let mut pt_auto: Plaintext = params.new_plaintext(log_q); let mut pt_auto: Plaintext = params.new_plaintext(log_q);
pt.at_mut(0).encode_vec_i64(log_base2k, log_k, &data, 32); pt.at_mut(0).encode_vec_i64(0, log_base2k, log_k, &data, 32);
module.vec_znx_automorphism(p, pt_auto.at_mut(0), pt.at(0)); module.vec_znx_automorphism(p, pt_auto.at_mut(0), pt.at(0));
encrypt_rlwe_sk( encrypt_rlwe_sk(
@@ -334,7 +334,7 @@ mod test {
// pt.at(0).print(pt.cols(), 16); // pt.at(0).print(pt.cols(), 16);
let noise_have: f64 = pt.at(0).std(log_base2k).log2(); let noise_have: f64 = pt.at(0).std(0, log_base2k).log2();
let var_msg: f64 = (params.xs() as f64) / params.n() as f64; let var_msg: f64 = (params.xs() as f64) / params.n() as f64;
let var_a_err: f64 = 1f64 / 12f64; let var_a_err: f64 = 1f64 / 12f64;

1
rlwe/src/decryptor.rs
View File

@@ -9,7 +9,6 @@ use base2k::{Module, SvpPPol, SvpPPolOps, VecZnx, VecZnxBigOps, VecZnxDft, VecZn
use std::cmp::min; use std::cmp::min;
pub struct Decryptor { pub struct Decryptor {
#[warn(dead_code)]
sk: SvpPPol, sk: SvpPPol,
} }

16
rlwe/src/elem.rs
View File

@@ -25,11 +25,11 @@ impl ElemVecZnx for Elem<VecZnx> {
let n: usize = module.n(); let n: usize = module.n();
assert!(bytes.len() >= Self::bytes_of(module, log_base2k, log_q, size)); assert!(bytes.len() >= Self::bytes_of(module, log_base2k, log_q, size));
let mut value: Vec<VecZnx> = Vec::new(); let mut value: Vec<VecZnx> = Vec::new();
let limbs: usize = (log_q + log_base2k - 1) / log_base2k; let cols: usize = (log_q + log_base2k - 1) / log_base2k;
let elem_size = VecZnx::bytes_of(n, limbs); let elem_size = VecZnx::bytes_of(n, size, cols);
let mut ptr: usize = 0; let mut ptr: usize = 0;
(0..size).for_each(|_| { (0..size).for_each(|_| {
value.push(VecZnx::from_bytes(n, limbs, &mut bytes[ptr..])); value.push(VecZnx::from_bytes(n, 1, cols, &mut bytes[ptr..]));
ptr += elem_size ptr += elem_size
}); });
Self { Self {
@@ -45,11 +45,11 @@ impl ElemVecZnx for Elem<VecZnx> {
let n: usize = module.n(); let n: usize = module.n();
assert!(bytes.len() >= Self::bytes_of(module, log_base2k, log_q, size)); assert!(bytes.len() >= Self::bytes_of(module, log_base2k, log_q, size));
let mut value: Vec<VecZnx> = Vec::new(); let mut value: Vec<VecZnx> = Vec::new();
let limbs: usize = (log_q + log_base2k - 1) / log_base2k; let cols: usize = (log_q + log_base2k - 1) / log_base2k;
let elem_size = VecZnx::bytes_of(n, limbs); let elem_size = VecZnx::bytes_of(n, 1, cols);
let mut ptr: usize = 0; let mut ptr: usize = 0;
(0..size).for_each(|_| { (0..size).for_each(|_| {
value.push(VecZnx::from_bytes_borrow(n, limbs, &mut bytes[ptr..])); value.push(VecZnx::from_bytes_borrow(n, 1, cols, &mut bytes[ptr..]));
ptr += elem_size ptr += elem_size
}); });
Self { Self {
@@ -135,9 +135,9 @@ impl<T: Infos> ElemCommon<T> for Elem<T> {
impl Elem<VecZnx> { impl Elem<VecZnx> {
pub fn new(module: &Module, log_base2k: usize, log_q: usize, rows: usize) -> Self { pub fn new(module: &Module, log_base2k: usize, log_q: usize, rows: usize) -> Self {
assert!(rows > 0); assert!(rows > 0);
let limbs: usize = (log_q + log_base2k - 1) / log_base2k; let cols: usize = (log_q + log_base2k - 1) / log_base2k;
let mut value: Vec<VecZnx> = Vec::new(); let mut value: Vec<VecZnx> = Vec::new();
(0..rows).for_each(|_| value.push(module.new_vec_znx(limbs))); (0..rows).for_each(|_| value.push(module.new_vec_znx(1, cols)));
Self { Self {
value, value,
log_q, log_q,

8
rlwe/src/gadget_product.rs
View File

@@ -206,7 +206,7 @@ mod test {
// Intermediate buffers // Intermediate buffers
// Input polynopmial, uniformly distributed // Input polynopmial, uniformly distributed
let mut a: VecZnx = params.module().new_vec_znx(params.cols_q()); let mut a: VecZnx = params.module().new_vec_znx(1, params.cols_q());
params params
.module() .module()
.fill_uniform(log_base2k, &mut a, params.cols_q(), &mut source_xa); .fill_uniform(log_base2k, &mut a, params.cols_q(), &mut source_xa);
@@ -217,7 +217,7 @@ mod test {
// Ideal output = a * s // Ideal output = a * s
let mut a_dft: VecZnxDft = params.module().new_vec_znx_dft(a.cols()); let mut a_dft: VecZnxDft = params.module().new_vec_znx_dft(a.cols());
let mut a_big: VecZnxBig = a_dft.as_vec_znx_big(); let mut a_big: VecZnxBig = a_dft.as_vec_znx_big();
let mut a_times_s: VecZnx = params.module().new_vec_znx(a.cols()); let mut a_times_s: VecZnx = params.module().new_vec_znx(1, a.cols());
// a * sk0 // a * sk0
params.module().svp_apply_dft(&mut a_dft, &sk0_svp_ppol, &a); params.module().svp_apply_dft(&mut a_dft, &sk0_svp_ppol, &a);
@@ -232,7 +232,7 @@ mod test {
// Iterates over all possible cols values for input/output polynomials and gadget ciphertext. // Iterates over all possible cols values for input/output polynomials and gadget ciphertext.
(1..a.cols() + 1).for_each(|a_cols| { (1..a.cols() + 1).for_each(|a_cols| {
let mut a_trunc: VecZnx = params.module().new_vec_znx(a_cols); let mut a_trunc: VecZnx = params.module().new_vec_znx(1, a_cols);
a_trunc.copy_from(&a); a_trunc.copy_from(&a);
(1..gadget_ct.cols() + 1).for_each(|b_cols| { (1..gadget_ct.cols() + 1).for_each(|b_cols| {
@@ -296,7 +296,7 @@ mod test {
// pt.at(0).print(pt.elem().cols(), 16); // pt.at(0).print(pt.elem().cols(), 16);
let noise_have: f64 = pt.at(0).std(log_base2k).log2(); let noise_have: f64 = pt.at(0).std(0, log_base2k).log2();
let var_a_err: f64; let var_a_err: f64;

2
rlwe/src/key_switching.rs
View File

@@ -1,6 +1,6 @@
use crate::ciphertext::Ciphertext; use crate::ciphertext::Ciphertext;
use crate::elem::ElemCommon; use crate::elem::ElemCommon;
use base2k::{Module, VecZnx, VecZnxBig, VecZnxBigOps, VecZnxDft, VecZnxDftOps, VmpPMat, VmpPMatOps, assert_alignement}; use base2k::{Module, VecZnx, VecZnxBigOps, VecZnxDftOps, VmpPMat, VmpPMatOps, assert_alignement};
use std::cmp::min; use std::cmp::min;
pub fn key_switch_tmp_bytes(module: &Module, log_base2k: usize, res_logq: usize, in_logq: usize, gct_logq: usize) -> usize { pub fn key_switch_tmp_bytes(module: &Module, log_base2k: usize, res_logq: usize, in_logq: usize, gct_logq: usize) -> usize {

4
rlwe/src/rgsw_product.rs
View File

@@ -193,7 +193,7 @@ mod test {
let mut pt: Plaintext = params.new_plaintext(log_q); let mut pt: Plaintext = params.new_plaintext(log_q);
let mut pt_rotate: Plaintext = params.new_plaintext(log_q); let mut pt_rotate: Plaintext = params.new_plaintext(log_q);
pt.at_mut(0).encode_vec_i64(log_base2k, log_k, &data, 32); pt.at_mut(0).encode_vec_i64(0, log_base2k, log_k, &data, 32);
module.vec_znx_rotate(k, pt_rotate.at_mut(0), pt.at_mut(0)); module.vec_znx_rotate(k, pt_rotate.at_mut(0), pt.at_mut(0));
@@ -222,7 +222,7 @@ mod test {
// pt.at(0).print(pt.cols(), 16); // pt.at(0).print(pt.cols(), 16);
let noise_have: f64 = pt.at(0).std(log_base2k).log2(); let noise_have: f64 = pt.at(0).std(0, log_base2k).log2();
let var_msg: f64 = 1f64 / params.n() as f64; // X^{k} let var_msg: f64 = 1f64 / params.n() as f64; // X^{k}
let var_a0_err: f64 = params.xe() * params.xe(); let var_a0_err: f64 = params.xe() * params.xe();

10
rlwe/src/trace.rs
View File

@@ -189,12 +189,12 @@ mod test {
let mut ct: Ciphertext<VecZnx> = params.new_ciphertext(log_q); let mut ct: Ciphertext<VecZnx> = params.new_ciphertext(log_q);
let mut pt: Plaintext = params.new_plaintext(log_q); let mut pt: Plaintext = params.new_plaintext(log_q);
pt.at_mut(0).encode_vec_i64(log_base2k, log_k, &data, 32); pt.at_mut(0).encode_vec_i64(0, log_base2k, log_k, &data, 32);
pt.at_mut(0).normalize(log_base2k, &mut tmp_bytes); pt.at_mut(0).normalize(log_base2k, &mut tmp_bytes);
pt.at(0).decode_vec_i64(log_base2k, log_k, &mut data); pt.at(0).decode_vec_i64(0, log_base2k, log_k, &mut data);
pt.at(0).print(pt.cols(), 16); pt.at(0).print(0, pt.cols(), 16);
encrypt_rlwe_sk( encrypt_rlwe_sk(
module, module,
@@ -227,9 +227,9 @@ mod test {
&mut tmp_bytes, &mut tmp_bytes,
); );
pt.at(0).print(pt.cols(), 16); pt.at(0).print(0, pt.cols(), 16);
pt.at(0).decode_vec_i64(log_base2k, log_k, &mut data); pt.at(0).decode_vec_i64(0, log_base2k, log_k, &mut data);
println!("trace: {:?}", &data[..16]); println!("trace: {:?}", &data[..16]);
} }