cleanup feature: default units per instruction (#26684)

This commit is contained in:
Jack May 2022-07-20 12:13:34 -07:00 committed by GitHub
parent 502f249904
commit 39a34db52a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 18 deletions

View File

@ -23,7 +23,6 @@ pub trait GetTransactionPriorityDetails {
let prioritization_fee_details = compute_budget let prioritization_fee_details = compute_budget
.process_instructions( .process_instructions(
instructions, instructions,
true, // use default units per instruction
true, // don't reject txs that use set compute unit price ix true, // don't reject txs that use set compute unit price ix
) )
.ok()?; .ok()?;

View File

@ -126,7 +126,6 @@ impl ComputeBudget {
pub fn process_instructions<'a>( pub fn process_instructions<'a>(
&mut self, &mut self,
instructions: impl Iterator<Item = (&'a Pubkey, &'a CompiledInstruction)>, instructions: impl Iterator<Item = (&'a Pubkey, &'a CompiledInstruction)>,
default_units_per_instruction: bool,
support_set_compute_unit_price_ix: bool, support_set_compute_unit_price_ix: bool,
) -> Result<PrioritizationFeeDetails, TransactionError> { ) -> Result<PrioritizationFeeDetails, TransactionError> {
let mut num_non_compute_budget_instructions: usize = 0; let mut num_non_compute_budget_instructions: usize = 0;
@ -221,18 +220,15 @@ impl ComputeBudget {
self.heap_size = Some(bytes as usize); self.heap_size = Some(bytes as usize);
} }
self.compute_unit_limit = if default_units_per_instruction { self.compute_unit_limit = updated_compute_unit_limit
updated_compute_unit_limit.or_else(|| { .or_else(|| {
Some( Some(
(num_non_compute_budget_instructions as u32) (num_non_compute_budget_instructions as u32)
.saturating_mul(DEFAULT_INSTRUCTION_COMPUTE_UNIT_LIMIT), .saturating_mul(DEFAULT_INSTRUCTION_COMPUTE_UNIT_LIMIT),
) )
}) })
} else { .unwrap_or(MAX_COMPUTE_UNIT_LIMIT)
updated_compute_unit_limit .min(MAX_COMPUTE_UNIT_LIMIT) as u64;
}
.unwrap_or(MAX_COMPUTE_UNIT_LIMIT)
.min(MAX_COMPUTE_UNIT_LIMIT) as u64;
Ok(prioritization_fee Ok(prioritization_fee
.map(|fee_type| PrioritizationFeeDetails::new(fee_type, self.compute_unit_limit)) .map(|fee_type| PrioritizationFeeDetails::new(fee_type, self.compute_unit_limit))
@ -275,11 +271,8 @@ mod tests {
Hash::default(), Hash::default(),
)); ));
let mut compute_budget = ComputeBudget::default(); let mut compute_budget = ComputeBudget::default();
let result = compute_budget.process_instructions( let result = compute_budget
tx.message().program_instructions_iter(), .process_instructions(tx.message().program_instructions_iter(), $type_change);
true,
$type_change,
);
assert_eq!($expected_result, result); assert_eq!($expected_result, result);
assert_eq!(compute_budget, $expected_budget); assert_eq!(compute_budget, $expected_budget);
}; };

View File

@ -108,8 +108,8 @@ use {
epoch_schedule::EpochSchedule, epoch_schedule::EpochSchedule,
feature, feature,
feature_set::{ feature_set::{
self, add_set_compute_unit_price_ix, default_units_per_instruction, self, add_set_compute_unit_price_ix, disable_fee_calculator,
disable_fee_calculator, enable_early_verification_of_account_modifications, FeatureSet, enable_early_verification_of_account_modifications, FeatureSet,
}, },
fee::FeeStructure, fee::FeeStructure,
fee_calculator::{FeeCalculator, FeeRateGovernor}, fee_calculator::{FeeCalculator, FeeRateGovernor},
@ -4542,7 +4542,6 @@ impl Bank {
Measure::start("compute_budget_process_transaction_time"); Measure::start("compute_budget_process_transaction_time");
let process_transaction_result = compute_budget.process_instructions( let process_transaction_result = compute_budget.process_instructions(
tx.message().program_instructions_iter(), tx.message().program_instructions_iter(),
feature_set.is_active(&default_units_per_instruction::id()),
feature_set.is_active(&add_set_compute_unit_price_ix::id()), feature_set.is_active(&add_set_compute_unit_price_ix::id()),
); );
compute_budget_process_transaction_time.stop(); compute_budget_process_transaction_time.stop();
@ -4837,7 +4836,6 @@ impl Bank {
let prioritization_fee_details = compute_budget let prioritization_fee_details = compute_budget
.process_instructions( .process_instructions(
message.program_instructions_iter(), message.program_instructions_iter(),
false,
support_set_compute_unit_price_ix, support_set_compute_unit_price_ix,
) )
.unwrap_or_default(); .unwrap_or_default();