diff --git a/src/fri/recursive_verifier.rs b/src/fri/recursive_verifier.rs index b8a7c20f..1baf598f 100644 --- a/src/fri/recursive_verifier.rs +++ b/src/fri/recursive_verifier.rs @@ -28,20 +28,23 @@ impl, const D: usize> CircuitBuilder { debug_assert_eq!(last_evals.len(), 1 << arity_bits); let g = F::primitive_root_of_unity(arity_bits); + let gt = self.constant(g); // The evaluation vector needs to be reordered first. let mut evals = last_evals.to_vec(); reverse_index_bits_in_place(&mut evals); let mut old_x_index_bits = self.split_le(old_x_index, arity_bits); old_x_index_bits.reverse(); - let evals = self.rotate_left_from_bits(&old_x_index_bits, &evals); + let start = self.exp_from_complement_bits(gt, &old_x_index_bits); + let start = self.mul(start, x); + let start = self.mul(start, gt); // The answer is gotten by interpolating {(x*g^i, P(x*g^i))} and evaluating at beta. let points = g .powers() .map(|y| { let yt = self.constant(y); - self.mul(x, yt) + self.mul(start, yt) }) .zip(evals) .collect::>(); diff --git a/src/fri/verifier.rs b/src/fri/verifier.rs index df0c33fc..89e35eb4 100644 --- a/src/fri/verifier.rs +++ b/src/fri/verifier.rs @@ -22,20 +22,21 @@ fn compute_evaluation, const D: usize>( last_evals: &[F::Extension], beta: F::Extension, ) -> F::Extension { - debug_assert_eq!(last_evals.len(), 1 << arity_bits); + let arity = 1 << arity_bits; + debug_assert_eq!(last_evals.len(), arity); let g = F::primitive_root_of_unity(arity_bits); // The evaluation vector needs to be reordered first. let mut evals = last_evals.to_vec(); reverse_index_bits_in_place(&mut evals); - evals.rotate_left(reverse_bits(old_x_index, arity_bits)); - + let rev_x_index = reverse_bits(old_x_index, arity_bits); + let start = x * g.exp((arity - rev_x_index) as u64); // The answer is gotten by interpolating {(x*g^i, P(x*g^i))} and evaluating at beta. let points = g .powers() .zip(evals) - .map(|(y, e)| ((x * y).into(), e)) + .map(|(y, e)| ((start * y).into(), e)) .collect::>(); let barycentric_weights = barycentric_weights(&points); interpolate(&points, beta, &barycentric_weights) diff --git a/src/gadgets/arithmetic.rs b/src/gadgets/arithmetic.rs index 6f85cdcf..605061ee 100644 --- a/src/gadgets/arithmetic.rs +++ b/src/gadgets/arithmetic.rs @@ -155,14 +155,13 @@ impl, const D: usize> CircuitBuilder { // TODO: Optimize this, maybe with a new gate. // TODO: Test - /// Exponentiate `base` to the power of `exponent`, where `exponent < 2^num_bits`. - pub fn exp(&mut self, base: Target, exponent: Target, num_bits: usize) -> Target { + /// Exponentiate `base` to the power of `exponent`, given by its little-endian bits. + pub fn exp_from_bits(&mut self, base: Target, exponent_bits: &[Target]) -> Target { let mut current = base; let one_ext = self.one_extension(); let mut product = self.one(); - let exponent_bits = self.split_le(exponent, num_bits); - for bit in exponent_bits.into_iter() { + for &bit in exponent_bits { let current_ext = self.convert_to_ext(current); let multiplicand = self.select(bit, current_ext, one_ext); product = self.mul(product, multiplicand.0[0]); @@ -172,6 +171,32 @@ impl, const D: usize> CircuitBuilder { product } + // TODO: Optimize this, maybe with a new gate. + // TODO: Test + /// Exponentiate `base` to the power of `exponent`, given by its little-endian bits. + pub fn exp_from_complement_bits(&mut self, base: Target, exponent_bits: &[Target]) -> Target { + let mut current = base; + let one_ext = self.one_extension(); + let mut product = self.one(); + + for &bit in exponent_bits { + let current_ext = self.convert_to_ext(current); + let multiplicand = self.select(bit, one_ext, current_ext); + product = self.mul(product, multiplicand.0[0]); + current = self.mul(current, current); + } + + product + } + + // TODO: Optimize this, maybe with a new gate. + // TODO: Test + /// Exponentiate `base` to the power of `exponent`, where `exponent < 2^num_bits`. + pub fn exp(&mut self, base: Target, exponent: Target, num_bits: usize) -> Target { + let exponent_bits = self.split_le(exponent, num_bits); + self.exp_from_bits(base, &exponent_bits) + } + /// Exponentiate `base` to the power of a known `exponent`. // TODO: Test pub fn exp_u64(&mut self, base: Target, exponent: u64) -> Target {