ProvableHQ/leo
1
1
Fork 0
mirror of https://github.com/ProvableHQ/leo.git synced 2024-09-22 11:17:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

perf: reduce allocations in signed_integer's CondSelectGadget impl

Browse Source 
Signed-off-by: ljedrz <ljedrz@gmail.com>
This commit is contained in:
ljedrz 2020-10-22 15:29:26 +02:00
parent f9e6b3caa8
commit 78f98f1f26
1 changed files with 8 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
gadgets/src/signed_integer/utilities/select.rs
View File

@ -51,23 +51,15 @@ macro_rules! select_int_impl {
let result = Self::alloc(cs.ns(|| "cond_select_result"), || result_val.get())?;
let expected_bits = first
.bits
.iter()
.zip(&second.bits)
.enumerate()
.map(|(i, (a, b))| {
Boolean::conditionally_select(
&mut cs.ns(|| format!("{}_cond_select_{}", <$gadget as Int>::SIZE, i)),
cond,
a,
b,
).unwrap()
})
.collect::<Vec<Boolean>>();
for (i, ((bit1, bit2), actual)) in first.bits.iter().zip(second.bits.iter()).zip(result.bits.iter()).enumerate() {
let expected = Boolean::conditionally_select(
&mut cs.ns(|| format!("{}_cond_select_{}", <$gadget as Int>::SIZE, i)),
cond,
bit1,
bit2,
).unwrap();
for (i, (actual, expected)) in result.bits.iter().zip(expected_bits.iter()).enumerate() {
actual.enforce_equal(&mut cs.ns(|| format!("selected_result_bit_{}", i)), expected)?;
actual.enforce_equal(&mut cs.ns(|| format!("selected_result_bit_{}", i)), &expected)?;
}
Ok(result)