12345678910111213141516171819 |
- use spiral_rs::poly::*;
- use spiral_rs::params::*;
- use spiral_rs::util::*;
- use rand::Rng;
- use rand::distributions::Standard;
- use criterion::{black_box, criterion_group, criterion_main, Criterion};
- fn criterion_benchmark(c: &mut Criterion) {
- let params = Params::init(2048, &vec![268369921u64, 249561089u64]);
- let mut rng = rand::thread_rng();
- let mut iter = rng.sample_iter(&Standard);
- let m1 = PolyMatrixNTT::random(¶ms, 2, 1, &mut iter);
- let m2 = PolyMatrixNTT::random(¶ms, 3, 2, &mut iter);
- let mut m3 = PolyMatrixNTT::zero(¶ms, 2, 2);
- c.bench_function("nttf 2048", |b| b.iter(|| multiply(black_box(&mut m3), black_box(&m1), black_box(&m2))));
- }
- criterion_group!(benches, criterion_benchmark);
- criterion_main!(benches);
|