fixed tests

This commit is contained in:
Jean-Philippe Bossuat
2025-05-26 13:57:19 +02:00
parent 4c3a568108
commit e5d6a6f828
2 changed files with 4 additions and 4 deletions

View File

@@ -1,7 +1,7 @@
use backend::{ use backend::{
Backend, FFT64, MatZnxDft, MatZnxDftAlloc, MatZnxDftOps, MatZnxDftToMut, MatZnxDftToRef, Module, ScalarZnx, ScalarZnxDft, Backend, FFT64, MatZnxDft, MatZnxDftAlloc, MatZnxDftOps, MatZnxDftToMut, MatZnxDftToRef, Module, ScalarZnx, ScalarZnxDft,
ScalarZnxDftToRef, ScalarZnxToRef, Scratch, VecZnxAlloc, VecZnxBigScratch, VecZnxDftAlloc, VecZnxDftToMut, VecZnxDftToRef, ScalarZnxDftToRef, ScalarZnxToRef, Scratch, VecZnxAlloc, VecZnxDftAlloc, VecZnxDftToMut, VecZnxDftToRef, VecZnxOps, ZnxInfos,
VecZnxOps, ZnxInfos, ZnxZero, ZnxZero,
}; };
use sampling::source::Source; use sampling::source::Source;

View File

@@ -40,9 +40,9 @@ fn test_encrypt_sk(log_n: usize, basek: usize, k: usize, sigma: f64, rank: usize
sk.fill_ternary_prob(0.5, &mut source_xs); sk.fill_ternary_prob(0.5, &mut source_xs);
let mut sk_dft: SecretKeyFourier<Vec<u8>, FFT64> = SecretKeyFourier::alloc(&module, rank); let mut sk_dft: SecretKeyFourier<Vec<u8>, FFT64> = SecretKeyFourier::alloc(&module, rank);
sk_dft.dft(generate_from_sksk); sk_dft.dft(&module, &sk);
tensor_key.encrypt_sk( tensor_key.generate_from_sk(
&module, &module,
&sk_dft, &sk_dft,
&mut source_xa, &mut source_xa,