forked from M-Labs/nalgebra
114 lines
3.3 KiB
Rust
114 lines
3.3 KiB
Rust
#![macro_use]
|
|
|
|
macro_rules! bench_binop(
|
|
($name: ident, $t1: ty, $t2: ty, $binop: ident) => {
|
|
fn $name(bh: &mut criterion::Criterion) {
|
|
use rand::SeedableRng;
|
|
let mut rng = IsaacRng::seed_from_u64(0);
|
|
let a = rng.gen::<$t1>();
|
|
let b = rng.gen::<$t2>();
|
|
|
|
bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
|
|
a.$binop(b)
|
|
}));
|
|
}
|
|
}
|
|
);
|
|
|
|
macro_rules! bench_binop_ref(
|
|
($name: ident, $t1: ty, $t2: ty, $binop: ident) => {
|
|
fn $name(bh: &mut criterion::Criterion) {
|
|
use rand::SeedableRng;
|
|
let mut rng = IsaacRng::seed_from_u64(0);
|
|
let a = rng.gen::<$t1>();
|
|
let b = rng.gen::<$t2>();
|
|
|
|
bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
|
|
a.$binop(&b)
|
|
}));
|
|
}
|
|
}
|
|
);
|
|
|
|
macro_rules! bench_binop_fn(
|
|
($name: ident, $t1: ty, $t2: ty, $binop: path) => {
|
|
fn $name(bh: &mut criterion::Criterion) {
|
|
use rand::SeedableRng;
|
|
let mut rng = IsaacRng::seed_from_u64(0);
|
|
let a = rng.gen::<$t1>();
|
|
let b = rng.gen::<$t2>();
|
|
|
|
bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
|
|
$binop(&a, &b)
|
|
}));
|
|
}
|
|
}
|
|
);
|
|
|
|
macro_rules! bench_unop_na(
|
|
($name: ident, $t: ty, $unop: ident) => {
|
|
fn $name(bh: &mut criterion::Criterion) {
|
|
const LEN: usize = 1 << 13;
|
|
|
|
use rand::SeedableRng;
|
|
let mut rng = IsaacRng::seed_from_u64(0);
|
|
|
|
let elems: Vec<$t> = (0usize .. LEN).map(|_| rng.gen::<$t>()).collect();
|
|
let mut i = 0;
|
|
|
|
bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
|
|
i = (i + 1) & (LEN - 1);
|
|
|
|
unsafe {
|
|
std::hint::black_box(na::$unop(elems.get_unchecked(i)))
|
|
}
|
|
}));
|
|
}
|
|
}
|
|
);
|
|
|
|
macro_rules! bench_unop(
|
|
($name: ident, $t: ty, $unop: ident) => {
|
|
fn $name(bh: &mut criterion::Criterion) {
|
|
const LEN: usize = 1 << 13;
|
|
|
|
use rand::SeedableRng;
|
|
let mut rng = IsaacRng::seed_from_u64(0);
|
|
|
|
let mut elems: Vec<$t> = (0usize .. LEN).map(|_| rng.gen::<$t>()).collect();
|
|
let mut i = 0;
|
|
|
|
bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
|
|
i = (i + 1) & (LEN - 1);
|
|
|
|
unsafe {
|
|
std::hint::black_box(elems.get_unchecked_mut(i).$unop())
|
|
}
|
|
}));
|
|
}
|
|
}
|
|
);
|
|
|
|
macro_rules! bench_construction(
|
|
($name: ident, $constructor: path, $( $args: ident: $types: ty),*) => {
|
|
fn $name(bh: &mut criterion::Criterion) {
|
|
const LEN: usize = 1 << 13;
|
|
|
|
use rand::SeedableRng;
|
|
let mut rng = IsaacRng::seed_from_u64(0);
|
|
|
|
$(let $args: Vec<$types> = (0usize .. LEN).map(|_| rng.gen::<$types>()).collect();)*
|
|
let mut i = 0;
|
|
|
|
bh.bench_function(stringify!($name), move |bh| bh.iter(|| {
|
|
i = (i + 1) & (LEN - 1);
|
|
|
|
unsafe {
|
|
let res = $constructor($(*$args.get_unchecked(i),)*);
|
|
std::hint::black_box(res)
|
|
}
|
|
}));
|
|
}
|
|
}
|
|
);
|