27 lines
912 B
Rust
27 lines
912 B
Rust
|
use divan::counter::BytesCount;
|
||
|
|
use divan::Bencher;
|
||
|
|
use kzgrs_backend::encoder::{DaEncoder, DaEncoderParams};
|
||
|
|
use rand::RngCore;
|
||
|
|
use std::hint::black_box;
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
divan::main()
|
||
|
|
}
|
||
|
|
|
||
|
|
const PARAMS: DaEncoderParams = DaEncoderParams::default_with(4096);
|
||
|
|
const ENCODER: DaEncoder = DaEncoder::new(PARAMS);
|
||
|
|
const KB: usize = 1024;
|
||
|
|
|
||
|
|
pub fn rand_data(elements_count: usize) -> Vec<u8> {
|
||
|
|
let mut buff = vec![0; elements_count * DaEncoderParams::MAX_BLS12_381_ENCODING_CHUNK_SIZE];
|
||
|
|
rand::thread_rng().fill_bytes(&mut buff);
|
||
|
|
buff
|
||
|
|
}
|
||
|
|
#[divan::bench(args = [32, 64, 128, 256, 512, 1024, 2048], sample_count = 1, sample_size = 1)]
|
||
|
|
fn encode(bencher: Bencher, size: usize) {
|
||
|
|
bencher
|
||
|
|
.with_inputs(|| rand_data(size * KB / DaEncoderParams::MAX_BLS12_381_ENCODING_CHUNK_SIZE))
|
||
|
|
.input_counter(|buff| BytesCount::new(buff.len()))
|
||
|
|
.bench_refs(|buff| black_box(ENCODER.encode(buff)));
|
||
|
|
}
|