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

First pass

Browse Source
This commit is contained in:
wborgeaud 2021-08-09 10:46:29 +02:00
parent a02ab2853a
commit f0f8320b93
1 changed files with 47 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
src/gadgets/arithmetic_extension.rs
View File

@ -12,6 +12,7 @@ use crate::iop::target::Target;
use crate::iop::witness::PartialWitness;
use crate::plonk::circuit_builder::CircuitBuilder;
use crate::util::bits_u64;
use crate::with_context;
impl<F: Extendable<D>, const D: usize> CircuitBuilder<F, D> {
pub fn double_arithmetic_extension(
@ -152,6 +153,29 @@ impl<F: Extendable<D>, const D: usize> CircuitBuilder<F, D> {
None
}
pub fn arithmetic_many_extension(
&mut self,
const_0: F,
const_1: F,
operands: Vec<[ExtensionTarget<D>; 3]>,
) -> Vec<ExtensionTarget<D>> {
let mut res = Vec::new();
for chunk in operands.chunks_exact(2) {
let [fm0, fm1, fa] = chunk[0];
let [sm0, sm1, sa] = chunk[1];
let arithm =
self.double_arithmetic_extension(const_0, const_1, fm0, fm1, fa, sm0, sm1, sa);
res.push(arithm.0);
res.push(arithm.1);
}
if operands.len().is_odd() {
let [m0, m1, a] = operands[operands.len() - 1];
res.push(self.arithmetic_extension(const_0, const_1, m0, m1, a));
}
res
}
pub fn add_extension(
&mut self,
a: ExtensionTarget<D>,
@ -325,17 +349,31 @@ impl<F: Extendable<D>, const D: usize> CircuitBuilder<F, D> {
a: ExtensionAlgebraTarget<D>,
b: ExtensionAlgebraTarget<D>,
) -> ExtensionAlgebraTarget<D> {
let mut res = [self.zero_extension(); D];
let w = self.constant(F::Extension::W);
let zero = self.zero_extension();
let mut ops = Vec::new();
let mut opsw = Vec::new();
for i in 0..D {
for j in 0..D {
res[(i + j) % D] = if i + j < D {
self.mul_add_extension(a.0[i], b.0[j], res[(i + j) % D])
} else {
let ai_bi = self.mul_extension(a.0[i], b.0[j]);
self.scalar_mul_add_extension(w, ai_bi, res[(i + j) % D])
}
for j in 0..D - i {
ops.push([a.0[i], b.0[j], zero]);
}
for j in D - i..D {
opsw.push([a.0[i], b.0[j], zero]);
}
}
let mut muls = self.arithmetic_many_extension(F::ONE, F::ONE, ops);
let mut mulsw = self.arithmetic_many_extension(F::Extension::W, F::ONE, opsw);
let mut toadd = vec![vec![]; D];
for i in 0..D {
for j in 0..D - i {
toadd[(i + j) % D].push(muls.remove(0));
}
for j in D - i..D {
toadd[(i + j) % D].push(mulsw.remove(0));
}
}
let mut res = [zero; D];
for i in 0..D {
res[i] = self.add_many_extension(&toadd[i]);
}
ExtensionAlgebraTarget(res)
}