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

fix sext

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-11-08 18:11:11 +01:00
parent 8f7cb9fbc5
commit e7bf8e9307
2 changed files with 41 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
poulpy-schemes/src/tfhe/bdd_arithmetic/tests/test_suite/fheuint.rs
View File

@@ -37,29 +37,52 @@ where
let mut a_enc: FheUint<Vec<u8>, u32> = FheUint::<Vec<u8>, u32>::alloc_from_infos(&glwe_infos);
for j in 0..3 {
for i in 0..32 {
let a: u32 = 0xFFFFFFFF >> i;
let a: u32 = 0x8483_8281;
a_enc.encrypt_sk(
module,
a,
sk,
&mut source_xa,
&mut source_xe,
scratch.borrow(),
);
a_enc.encrypt_sk(
module,
a,
sk,
&mut source_xa,
&mut source_xe,
scratch.borrow(),
);
a_enc.sext(module, j, keys, scratch.borrow());
a_enc.sext(module, j, keys, scratch.borrow());
// println!("{:08x} -> {:08x} {:08x}", a, sext(a, j), a_enc.decrypt(module, sk, scratch.borrow()));
// println!("{:08x} -> {:08x} {:08x}", a, sext(a, j), a_enc.decrypt(module, sk, scratch.borrow()));
assert_eq!(
sext(a, ((1 + j as u32) << 3) - 1),
a_enc.decrypt(module, sk, scratch.borrow())
);
}
assert_eq!(sext(a, j), a_enc.decrypt(module, sk, scratch.borrow()));
}
for j in 0..3 {
let a: u32 = 0x4443_4241;
a_enc.encrypt_sk(
module,
a,
sk,
&mut source_xa,
&mut source_xe,
scratch.borrow(),
);
a_enc.sext(module, j, keys, scratch.borrow());
// println!("{:08x} -> {:08x} {:08x}", a, sext(a, j), a_enc.decrypt(module, sk, scratch.borrow()));
assert_eq!(
sext(a, ((1 + j as u32) << 3) - 1),
a_enc.decrypt(module, sk, scratch.borrow())
);
}
}
pub fn sext(x: u32, byte: usize) -> u32 {
x | (((x >> (byte << 3)) & 1) * (0xFFFF_FFFF << (byte << 3)))
pub(crate) fn sext(x: u32, bits: u32) -> u32 {
let lo: u32 = x << (u32::BITS - bits) >> (u32::BITS - bits);
let hi: u32 = ((x >> bits) & 1) * (0xFFFF_FFFF << bits);
hi | lo
}
pub fn test_fhe_uint_splice_u8<BRA: BlindRotationAlgo, BE: Backend>(test_context: &TestContext<BRA, BE>)