69 lines
2.0 KiB
Rust
69 lines
2.0 KiB
Rust
use criterion::Criterion;
|
|
use criterion::{black_box, criterion_group, criterion_main, Benchmark};
|
|
// use env_logger::{Builder, Env};
|
|
use test_harness::BeaconChainHarness;
|
|
use types::{ChainSpec, Hash256};
|
|
|
|
fn mid_epoch_state_transition(c: &mut Criterion) {
|
|
// Builder::from_env(Env::default().default_filter_or("debug")).init();
|
|
|
|
let validator_count = 1000;
|
|
let mut rig = BeaconChainHarness::new(ChainSpec::foundation(), validator_count);
|
|
|
|
let epoch_depth = (rig.spec.epoch_length * 2) + (rig.spec.epoch_length / 2);
|
|
|
|
for _ in 0..epoch_depth {
|
|
rig.advance_chain_with_block();
|
|
}
|
|
|
|
let state = rig.beacon_chain.state.read().clone();
|
|
|
|
assert!((state.slot + 1) % rig.spec.epoch_length != 0);
|
|
|
|
c.bench_function("mid-epoch state transition 10k validators", move |b| {
|
|
let state = state.clone();
|
|
b.iter(|| {
|
|
let mut state = state.clone();
|
|
black_box(state.per_slot_processing(Hash256::zero(), &rig.spec))
|
|
})
|
|
});
|
|
}
|
|
|
|
fn epoch_boundary_state_transition(c: &mut Criterion) {
|
|
// Builder::from_env(Env::default().default_filter_or("debug")).init();
|
|
|
|
let validator_count = 10000;
|
|
let mut rig = BeaconChainHarness::new(ChainSpec::foundation(), validator_count);
|
|
|
|
let epoch_depth = rig.spec.epoch_length * 2;
|
|
|
|
for _ in 0..(epoch_depth - 1) {
|
|
rig.advance_chain_with_block();
|
|
}
|
|
|
|
let state = rig.beacon_chain.state.read().clone();
|
|
|
|
assert_eq!((state.slot + 1) % rig.spec.epoch_length, 0);
|
|
|
|
c.bench(
|
|
"routines",
|
|
Benchmark::new("routine_1", move |b| {
|
|
let state = state.clone();
|
|
b.iter(|| {
|
|
let mut state = state.clone();
|
|
black_box(black_box(
|
|
state.per_slot_processing(Hash256::zero(), &rig.spec),
|
|
))
|
|
})
|
|
})
|
|
.sample_size(5), // sample size is low because function is sloooow.
|
|
);
|
|
}
|
|
|
|
criterion_group!(
|
|
benches,
|
|
mid_epoch_state_transition,
|
|
epoch_boundary_state_transition
|
|
);
|
|
criterion_main!(benches);
|