int
main(void)
{
long l, len = 20;
long runs[] = {
100000000, 1000000, 1000000, 1000000, 100000,
100000, 10000, 10000, 10000, 1000,
100, 100, 10, 1, 1,
1, 1, 1, 1, 1
};
long N[] = {
1, 2, 4, 8, 16,
32, 64, 128, 256, 512,
1024, WORD(1) << 11, WORD(1) << 12, WORD(1) << 13, WORD(1) << 14,
WORD(1) << 15, WORD(1) << 16, WORD(1) << 17, WORD(1) << 18, WORD(1) << 19
};
long T[20] = {0};
flint_printf("Benchmark for p-adic exponential (rectangular).\n");
fflush(stdout);
for (l = 0; l < FLINT_MIN(17, len); l++)
{
FLINT_TEST_INIT(state);
long n = N[l], r;
clock_t c0, c1;
long double cputime;
fmpz_t p;
padic_ctx_t ctx;
padic_t d, z;
fmpz_init_set_ui(p, 17);
padic_ctx_init(ctx, p, n, n, PADIC_VAL_UNIT);
padic_init(d);
padic_init(z);
if (n > 1)
{
fmpz_t f = {WORD(3)}, pow;
fmpz_init(pow);
fmpz_pow_ui(pow, p, n - 1);
fmpz_pow_ui(padic_unit(d), f, 3 * n);
fmpz_mod(padic_unit(d), padic_unit(d), pow);
padic_val(d) = 1;
fmpz_clear(pow);
}
c0 = clock();
for (r = runs[l]; (r); r--)
{
padic_exp_rectangular(z, d, ctx);
padic_zero(z);
}
c1 = clock();
cputime = (long double) (c1 - c0) / (long double) CLOCKS_PER_SEC;
T[l] = (slong) (cputime * (1000000000 / runs[l]));
flint_printf("%2ld, %4XYXYXYXY, %9ld, %wd\n",
l, cputime, runs[l], T[l]);
padic_clear(d);
padic_clear(z);
fmpz_clear(p);
padic_ctx_clear(ctx);
flint_randclear(state);
}
flint_printf("Output as a list:\n");
for (l = 0; l < len; l++)
flint_printf("%wd, ", T[l]);
flint_printf("\n");
}
int
main(void)
{
int i, result;
flint_rand_t state;
printf("exp_rectangular... ");
fflush(stdout);
flint_randinit(state);
/** p == 2 *******************************************************************/
/* Check aliasing: a = exp(a) */
for (i = 0; i < 10000; i++)
{
fmpz_t p;
long N;
padic_ctx_t ctx;
padic_t a, b;
int ans1, ans2;
fmpz_init(p);
fmpz_set_ui(p, 2);
N = n_randint(state, 50) + 1;
padic_ctx_init(ctx, p, N, PADIC_SERIES);
padic_init(a, ctx);
padic_init(b, ctx);
padic_randtest(a, state, ctx);
ans1 = padic_exp_rectangular(b, a, ctx);
ans2 = padic_exp_rectangular(a, a, ctx);
result = ((ans1 == ans2) && (!ans1 || padic_equal(a, b, ctx)));
if (!result)
{
printf("FAIL (aliasing):\n\n");
printf("a = "), padic_print(a, ctx), printf("\n");
printf("b = "), padic_print(b, ctx), printf("\n");
printf("ans1 = %d\n", ans1);
printf("ans2 = %d\n", ans2);
abort();
}
padic_clear(a, ctx);
padic_clear(b, ctx);
fmpz_clear(p);
padic_ctx_clear(ctx);
}
/* Check aliasing: exp(a + b) == exp(a) exp(b) */
for (i = 0; i < 10000; i++)
{
fmpz_t p;
long N;
padic_ctx_t ctx;
padic_t a, b, c, d, e, f, g;
int ans1, ans2, ans3;
fmpz_init(p);
fmpz_set_ui(p, 2);
N = n_randint(state, 10) + 1;
padic_ctx_init(ctx, p, N, PADIC_SERIES);
padic_init(a, ctx);
padic_init(b, ctx);
padic_init(c, ctx);
padic_init(d, ctx);
padic_init(e, ctx);
padic_init(f, ctx);
padic_init(g, ctx);
padic_randtest(a, state, ctx);
padic_randtest(b, state, ctx);
padic_add(c, a, b, ctx);
ans1 = padic_exp_rectangular(d, a, ctx);
ans2 = padic_exp_rectangular(e, b, ctx);
padic_mul(f, d, e, ctx);
ans3 = padic_exp_rectangular(g, c, ctx);
result = (!ans1 || !ans2 || (ans3 && padic_equal(f, g, ctx)));
if (!result)
{
printf("FAIL (functional equation):\n\n");
printf("a = "), padic_print(a, ctx), printf("\n");
printf("b = "), padic_print(b, ctx), printf("\n");
printf("c = a + b = "), padic_print(c, ctx), printf("\n");
printf("d = exp(a) = "), padic_print(d, ctx), printf("\n");
printf("e = exp(b) = "), padic_print(e, ctx), printf("\n");
printf("f = exp(a) exp(b) = "), padic_print(f, ctx), printf("\n");
printf("g = exp(a + b) = "), padic_print(g, ctx), printf("\n");
abort();
}
padic_clear(a, ctx);
padic_clear(b, ctx);
padic_clear(c, ctx);
padic_clear(d, ctx);
padic_clear(e, ctx);
padic_clear(f, ctx);
padic_clear(g, ctx);
fmpz_clear(p);
padic_ctx_clear(ctx);
}
/** p > 2 ********************************************************************/
/* Check aliasing: a = exp(a) */
for (i = 0; i < 10000; i++)
{
fmpz_t p;
long N;
padic_ctx_t ctx;
padic_t a, b;
int ans1, ans2;
fmpz_init(p);
fmpz_set_ui(p, n_randprime(state, 5, 1));
N = n_randint(state, 50) + 1;
padic_ctx_init(ctx, p, N, PADIC_SERIES);
padic_init(a, ctx);
padic_init(b, ctx);
padic_randtest(a, state, ctx);
ans1 = padic_exp_rectangular(b, a, ctx);
ans2 = padic_exp_rectangular(a, a, ctx);
result = ((ans1 == ans2) && (!ans1 || padic_equal(a, b, ctx)));
if (!result)
{
printf("FAIL (aliasing):\n\n");
printf("a = "), padic_print(a, ctx), printf("\n");
printf("b = "), padic_print(b, ctx), printf("\n");
printf("ans1 = %d\n", ans1);
printf("ans2 = %d\n", ans2);
abort();
}
padic_clear(a, ctx);
padic_clear(b, ctx);
fmpz_clear(p);
padic_ctx_clear(ctx);
}
/* Check aliasing: exp(a + b) == exp(a) exp(b) */
for (i = 0; i < 10000; i++)
{
fmpz_t p;
long N;
padic_ctx_t ctx;
padic_t a, b, c, d, e, f, g;
int ans1, ans2, ans3;
fmpz_init(p);
fmpz_set_ui(p, n_randprime(state, 5, 1));
N = n_randint(state, 10) + 1;
padic_ctx_init(ctx, p, N, PADIC_SERIES);
padic_init(a, ctx);
padic_init(b, ctx);
padic_init(c, ctx);
padic_init(d, ctx);
padic_init(e, ctx);
padic_init(f, ctx);
padic_init(g, ctx);
padic_randtest(a, state, ctx);
padic_randtest(b, state, ctx);
padic_add(c, a, b, ctx);
ans1 = padic_exp_rectangular(d, a, ctx);
ans2 = padic_exp_rectangular(e, b, ctx);
padic_mul(f, d, e, ctx);
ans3 = padic_exp_rectangular(g, c, ctx);
result = (!ans1 || !ans2 || (ans3 && padic_equal(f, g, ctx)));
if (!result)
{
printf("FAIL (functional equation):\n\n");
printf("a = "), padic_print(a, ctx), printf("\n");
printf("b = "), padic_print(b, ctx), printf("\n");
printf("c = a + b = "), padic_print(c, ctx), printf("\n");
printf("d = exp(a) = "), padic_print(d, ctx), printf("\n");
printf("e = exp(b) = "), padic_print(e, ctx), printf("\n");
printf("f = exp(a) exp(b) = "), padic_print(f, ctx), printf("\n");
printf("g = exp(a + b) = "), padic_print(g, ctx), printf("\n");
abort();
}
padic_clear(a, ctx);
padic_clear(b, ctx);
padic_clear(c, ctx);
padic_clear(d, ctx);
padic_clear(e, ctx);
padic_clear(f, ctx);
padic_clear(g, ctx);
fmpz_clear(p);
padic_ctx_clear(ctx);
}
flint_randclear(state);
_fmpz_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
