gatenative 0.2.0

The library to execute natively Gate circuits.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use gategen::boolvar::*;
use gategen::gatesim::*;
use gategen::intvar::*;
use gatenative::mapper::*;
use gatenative::opencl_build_exec::*;
use gatenative::*;
use opencl3::device::{get_all_devices, Device, CL_DEVICE_TYPE_GPU};

// generate circuit
fn mul_add_circuit() -> Circuit<u32> {
    call32(|| {
        let a = U10Var32::var();
        let b = U10Var32::var();
        let c = U10Var32::var();
        let r = &a * &b + &c;
        // Circuit has 30-bit input divided into:
        // 0..10 - 'a' argument
        // 10..20 - 'b' argument
        // 20..30 - 'c' argument
        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"),
    );
    // Create circuit.
    let circuit = mul_add_circuit();
    // Create builder.
    let builder = OpenCLBuilder::new(&device, None);
    // Create basic mapper builder.
    let mut builder = OpenCLBasicMapperBuilder::new(builder);
    // Put transform helpers to code
    builder.transform_helpers();
    // Add circuit to builder.
    builder.add_with_config(
        "mul_add",
        circuit,
        CodeConfig::new()
            // Assign circuit's 6 highest bits to arg input.
            .arg_inputs(Some(&(24..30).collect::<Vec<_>>()))
            // Assign circuit 24 lowest bits to element index.
            .elem_inputs(Some(&(0..24).collect::<Vec<_>>()))
            // Sets aggr output code that sums all results.
            // Because is possible overflow then uses 16 differents accumulators
            .aggr_output_code(Some(
                r##"{
    size_t i, j;
    // buf are accumulators
    global uint* buf = (global uint*)output;
    uint 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++)
            atomic_add(buf + j, temp[i + j]);
}"##,
            ))
            // Sets aggregated output with length 16 32-bit words
            .aggr_output_len(Some(16)),
    );
    let mut execs = builder.build()?;
    // Prepare empty input for execution.
    let input = execs[0].new_data(16);
    // Execute simulation for all combinations of inputs.
    let output = execs[0].execute_direct(
        &input,
        0u64,
        // just sum all outputs
        |sum, _, output, arg| {
            eprintln!("Arg Input: {}", arg);
            // Make sum of array elements.
            sum + output.into_iter().map(|x| u64::from(*x)).sum::<u64>()
        },
        // No stop.
        |_| false,
    )?;
    println!("Sum: {}", output);
    Ok(())
}