use gategen::boolvar::*;
use gategen::gatesim::*;
use gategen::intvar::*;
use gatenative::cpu_build_exec::*;
use gatenative::mapper::*;
use gatenative::*;
fn mul_add_circuit() -> Circuit<u32> {
call32(|| {
let a = U10Var32::var();
let b = U10Var32::var();
let c = U10Var32::var();
let r = &a * &b + &c;
r.to_translated_circuit(a.concat(b).concat(c).iter())
})
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let circuit = mul_add_circuit();
let builder = CPUBuilder::new(None);
let mut builder = CPUBasicMapperBuilder::new(builder);
builder.transform_helpers();
builder.add_with_config(
"mul_add",
circuit,
CodeConfig::new()
.arg_inputs(Some(&(24..30).collect::<Vec<_>>()))
.elem_inputs(Some(&(0..24).collect::<Vec<_>>()))
.aggr_output_code(Some(
r##"{
size_t i, j;
// buf are accumulators
uint32_t* buf = (uint32_t*)output;
uint32_t temp[TYPE_LEN];
// transform to 32-bit word array.
OUTPUT_TRANSFORM_B10(temp, o0, o1, o2, o3, o4, o5, o6, o7, o8, o9);
for (i = 0; i < TYPE_LEN; i += 16)
for (j = 0; j < 16; j++)
__sync_fetch_and_add(buf + j, temp[i + j]);
}"##,
))
.aggr_output_len(Some(16)),
);
let mut execs = builder.build()?;
let input = execs[0].new_data(16);
let output = execs[0].execute_direct(
&input,
0u64,
|sum, _, output, arg| {
eprintln!("Arg Input: {}", arg);
sum + output.into_iter().map(|x| u64::from(*x)).sum::<u64>()
},
|_| false,
)?;
println!("Sum: {}", output);
Ok(())
}