bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

add RPO to BB CFG cache

Browse source
This commit is contained in:
Rémy Rakic 2023-06-14 19:56:35 +00:00
parent e596579066
commit 066d38190b
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
compiler/rustc_middle/src/mir/basic_blocks.rs
View file

@ -27,6 +27,7 @@ struct Cache {
switch_sources: OnceCell<SwitchSources>, switch_sources: OnceCell<SwitchSources>,
is_cyclic: OnceCell<bool>, is_cyclic: OnceCell<bool>,
postorder: OnceCell<Vec<BasicBlock>>, postorder: OnceCell<Vec<BasicBlock>>,
reverse_postorder: OnceCell<Vec<BasicBlock>>,
dominators: OnceCell<Dominators<BasicBlock>>, dominators: OnceCell<Dominators<BasicBlock>>,
} }
@ -70,6 +71,17 @@ impl<'tcx> BasicBlocks<'tcx> {
}) })
} }
/// Returns basic blocks in a reverse postorder.
#[inline]
pub fn reverse_postorder(&self) -> &[BasicBlock] {
self.cache.reverse_postorder.get_or_init(|| {
let mut rpo: Vec<_> =
Postorder::new(&self.basic_blocks, START_BLOCK).map(|(bb, _)| bb).collect();
rpo.reverse();
rpo
})
}
/// `switch_sources()[&(target, switch)]` returns a list of switch /// `switch_sources()[&(target, switch)]` returns a list of switch
/// values that lead to a `target` block from a `switch` block. /// values that lead to a `target` block from a `switch` block.
#[inline] #[inline]