Use saturating math for timings (#18735)

This commit is contained in:
Jack May
2021-07-19 11:05:10 -07:00
committed by GitHub
parent d9970c0a7d
commit 7ed8792647
2 changed files with 34 additions and 21 deletions

View File

@ -169,12 +169,16 @@ pub struct ExecuteTimings {
} }
impl ExecuteTimings { impl ExecuteTimings {
pub fn accumulate(&mut self, other: &ExecuteTimings) { pub fn accumulate(&mut self, other: &ExecuteTimings) {
self.check_us += other.check_us; self.check_us = self.check_us.saturating_add(other.check_us);
self.load_us += other.load_us; self.load_us = self.load_us.saturating_add(other.load_us);
self.execute_us += other.execute_us; self.execute_us = self.execute_us.saturating_add(other.execute_us);
self.store_us += other.store_us; self.store_us = self.store_us.saturating_add(other.store_us);
self.total_batches_len += other.total_batches_len; self.total_batches_len = self
self.num_execute_batches += other.num_execute_batches; .total_batches_len
.saturating_add(other.total_batches_len);
self.num_execute_batches = self
.num_execute_batches
.saturating_add(other.num_execute_batches);
self.details.accumulate(&other.details); self.details.accumulate(&other.details);
} }
} }
@ -2768,8 +2772,8 @@ impl Bank {
.details .details
.per_program_timings .per_program_timings
.iter() .iter()
.fold(0, |acc, (_, program_timing)| { .fold(0, |acc: u64, (_, program_timing)| {
acc + program_timing.accumulated_units acc.saturating_add(program_timing.accumulated_units)
}); });
debug!("simulate_transaction: {:?}", timings); debug!("simulate_transaction: {:?}", timings);
@ -3329,9 +3333,9 @@ impl Bank {
execution_time.as_us(), execution_time.as_us(),
sanitized_txs.len(), sanitized_txs.len(),
); );
timings.check_us += check_time.as_us(); timings.check_us = timings.check_us.saturating_add(check_time.as_us());
timings.load_us += load_time.as_us(); timings.load_us = timings.load_us.saturating_add(load_time.as_us());
timings.execute_us += execution_time.as_us(); timings.execute_us = timings.execute_us.saturating_add(execution_time.as_us());
let mut tx_count: u64 = 0; let mut tx_count: u64 = 0;
let err_count = &mut error_counters.total; let err_count = &mut error_counters.total;
@ -3537,7 +3541,7 @@ impl Bank {
write_time.as_us(), write_time.as_us(),
sanitized_txs.len() sanitized_txs.len()
); );
timings.store_us += write_time.as_us(); timings.store_us = timings.store_us.saturating_add(write_time.as_us());
self.update_transaction_statuses(sanitized_txs, executed); self.update_transaction_statuses(sanitized_txs, executed);
let fee_collection_results = self let fee_collection_results = self
.filter_program_errors_and_collect_fee(sanitized_txs.as_transactions_iter(), executed); .filter_program_errors_and_collect_fee(sanitized_txs.as_transactions_iter(), executed);

View File

@ -76,7 +76,6 @@ pub struct ExecuteDetailsTimings {
pub data_size_changed: usize, pub data_size_changed: usize,
pub per_program_timings: HashMap<Pubkey, ProgramTiming>, pub per_program_timings: HashMap<Pubkey, ProgramTiming>,
} }
impl ExecuteDetailsTimings { impl ExecuteDetailsTimings {
pub fn accumulate(&mut self, other: &ExecuteDetailsTimings) { pub fn accumulate(&mut self, other: &ExecuteDetailsTimings) {
self.serialize_us += other.serialize_us; self.serialize_us += other.serialize_us;
@ -89,11 +88,21 @@ impl ExecuteDetailsTimings {
self.data_size_changed += other.data_size_changed; self.data_size_changed += other.data_size_changed;
for (id, other) in &other.per_program_timings { for (id, other) in &other.per_program_timings {
let program_timing = self.per_program_timings.entry(*id).or_default(); let program_timing = self.per_program_timings.entry(*id).or_default();
program_timing.accumulated_us += other.accumulated_us; program_timing.accumulated_us = program_timing
program_timing.accumulated_units += other.accumulated_units; .accumulated_us
program_timing.count += other.count; .saturating_add(other.accumulated_us);
program_timing.accumulated_units = program_timing
.accumulated_units
.saturating_add(other.accumulated_units);
program_timing.count = program_timing.count.saturating_add(other.count);
} }
} }
pub fn accumulate_program(&mut self, program_id: &Pubkey, us: u64, units: u64) {
let program_timing = self.per_program_timings.entry(*program_id).or_default();
program_timing.accumulated_us = program_timing.accumulated_us.saturating_add(us);
program_timing.accumulated_units = program_timing.accumulated_units.saturating_add(units);
program_timing.count = program_timing.count.saturating_add(1);
}
} }
// The relevant state of an account before an Instruction executes, used // The relevant state of an account before an Instruction executes, used
@ -1262,11 +1271,11 @@ impl MessageProcessor {
time.stop(); time.stop();
let post_remaining_units = compute_meter.borrow().get_remaining(); let post_remaining_units = compute_meter.borrow().get_remaining();
let program_id = instruction.program_id(&message.account_keys); timings.accumulate_program(
let program_timing = timings.per_program_timings.entry(*program_id).or_default(); instruction.program_id(&message.account_keys),
program_timing.accumulated_us += time.as_us(); time.as_us(),
program_timing.accumulated_units += pre_remaining_units - post_remaining_units; pre_remaining_units - post_remaining_units,
program_timing.count += 1; );
err?; err?;
} }