gnosis/lighthouse
1
0
Fork 0
mirror of https://github.com/sigp/lighthouse.git synced 2026-04-17 04:48:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix EpochCache handling in ef-tests (#4454)

Browse Source
This commit is contained in:
Michael Sproul
2023-07-01 09:53:06 +10:00
committed by GitHub
parent 2df714e2cd
commit 835fa70ea4
7 changed files with 32 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
consensus/state_processing/src/epoch_cache.rs
View File

@@ -23,6 +23,7 @@ pub fn initialize_epoch_cache<E: EthSpec>(
}
// Compute base rewards.
state.build_total_active_balance_cache_at(epoch, spec)?;
let total_active_balance = state.get_total_active_balance_at_epoch(epoch)?;
let sqrt_total_active_balance = SqrtTotalActiveBalance::new(total_active_balance);
let base_reward_per_increment = BaseRewardPerIncrement::new(total_active_balance, spec)?;

2
consensus/state_processing/src/per_epoch_processing/altair.rs
View File

@@ -28,6 +28,8 @@ pub fn process_epoch<T: EthSpec>(
state.build_committee_cache(RelativeEpoch::Previous, spec)?;
state.build_committee_cache(RelativeEpoch::Current, spec)?;
state.build_committee_cache(RelativeEpoch::Next, spec)?;
state.build_total_active_balance_cache_at(state.current_epoch(), spec)?;
initialize_epoch_cache(state, state.current_epoch(), spec)?;
// Pre-compute participating indices and total balances.
let mut participation_cache = ParticipationCache::new(state, spec)?;

2
consensus/state_processing/src/per_epoch_processing/base.rs
View File

@@ -24,6 +24,8 @@ pub fn process_epoch<T: EthSpec>(
state.build_committee_cache(RelativeEpoch::Previous, spec)?;
state.build_committee_cache(RelativeEpoch::Current, spec)?;
state.build_committee_cache(RelativeEpoch::Next, spec)?;
state.build_total_active_balance_cache_at(state.current_epoch(), spec)?;
initialize_epoch_cache(state, state.current_epoch(), spec)?;
// Load the struct we use to assign validators into sets based on their participation.
//

2
consensus/state_processing/src/per_epoch_processing/capella.rs
View File

@@ -24,6 +24,8 @@ pub fn process_epoch<T: EthSpec>(
state.build_committee_cache(RelativeEpoch::Previous, spec)?;
state.build_committee_cache(RelativeEpoch::Current, spec)?;
state.build_committee_cache(RelativeEpoch::Next, spec)?;
state.build_total_active_balance_cache_at(state.current_epoch(), spec)?;
initialize_epoch_cache(state, state.current_epoch(), spec)?;
// Pre-compute participating indices and total balances.
let mut participation_cache = ParticipationCache::new(state, spec)?;

27
consensus/types/src/beacon_state.rs
View File

@@ -1468,10 +1468,24 @@ impl<T: EthSpec> BeaconState<T> {
}
/// Build the total active balance cache.
fn build_total_active_balance_cache(&mut self, spec: &ChainSpec) -> Result<(), Error> {
let current_epoch = self.current_epoch();
let total_active_balance = self.compute_total_active_balance(current_epoch, spec)?;
*self.total_active_balance_mut() = Some((current_epoch, total_active_balance));
pub fn build_total_active_balance_cache_at(
&mut self,
epoch: Epoch,
spec: &ChainSpec,
) -> Result<(), Error> {
if self.get_total_active_balance_at_epoch(epoch).is_err() {
self.force_build_total_active_balance_cache_at(epoch, spec)?;
}
Ok(())
}
pub fn force_build_total_active_balance_cache_at(
&mut self,
epoch: Epoch,
spec: &ChainSpec,
) -> Result<(), Error> {
let total_active_balance = self.compute_total_active_balance(epoch, spec)?;
*self.total_active_balance_mut() = Some((epoch, total_active_balance));
Ok(())
}
@@ -1552,6 +1566,7 @@ impl<T: EthSpec> BeaconState<T> {
self.drop_committee_cache(RelativeEpoch::Next)?;
self.drop_pubkey_cache();
*self.exit_cache_mut() = ExitCache::default();
*self.epoch_cache_mut() = EpochCache::default();
Ok(())
}
@@ -1580,7 +1595,7 @@ impl<T: EthSpec> BeaconState<T> {
}
if self.total_active_balance().is_none() && relative_epoch == RelativeEpoch::Current {
self.build_total_active_balance_cache(spec)?;
self.build_total_active_balance_cache_at(self.current_epoch(), spec)?;
}
Ok(())
}
@@ -1874,7 +1889,7 @@ impl<T: EthSpec> BeaconState<T> {
// Ensure total active balance cache remains built whenever current committee
// cache is built.
if epoch == self.current_epoch() {
self.build_total_active_balance_cache(spec)?;
self.build_total_active_balance_cache_at(self.current_epoch(), spec)?;
}
}
}