rs-basic/members/revsqrt/benches/rsqrt-bench.rs

60 lines
1.6 KiB
Rust

use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion, Throughput};
use revsqrt::{fast_inverse_sqrt, regular_inverse_sqrt};
const SIZE: f32 = 1337.1337;
const FCONST: f32 = 1_024.124_8;
const FCONST1: f32 = 4_025.724_9;
pub fn single_input(c: &mut Criterion) {
c.bench_with_input(BenchmarkId::new("regular rsqrt", SIZE), &SIZE, |b, &s| {
b.iter(|| regular_inverse_sqrt(s))
});
c.bench_with_input(BenchmarkId::new("fast rsqrt", SIZE), &SIZE, |b, &s| {
b.iter(|| fast_inverse_sqrt(s))
});
}
pub fn multi_input(c: &mut Criterion) {
let mut group = c.benchmark_group("multi_input");
for size in [
FCONST,
2.2 * FCONST,
4.24 * FCONST,
8.64 * FCONST,
16.12 * FCONST,
]
.iter()
{
group.throughput(Throughput::Bytes(*size as u64));
group.bench_with_input(
BenchmarkId::new("regular rsqrt mixed input", FCONST),
size,
|b, &size| {
b.iter(|| regular_inverse_sqrt(size));
},
);
}
for size in [
FCONST1,
2.2 * FCONST1,
4.24 * FCONST1,
8.64 * FCONST1,
16.12 * FCONST1,
]
.iter()
{
group.throughput(Throughput::Bytes(*size as u64));
group.bench_with_input(
BenchmarkId::new("fast rsqrt mixed input", FCONST1),
size,
|b, &size| {
b.iter(|| fast_inverse_sqrt(size));
},
);
}
group.finish();
}
criterion_group!(benches, single_input, multi_input);
criterion_main!(benches);