gatenative 0.2.2

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
84
85
86
87
88
89
90
91
92
93
94

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};

// generate circuit
fn mul_add_circuit() -> Circuit<u32> {
    call32(|| {
        let a = U16Var32::var();
        let b = U16Var32::var();
        let c = U16Var32::var();
        let r = &a * &b + &c;
        // Circuit has 48-bit input divided into:
        // 0..16 - 'a' argument
        // 16..32 - 'b' argument
        // 32..48 - '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"),
    );
    // Set argument constants.
    let b_start = 4782u16;
    let c_start = 18941u16;
    // Set argument constants.
    let param_a = 0x4da0u16;
    let param_b = 0x8725u16;
    let param_c = 0xe052u16;
    // Create circuit.
    let circuit = mul_add_circuit();
    // Create circuit.
    let mut builder = OpenCLBuilder::new(&device, None);
    // Put transform helpers to code
    builder.transform_helpers();
    // Add circuit to builder.
    builder.add_with_config(
        "mul_add",
        circuit,
        CodeConfig::new()
            // Assign circuit's inputs 'b' and 'c' to arg input ('c' higher).
            .arg_inputs(Some(&(16..48).collect::<Vec<_>>()))
            // Set pop input buffer (in input data) to 3 words
            .pop_input_len(Some(3))
            // Sets pop_input_code that populate 'a' input for circuit from 3 parameters
            // stored in input data and transforms input to circuit inputs.
            .pop_input_code(Some(
                r##"{
    // Get paramareters from input
    const global uint* params = (const global uint*)input;
    const uint param_a = params[0];
    const uint param_b = params[1];
    const uint param_c = params[2];
    size_t i;
    uint temp[TYPE_LEN];
    // generate values of function: (x xor param_a) + (x and param_b) + (x or param_c).
    for (i = 0; i < TYPE_LEN; i++) {
        const uint id = idx*TYPE_LEN + i;
        temp[i] = ((id^param_a) + (id&param_b) + (id|param_c)) & 0xffff;
    }
    INPUT_TRANSFORM_B16(i0, i1, i2, i3, i4, i5, i6, i7,
                        i8, i9, i10, i11, i12, i13, i14, i15, temp);
}"##,
            ))
            // Sets aggr_output_len to 2**16 32-bit words.
            .aggr_output_len(Some(1 << 16))
            // Sets aggr output code that transforms output.
            // Output will be stored to output data (and that is additional buffer).
            .aggr_output_code(Some(
                r##"{
    global uint* output_u32 = (global uint*)output;
    OUTPUT_TRANSFORM_B16(output_u32 + TYPE_LEN*idx, o0, o1, o2, o3, o4, o5, o6, o7,
                        o8, o9, o10, o11, o12, o13, o14, o15);
}"##,
            )),
    );
    let mut execs = builder.build()?;
    // Prepare empty input for execution.
    let input = execs[0].new_data_from_slice(&[param_a.into(), param_b.into(), param_c.into()]);
    // Execute simulation. Set 'b' to b_start and 'c' to c_start by arg input.
    let output = execs[0].execute(&input, ((c_start as u64) << 16) | (b_start as u64))?;
    // Release output data holder - just get its data.
    let output = output.release();
    // Print that data
    for (i, v) in output.into_iter().enumerate() {
        println!("{}: {}", i, v);
    }
    Ok(())
}