Measure benchmark runtime in cycles per byte on x86

Author: milkey-mouse

Cargo.toml

@@ -4,15 +4,19 @@ version = "0.1.0"
 authors = ["Heinz N. Gies <[email protected]>"]
 edition = "2018"
 
-# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
-
+[features]
+default = ["cpb"]
+cpb = []
 
 [dev-dependencies]
 proptest = "0.10"
 criterion = "0.3"
 mimalloc = "0.1"
 core_affinity = "*"
 
+[target.'cfg(any(target_arch = "x86_64", target_arch = "x86"))'.dev-dependencies]
+criterion-cycles-per-byte = "0.1"
+
 [[bench]]
 name = "criterion_bench"
 harness = false

benches/criterion_bench.rs

@@ -1,15 +1,18 @@
-extern crate core_affinity;
-#[macro_use]
-extern crate criterion;
-
 use mimalloc::MiMalloc;
 #[global_allocator]
 static GLOBAL: MiMalloc = MiMalloc;
 
-use criterion::{measurement::Measurement, Criterion, Throughput};
+use criterion::{criterion_group, criterion_main, measurement::Measurement, Criterion, Throughput};
+
+#[cfg(all(any(target_arch = "x86_64", target_arch = "x86"), feature = "cpb"))]
+use criterion_cycles_per_byte::CyclesPerByte;
+
 use std::{fs, str};
 
 fn bench_file<T: Measurement>(c: &mut Criterion<T>, name: &str, is_valid: bool) {
+    let core_ids = core_affinity::get_core_ids().unwrap();
+    core_affinity::set_for_current(core_ids[0]);
+
     let buf = fs::read(format!("data/{}.data", name)).unwrap();
 
     let mut group = c.benchmark_group(name);
@@ -47,5 +50,11 @@ fn bench_all<T: Measurement>(c: &mut Criterion<T>) {
     bench_file(c, "ascii_sample_ok", true);
 }
 
-criterion_group!(benches, bench_all);
+#[cfg(all(any(target_arch = "x86_64", target_arch = "x86"), feature = "cpb"))]
+criterion_group! {
+    name = benches;
+    config = Criterion::default().with_measurement(CyclesPerByte);
+    targets = bench_all
+}
+
 criterion_main!(benches);