flint-sys 0.9.0

Bindings to the FLINT C library
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

/*
    Copyright (C) 2023 Albin Ahlbäck

    This file is part of FLINT.

    FLINT is free software: you can redistribute it and/or modify it under
    the terms of the GNU Lesser General Public License (LGPL) as published
    by the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.  See <https://www.gnu.org/licenses/>.
*/

#include "mpn_extras.h"

#if FLINT_HAVE_ASSEMBLY_x86_64_adx

mp_limb_t flint_mpn_sqr_1(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_2(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_3(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_4(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_5(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_6(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_7(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_8(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_9(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_10(mp_ptr, mp_srcptr);

mp_limb_t flint_mpn_mul_7_2(mp_ptr, mp_srcptr, mp_srcptr);
mp_limb_t flint_mpn_mul_7_3(mp_ptr, mp_srcptr, mp_srcptr);
mp_limb_t flint_mpn_mul_7_4(mp_ptr, mp_srcptr, mp_srcptr);
mp_limb_t flint_mpn_mul_7_5(mp_ptr, mp_srcptr, mp_srcptr);
mp_limb_t flint_mpn_mul_7_6(mp_ptr, mp_srcptr, mp_srcptr);
mp_limb_t flint_mpn_mul_7_7(mp_ptr, mp_srcptr, mp_srcptr);
mp_limb_t flint_mpn_mul_8_7(mp_ptr, mp_srcptr, mp_srcptr);

static mp_limb_t flint_mpn_sqr_11(mp_ptr res, mp_srcptr x)
{
    mp_limb_t t[14];
    flint_mpn_sqr_7(res, x);
    flint_mpn_sqr_4(res + 14, x + 7);
    flint_mpn_mul_7_4(t, x, x + 7);
    t[11] = 0;
    t[12] = 0;
    t[13] = 0;
    res[21] += mpn_addmul_1(res + 7, t, 14, 2);
    return res[21];
}

static mp_limb_t flint_mpn_sqr_12(mp_ptr res, mp_srcptr x)
{
    mp_limb_t t[12];
    mp_limb_t cy;
    flint_mpn_sqr_7(res, x);
    flint_mpn_sqr_5(res + 14, x + 7);
    flint_mpn_mul_7_5(t, x, x + 7);
    cy = mpn_addmul_1(res + 7, t, 12, 2);
    mpn_add_1(res + 19, res + 19, 5, cy);
    return res[23];
}

static mp_limb_t flint_mpn_sqr_13(mp_ptr res, mp_srcptr x)
{
    mp_limb_t t[13];
    mp_limb_t cy;
    flint_mpn_sqr_7(res, x);
    flint_mpn_sqr_6(res + 14, x + 7);
    flint_mpn_mul_7_6(t, x, x + 7);
    cy = mpn_addmul_1(res + 7, t, 13, 2);
    mpn_add_1(res + 20, res + 20, 6, cy);
    return res[25];
}

static mp_limb_t flint_mpn_sqr_14(mp_ptr res, mp_srcptr x)
{
    mp_limb_t t[14];
    mp_limb_t cy;
    flint_mpn_sqr_7(res, x);
    flint_mpn_sqr_7(res + 14, x + 7);
    flint_mpn_mul_7_7(t, x, x + 7);
    cy = mpn_addmul_1(res + 7, t, 14, 2);
    mpn_add_1(res + 21, res + 21, 7, cy);
    return res[27];
}

const flint_mpn_sqr_func_t flint_mpn_sqr_func_tab[] = {
    NULL,
    flint_mpn_sqr_1,
    flint_mpn_sqr_2,
    flint_mpn_sqr_3,
    flint_mpn_sqr_4,
    flint_mpn_sqr_5,
    flint_mpn_sqr_6,
    flint_mpn_sqr_7,
    flint_mpn_sqr_8,
    flint_mpn_sqr_9,
    flint_mpn_sqr_10,
    flint_mpn_sqr_11,
    flint_mpn_sqr_12,
    flint_mpn_sqr_13,
    flint_mpn_sqr_14
};

#elif FLINT_HAVE_ASSEMBLY_armv8
mp_limb_t flint_mpn_sqr_1(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_2(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_3(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_4(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_5(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_6(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_7(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_8(mp_ptr, mp_srcptr);
mp_limb_t flint_mpn_sqr_9(mp_ptr, mp_srcptr);

const flint_mpn_sqr_func_t flint_mpn_sqr_func_tab[] =
{
    NULL,
    flint_mpn_sqr_1,
    flint_mpn_sqr_2,
    flint_mpn_sqr_3,
    flint_mpn_sqr_4,
    flint_mpn_sqr_5,
    flint_mpn_sqr_6,
    flint_mpn_sqr_7,
    flint_mpn_sqr_8,
    flint_mpn_sqr_9
};
#else

/* Currently generic C code performs worse than GMP. */

const flint_mpn_sqr_func_t flint_mpn_sqr_func_tab[] = {
    NULL,
};

#endif