Skip to content

Commit 556b244

Browse files
committed
Added benchmarks for trialdivision and sieve
1 parent 5b60e55 commit 556b244

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

benches/benches.rs

+16-5
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
11
use criterion::{black_box, criterion_group, criterion_main, BenchmarkId, Criterion, Throughput};
22

3-
use primes::{PrimeSet, TrialDivision};
3+
use primes::{PrimeSet, Sieve, TrialDivision};
44

55
fn bench_primes(c: &mut Criterion) {
6-
let mut group = c.benchmark_group("TrialDivision::find");
7-
86
let mut sizes: Vec<u64> = Vec::new();
97
for &base in &[5_000, 50_000] {
108
for size in 2..=20 {
@@ -14,12 +12,25 @@ fn bench_primes(c: &mut Criterion) {
1412
sizes.sort();
1513
sizes.dedup();
1614

15+
let mut group = c.benchmark_group("TrialDivision::find");
16+
for &size in sizes.iter() {
17+
group.throughput(Throughput::Elements(size));
18+
group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, &size| {
19+
b.iter(|| {
20+
let mut pset = TrialDivision::new();
21+
black_box(pset.find(size))
22+
})
23+
});
24+
}
25+
group.finish();
26+
27+
let mut group = c.benchmark_group("Sieve::find");
1728
for &size in sizes.iter() {
1829
group.throughput(Throughput::Elements(size));
1930
group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, &size| {
2031
b.iter(|| {
21-
let mut td = TrialDivision::new();
22-
black_box(td.find(size))
32+
let mut pset = Sieve::new();
33+
black_box(pset.find(size))
2334
})
2435
});
2536
}

0 commit comments

Comments
 (0)