p3_baby_bear/aarch64_neon/
poseidon2.rs

Help
1//! Eventually this will hold a vectorized Neon implementation of Poseidon2 for PackedBabyBearNeon
2//! Currently this is essentially a placeholder to allow compilation and testing on Neon devices.
3//!
4//! Converting the AVX2/AVX512 code across to Neon is on the TODO list.
5
6#[cfg(test)]
7mod tests {
8    use p3_field::FieldAlgebra;
9    use p3_symmetric::Permutation;
10    use rand::Rng;
11
12    use crate::{BabyBear, PackedBabyBearNeon, Poseidon2BabyBear};
13
14    type F = BabyBear;
15    type Perm16 = Poseidon2BabyBear<16>;
16    type Perm24 = Poseidon2BabyBear<24>;
17
18    /// Test that the output is the same as the scalar version on a random input.
19    #[test]
20    fn test_neon_poseidon2_width_16() {
21        let mut rng = rand::thread_rng();
22
23        // Our Poseidon2 implementation.
24        let poseidon2 = Perm16::new_from_rng_128(&mut rng);
25
26        let input: [F; 16] = rng.gen();
27
28        let mut expected = input;
29        poseidon2.permute_mut(&mut expected);
30
31        let mut neon_input = input.map(PackedBabyBearNeon::from_f);
32        poseidon2.permute_mut(&mut neon_input);
33
34        let neon_output = neon_input.map(|x| x.0[0]);
35
36        assert_eq!(neon_output, expected);
37    }
38
39    /// Test that the output is the same as the scalar version on a random input.
40    #[test]
41    fn test_neon_poseidon2_width_24() {
42        let mut rng = rand::thread_rng();
43
44        // Our Poseidon2 implementation.
45        let poseidon2 = Perm24::new_from_rng_128(&mut rng);
46
47        let input: [F; 24] = rng.gen();
48
49        let mut expected = input;
50        poseidon2.permute_mut(&mut expected);
51
52        let mut neon_input = input.map(PackedBabyBearNeon::from_f);
53        poseidon2.permute_mut(&mut neon_input);
54
55        let neon_output = neon_input.map(|x| x.0[0]);
56
57        assert_eq!(neon_output, expected);
58    }
59}
p3_baby_bear/aarch64_neon/poseidon2.rs

p3_baby_bear/aarch64_neon/
poseidon2.rs
