use gategen::boolvar::*;
use gategen::gatesim::*;
use gategen::intvar::*;
use gatenative::{opencl_build_exec::*, *};
use opencl3::device::{get_all_devices, Device, CL_DEVICE_TYPE_GPU};
fn mul_add_circuit() -> Circuit<u32> {
call32(|| {
let a = U16Var32::var();
let b = U16Var32::var();
let c = U16Var32::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 device = Device::new(
*get_all_devices(CL_DEVICE_TYPE_GPU)?
.get(0)
.expect("No OpenCL devices"),
);
let circuit = mul_add_circuit();
let mut builder = OpenCLBuilder::new(&device, None);
builder.add_simple("mul_add", circuit);
let mut execs = builder.build()?;
let mut it = execs[0].input_transformer(
96,
&((0..16).chain(32..48).chain(64..80).collect::<Vec<_>>()),
)?;
let mut ot = execs[0].output_transformer(32, &((0..16).collect::<Vec<_>>()))?;
let input = execs[0].new_data_from_vec(
(0..4096u32)
.map(|x| [(x + 3) & 0xffff, (x + 1489u32) & 0xffff, (5 * x) & 0xffff])
.flatten()
.collect::<Vec<_>>(),
);
let input = it.transform(&input)?;
let output = execs[0].execute(&input, 0)?;
let output = ot.transform(&output)?;
let output = output.release();
for (i, v) in output.into_iter().enumerate() {
println!("{}: {}", i, v);
}
Ok(())
}