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

Support for bivariate convolution & normalization with offset (#126)

Browse Source 
* Add bivariate-convolution
* Add pair-wise convolution + tests + benches
* Add take_cnv_pvec_[left/right] to Scratch & updated CHANGELOG.md
* cross-base2k normalization with positive offset
* clippy & fix CI doctest avx compile error
* more streamlined bounds derivation for normalization
* Working cross-base2k normalization with pos/neg offset
* Update normalization API & tests
* Add glwe tensoring test
* Add relinearization + preliminary test
* Fix GGLWEToGGSW key infos
* Add (X,Y) convolution by const (1, Y) poly
* Faster normalization test + add bench for cnv_by_const
* Update changelog
This commit is contained in:
Jean-Philippe Bossuat
2025-12-21 16:56:42 +01:00
committed by GitHub
parent 76424d0ab5
commit 4e90e08a71
219 changed files with 6571 additions and 5041 deletions
Download Patch File Download Diff File Expand all files Collapse all files

128
poulpy-schemes/benches/bdd_arithmetic.rs
View File

@@ -91,14 +91,7 @@ where
// Circuit bootstrapping evaluation key
let mut cbt_key: CircuitBootstrappingKey<Vec<u8>, BRA> =
CircuitBootstrappingKey::alloc_from_infos(&params.bdd_layout.cbt_layout);
cbt_key.encrypt_sk(
&module,
&sk_lwe,
&sk_glwe,
&mut source_xa,
&mut source_xe,
scratch.borrow(),
);
cbt_key.encrypt_sk(&module, &sk_lwe, &sk_glwe, &mut source_xa, &mut source_xe, scratch.borrow());
let mut cbt_key_prepared: CircuitBootstrappingKeyPrepared<Vec<u8>, BRA, BE> =
CircuitBootstrappingKeyPrepared::alloc_from_infos(&module, &params.bdd_layout.cbt_layout);
@@ -108,14 +101,7 @@ where
sk_glwe_prepared.prepare(&module, &sk_glwe);
let mut bdd_key: BDDKey<Vec<u8>, BRA> = BDDKey::alloc_from_infos(&params.bdd_layout);
bdd_key.encrypt_sk(
&module,
&sk_lwe,
&sk_glwe,
&mut source_xa,
&mut source_xe,
scratch.borrow(),
);
bdd_key.encrypt_sk(&module, &sk_lwe, &sk_glwe, &mut source_xa, &mut source_xe, scratch.borrow());
let input_a = 255_u32;
let input_b = 30_u32;
@@ -332,95 +318,45 @@ where
};
// Benchmark each operation
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"add",
|c_enc, module, a, b, key, scratch| {
c_enc.add(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "add", |c_enc, module, a, b, key, scratch| {
c_enc.add(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"sub",
|c_enc, module, a, b, key, scratch| {
c_enc.sub(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "sub", |c_enc, module, a, b, key, scratch| {
c_enc.sub(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"sll",
|c_enc, module, a, b, key, scratch| {
c_enc.sll(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "sll", |c_enc, module, a, b, key, scratch| {
c_enc.sll(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"sra",
|c_enc, module, a, b, key, scratch| {
c_enc.sra(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "sra", |c_enc, module, a, b, key, scratch| {
c_enc.sra(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"srl",
|c_enc, module, a, b, key, scratch| {
c_enc.srl(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "srl", |c_enc, module, a, b, key, scratch| {
c_enc.srl(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"slt",
|c_enc, module, a, b, key, scratch| {
c_enc.slt(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "slt", |c_enc, module, a, b, key, scratch| {
c_enc.slt(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"sltu",
|c_enc, module, a, b, key, scratch| {
c_enc.sltu(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "sltu", |c_enc, module, a, b, key, scratch| {
c_enc.sltu(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"or",
|c_enc, module, a, b, key, scratch| {
c_enc.or(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "or", |c_enc, module, a, b, key, scratch| {
c_enc.or(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"and",
|c_enc, module, a, b, key, scratch| {
c_enc.and(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "and", |c_enc, module, a, b, key, scratch| {
c_enc.and(module, a, b, key, scratch);
});
bench_operation::<BE, BRA, _>(
&mut group,
&params,
"xor",
|c_enc, module, a, b, key, scratch| {
c_enc.xor(module, a, b, key, scratch);
},
);
bench_operation::<BE, BRA, _>(&mut group, &params, "xor", |c_enc, module, a, b, key, scratch| {
c_enc.xor(module, a, b, key, scratch);
});
group.finish();
}