poulpy-cpu-ref 0.6.0

Portable reference CPU implementations of poulpy-hal open extension points
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use poulpy_core::{
    api::{GLWEMulXpMinusOne, GLWERotate},
    layouts::{Base2K, Degree, GLWE, GLWELayout, ModuleCoreAlloc, Rank, TorusPrecision},
};
use poulpy_hal::{
    layouts::{FillUniform, Module},
    source::Source,
};

use crate::{FFT64Ref, hal_impl::delegating_backend::DelegatingFFT64Ref};

fn sample_glwe() -> GLWE<Vec<u8>> {
    let layout = GLWELayout {
        n: Degree(256),
        base2k: Base2K(17),
        k: TorusPrecision(50),
        rank: Rank(2),
    };
    let module: Module<FFT64Ref> = Module::new(256);
    let mut ct: GLWE<Vec<u8>> = module.glwe_alloc_from_infos(&layout);
    let mut source = Source::new([7u8; 32]);
    ct.fill_uniform(40, &mut source);
    ct
}

#[test]
fn delegating_backend_manual_family_matches_fft64_ref() {
    let module_delegating: Module<DelegatingFFT64Ref> = Module::new(256);
    let module_ref: Module<FFT64Ref> = Module::new(256);

    let input = sample_glwe();
    let mut delegating_out = module_delegating.glwe_alloc_from_infos(&input);
    let mut ref_out = module_ref.glwe_alloc_from_infos(&input);

    module_delegating.glwe_mul_xp_minus_one(-7, &mut delegating_out, &input);
    module_ref.glwe_mul_xp_minus_one(-7, &mut ref_out, &input);

    assert_eq!(delegating_out, ref_out);
}

#[test]
fn delegating_backend_delegated_family_matches_fft64_ref() {
    let module_delegating: Module<DelegatingFFT64Ref> = Module::new(256);
    let module_ref: Module<FFT64Ref> = Module::new(256);

    let input = sample_glwe();
    let mut delegating_out = module_delegating.glwe_alloc_from_infos(&input);
    let mut ref_out = module_ref.glwe_alloc_from_infos(&input);

    module_delegating.glwe_rotate(11, &mut delegating_out, &input);
    module_ref.glwe_rotate(11, &mut ref_out, &input);

    assert_eq!(delegating_out, ref_out);
}