This commit is contained in:
Nicholas Ward 2021-10-04 14:17:28 -07:00
parent ffb544e4a5
commit e48e0a4a58
2 changed files with 47 additions and 10 deletions

View File

@ -19,7 +19,11 @@ impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
}
pub fn add_virtual_u32_targets(&self, n: usize) -> Vec<U32Target> {
self.add_virtual_targets(n).iter().cloned().map(U32Target).collect()
self.add_virtual_targets(n)
.iter()
.cloned()
.map(U32Target)
.collect()
}
pub fn zero_u32(&self) -> U32Target {
@ -43,18 +47,45 @@ impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
};
let gate_index = self.add_gate(gate.clone(), vec![]);
(gate_index, 0)
},
}
Some((gate_index, copy)) => (gate_index, copy),
};
let output_low = self.add_virtual_u32_target();
let output_high = self.add_virtual_u32_target();
self.connect(Target::wire(gate_index, U32ArithmeticGate::<F, D>::wire_ith_multiplicand_0(copy)), x.0);
self.connect(Target::wire(gate_index, U32ArithmeticGate::<F, D>::wire_ith_multiplicand_1(copy)), y.0);
self.connect(Target::wire(gate_index, U32ArithmeticGate::<F, D>::wire_ith_addend(copy)), z.0);
self.connect(Target::wire(gate_index, U32ArithmeticGate::<F, D>::wire_ith_output_low_half(copy)), output_low.0);
self.connect(Target::wire(gate_index, U32ArithmeticGate::<F, D>::wire_ith_output_high_half(copy)), output_high.0);
self.connect(
Target::wire(
gate_index,
U32ArithmeticGate::<F, D>::wire_ith_multiplicand_0(copy),
),
x.0,
);
self.connect(
Target::wire(
gate_index,
U32ArithmeticGate::<F, D>::wire_ith_multiplicand_1(copy),
),
y.0,
);
self.connect(
Target::wire(gate_index, U32ArithmeticGate::<F, D>::wire_ith_addend(copy)),
z.0,
);
self.connect(
Target::wire(
gate_index,
U32ArithmeticGate::<F, D>::wire_ith_output_low_half(copy),
),
output_low.0,
);
self.connect(
Target::wire(
gate_index,
U32ArithmeticGate::<F, D>::wire_ith_output_high_half(copy),
),
output_high.0,
);
self.current_u32_arithmetic_gate = Some((gate_index, 0));
@ -65,7 +96,12 @@ impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
self.add_mul_u32(a, self.one_u32(), b)
}
pub fn add_three_u32(&mut self, a: U32Target, b: U32Target, c: U32Target) -> (U32Target, U32Target) {
pub fn add_three_u32(
&mut self,
a: U32Target,
b: U32Target,
c: U32Target,
) -> (U32Target, U32Target) {
let (init_low, carry1) = self.add_u32(a, b);
let (final_low, carry2) = self.add_u32(c, init_low);
let (combined_carry, _zero) = self.add_u32(carry1, carry2);

View File

@ -1,7 +1,8 @@
use num::bigint::BigUint;
use std::collections::BTreeMap;
use std::marker::PhantomData;
use num::bigint::BigUint;
use crate::field::field_types::RichField;
use crate::field::{extension_field::Extendable, field_types::Field};
use crate::gadgets::arithmetic_u32::U32Target;
@ -34,7 +35,7 @@ impl<F: RichField + Extendable<D>, const D: usize> CircuitBuilder<F, D> {
combined_limbs[i] = new_limb;
}
combined_limbs[num_limbs] = carry;
NonNativeTarget {
modulus,
limbs: combined_limbs,