79 lines
2.3 KiB
Rust
79 lines
2.3 KiB
Rust
#![feature(test)]
|
|
|
|
extern crate test;
|
|
use solana_ledger::shred::Shred;
|
|
use solana_ledger::shred::SIZE_OF_DATA_SHRED_PAYLOAD;
|
|
use solana_ledger::sigverify_shreds::{
|
|
sign_shreds_cpu, sign_shreds_gpu, sign_shreds_gpu_pinned_keypair,
|
|
};
|
|
use solana_perf::packet::{Packet, Packets};
|
|
use solana_perf::recycler_cache::RecyclerCache;
|
|
use solana_sdk::signature::Keypair;
|
|
use std::sync::Arc;
|
|
use test::Bencher;
|
|
|
|
const NUM_PACKETS: usize = 256;
|
|
const NUM_BATCHES: usize = 1;
|
|
#[bench]
|
|
fn bench_sigverify_shreds_sign_gpu(bencher: &mut Bencher) {
|
|
let recycler_cache = RecyclerCache::default();
|
|
|
|
let mut packets = Packets::default();
|
|
packets.packets.set_pinnable();
|
|
let slot = 0xdead_c0de;
|
|
// need to pin explicitly since the resize will not cause re-allocation
|
|
packets.packets.reserve_and_pin(NUM_PACKETS);
|
|
packets.packets.resize(NUM_PACKETS, Packet::default());
|
|
for p in packets.packets.iter_mut() {
|
|
let shred = Shred::new_from_data(
|
|
slot,
|
|
0xc0de,
|
|
0xdead,
|
|
Some(&[5; SIZE_OF_DATA_SHRED_PAYLOAD]),
|
|
true,
|
|
true,
|
|
1,
|
|
2,
|
|
0,
|
|
);
|
|
shred.copy_to_packet(p);
|
|
}
|
|
let mut batch = vec![packets; NUM_BATCHES];
|
|
let keypair = Keypair::new();
|
|
let pinned_keypair = sign_shreds_gpu_pinned_keypair(&keypair, &recycler_cache);
|
|
let pinned_keypair = Some(Arc::new(pinned_keypair));
|
|
//warmup
|
|
for _ in 0..100 {
|
|
sign_shreds_gpu(&keypair, &pinned_keypair, &mut batch, &recycler_cache);
|
|
}
|
|
bencher.iter(|| {
|
|
sign_shreds_gpu(&keypair, &pinned_keypair, &mut batch, &recycler_cache);
|
|
})
|
|
}
|
|
|
|
#[bench]
|
|
fn bench_sigverify_shreds_sign_cpu(bencher: &mut Bencher) {
|
|
let mut packets = Packets::default();
|
|
let slot = 0xdead_c0de;
|
|
packets.packets.resize(NUM_PACKETS, Packet::default());
|
|
for p in packets.packets.iter_mut() {
|
|
let shred = Shred::new_from_data(
|
|
slot,
|
|
0xc0de,
|
|
0xdead,
|
|
Some(&[5; SIZE_OF_DATA_SHRED_PAYLOAD]),
|
|
true,
|
|
true,
|
|
1,
|
|
2,
|
|
0,
|
|
);
|
|
shred.copy_to_packet(p);
|
|
}
|
|
let mut batch = vec![packets; NUM_BATCHES];
|
|
let keypair = Keypair::new();
|
|
bencher.iter(|| {
|
|
sign_shreds_cpu(&keypair, &mut batch);
|
|
})
|
|
}
|