mirror of
https://github.com/arnaucube/poulpy.git
synced 2026-02-10 05:06:44 +01:00
wip
This commit is contained in:
Jean-Philippe Bossuat
122
math/benches/ntt.rs
Normal file
122
math/benches/ntt.rs
Normal file
@@ -0,0 +1,122 @@
|
||||
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
|
||||
use math::{modulus::prime::Prime,dft::ntt::Table};
|
||||
use math::dft::DFT;
|
||||
|
||||
fn forward_inplace(c: &mut Criterion) {
|
||||
fn runner(prime_instance: Prime<u64>, nth_root: u64) -> Box<dyn FnMut()> {
|
||||
let ntt_table: Table<u64> = Table::<u64>::new(prime_instance, nth_root);
|
||||
let mut a: Vec<u64> = vec![0; (nth_root >> 1) as usize];
|
||||
for i in 0..a.len(){
|
||||
a[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
ntt_table.forward_inplace::<false>(&mut a)
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("forward_inplace");
|
||||
for log_nth_root in 11..18 {
|
||||
|
||||
let prime_instance: Prime<u64> = Prime::<u64>::new(0x1fffffffffe00001, 1);
|
||||
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(prime_instance, 1<<log_nth_root)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, 1<<(log_nth_root-1));
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn forward_inplace_lazy(c: &mut Criterion) {
|
||||
fn runner(prime_instance: Prime<u64>, nth_root: u64) -> Box<dyn FnMut()> {
|
||||
let ntt_table: Table<u64> = Table::<u64>::new(prime_instance, nth_root);
|
||||
let mut a: Vec<u64> = vec![0; (nth_root >> 1) as usize];
|
||||
for i in 0..a.len(){
|
||||
a[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
ntt_table.forward_inplace_lazy(&mut a)
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("forward_inplace_lazy");
|
||||
for log_nth_root in 11..17 {
|
||||
|
||||
let prime_instance: Prime<u64> = Prime::<u64>::new(0x1fffffffffe00001, 1);
|
||||
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(prime_instance, 1<<log_nth_root)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, 1<<(log_nth_root-1));
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn backward_inplace(c: &mut Criterion) {
|
||||
fn runner(prime_instance: Prime<u64>, nth_root: u64) -> Box<dyn FnMut()> {
|
||||
let ntt_table: Table<u64> = Table::<u64>::new(prime_instance, nth_root);
|
||||
let mut a: Vec<u64> = vec![0; (nth_root >> 1) as usize];
|
||||
for i in 0..a.len(){
|
||||
a[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
ntt_table.backward_inplace::<false>(&mut a)
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("backward_inplace");
|
||||
for log_nth_root in 11..18 {
|
||||
|
||||
let prime_instance: Prime<u64> = Prime::<u64>::new(0x1fffffffffe00001, 1);
|
||||
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(prime_instance, 1<<log_nth_root)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, 1<<(log_nth_root-1));
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn backward_inplace_lazy(c: &mut Criterion) {
|
||||
fn runner(prime_instance: Prime<u64>, nth_root: u64) -> Box<dyn FnMut()> {
|
||||
let ntt_table: Table<u64> = Table::<u64>::new(prime_instance, nth_root);
|
||||
let mut a: Vec<u64> = vec![0; (nth_root >> 1) as usize];
|
||||
for i in 0..a.len(){
|
||||
a[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
ntt_table.backward_inplace::<true>(&mut a)
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("backward_inplace_lazy");
|
||||
for log_nth_root in 11..17 {
|
||||
|
||||
let prime_instance: Prime<u64> = Prime::<u64>::new(0x1fffffffffe00001, 1);
|
||||
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(prime_instance, 1<<log_nth_root)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, 1<<(log_nth_root-1));
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
criterion_group!(benches, forward_inplace, forward_inplace_lazy, backward_inplace, backward_inplace_lazy);
|
||||
criterion_main!(benches);
|
||||
109
math/benches/operations.rs
Normal file
109
math/benches/operations.rs
Normal file
@@ -0,0 +1,109 @@
|
||||
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
|
||||
use math::ring::Ring;
|
||||
use math::modulus::VecOperations;
|
||||
use math::modulus::montgomery::Montgomery;
|
||||
use math::modulus::ONCE;
|
||||
use math::CHUNK;
|
||||
|
||||
fn vec_add_unary(c: &mut Criterion) {
|
||||
fn runner(r: Ring<u64>) -> Box<dyn FnMut()> {
|
||||
|
||||
let mut p0: math::poly::Poly<u64> = r.new_poly();
|
||||
let mut p1: math::poly::Poly<u64> = r.new_poly();
|
||||
for i in 0..p0.n(){
|
||||
p0.0[i] = i as u64;
|
||||
p1.0[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
r.modulus.vec_add_unary_assign::<CHUNK, ONCE>(&p0.0, &mut p1.0);
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("add_vec_unary");
|
||||
for log_n in 11..17 {
|
||||
|
||||
let n: usize = 1<<log_n as usize;
|
||||
let q_base: u64 = 0x1fffffffffe00001u64;
|
||||
let q_power: usize = 1usize;
|
||||
let r: Ring<u64> = Ring::<u64>::new(n, q_base, q_power);
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(r)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, n);
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn vec_mul_montgomery_external_unary_assign(c: &mut Criterion) {
|
||||
fn runner(r: Ring<u64>) -> Box<dyn FnMut()> {
|
||||
|
||||
let mut p0: math::poly::Poly<Montgomery<u64>> = r.new_poly();
|
||||
let mut p1: math::poly::Poly<u64> = r.new_poly();
|
||||
for i in 0..p0.n(){
|
||||
p0.0[i] = r.modulus.montgomery.prepare::<ONCE>(i as u64);
|
||||
p1.0[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
r.modulus.vec_mul_montgomery_external_unary_assign::<CHUNK, ONCE>(&p0.0, &mut p1.0);
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("mul_vec_montgomery_external_unary_assign");
|
||||
for log_n in 11..17 {
|
||||
|
||||
let n: usize = 1<<log_n as usize;
|
||||
let q_base: u64 = 0x1fffffffffe00001u64;
|
||||
let q_power: usize = 1usize;
|
||||
let r: Ring<u64> = Ring::<u64>::new(n, q_base, q_power);
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(r)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, n);
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn vec_mul_montgomery_external_binary_assign(c: &mut Criterion) {
|
||||
fn runner(r: Ring<u64>) -> Box<dyn FnMut()> {
|
||||
|
||||
let mut p0: math::poly::Poly<Montgomery<u64>> = r.new_poly();
|
||||
let mut p1: math::poly::Poly<u64> = r.new_poly();
|
||||
let mut p2: math::poly::Poly<u64> = r.new_poly();
|
||||
for i in 0..p0.n(){
|
||||
p0.0[i] = r.modulus.montgomery.prepare::<ONCE>(i as u64);
|
||||
p1.0[i] = i as u64;
|
||||
}
|
||||
Box::new(move || {
|
||||
r.modulus.vec_mul_montgomery_external_binary_assign::<CHUNK,ONCE>(&p0.0, & p1.0, &mut p2.0);
|
||||
})
|
||||
}
|
||||
|
||||
let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> = c.benchmark_group("mul_vec_montgomery_external_binary_assign");
|
||||
for log_n in 11..17 {
|
||||
|
||||
let n: usize = 1<<log_n as usize;
|
||||
let q_base: u64 = 0x1fffffffffe00001u64;
|
||||
let q_power: usize = 1usize;
|
||||
let r: Ring<u64> = Ring::<u64>::new(n, q_base, q_power);
|
||||
let runners = [
|
||||
("prime", {
|
||||
runner(r)
|
||||
}),
|
||||
];
|
||||
for (name, mut runner) in runners {
|
||||
let id = BenchmarkId::new(name, n);
|
||||
b.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
criterion_group!(benches, vec_add_unary, vec_mul_montgomery_external_unary_assign, vec_mul_montgomery_external_binary_assign);
|
||||
criterion_main!(benches);
|
||||
Reference in New Issue
Block a user