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

Support treeification of adjacent side-effecting ops in some cases.

Browse source
This commit is contained in:
Chris Fallin 2024-05-07 15:22:18 -07:00 committed by Graham Kelly
parent 908ad937e1
commit c960d728bf
1 changed files with 47 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/backend/treeify.rs
View file

@ -41,8 +41,10 @@ impl Trees {
let mut remat = HashSet::default();
let mut multi_use = HashSet::default();
for (value, def) in body.values.entries() {
match def {
for block_def in body.blocks.values() {
let mut last_non_pure = None;
for &value in &block_def.insts {
match &body.values[value] {
&ValueDef::Operator(op, args, _) => {
// Ignore operators with invalid args: these must
// always be unreachable.
@ -69,13 +71,17 @@ impl Trees {
} else if let Some(old_owner) = owner.remove(&arg) {
owned.remove(&old_owner);
multi_use.insert(arg);
} else if Self::is_movable(body, arg) {
} else if Self::is_movable(body, arg) || Some(arg) == last_non_pure {
let pos = u16::try_from(i).unwrap();
let value_arg = ValueArg(value, pos);
owner.insert(arg, value_arg);
owned.insert(value_arg, arg);
}
}
if !op.is_pure() {
last_non_pure = Some(value);
}
}
&ValueDef::PickOutput(..) => {
// Can ignore use: multi-arity values are never treeified.
@ -87,6 +93,7 @@ impl Trees {
| &ValueDef::None => {}
}
}
}
for block in body.blocks.values() {
block.terminator.visit_uses(|u| {
let u = body.resolve_alias(u);