logos-storage/plonky2
1
0
Fork 0
mirror of https://github.com/logos-storage/plonky2.git synced 2026-01-08 16:53:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix alignment assumptions in AVX2 Poseidon

Browse Source
This commit is contained in:
Jakub Nabaglo 2021-09-17 10:04:05 -07:00
parent 7be26a708b
commit d3de2b5518
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/hash/poseidon_avx2.rs
View File

@ -214,10 +214,10 @@ pub unsafe fn crandall_poseidon_const_avx2<const PACKED_WIDTH: usize>(
round_constants: [u64; 4 * PACKED_WIDTH],
) {
let packed_state = PackedCrandallAVX2::pack_slice_mut(state);
let packed_round_constants =
std::slice::from_raw_parts((&round_constants).as_ptr().cast::<__m256i>(), PACKED_WIDTH);
for i in 0..PACKED_WIDTH {
packed_state[i] = packed_state[i].add_canonical_u64(packed_round_constants[i]);
let constants_ptr = (&round_constants[4 * i..4 * i + 4]).as_ptr();
let packed_constants = _mm256_loadu_si256(constants_ptr.cast::<__m256i>());
packed_state[i] = packed_state[i].add_canonical_u64(packed_constants);
}
}