-
Notifications
You must be signed in to change notification settings - Fork 155
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
internal: add some initial input benchmarks
- Loading branch information
1 parent
2f4f80f
commit 857e088
Showing
2 changed files
with
112 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,107 @@ | ||
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion}; | ||
use salsa::Setter; | ||
|
||
#[salsa::db] | ||
pub trait Db: salsa::Database {} | ||
|
||
#[salsa::db] | ||
#[derive(Default)] | ||
pub struct Database { | ||
storage: salsa::Storage<Self>, | ||
} | ||
|
||
#[salsa::db] | ||
impl salsa::Database for Database {} | ||
|
||
#[salsa::db] | ||
impl Db for Database {} | ||
|
||
#[salsa::input] | ||
pub struct Input { | ||
pub text: String, | ||
} | ||
|
||
#[salsa::tracked] | ||
pub fn length(db: &dyn Db, input: Input) -> usize { | ||
input.text(db).len() | ||
} | ||
|
||
#[salsa::interned] | ||
pub struct InternedInput<'db> { | ||
pub text: String, | ||
} | ||
|
||
#[salsa::tracked] | ||
pub fn interned_length<'db>(db: &'db dyn Db, input: InternedInput<'db>) -> usize { | ||
input.text(db).len() | ||
} | ||
|
||
fn mutating_inputs(c: &mut Criterion) { | ||
let mut group: criterion::BenchmarkGroup<criterion::measurement::WallTime> = | ||
c.benchmark_group("Mutating Inputs"); | ||
|
||
let mut db = Database::default(); | ||
|
||
for n in &[10, 20, 30] { | ||
let base_string = "hello, world!".to_owned(); | ||
let base_len = base_string.len(); | ||
|
||
let string = std::iter::repeat(base_string.clone()) | ||
.take(*n) | ||
.collect::<String>(); | ||
let new_len = string.len(); | ||
|
||
group.bench_function(BenchmarkId::new("mutating", n), |b| { | ||
b.iter(|| { | ||
let input = Input::new(&db, base_string.clone()); | ||
let actual_len = length(&db, input); | ||
assert_eq!(base_len, actual_len); | ||
|
||
input.set_text(&mut db).to(string.clone()); | ||
let actual_len = length(&db, input); | ||
assert_eq!(new_len, actual_len); | ||
}) | ||
}); | ||
} | ||
|
||
group.finish(); | ||
} | ||
|
||
fn inputs(c: &mut Criterion) { | ||
let mut group: criterion::BenchmarkGroup<criterion::measurement::WallTime> = | ||
c.benchmark_group("Inputs"); | ||
let db = Database::default(); | ||
|
||
group.bench_function(BenchmarkId::new("new", "InternedInput"), |b| { | ||
b.iter(|| { | ||
let input: InternedInput = InternedInput::new(&db, "hello, world!".to_owned()); | ||
interned_length(&db, input); | ||
}) | ||
}); | ||
|
||
group.bench_function(BenchmarkId::new("amortized", "InternedInput"), |b| { | ||
let input = InternedInput::new(&db, "hello, world!".to_owned()); | ||
let _ = interned_length(&db, input); | ||
|
||
b.iter(|| interned_length(&db, input)); | ||
}); | ||
|
||
group.bench_function(BenchmarkId::new("new", "Input"), |b| { | ||
b.iter(|| { | ||
let input = Input::new(&db, "hello, world!".to_owned()); | ||
length(&db, input); | ||
}) | ||
}); | ||
|
||
group.bench_function(BenchmarkId::new("amortized", "Input"), |b| { | ||
let input = Input::new(&db, "hello, world!".to_owned()); | ||
let _ = length(&db, input); | ||
|
||
b.iter(|| length(&db, input)); | ||
}); | ||
|
||
group.finish(); | ||
} | ||
|
||
criterion_group!(benches, mutating_inputs, inputs); | ||
criterion_main!(benches); |