github/zerokit
1
1
Fork 0
mirror of https://github.com/vacp2p/zerokit.git synced 2026-01-10 06:58:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add batch setup to the imt setup group

Browse Source
This commit is contained in:
Ben-PH
2025-05-12 18:56:47 +02:00
parent a9838697c2
commit 702ac11a7c
1 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
utils/benches/imt_benchy.rs
View File

@@ -102,11 +102,11 @@ fn benchy_prototype_code() {
}
pub fn imt_benchy(c: &mut Criterion) {
let mut group = c.benchmark_group("markle tree setup");
let mut group = c.benchmark_group("merkle tree setup");
for size in [7u32, 13, 17].iter() {
group.bench_with_input(
BenchmarkId::new("Lean IMT setup", size),
BenchmarkId::new("Lean IMT setup incremental", size),
size,
|b, &size| {
b.iter_batched(
@@ -129,6 +129,29 @@ pub fn imt_benchy(c: &mut Criterion) {
)
},
);
group.bench_with_input(
BenchmarkId::new("Lean IMT setup batch", size),
size,
|b, &size| {
b.iter_batched(
// Setup: create values for each benchmark iteration
|| {
let data_source: Vec<[u8; 32]> = HashMockStream::seeded_stream(42)
.take(size as usize)
.collect()
;
let tree = HashedLeanIMT::<32, BenchyIFTHasher>::new(&[], BenchyIFTHasher)
.unwrap();
(tree, data_source)
},
// Actual benchmark
|(mut tree, data_source)| {
tree.insert_many(&data_source[..])
},
BatchSize::SmallInput,
)
},
);
}
group.finish();
}