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

/*
    Copyright (C) 2013 William Hart
    Copyright (C) 2025 Fredrik Johansson

    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 "test_helpers.h"
#include "mpn_extras.h"

TEST_FUNCTION_START(flint_mpn_mulmod_precond_matrix, state)
{
    int i;
    slong n, npre;

    mp_ptr a, apre, b, d, dnormed, dinv, r1, r2, t, u;
    ulong norm;

    for (i = 0; i < 10000 * flint_test_multiplier(); i++)
    {
        n = 2 + n_randint(state, 10);
        npre = flint_mpn_mulmod_precond_matrix_alloc(n);

        a = flint_malloc(sizeof(mp_limb_t) * n);
        apre = flint_malloc(sizeof(mp_limb_t) * npre);
        b = flint_malloc(sizeof(mp_limb_t) * n);
        d = flint_malloc(sizeof(mp_limb_t) * n);
        dnormed = flint_malloc(sizeof(mp_limb_t) * n);
        dinv = flint_malloc(sizeof(mp_limb_t) * n);
        t = flint_malloc(sizeof(mp_limb_t) * 2 * n);
        u = flint_malloc(sizeof(mp_limb_t) * (n + 1));
        r1 = flint_malloc(sizeof(mp_limb_t) * n);
        r2 = flint_malloc(sizeof(mp_limb_t) * n);

        flint_mpn_rrandom(d, state, n);
        while (d[n - 1] == 0)
            flint_mpn_rrandom(d + n - 1, state, 1);

        if (n_randint(state, 2))
            d[n - 1] |= (UWORD(1) << (FLINT_BITS - 1));

        norm = flint_clz(d[n - 1]);
        if (norm == 0)
            mpn_copyi(dnormed, d, n);
        else
            mpn_lshift(dnormed, d, n, norm);

        flint_mpn_preinvn(dinv, dnormed, n);

        /* reduce a, b mod d */
        flint_mpn_rrandom(a, state, n);
        mpn_tdiv_qr(t, a, 0, a, n, d, n);
        flint_mpn_rrandom(b, state, n);
        mpn_tdiv_qr(t, b, 0, b, n, d, n);

        mpn_mul_n(t, a, b, n);
        mpn_tdiv_qr(u, r1, 0, t, 2 * n, d, n);

        flint_mpn_mulmod_precond_matrix_precompute(apre, a, n, dnormed, dinv, norm);
        flint_mpn_mulmod_precond_matrix(r2, apre, b, n, dnormed, dinv, norm);

        if (mpn_cmp(r1, r2, n))
        {
            flint_printf("FAIL\n");
            flint_printf("n = %wd, norm = %wu\n", n, norm);
            flint_printf("d = "); flint_mpn_debug(d, n);
            flint_printf("a = "); flint_mpn_debug(a, n);
            flint_printf("apre = "); flint_mpn_debug(apre, npre);
            flint_printf("b = "); flint_mpn_debug(b, n);
            flint_printf("r1 = "); flint_mpn_debug(r1, n);
            flint_printf("r2 = "); flint_mpn_debug(r2, n);
            flint_abort();
        }

        flint_free(a);
        flint_free(apre);
        flint_free(b);
        flint_free(d);
        flint_free(dnormed);
        flint_free(dinv);
        flint_free(t);
        flint_free(u);
        flint_free(r1);
        flint_free(r2);
    }

    TEST_FUNCTION_END(state);
}