feat!: add warmup runs

CodSpeedHQ · Jul 4, 2023 · 6f198a6 · 6f198a6
1 parent 9c6a5a8
commit 6f198a6
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/crates/bencher_compat/src/compat.rs b/crates/bencher_compat/src/compat.rs
@@ -1,4 +1,4 @@
-use codspeed::codspeed::{black_box, CodSpeed};
+use codspeed::codspeed::{black_box, CodSpeed, WARMUP_RUNS};
 
 pub struct Bencher {
     pub bytes: u64,
@@ -24,6 +24,9 @@ impl Bencher {
         F: FnMut() -> T,
     {
         let uri = self.current_uri.as_str();
+        for _ in 0..WARMUP_RUNS {
+            black_box(inner());
+        }
         self.codspeed.start_benchmark(uri);
         black_box(inner());
         self.codspeed.end_benchmark();

diff --git a/crates/codspeed/src/codspeed.rs b/crates/codspeed/src/codspeed.rs
@@ -4,6 +4,8 @@ use colored::Colorize;
 
 use crate::measurement;
 
+pub const WARMUP_RUNS: u32 = 5;
+
 //TODO: use std::hint::black_box when it's stable
 pub fn black_box<T>(dummy: T) -> T {
     unsafe {

diff --git a/crates/criterion_compat/src/compat/bencher.rs b/crates/criterion_compat/src/compat/bencher.rs
@@ -20,6 +20,9 @@ impl Bencher {
         R: FnMut() -> O,
     {
         let mut codspeed = self.codspeed.borrow_mut();
+        for _ in 0..codspeed::codspeed::WARMUP_RUNS {
+            black_box(routine());
+        }
         codspeed.start_benchmark(self.uri.as_str());
         black_box(routine());
         codspeed.end_benchmark();
@@ -44,6 +47,11 @@ impl Bencher {
         R: FnMut(I) -> O,
     {
         let mut codspeed = self.codspeed.borrow_mut();
+        for _ in 0..codspeed::codspeed::WARMUP_RUNS {
+            let input = black_box(setup());
+            let output = routine(input);
+            drop(black_box(output));
+        }
         let input = black_box(setup());
         codspeed.start_benchmark(self.uri.as_str());
         let output = routine(input);
@@ -82,8 +90,15 @@ impl Bencher {
         R: FnMut(&mut I) -> O,
     {
         let mut codspeed = self.codspeed.borrow_mut();
-        let mut input = black_box(setup());
 
+        for _ in 0..codspeed::codspeed::WARMUP_RUNS {
+            let mut input = black_box(setup());
+            let output = routine(&mut input);
+            drop(black_box(output));
+            drop(black_box(input));
+        }
+
+        let mut input = black_box(setup());
         codspeed.start_benchmark(self.uri.as_str());
         let output = routine(&mut input);
         codspeed.end_benchmark();