diff --git a/crates/bencher_compat/src/compat.rs b/crates/bencher_compat/src/compat.rs index 76a76033..35147788 100644 --- a/crates/bencher_compat/src/compat.rs +++ b/crates/bencher_compat/src/compat.rs @@ -38,8 +38,9 @@ impl Bencher { black_box(inner()); } self.codspeed.start_benchmark(uri.as_str()); - black_box(inner()); + let output = inner(); self.codspeed.end_benchmark(); + black_box(output); } } diff --git a/crates/criterion_compat/src/compat/bencher.rs b/crates/criterion_compat/src/compat/bencher.rs index 6eaaab0d..c50af643 100644 --- a/crates/criterion_compat/src/compat/bencher.rs +++ b/crates/criterion_compat/src/compat/bencher.rs @@ -67,11 +67,11 @@ impl<'a> Bencher<'a> { for i in 0..codspeed::codspeed::WARMUP_RUNS + 1 { let input = black_box(setup()); let output = if i < codspeed::codspeed::WARMUP_RUNS { - black_box(routine(input)) + routine(input) } else { let input = black_box(setup()); codspeed.start_benchmark(self.uri.as_str()); - let output = black_box(routine(input)); + let output = routine(input); codspeed.end_benchmark(); output }; @@ -219,10 +219,10 @@ impl<'a, 'b, A: AsyncExecutor> AsyncBencher<'a, 'b, A> { for i in 0..codspeed::codspeed::WARMUP_RUNS + 1 { let input = black_box(setup()); let output = if i < codspeed::codspeed::WARMUP_RUNS { - black_box(routine(input).await) + routine(input).await } else { codspeed.start_benchmark(b.uri.as_str()); - let output = black_box(routine(input).await); + let output = routine(input).await; codspeed.end_benchmark(); output }; diff --git a/crates/divan_compat/src/compat/bench/mod.rs b/crates/divan_compat/src/compat/bench/mod.rs index 48910ed1..5ffd3fd3 100644 --- a/crates/divan_compat/src/compat/bench/mod.rs +++ b/crates/divan_compat/src/compat/bench/mod.rs @@ -97,8 +97,9 @@ where let mut gen_input = self.config.gen_input.borrow_mut(); let input = gen_input(); codspeed.start_benchmark(self.uri.as_str()); - divan::black_box(benched(input)); + let output = benched(input); codspeed.end_benchmark(); + divan::black_box(output); } pub fn bench_local_refs(self, mut benched: B) @@ -110,7 +111,8 @@ where let mut input = gen_input(); codspeed.start_benchmark(self.uri.as_str()); - divan::black_box(benched(&mut input)); + let output = benched(&mut input); codspeed.end_benchmark(); + divan::black_box(output); } }