You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.2 KiB
100 lines
3.2 KiB
4 years ago
|
#include <stdio.h>
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <cybozu/inttype.hpp>
|
||
|
|
#include <cybozu/benchmark.hpp>
|
||
|
|
#include <cybozu/xorshift.hpp>
|
||
|
|
|
||
|
|
typedef size_t Unit;
|
||
|
|
|
||
|
|
template<size_t N>
|
||
|
|
void mulPre(Unit*, const Unit*, const Unit*);
|
||
|
|
|
||
|
|
template<size_t N>
|
||
|
|
void sqrPre(Unit*, const Unit*);
|
||
|
|
|
||
|
|
template<size_t N>
|
||
|
|
void mod(Unit*, const Unit*, const Unit *);
|
||
|
|
|
||
|
|
template<size_t N>
|
||
|
|
void mont(Unit*, const Unit*, const Unit*, const Unit *);
|
||
|
|
|
||
|
|
#define MCL_FP_DEF_FUNC_SUB(n, suf) \
|
||
|
|
extern "C" { \
|
||
|
|
void mcl_fp_add ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fp_addNF ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fp_sub ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fp_subNF ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fp_shr1_ ## n ## suf(Unit*y, const Unit* x); \
|
||
|
|
Unit mcl_fp_addPre ## n ## suf(Unit* z, const Unit* x, const Unit* y); \
|
||
|
|
Unit mcl_fp_subPre ## n ## suf(Unit* z, const Unit* x, const Unit* y); \
|
||
|
|
void mcl_fp_mulUnitPre ## n ## suf(Unit* z, const Unit* x, Unit y); \
|
||
|
|
void mcl_fpDbl_mulPre ## n ## suf(Unit* z, const Unit* x, const Unit* y); \
|
||
|
|
void mcl_fpDbl_sqrPre ## n ## suf(Unit* y, const Unit* x); \
|
||
|
|
void mcl_fp_mont ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fp_montNF ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fp_montRed ## n ## suf(Unit* z, const Unit* xy, const Unit* p); \
|
||
|
|
void mcl_fp_montRedNF ## n ## suf(Unit* z, const Unit* xy, const Unit* p); \
|
||
|
|
void mcl_fpDbl_add ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
void mcl_fpDbl_sub ## n ## suf(Unit* z, const Unit* x, const Unit* y, const Unit* p); \
|
||
|
|
} \
|
||
|
|
template<>void mulPre<n>(Unit *z, const Unit *x, const Unit *y) { mcl_fpDbl_mulPre ## n ## suf(z, x, y); } \
|
||
|
|
template<>void sqrPre<n>(Unit *z, const Unit *x) { mcl_fpDbl_sqrPre ## n ## suf(z, x); } \
|
||
|
|
template<>void mod<n>(Unit *z, const Unit *x, const Unit *p) { mcl_fp_montRedNF ## n ## suf(z, x, p); } \
|
||
|
|
template<>void mont<n>(Unit *z, const Unit *x, const Unit *y, const Unit *p) { mcl_fp_montNF ## n ## suf(z, x, y, p); }
|
||
|
|
|
||
|
|
MCL_FP_DEF_FUNC_SUB(4, L)
|
||
|
|
MCL_FP_DEF_FUNC_SUB(5, L)
|
||
|
|
MCL_FP_DEF_FUNC_SUB(6, L)
|
||
|
|
MCL_FP_DEF_FUNC_SUB(7, L)
|
||
|
|
MCL_FP_DEF_FUNC_SUB(8, L)
|
||
|
|
#if CYBOZU_OS_BIT == 32
|
||
|
|
MCL_FP_DEF_FUNC_SUB(12)
|
||
|
|
MCL_FP_DEF_FUNC_SUB(16)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
template<class RG, class T>
|
||
|
|
void setRand(T *x, size_t n, RG& rg)
|
||
|
|
{
|
||
|
|
for (size_t i = 0; i < n; i++) {
|
||
|
|
if (sizeof(T) == 4) {
|
||
|
|
x[i] = rg.get32();
|
||
|
|
} else {
|
||
|
|
x[i] = rg.get64();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
template<size_t N>
|
||
|
|
void bench(Unit *x, Unit *y, const Unit *p)
|
||
|
|
{
|
||
|
|
printf("N=%zd\n", N);
|
||
|
|
Unit xx[N * 2], yy[N * 2];
|
||
|
|
const int C = 1000;
|
||
|
|
CYBOZU_BENCH_C("mulPre", C, mulPre<N>, xx, x, y);
|
||
|
|
CYBOZU_BENCH_C("sqrPre", C, sqrPre<N>, yy, x);
|
||
|
|
CYBOZU_BENCH_C("mod ", C, mod<N>, yy, xx, p);
|
||
|
|
CYBOZU_BENCH_C("mont ", C, mont<N>, yy, x, y, p);
|
||
|
|
}
|
||
|
|
|
||
|
|
int main()
|
||
|
|
{
|
||
|
|
printf("sizeof(Unit)=%zd\n", sizeof(Unit));
|
||
|
|
const size_t maxN = 16;
|
||
|
|
Unit x[maxN], y[maxN], p[maxN + 1];
|
||
|
|
cybozu::XorShift rg;
|
||
|
|
setRand(x, maxN, rg);
|
||
|
|
setRand(y, maxN, rg);
|
||
|
|
setRand(p, maxN + 1, rg);
|
||
|
|
bench<4>(x, y, p + 1);
|
||
|
|
bench<5>(x, y, p + 1);
|
||
|
|
bench<6>(x, y, p + 1);
|
||
|
|
bench<7>(x, y, p + 1);
|
||
|
|
bench<8>(x, y, p + 1);
|
||
|
|
#if CYBOZU_OS_BIT == 32
|
||
|
|
bench<12>(x, y, p + 1);
|
||
|
|
bench<16>(x, y, p + 1);
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|