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

fmt

Browse Source
This commit is contained in:
Nicholas Ward 2022-06-03 11:32:27 -07:00
parent 80d5e5375e
commit dc082139c5
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
evm/src/keccak/keccak_stark.rs
View File

@ -508,8 +508,8 @@ impl<F: RichField + Extendable<D>, const D: usize> Stark<F, D> for KeccakStark<F
mod tests { mod tests {
use anyhow::Result; use anyhow::Result;
use keccak_rust::{KeccakF, StateBitsWidth}; use keccak_rust::{KeccakF, StateBitsWidth};
use plonky2::plonk::config::{GenericConfig, PoseidonGoldilocksConfig};
use plonky2::field::field_types::Field; use plonky2::field::field_types::Field;
use plonky2::plonk::config::{GenericConfig, PoseidonGoldilocksConfig};
use rand::Rng; use rand::Rng;
use crate::keccak::keccak_stark::{KeccakStark, INPUT_LIMBS}; use crate::keccak::keccak_stark::{KeccakStark, INPUT_LIMBS};
@ -566,19 +566,28 @@ mod tests {
let base = F::from_canonical_u64(1 << 32); let base = F::from_canonical_u64(1 << 32);
for x in 0..5 { for x in 0..5 {
for y in 0..5 { for y in 0..5 {
output.push(last_row[reg_a_prime_prime_prime(x, y)] + base * last_row[reg_a_prime_prime_prime(x, y) + 1]); output.push(
last_row[reg_a_prime_prime_prime(x, y)]
+ base * last_row[reg_a_prime_prime_prime(x, y) + 1],
);
} }
} }
let mut expected: [[u64; 5]; 5] = [input[0..5].try_into().unwrap(), let mut expected: [[u64; 5]; 5] = [
input[5..10].try_into().unwrap(), input[0..5].try_into().unwrap(),
input[10..15].try_into().unwrap(), input[5..10].try_into().unwrap(),
input[15..20].try_into().unwrap(), input[10..15].try_into().unwrap(),
input[20..25].try_into().unwrap()]; input[15..20].try_into().unwrap(),
input[20..25].try_into().unwrap(),
];
let keccak = KeccakF::new(StateBitsWidth::F1600); let keccak = KeccakF::new(StateBitsWidth::F1600);
keccak.permutations(&mut expected); keccak.permutations(&mut expected);
let expected_flattened: Vec<_> = expected.iter().flatten().map(|&x| F::from_canonical_u64(x)).collect(); let expected_flattened: Vec<_> = expected
.iter()
.flatten()
.map(|&x| F::from_canonical_u64(x))
.collect();
assert_eq!(output, expected_flattened); assert_eq!(output, expected_flattened);