blockworks-foundation
/
solana-with-rpc-optimizations
mirror of https://github.com/blockworks-foundation/solana-with-rpc-optimizations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Support --block-production-method in banking-bench (#269)

This commit is contained in:
Andrew Fitzgerald 2024-03-22 09:12:21 -07:00 committed by GitHub
parent 91b1ee3df6
commit 84639441c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
banking-bench/src/main.rs
View File

@ -9,6 +9,7 @@ use {
solana_core::{
banking_stage::BankingStage,
banking_trace::{BankingPacketBatch, BankingTracer, BANKING_TRACE_DIR_DEFAULT_BYTE_LIMIT},
validator::BlockProductionMethod,
},
solana_gossip::cluster_info::{ClusterInfo, Node},
solana_ledger::{
@ -279,6 +280,14 @@ fn main() {
.takes_value(true)
.help("Number of batches to send in each iteration"),
)
.arg(
Arg::with_name("block_production_method")
.long("block-production-method")
.value_name("METHOD")
.takes_value(true)
.possible_values(BlockProductionMethod::cli_names())
.help(BlockProductionMethod::cli_message()),
)
.arg(
Arg::new("num_banking_threads")
.long("num-banking-threads")
@ -306,6 +315,9 @@ fn main() {
)
.get_matches();
let block_production_method = matches
.value_of_t::<BlockProductionMethod>("block_production_method")
.unwrap_or_default();
let num_banking_threads = matches
.value_of_t::<u32>("num_banking_threads")
.unwrap_or_else(|_| BankingStage::num_threads());
@ -448,7 +460,8 @@ fn main() {
DEFAULT_TPU_CONNECTION_POOL_SIZE,
),
};
let banking_stage = BankingStage::new_thread_local_multi_iterator(
let banking_stage = BankingStage::new_num_threads(
block_production_method,
&cluster_info,
&poh_recorder,
non_vote_receiver,