Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

bench: use black box on integer test, change for macros #360

Merged
merged 1 commit into from
Oct 28, 2024
Merged

bench: use black box on integer test, change for macros #360

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
86 changes: 36 additions & 50 deletions benches/integer.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,10 @@
// Modifications:
// Adds other rasn codecs to the benchmark
// Adds optional and extended fields for `Color`
// Use macros to generate the tests
// Wrap with black_box to prevent the compiler from optimizing out the code

use criterion::{criterion_group, criterion_main, Criterion};
use criterion::{black_box, criterion_group, criterion_main, Criterion};
use rasn::{ber, oer, uper};

#[allow(non_camel_case_types, non_snake_case, non_upper_case_globals, unused)]
Expand Down Expand Up @@ -117,61 +119,45 @@ pub fn build_sample_rasn() -> world3d::World {
World { depth }
}

fn uper_rasn_enc(c: &mut Criterion) {
c.bench_function("RASN/encode UPER - sample.asn", |b| {
b.iter(|| {
let w = build_sample_rasn();
let _ = uper::encode(&w).unwrap();
})
});
}
fn oer_rasn_enc(c: &mut Criterion) {
c.bench_function("RASN/encode OER - sample.asn", |b| {
b.iter(|| {
let w = build_sample_rasn();
let _ = oer::encode(&w).unwrap();
})
});
}
fn ber_rasn_enc(c: &mut Criterion) {
c.bench_function("RASN/encode BER - sample.asn", |b| {
b.iter(|| {
let w = build_sample_rasn();
let _ = ber::encode(&w).unwrap();
})
});
macro_rules! rasn_enc_fn {
($fn_name:ident, $codec:ident) => {
fn $fn_name(c: &mut Criterion) {
let items = build_sample_rasn();
c.bench_function(
&format!(
"RASN/encode {} - extended integer sample",
stringify!($codec).to_uppercase()
),
|b| b.iter(|| black_box($codec::encode(&items).unwrap())),
);
}
};
}

fn uper_rasn_dec(c: &mut Criterion) {
let w = build_sample_rasn();
let encoded = uper::encode(&w).unwrap();
macro_rules! rasn_dec_fn {
($fn_name:ident, $codec:ident) => {
fn $fn_name(c: &mut Criterion) {
let w = build_sample_rasn();
let encoded = $codec::encode(&w).unwrap();

c.bench_function("RASN/decode UPER - sample.asn", |b| {
b.iter(|| {
let _ = uper::decode::<world3d::World>(&encoded).unwrap();
})
});
c.bench_function(
&format!(
"RASN/decode {} - extended integer sample",
stringify!($codec).to_uppercase()
),
|b| b.iter(|| black_box($codec::decode::<world3d::World>(&encoded).unwrap())),
);
}
};
}
fn oer_rasn_dec(c: &mut Criterion) {
let w = build_sample_rasn();
let encoded = oer::encode(&w).unwrap();

c.bench_function("RASN/decode OER - sample.asn", |b| {
b.iter(|| {
let _ = oer::decode::<world3d::World>(&encoded).unwrap();
})
});
}
fn ber_rasn_dec(c: &mut Criterion) {
let w = build_sample_rasn();
let encoded = ber::encode(&w).unwrap();
rasn_enc_fn!(uper_rasn_enc, uper);
rasn_enc_fn!(oer_rasn_enc, oer);
rasn_enc_fn!(ber_rasn_enc, ber);

c.bench_function("RASN/decode BER - sample.asn", |b| {
b.iter(|| {
let _ = ber::decode::<world3d::World>(&encoded).unwrap();
})
});
}
rasn_dec_fn!(uper_rasn_dec, uper);
rasn_dec_fn!(oer_rasn_dec, oer);
rasn_dec_fn!(ber_rasn_dec, ber);

criterion_group!(
benches,
Expand Down
Loading