fix grind step

starkware-libs · Jan 10, 2025 · 2db1f86 · 2db1f86
1 parent 14cf9d2
commit 2db1f86
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crates/prover/src/core/backend/simd/grind.rs b/crates/prover/src/core/backend/simd/grind.rs
@@ -9,7 +9,7 @@ use rayon::prelude::*;
 
 use super::blake2s::compress16;
 use super::SimdBackend;
-use crate::core::backend::simd::m31::{PackedM31, LOG_N_LANES, N_LANES};
+use crate::core::backend::simd::m31::{PackedM31, N_LANES};
 use crate::core::backend::simd::poseidon31::permute;
 use crate::core::channel::{Blake2sChannel, Poseidon31Channel};
 #[cfg(not(target_arch = "wasm32"))]
@@ -114,7 +114,7 @@ fn grind_poseidon31(digest: &[M31; 8], hi: u64, pow_bits: u32) -> Option<u64> {
     let high_start = hi << GRIND_LOW_BITS;
 
     for low in (0..(1 << GRIND_LOW_BITS)).step_by(N_LANES) {
-        let start = high_start + (low << LOG_N_LANES);
+        let start = high_start + low;
 
         attempt[0] = PackedM31::from_array(std::array::from_fn(|i| {
             M31::from_u32_unchecked(((start + i as u64) % ((1 << 22) - 1)) as u32)