Skip to content

Commit 03594e4

Browse files
authored
ForwardingStage hookup, BankingStage forwarding removal (anza-xyz#4518)
1 parent 0e7b296 commit 03594e4

27 files changed

+195
-2638
lines changed

banking-bench/src/main.rs

-22
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ use {
77
log::*,
88
rand::{thread_rng, Rng},
99
rayon::prelude::*,
10-
solana_client::connection_cache::ConnectionCache,
1110
solana_core::{
1211
banking_stage::{update_bank_forks_and_poh_recorder_for_new_tpu_bank, BankingStage},
1312
banking_trace::{BankingTracer, Channels, BANKING_TRACE_DIR_DEFAULT_BYTE_LIMIT},
@@ -37,7 +36,6 @@ use {
3736
transaction::Transaction,
3837
},
3938
solana_streamer::socket::SocketAddrSpace,
40-
solana_tpu_client::tpu_client::DEFAULT_TPU_CONNECTION_POOL_SIZE,
4139
std::{
4240
sync::{atomic::Ordering, Arc, RwLock},
4341
thread::sleep,
@@ -304,12 +302,6 @@ fn main() {
304302
.takes_value(true)
305303
.help("Number of threads to use in the banking stage"),
306304
)
307-
.arg(
308-
Arg::new("tpu_disable_quic")
309-
.long("tpu-disable-quic")
310-
.takes_value(false)
311-
.help("Disable forwarding messages to TPU using QUIC"),
312-
)
313305
.arg(
314306
Arg::new("simulate_mint")
315307
.long("simulate-mint")
@@ -468,18 +460,6 @@ fn main() {
468460
gossip_vote_sender,
469461
gossip_vote_receiver,
470462
} = banking_tracer.create_channels(false);
471-
let tpu_disable_quic = matches.is_present("tpu_disable_quic");
472-
let connection_cache = if tpu_disable_quic {
473-
ConnectionCache::with_udp(
474-
"connection_cache_banking_bench_udp",
475-
DEFAULT_TPU_CONNECTION_POOL_SIZE,
476-
)
477-
} else {
478-
ConnectionCache::new_quic(
479-
"connection_cache_banking_bench_quic",
480-
DEFAULT_TPU_CONNECTION_POOL_SIZE,
481-
)
482-
};
483463
let banking_stage = BankingStage::new_num_threads(
484464
block_production_method,
485465
transaction_struct,
@@ -492,10 +472,8 @@ fn main() {
492472
None,
493473
replay_vote_sender,
494474
None,
495-
Arc::new(connection_cache),
496475
bank_forks.clone(),
497476
&prioritization_fee_cache,
498-
false,
499477
);
500478

501479
// This is so that the signal_receiver does not go out of scope after the closure.

core/benches/banking_stage.rs

-3
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ use {
1818
log::*,
1919
rand::{thread_rng, Rng},
2020
rayon::prelude::*,
21-
solana_client::connection_cache::ConnectionCache,
2221
solana_core::{
2322
banking_stage::{
2423
committer::Committer,
@@ -316,10 +315,8 @@ fn bench_banking(
316315
None,
317316
s,
318317
None,
319-
Arc::new(ConnectionCache::new("connection_cache_test")),
320318
bank_forks,
321319
&Arc::new(PrioritizationFeeCache::new(0u64)),
322-
false,
323320
);
324321

325322
let chunk_len = verified.len() / CHUNKS;

core/benches/forwarder.rs

-177
This file was deleted.

core/benches/sigverify_stage.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ fn bench_sigverify_stage(bencher: &mut Bencher, use_same_tx: bool) {
157157
trace!("start");
158158
let (packet_s, packet_r) = unbounded();
159159
let (verified_s, verified_r) = BankingTracer::channel_for_test();
160-
let verifier = TransactionSigVerifier::new(verified_s);
160+
let verifier = TransactionSigVerifier::new(verified_s, None);
161161
let stage = SigVerifyStage::new(packet_r, verifier, "solSigVerBench", "bench");
162162

163163
bencher.iter(move || {
@@ -231,7 +231,7 @@ fn prepare_batches(discard_factor: i32) -> (Vec<PacketBatch>, usize) {
231231
fn bench_shrink_sigverify_stage_core(bencher: &mut Bencher, discard_factor: i32) {
232232
let (batches0, num_valid_packets) = prepare_batches(discard_factor);
233233
let (verified_s, _verified_r) = BankingTracer::channel_for_test();
234-
let verifier = TransactionSigVerifier::new(verified_s);
234+
let verifier = TransactionSigVerifier::new(verified_s, None);
235235

236236
let mut c = 0;
237237
let mut total_shrink_time = 0;

core/src/banking_simulation.rs

-4
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ use {
1616
crossbeam_channel::{unbounded, Sender},
1717
itertools::Itertools,
1818
log::*,
19-
solana_client::connection_cache::ConnectionCache,
2019
solana_gossip::{
2120
cluster_info::{ClusterInfo, Node},
2221
contact_info::ContactInfoQuery,
@@ -787,7 +786,6 @@ impl BankingSimulator {
787786
gossip_vote_receiver,
788787
} = retracer.create_channels(false);
789788

790-
let connection_cache = Arc::new(ConnectionCache::new("connection_cache_sim"));
791789
let (replay_vote_sender, _replay_vote_receiver) = unbounded();
792790
let (retransmit_slots_sender, retransmit_slots_receiver) = unbounded();
793791
let shred_version = compute_shred_version(
@@ -833,10 +831,8 @@ impl BankingSimulator {
833831
None,
834832
replay_vote_sender,
835833
None,
836-
connection_cache,
837834
bank_forks.clone(),
838835
prioritization_fee_cache,
839-
false,
840836
);
841837

842838
let (&_slot, &raw_base_event_time) = freeze_time_by_slot

0 commit comments

Comments
 (0)