zkevm_circuits 0.153.9

ZKsync Era circuits for EraVM
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

use super::*;
use boojum::cs::implementations::lookup_table::LookupTable;
use boojum::field::SmallField;

pub const REG_IDX_TO_BITMASK_TABLE_NAME: &'static str = "Register index to bitmask table";
pub const UMA_SHIFT_TO_BITMASK_TABLE_NAME: &'static str = "UMA shift to bitmask table";
pub const VM_SUBPC_TO_BITMASK_TABLE_NAME: &'static str = "Sub PC to bitmask table";

#[derive(Derivative)]
#[derivative(Clone, Copy, Debug, PartialEq, Eq)]
pub struct RegisterIndexToBitmaskTable;

#[derive(Derivative)]
#[derivative(Clone, Copy, Debug, PartialEq, Eq)]
pub struct UMAShiftToBitmaskTable;

#[derive(Derivative)]
#[derivative(Clone, Copy, Debug, PartialEq, Eq)]
pub struct VMSubPCToBitmaskTable;

pub fn create_integer_to_bitmask_table<F: SmallField>(
    num_bits: usize,
    name: &'static str,
) -> LookupTable<F, 3> {
    assert!(num_bits <= 16);
    let mut all_keys = Vec::with_capacity(1 << num_bits);
    for integer in 0..(1u64 << num_bits) {
        let key = smallvec::smallvec![F::from_u64_unchecked(integer as u64)];
        all_keys.push(key);
    }

    LookupTable::new_from_keys_and_generation_function(&all_keys, name.to_string(), 1, |keys| {
        let a = keys[0].as_u64_reduced();

        let result = if a == 0 {
            0u64
        } else {
            1u64 << (a - 1) // 1 in some position
        };

        smallvec::smallvec![F::from_u64_unchecked(result), F::ZERO]
    })
}

pub fn create_integer_set_ith_bit_table<F: SmallField>(
    num_bits: usize,
    name: &'static str,
) -> LookupTable<F, 3> {
    assert!(num_bits <= 16);
    let mut all_keys = Vec::with_capacity(1 << num_bits);
    for integer in 0..(1u64 << num_bits) {
        let key = smallvec::smallvec![F::from_u64_unchecked(integer as u64)];
        all_keys.push(key);
    }

    LookupTable::new_from_keys_and_generation_function(&all_keys, name.to_string(), 1, |keys| {
        let a = keys[0].as_u64_reduced();

        let result = 1u64 << a; // 1 in some position

        smallvec::smallvec![F::from_u64_unchecked(result), F::ZERO]
    })
}

pub fn create_subpc_bitmask_table<F: SmallField>() -> LookupTable<F, 3> {
    create_integer_to_bitmask_table(2, VM_SUBPC_TO_BITMASK_TABLE_NAME)

    // let num_bits = 2;
    // let mut all_keys = Vec::with_capacity(1 << num_bits);
    // for integer in 0..(1u64 << num_bits) {
    //     let key = smallvec::smallvec![F::from_u64_unchecked(integer as u64)];
    //     all_keys.push(key);
    // }

    // LookupTable::new_from_keys_and_generation_function(
    //     &all_keys,
    //     VM_SUBPC_TO_BITMASK_TABLE_NAME.to_string(),
    //     1,
    //     |keys| {
    //         let a = keys[0].as_u64_reduced();

    //         let result = if a == 0 {
    //             0u64
    //         } else {
    //             1u64 << (a - 1) // 1 in some position
    //         };

    //         smallvec::smallvec![F::from_u64_unchecked(result), F::ZERO]
    //     },
    // )
}