ark-ff-macros 0.4.0-alpha

A library for generating x86-64 assembly for finite field multiplication
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

use quote::quote;

pub(super) fn add_with_carry_impl(num_limbs: usize) -> proc_macro2::TokenStream {
    let mut body = proc_macro2::TokenStream::new();
    body.extend(quote! {
        use ark_ff::biginteger::arithmetic::adc_for_add_with_carry as adc;
        let mut carry = 0;
    });
    for i in 0..num_limbs {
        body.extend(quote! {
            carry = adc(&mut a.0[#i], b.0[#i], carry);
        });
    }
    body.extend(quote! {
        carry != 0
    });
    quote! {
        #[inline(always)]
        fn __add_with_carry(
            a: &mut B,
            b: & B,
        ) -> bool {
            #body
        }
    }
    .into()
}

pub(super) fn sub_with_borrow_impl(num_limbs: usize) -> proc_macro2::TokenStream {
    let mut body = proc_macro2::TokenStream::new();
    body.extend(quote! {
        use ark_ff::biginteger::arithmetic::sbb_for_sub_with_borrow as sbb;
        let mut borrow = 0;
    });
    for i in 0..num_limbs {
        body.extend(quote! {
            borrow = sbb(&mut a.0[#i], b.0[#i], borrow);
        });
    }
    body.extend(quote! {
        borrow != 0
    });
    quote! {
        #[inline(always)]
        fn __sub_with_borrow(
            a: &mut B,
            b: & B,
        ) -> bool {
            #body
        }
    }
    .into()
}

pub(super) fn subtract_modulus_impl(
    modulus: &proc_macro2::TokenStream,
) -> proc_macro2::TokenStream {
    quote! {
        #[inline(always)]
        fn __subtract_modulus(a: &mut F) {
            if a.is_geq_modulus() {
                __sub_with_borrow(&mut a.0, &#modulus);
            }
        }
    }
}