portal/waffle
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

WIP.

Browse source
This commit is contained in:
Chris Fallin 2022-12-01 20:36:28 -08:00
parent 77141d2177
commit 4140ed9ea6
1 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/ir/func.rs
View file

@ -84,7 +84,7 @@ impl FunctionBody {
id
}
fn add_edge(&mut self, from: Block, to: Block) {
pub fn add_edge(&mut self, from: Block, to: Block) {
let succ_pos = self.blocks[from].succs.len();
let pred_pos = self.blocks[to].preds.len();
self.blocks[from].succs.push(to);
@ -171,6 +171,7 @@ impl FunctionBody {
}
pub fn set_terminator(&mut self, block: Block, terminator: Terminator) {
debug_assert_eq!(&self.blocks[block].terminator, &Terminator::None);
log::trace!("block {} terminator {:?}", block, terminator);
terminator.visit_successors(|succ| {
self.add_edge(block, succ);
@ -251,9 +252,17 @@ impl FunctionBody {
};
for (i, &inst) in block_def.insts.iter().enumerate() {
self.values[inst].visit_uses(|u| {
visit_use(u, Some(i), Some(inst));
});
match &self.values[inst] {
&ValueDef::Operator(_, ref args, _) => {
for &arg in args {
visit_use(arg, Some(i), Some(inst));
}
}
&ValueDef::PickOutput(val, _, _) => {
visit_use(val, Some(i), Some(inst));
}
_ => {}
}
}
let terminator_idx = block_def.insts.len();
block_def.terminator.visit_uses(|u| {
@ -290,7 +299,7 @@ pub struct BlockDef {
pub params: Vec<(Type, Value)>,
}
#[derive(Clone, Debug)]
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct BlockTarget {
pub block: Block,
pub args: Vec<Value>,
@ -307,7 +316,7 @@ impl std::fmt::Display for BlockTarget {
}
}
#[derive(Clone, Debug)]
#[derive(Clone, Debug, PartialEq, Eq)]
pub enum Terminator {
Br {
target: BlockTarget,