Skip to main content

rucc_opt/
simplify_cfg.rs

1//! Control flow simplification: unreachable blocks go, a branch that only ever goes one way
2//! becomes a jump, a block that does nothing but jump somewhere else stops being in the way, a
3//! block parameter that is the same value on every way in stops being a parameter, and a block
4//! with one way in is folded into the block above it.
5//!
6//! Design: `spec/optimizer/21-cfg-simplification.md`, and section 6.5 of
7//! `spec/optimizer/06-cfg-and-dominators.md`, which states the rule for the whole optimizer, that
8//! a block the entry does not reach is invisible to every analysis and is deleted here rather than
9//! by whichever pass happened to notice it.
10//!
11//! # The order
12//!
13//! Section 21.4, and it is an order rather than a loop. Unreachable removal, then the branches,
14//! then the straightening, then merging, each once. Running the four to a fixed point would cost a
15//! walk of the function for every pass over it and buy back a case nobody has: what merging leaves
16//! behind is a bigger block, and a bigger block does not make a branch foldable that was not
17//! foldable before. The pipeline runs this pass more than once anyway, so the second chance is a
18//! pass boundary away rather than a loop away, and that is a chance the pass manager can count and
19//! print.
20//!
21//! Step three is the exception, and it is the spec's exception rather than one taken here. Taking a
22//! forwarder out gives the block below it a way in it did not have, which can be the way in that
23//! makes one of its parameters the same value from everywhere; and taking a parameter away can be
24//! what leaves a block empty enough to be a forwarder. So the two run together on one worklist,
25//! which is a fixed point over a step and not over the pass.
26//!
27//! # The parameter nothing reads
28//!
29//! Section 21.2 is about a parameter that is the same value every way in, and there is a second
30//! kind that goes, which is one nothing reads at all. `crate::dce` is the pass that removes what
31//! nothing reads and it says in its own documentation why this one is not its job: a count driven
32//! to zero does not see a loop counter, because the counter's only reader is the addition that
33//! produces the value handed back to the counter. The count never reaches zero and the whole cycle
34//! is dead anyway.
35//!
36//! So it is answered the other way round, by asking what is live rather than what is dead.
37//! Something is live if an instruction that has to happen reads it, and then live spreads: the
38//! operands of a live instruction are live, and the argument every edge passes in a live
39//! parameter's place is live. Anything the spread does not reach is not read by anything that
40//! happens, and a parameter it does not reach goes along with the argument in its place on every
41//! edge into the block. What that strands is an addition whose result nobody wants any more, which
42//! is exactly the shape `crate::dce` was already good at.
43//!
44//! Starting from nothing live rather than from everything live is what breaks the cycle, and it is
45//! the same optimistic reading section 14.1 takes and section 21.2 takes one paragraph up. The
46//! risk in reading optimistically is claiming something is dead when it is not, so the seeds are
47//! generous: a terminator or an instruction with effects makes its operands live whatever else is
48//! true, and the entry block's parameters are the function's own and stay whether anybody reads
49//! them or not.
50//!
51//! `crate::ivopts` is what makes this worth having. Section 28.4 of
52//! `spec/optimizer/28-induction-variables.md` has a loop stop asking its counter anything, and
53//! before this the counter went on being incremented round a loop that had no other use for it.
54//!
55//! What the addition is left reading matters, and this used to leave it reading the parameter that
56//! had just gone. The argument for that was that the addition is dead and `crate::dce` is the pass
57//! for what is dead, and the argument is true at every level except the one where it counts:
58//! `-O0` runs this pass and nothing else, so nothing came along behind it and the operand reached
59//! the printer as a use with no definition. So the addition goes here, with the parameter that
60//! stranded it, and the `strand` function below carries the argument for why that is always safe.
61//!
62//! Cross jumping is the one transformation of section 21.1 that is not here at all, and that is
63//! section 21.1's last paragraph telling us not to: it costs a branch to save a copy, so it belongs
64//! at the machine level under `-Os`, which is document 37.
65//!
66//! # What a forwarder is allowed to be
67//!
68//! Section 21.1 wants four things of a block before its predecessors are pointed past it: one
69//! successor, the successor is not the exit, the successor is not the block itself, and the edge
70//! out is not abnormal. Then it adds the one the block parameter form needs, which is that the
71//! arguments the block passes on all dominate every predecessor of it, because those predecessors
72//! are the ones that will be passing them.
73//!
74//! Requiring the block to have no parameters of its own discharges that last one without a
75//! dominator tree, and the argument is short. A value defined in a block that dominates the
76//! forwarder dominates every predecessor of it too: take any path to a predecessor, follow the edge
77//! to the forwarder, and the definition is somewhere on the result, which is either before the
78//! predecessor or is the forwarder itself. A block with no parameters and no instructions defines
79//! nothing, so the second case cannot arise and the first is the condition.
80//!
81//! The requirement earns something else as well. A parameter of the forwarder could be read by a
82//! block below it, which is legal exactly when the forwarder dominates that block, and pointing the
83//! predecessors past would leave that read with nothing to read. Insisting on no parameters is one
84//! rule that answers both, and a forwarder that has one gets taken apart by the other half of the
85//! worklist first.
86//!
87//! There is no exit block in this IR, so the second condition is not a condition. Abnormal edges
88//! are the ones into a block whose address is taken, which arrive from an `indirect_br` the graph
89//! reads from the other end, and those blocks are refused here the same way they are refused
90//! everywhere else in this pass.
91//!
92//! One condition is here that the section does not ask for. A forwarder that passes arguments on,
93//! and that is arrived at from a block which branches, is the block the moves for those arguments
94//! go in. Take it out and the edge it was on becomes one that goes out of a block with two ways out
95//! and into a block with two ways in, which has no end of a block to put a move at, so the back end
96//! splits it and puts an empty block back. The block comes back at the end of the layout instead of
97//! where it was, the jump that was free because it fell through is a jump that is taken, and the
98//! value the move was carrying is live across more of the function. On the corpus at -O2 that costs
99//! more than the block is worth, so a forwarder in that position stays. A forwarder that carries
100//! nothing is taken out whatever the edges look like, since there is no move to find a place for.
101//!
102//! # Why this is not only an optimization
103//!
104//! Issue 359 is a program that does not link:
105//!
106//! ```c
107//! extern void link_error(void);
108//! void foo(int x) {
109//!     switch (x) {
110//!     case 0:
111//!         if (0) { link_error(); case 1: bar(); }
112//!     }
113//! }
114//! ```
115//!
116//! Nothing calls `link_error`, so a compiler that emits the call produces an object file that
117//! does not link, and the difference between the two compilers is not how fast the program runs.
118//! The file is in a suite of forty years of compiler bugs for the reason the `case 1:` is where
119//! it is: control does reach `bar` through the switch, and it reaches it from inside the body of
120//! the dead `if`. A compiler that deletes the compound statement gets this as wrong as one that
121//! keeps all of it.
122//!
123//! Doing it in two steps is what makes that come out right without a special case for it. The
124//! branch on the constant becomes a jump, which takes the edge into the dead arm away, and then
125//! reachability from the entry decides what is left. The block holding `bar` has an edge from the
126//! `switch` and stays. The block holding `link_error` has no edges at all and goes.
127//!
128//! # The condition it can read
129//!
130//! A constant, and a comparison of two constants. The second is here rather than in
131//! [`crate::fold`] because folding a comparison would produce an `i1` standing on its own, which
132//! is issue 352 and does not lower, so the pass that folds arithmetic deliberately leaves
133//! comparisons alone. Reading one to decide which way a branch goes produces no `i1` at all: the
134//! comparison is left exactly where it was, used by nothing, and [`crate::dce`] takes it out.
135//!
136//! # Fuel
137//!
138//! Fuel is charged for each branch that folds and for each block that is merged away, and not for
139//! the blocks that go because nothing reaches them. Removal is the second half of the
140//! transformation that was already paid for rather than a transformation of its own, and a fuel
141//! limit that could stop between the two halves would hand the verifier a block nothing reaches.
142//! Section 41.5 of `spec/optimizer/41-correctness.md` asks for fuel that is monotonic, which means
143//! each step being all of one change and not part of one.
144//!
145//! That reasoning covers the blocks a fold stranded. It does not cover the ones that arrived
146//! unreachable, and those are not charged for either, for a different reason: section 6.5 makes
147//! removing them this pass's standing obligation rather than an optimization, everything below
148//! reads the graph as though they are not there, and a bisection that turned the obligation off
149//! would be bisecting over a function the rest of the optimizer does not believe in.
150
151use std::collections::{HashMap, HashSet, VecDeque};
152
153use rucc_base::Idx;
154use rucc_ir::{Block, BlockCall, Def, Extra, Func, Inst, Opcode, Value};
155
156use crate::fold::constant;
157use crate::{Analyses, Fuel, Pass, Preserved, Stats, uses};
158
159/// Recorded once for each branch that turned into a jump.
160const FOLDED: &str = "branch on a condition that is always the same way replaced by a jump";
161
162/// Recorded once for each block that went with it.
163pub(crate) const REMOVED: &str = "block nothing reaches removed";
164
165/// Recorded once for each block folded into the one above it.
166const MERGED: &str = "block with one way into it merged into the block above it";
167
168/// Recorded once for each block that did nothing but jump and is no longer in the way.
169const FORWARDED: &str = "block that only jumped somewhere else removed and its edges pointed past";
170
171/// Recorded once for each block parameter that turned out to be one value.
172const SAME_EVERY_WAY: &str = "block parameter that arrives as the same value every way in removed";
173
174/// Recorded for a branch that would have folded if there had been fuel for it.
175const NO_FUEL: &str = "branch on a known condition left alone, the pass ran out of fuel";
176
177/// Recorded for a block that would have been merged if there had been fuel for it.
178const NO_FUEL_MERGE: &str = "block with one way into it left alone, the pass ran out of fuel";
179
180/// Recorded for a forwarder that would have gone if there had been fuel for it.
181const NO_FUEL_FORWARD: &str =
182    "block that only jumped somewhere else kept, the pass ran out of fuel";
183
184/// Recorded for a block parameter that would have gone if there had been fuel for it.
185const NO_FUEL_PARAM: &str = "block parameter that is one value kept, the pass ran out of fuel";
186
187/// Recorded once for each block parameter that nothing turned out to read.
188const NOTHING_READS_IT: &str = "block parameter nothing reads removed, and the argument on every \
189                                edge that was feeding it";
190
191/// Recorded for a parameter nothing reads that would have gone if there had been fuel for it.
192const NO_FUEL_UNREAD: &str = "block parameter nothing reads kept, the pass ran out of fuel";
193
194/// The pass.
195#[derive(Debug, Clone, Copy, PartialEq, Eq)]
196pub struct SimplifyCfg;
197
198impl Pass for SimplifyCfg {
199    fn name(&self) -> &'static str {
200        "simplify-cfg"
201    }
202
203    fn describe(&self) -> &'static str {
204        "unreachable blocks go, a branch that only goes one way becomes a jump, a block that only \
205         jumps stops being in the way, and a block with one way in is merged into the one above it"
206    }
207
208    fn preserves(&self) -> Preserved {
209        // Nothing at all, and this is the pass the declaration exists for. An edge moves, so the
210        // graph is a different graph, and everything built on the graph was about the old one.
211        Preserved::NONE
212    }
213
214    fn run(&self, func: &mut Func, an: &mut Analyses, fuel: &mut Fuel) -> Stats {
215        let mut stats = Stats::new();
216        // Step one, and it is first for a reason beyond tidiness: a branch in a block nothing
217        // reaches is a branch nothing executes, and folding one would spend fuel on a change
218        // nobody can see and charge the two steps below for walking blocks that are not there.
219        sweep(func, an, &mut stats);
220        let mut folded = false;
221        // Nothing bound, because this step asks where a branch goes whichever way control arrived
222        // at it. Binding a block's parameters to one edge's arguments is the question
223        // [`crate::thread`] asks, and it is a different question with a different answer.
224        let unbound = Bindings::new();
225        for block in func.blocks().collect::<Vec<Block>>() {
226            let Some(term) = func.terminator(block) else { continue };
227            let Some(taken) = taken(func, term, &unbound) else { continue };
228            if !fuel.take() {
229                // Out of fuel stops the transforming and not the looking, the same way the other
230                // passes treat it, so that the walk is the same walk at every fuel setting.
231                stats.missed(NO_FUEL);
232                continue;
233            }
234            jump_to(func, term, taken);
235            stats.optimized(FOLDED);
236            folded = true;
237        }
238        if folded {
239            // The second sweep section 21.4 folds into step two. The cache is holding answers
240            // about the function as it was a moment ago, and the manager clears it after the pass
241            // returns, which is too late for the pass itself.
242            an.clear();
243            sweep(func, an, &mut stats);
244        }
245        let mut forward = HashMap::new();
246        // Step three, and it keeps its own record of the edges rather than asking for the graph,
247        // because it changes the edges as it goes and a cached answer would be about the shape the
248        // function had one forwarder ago. The parameters nothing reads go first, for the reason
249        // this module's documentation gives, which is that a block they leave empty is a forwarder
250        // and the worklist below is what takes forwarders out.
251        let dropped = drop_unread(func, fuel, &mut stats);
252        if straighten(func, fuel, &mut stats, &mut forward) || dropped {
253            an.clear();
254        }
255        // Merging reads which blocks have one predecessor, so it has to run on the graph as it is
256        // after the stranded ones have gone. A block kept alive only by an edge from a block
257        // nothing reaches looks like it has two ways in until that block is out of the function.
258        for chain in chains(func, an) {
259            for (at, &block) in chain.iter().enumerate().skip(1) {
260                if !fuel.take() {
261                    // The rest of the chain goes with it. A block is merged into the one at the
262                    // head of its chain, and it can only get there once everything between them
263                    // has already arrived.
264                    for _ in at..chain.len() {
265                        stats.missed(NO_FUEL_MERGE);
266                    }
267                    break;
268                }
269                merge(func, chain[0], block, &mut forward);
270                stats.optimized(MERGED);
271            }
272        }
273        if !forward.is_empty() {
274            // Once, for every parameter every merge bound, rather than a walk of the function per
275            // block merged.
276            uses::substitute(func, &forward);
277        }
278        stats
279    }
280}
281
282/// What a block's parameters hold along one particular edge into it.
283///
284/// Empty is the honest answer for a question asked about a block rather than about an edge, and it
285/// is what this pass passes, since a branch it folds has to fold whichever way control arrived.
286/// [`crate::thread`] asks the same question one edge at a time and fills this in, which is the
287/// whole difference between folding a branch and threading one.
288pub(crate) type Bindings = HashMap<Value, Value>;
289
290/// The value this one stands for along the edge, which is itself when the edge says nothing.
291fn resolve(subst: &Bindings, value: Value) -> Value {
292    subst.get(&value).copied().unwrap_or(value)
293}
294
295/// Where this terminator always goes, if it always goes to one place.
296///
297/// `None` is every reason not to fold and does not say which, because the answer to all of them
298/// is to leave the branch alone.
299///
300/// Shared with [`crate::thread`], which asks it under a `subst` that binds the block's parameters
301/// to what one edge into the block carries. Two answers about when a branch is decided would be
302/// two compilers, and the threading pass would be the one nobody checked.
303pub(crate) fn taken(func: &Func, term: Inst, subst: &Bindings) -> Option<BlockCall> {
304    let data = &func[term];
305    let arg = *func[data.args].first()?;
306    match data.opcode {
307        Opcode::BrIf => {
308            let Extra::Targets(targets) = data.extra else { return None };
309            if let Some(call) = one_place(func, &func[targets]) {
310                return Some(call);
311            }
312            // The first target is the one taken when the condition is one, which is what
313            // `Builder::br_if` writes and what the printer reads back.
314            let arm = usize::from(!known(func, arg, subst)?);
315            func[targets].get(arm).copied()
316        }
317        Opcode::Switch => {
318            let Extra::Switch(at) = data.extra else { return None };
319            let info = func[at];
320            if let Some(call) = one_place(func, &func[info.targets]) {
321                return Some(call);
322            }
323            let (value, _) = constant(func, resolve(subst, arg))?;
324            // The default is the first target and the cases follow it in the order their values
325            // are in, so the target for a case that matches is one past the value's own place.
326            let case = func[info.cases].iter().position(|it| *it == value);
327            func[info.targets].get(case.map_or(0, |case| case + 1)).copied()
328        }
329        _ => None,
330    }
331}
332
333/// The one edge every arm of a branch is, when they are all the same edge.
334///
335/// Section 21.1's branch simplification, the half of it that is not about a constant. A branch
336/// whose arms all go to the same block with the same arguments goes there whatever the condition
337/// says, so it is a jump, and the condition becomes something nothing reads for
338/// [`crate::dce`] to take out.
339///
340/// The arguments have to match and not only the block. Two edges to one block carrying different
341/// arguments are two different edges, and that is the whole reason this IR passes arguments along
342/// an edge rather than writing a phi in the block: `if (c) goto L(1); else goto L(2);` is a real
343/// program and turning it into a jump would have to pick one of the two numbers.
344fn one_place(func: &Func, calls: &[BlockCall]) -> Option<BlockCall> {
345    let &first = calls.first()?;
346    let same = |call: &BlockCall| call.block == first.block && func[call.args] == func[first.args];
347    calls[1..].iter().all(same).then_some(first)
348}
349
350/// Rewrites the terminator as a jump to that one of its targets.
351///
352/// In place, and the target keeps the arguments it already had, because the arguments belong to
353/// the edge and the edge is the one that survives.
354pub(crate) fn jump_to(func: &mut Func, term: Inst, call: BlockCall) {
355    let targets = func.push_block_calls(&[call]);
356    let args = func.push_values(&[]);
357    let data = &mut func[term];
358    data.opcode = Opcode::Jump;
359    data.args = args;
360    data.extra = Extra::Targets(targets);
361}
362
363/// Takes every block the entry does not reach out of the function.
364///
365/// The cache goes with them, because what it is holding is answers about a function that had them
366/// in it, and the pass is not finished asking.
367///
368/// Shared with [`crate::thread`]. Section 6.5 makes this a standing obligation of whichever pass
369/// stranded the block rather than an optimization of this one, the verifier holds every pass to it,
370/// and a second walk written next door would be a second answer about what reachable means.
371pub(crate) fn sweep(func: &mut Func, an: &mut Analyses, stats: &mut Stats) {
372    let gone = stranded(func, an);
373    if gone.is_empty() {
374        return;
375    }
376    for block in gone {
377        func.remove_block(block);
378        stats.optimized(REMOVED);
379    }
380    an.clear();
381}
382
383/// The blocks the entry cannot reach, in block order.
384///
385/// This is reachability as the verifier counts it, which is over the edges the terminators name
386/// and additionally over the blocks a `block_addr` mentions. A block whose address is taken is
387/// arrived at by an `indirect_br` somewhere, and that instruction lists every block the address
388/// can hold, so the edge is already in the graph from the place control really leaves. What the
389/// graph does not carry is the `block_addr` itself, and deleting the block under one would leave
390/// an instruction naming a block that is not there.
391fn stranded(func: &Func, an: &mut Analyses) -> Vec<Block> {
392    let cfg = an.cfg(func);
393    let Some(entry) = cfg.entry() else { return Vec::new() };
394    let mut seen = vec![false; cfg.capacity()];
395    seen[entry.index()] = true;
396    let mut stack = vec![entry];
397    let mut reached = Vec::new();
398    while let Some(block) = stack.pop() {
399        for &succ in cfg.successors(block) {
400            if !seen[succ.index()] {
401                seen[succ.index()] = true;
402                stack.push(succ);
403            }
404        }
405        reached.push(block);
406    }
407    // The addresses in a second walk over the blocks the first one reached, because an address
408    // taken in a block nothing reaches is an address nothing takes.
409    let mut next = reached;
410    while !next.is_empty() {
411        let mut found = Vec::new();
412        for block in next {
413            for inst in func.insts(block) {
414                if func[inst].opcode != Opcode::BlockAddr {
415                    continue;
416                }
417                for call in func.successors(inst) {
418                    if !seen[call.block.index()] {
419                        seen[call.block.index()] = true;
420                        found.push(call.block);
421                    }
422                }
423            }
424        }
425        // Everything the newly kept blocks reach is kept too, which is what makes this a fixed
426        // point rather than one extra step.
427        let mut stack = found.clone();
428        while let Some(block) = stack.pop() {
429            for &succ in cfg.successors(block) {
430                if !seen[succ.index()] {
431                    seen[succ.index()] = true;
432                    stack.push(succ);
433                    found.push(succ);
434                }
435            }
436        }
437        next = found;
438    }
439    func.blocks().filter(|block| !seen[block.index()]).collect()
440}
441
442/// Where every edge that arrives at a block was written down, and which block it left.
443///
444/// A block call rather than a predecessor, because both halves of step three edit the edge and
445/// neither of them can find it again from the block it goes to. Redirecting one wants the slot in
446/// the pool, and taking a block parameter away wants the slot too, so this is what the step keeps
447/// instead of a [`crate::Cfg`].
448pub(crate) type Edges = HashMap<Block, Vec<(Block, Idx<BlockCall>)>>;
449
450/// Every edge in the function, filed under the block it arrives at.
451///
452/// Terminators only. A `block_addr` names a block and is not an edge, which is the same
453/// distinction [`stranded`] draws from the other side.
454///
455/// Shared with [`crate::thread`], which edits edges as well and so wants the slot in the pool for
456/// the same reason this step does.
457pub(crate) fn incoming(func: &Func) -> Edges {
458    let mut edges: Edges = HashMap::new();
459    for block in func.blocks() {
460        let Some(term) = func.terminator(block) else { continue };
461        for at in func.target_list(term).iter() {
462            edges.entry(func[at].block).or_default().push((block, at));
463        }
464    }
465    edges
466}
467
468/// Takes out every block parameter nothing reads, and the argument in its place on every edge.
469///
470/// This module's documentation is the argument. [`live`] is where the reading is done and this is
471/// what acts on it: one walk, in block order, because the answer is about the whole function and a
472/// worklist would only be asking the same question again.
473///
474/// # Fuel
475///
476/// One unit per parameter, and running out stops the removals rather than the looking, which this
477/// step can do because it has already worked out the whole answer. The parameters of a block are
478/// taken together once they have all been paid for, since an argument list that has lost some of
479/// its entries and not others is a function the verifier refuses.
480fn drop_unread(func: &mut Func, fuel: &mut Fuel, stats: &mut Stats) -> bool {
481    let Some(entry) = func.entry() else { return false };
482    let live = live(func, entry, &addressed(func));
483    let edges = incoming(func);
484    let mut changed = false;
485    let mut gone: HashSet<Value> = HashSet::new();
486    for block in func.blocks().collect::<Vec<Block>>() {
487        let mut taking = Vec::new();
488        for (index, &param) in func[block].params.iter().enumerate() {
489            if live.contains(&param) {
490                continue;
491            }
492            if !fuel.take() {
493                stats.missed(NO_FUEL_UNREAD);
494                continue;
495            }
496            taking.push(index);
497        }
498        if taking.is_empty() {
499            continue;
500        }
501        for _ in &taking {
502            stats.optimized(NOTHING_READS_IT);
503        }
504        gone.extend(taking.iter().map(|&index| func[block].params[index]));
505        take_params(func, block, &taking, edges.get(&block));
506        changed = true;
507    }
508    if !gone.is_empty() {
509        strand(func, gone);
510    }
511    changed
512}
513
514/// Takes out the instructions that were reading a parameter the step above removed.
515///
516/// Without this the step above leaves a use of a value nothing defines, which is a function the
517/// verifier is entitled to refuse and a printed IR that does not read back. It was written the
518/// other way round, on the argument that what is stranded is dead and [`crate::dce`] is the pass
519/// that takes what is dead. That holds at every level except the one that matters most, because
520/// `-O0` runs this pass and no other, so at `-O0` nothing came along afterwards and the dangling
521/// operand reached the printer. That was issue 1016.
522///
523/// Every instruction this removes is dead, and the proof is [`live`] read backwards. Live spreads
524/// from a result to the operands that produced it, so an instruction whose result were live would
525/// have made its operands live, and the parameter it reads was removed exactly because nothing
526/// made it live. A terminator's operands and an effectful instruction's operands are live by
527/// seeding, so neither of those can be reading a removed parameter either. The same argument
528/// applies again to the results of what goes, which is why this follows the chain.
529///
530/// # Fuel
531///
532/// None, for the reason this module's documentation gives about the blocks a fold strands: this is
533/// the second half of a transformation that has already been paid for, and a budget that could
534/// stop between the two halves would hand the verifier a use with no definition.
535fn strand(func: &mut Func, mut gone: HashSet<Value>) {
536    loop {
537        let mut spread = false;
538        for block in func.blocks().collect::<Vec<Block>>() {
539            for inst in func.insts(block).collect::<Vec<Inst>>() {
540                if !func[func[inst].args].iter().any(|value| gone.contains(value)) {
541                    continue;
542                }
543                let results: Vec<Value> = func[inst].results().collect();
544                for result in results {
545                    spread |= gone.insert(result);
546                }
547                func.remove_inst(inst);
548            }
549        }
550        // A second walk only when something this removed produced a value of its own, since that
551        // value may be read further down and layout order is not a promise about where.
552        if !spread {
553            return;
554        }
555    }
556}
557
558/// Every value something that happens reads, worked out from nothing live outwards.
559///
560/// The seeds are the operands of the instructions that have to happen, which are the terminators
561/// and the ones with effects. A terminator's own operands are seeds and the arguments it passes to
562/// the blocks it branches to are not, and that split is the whole point: an argument is read only
563/// if the parameter it lands in is read, so it waits for that parameter to be reached.
564///
565/// Then live spreads two ways. From a value an instruction produced, to that instruction's
566/// operands, because producing it meant reading them. From a parameter, to the argument in its
567/// place on every edge into the block, because arriving there meant passing them.
568///
569/// The entry block's parameters are the function's own and are live by declaration rather than by
570/// use, and so are the parameters of a block whose address is taken, because an `indirect_br` is a
571/// way in that this reads from the wrong end.
572fn live(func: &Func, entry: Block, addressed: &HashSet<Block>) -> HashSet<Value> {
573    let mut where_from: HashMap<Value, (Block, usize)> = HashMap::new();
574    let mut live: HashSet<Value> = HashSet::new();
575    let mut work: Vec<Value> = Vec::new();
576    let seed = |value: Value, live: &mut HashSet<Value>, work: &mut Vec<Value>| {
577        if live.insert(value) {
578            work.push(value);
579        }
580    };
581    for block in func.blocks() {
582        let held = block == entry || addressed.contains(&block);
583        for (index, &param) in func[block].params.iter().enumerate() {
584            where_from.insert(param, (block, index));
585            if held {
586                seed(param, &mut live, &mut work);
587            }
588        }
589        for inst in func.insts(block) {
590            if !func.is_terminator(inst) && !func[inst].opcode.has_effects() {
591                continue;
592            }
593            for &value in &func[func[inst].args] {
594                seed(value, &mut live, &mut work);
595            }
596        }
597    }
598
599    let edges = incoming(func);
600    while let Some(value) = work.pop() {
601        match func[value].def {
602            Def::Result { inst, .. } => {
603                for &operand in &func[func[inst].args] {
604                    seed(operand, &mut live, &mut work);
605                }
606            }
607            Def::Param { .. } => {
608                let Some(&(block, index)) = where_from.get(&value) else { continue };
609                for &(_, at) in edges.get(&block).into_iter().flatten() {
610                    let Some(&arg) = func[func[at].args].get(index) else { continue };
611                    seed(arg, &mut live, &mut work);
612                }
613            }
614        }
615    }
616    live
617}
618
619/// Section 21.4's step three, both halves of it, on one worklist. Says whether anything changed.
620///
621/// Forwarder removal and redundant block parameter removal are one step because each is the other's
622/// reason to look again. Pointing a block's predecessors past it hands the block below several ways
623/// in where there was one, and a parameter that was obviously one value may stop being one, or
624/// several arguments that were the same may now arrive together and make one; taking a parameter
625/// away can leave a block with nothing but its jump, which is the whole of what a forwarder is.
626///
627/// A block goes back on the worklist when an edge into it or out of it moved, and the loop stops
628/// when nothing has moved. That is a fixed point, and it is the one section 21.4 asks for, over the
629/// step rather than over the pass.
630///
631/// What this does not do is put the two halves in a particular order within a block. Parameters
632/// first is not a policy, it is the only order that gets a forwarder with a redundant parameter in
633/// one visit rather than two.
634///
635/// # Fuel
636///
637/// One unit for each forwarder and one for each parameter, and the first refusal is where the step
638/// stops rather than where it starts skipping. The other steps go on looking after they run out and
639/// say so once for each thing they did not do, which they can because each of them walks the
640/// function once. This one comes back to a block whenever an edge near it moved, so a refusal
641/// counted per visit would count one opportunity several times and the number would say more about
642/// the shape of the worklist than about the function. A budget that has reached zero is not going
643/// to have anything in it later, so there is one refusal recorded and it is the true one.
644fn straighten(
645    func: &mut Func,
646    fuel: &mut Fuel,
647    stats: &mut Stats,
648    forward: &mut HashMap<Value, Value>,
649) -> bool {
650    let Some(entry) = func.entry() else { return false };
651    let addressed = addressed(func);
652    let mut edges = incoming(func);
653    let mut work: VecDeque<Block> = func.blocks().collect();
654    let mut queued: HashSet<Block> = work.iter().copied().collect();
655    let mut gone: HashSet<Block> = HashSet::new();
656    let mut changed = false;
657    while let Some(block) = work.pop_front() {
658        queued.remove(&block);
659        if gone.contains(&block) {
660            continue;
661        }
662        let mut starved = false;
663        if block != entry {
664            let drop = redundant(func, block, edges.get(&block), forward);
665            let mut taking = Vec::new();
666            for (index, value) in drop {
667                if !fuel.take() {
668                    stats.missed(NO_FUEL_PARAM);
669                    starved = true;
670                    break;
671                }
672                // Through what an earlier one already decided, the same way merging does, because
673                // a parameter can be redundant on an argument that is on its way somewhere else.
674                let value = uses::chase(forward, value);
675                forward.insert(func[block].params[index], value);
676                taking.push(index);
677                stats.optimized(SAME_EVERY_WAY);
678            }
679            if !taking.is_empty() {
680                take_params(func, block, &taking, edges.get(&block));
681                // Itself, because a block that has run out of parameters may be a forwarder now,
682                // and because a parameter can be redundant on one that just went.
683                requeue(block, &mut work, &mut queued);
684                // And the blocks below, because a parameter passed straight on down is the shape
685                // section 21.2 means by one removal making the next one possible.
686                if let Some(term) = func.terminator(block) {
687                    for call in func.successors(term).collect::<Vec<BlockCall>>() {
688                        requeue(call.block, &mut work, &mut queued);
689                    }
690                }
691                changed = true;
692            }
693        }
694        // What was already paid for is applied first, and then the step stops, because a block
695        // whose parameters half went is a block whose edges have to agree with it.
696        if starved {
697            break;
698        }
699        let Some((term, into, args)) = forwards(func, block, entry, &addressed, &edges) else {
700            continue;
701        };
702        if !fuel.take() {
703            stats.missed(NO_FUEL_FORWARD);
704            break;
705        }
706        // The block's own edge stops existing along with the block, and it has to come out of the
707        // record before its predecessors' edges go in, or the block below would be told it has a
708        // way in from a block that is not there.
709        let out = func.target_list(term).iter().next().expect("a jump has a target");
710        if let Some(list) = edges.get_mut(&into) {
711            list.retain(|&(_, at)| at != out);
712        }
713        let ins = edges.remove(&block).unwrap_or_default();
714        for &(_, at) in &ins {
715            // A list of its own for each edge rather than one shared between them, because a
716            // later substitution rewrites a list in place and a shared one would be rewritten
717            // once for every edge that named it.
718            let call = func[at];
719            let args = func.push_values(&args);
720            func.set_block_call(at, BlockCall { block: into, args, ..call });
721        }
722        edges.entry(into).or_default().extend(ins.iter().copied());
723        func.remove_block(block);
724        gone.insert(block);
725        stats.optimized(FORWARDED);
726        changed = true;
727        requeue(into, &mut work, &mut queued);
728        for &(from, _) in &ins {
729            requeue(from, &mut work, &mut queued);
730        }
731    }
732    changed
733}
734
735/// Puts a block back on the worklist, if it is not on it already.
736fn requeue(block: Block, work: &mut VecDeque<Block>, queued: &mut HashSet<Block>) {
737    if queued.insert(block) {
738        work.push_back(block);
739    }
740}
741
742/// Which of a block's parameters arrive as the same value every way in, and what that value is.
743///
744/// Section 21.2. A parameter that is `x` from one edge and `x` from every other is not carrying
745/// anything, it is spelling `x` a second way, and document 12's hash consing cannot see through the
746/// spelling, so two equal values look different for as long as it is there.
747///
748/// The one subtlety is an argument that is the parameter itself, which is what a loop header looks
749/// like: the preheader passes `init` and the latch passes the parameter back. Reading that
750/// literally says two different values and the answer is `init`, because a value that can only ever
751/// be itself or `init` was `init` to begin with. So a self reference is not an argument for this
752/// purpose, which is the same optimistic reading section 14.1 takes.
753///
754/// A block with no way in gets nothing said about it. That is an unreachable block, [`sweep`] has
755/// already run, and answering `init` for a parameter with no arguments at all would be inventing
756/// one.
757fn redundant(
758    func: &Func,
759    block: Block,
760    ins: Option<&Vec<(Block, Idx<BlockCall>)>>,
761    forward: &HashMap<Value, Value>,
762) -> Vec<(usize, Value)> {
763    let Some(ins) = ins.filter(|ins| !ins.is_empty()) else { return Vec::new() };
764    let mut found = Vec::new();
765    for (index, &param) in func[block].params.iter().enumerate() {
766        let mut only = None;
767        let mut agree = true;
768        for &(_, at) in ins {
769            let list = func[at].args;
770            let Some(&arg) = func[list].get(index) else {
771                // Fewer arguments than parameters is a function the verifier will refuse, and
772                // guessing what the missing one was is not this pass's job.
773                agree = false;
774                break;
775            };
776            let arg = uses::chase(forward, arg);
777            if arg == param {
778                continue;
779            }
780            match only {
781                None => only = Some(arg),
782                Some(seen) if seen == arg => {}
783                Some(_) => {
784                    agree = false;
785                    break;
786                }
787            }
788        }
789        if !agree {
790            continue;
791        }
792        if let Some(value) = only {
793            found.push((index, value));
794        }
795    }
796    found
797}
798
799/// Drops those parameters of a block and the arguments in their places on every edge into it.
800///
801/// Both halves together, because a block whose parameters and arguments disagree in number is one
802/// the verifier refuses, and section 21.6 says that is the most common bug in this document.
803fn take_params(
804    func: &mut Func,
805    block: Block,
806    taking: &[usize],
807    ins: Option<&Vec<(Block, Idx<BlockCall>)>>,
808) {
809    for &(_, at) in ins.into_iter().flatten() {
810        let call = func[at];
811        let kept: Vec<Value> = func[call.args]
812            .iter()
813            .enumerate()
814            .filter(|(index, _)| !taking.contains(index))
815            .map(|(_, &value)| value)
816            .collect();
817        let args = func.push_values(&kept);
818        func.set_block_call(at, BlockCall { args, ..call });
819    }
820    let mut index = 0;
821    func.retain_params(block, |_| {
822        let keep = !taking.contains(&index);
823        index += 1;
824        keep
825    });
826}
827
828/// Where a block forwards to and what it passes on, when it is a forwarder.
829///
830/// The conditions are in this module's documentation, and every one of them is a `None` here. What
831/// comes back is the terminator, the block below, and the arguments the jump was carrying, which
832/// are what each of the block's predecessors will be carrying instead.
833fn forwards(
834    func: &Func,
835    block: Block,
836    entry: Block,
837    addressed: &HashSet<Block>,
838    edges: &Edges,
839) -> Option<(Inst, Block, Vec<Value>)> {
840    if block == entry || addressed.contains(&block) || !func[block].params.is_empty() {
841        return None;
842    }
843    let term = func.terminator(block)?;
844    if func[term].opcode != Opcode::Jump {
845        return None;
846    }
847    // Nothing above the jump, which is what "no instructions" means once the jump is counted as
848    // one of them.
849    if func.insts(block).count() != 1 {
850        return None;
851    }
852    let call = func.successors(term).next()?;
853    if call.block == block {
854        return None;
855    }
856    if carrying(func, block, call.block, func[call.args].len(), edges) {
857        return None;
858    }
859    Some((term, call.block, func[call.args].to_vec()))
860}
861
862/// Whether taking this forwarder out would put arguments on an edge that has nowhere to move them.
863///
864/// An edge carries values when the block it arrives at takes parameters, and giving a parameter its
865/// value is a move that has to happen on the edge itself. An edge out of a block that goes two ways
866/// and into a block arrived at two ways has no block to put that move in, so the back end splits it
867/// and puts an empty block back on it, which is `rucc_codegen::split::critical`. A forwarder that
868/// carries arguments and whose predecessor branches is already that block, sitting in the place the
869/// layout wants it rather than at the end where the splitter has to append it. Taking it out and
870/// having it put back costs a jump and a longer live range, and the measurement on the corpus says
871/// it costs enough to see, so it is not taken out.
872///
873/// A forwarder that carries nothing is removed whatever the edges look like, because there is no
874/// move to find a place for and the splitter would leave the edge alone as well.
875fn carrying(func: &Func, block: Block, into: Block, args: usize, edges: &Edges) -> bool {
876    if args == 0 {
877        return false;
878    }
879    let ins = edges.get(&block).map_or(0, Vec::len);
880    let after = edges.get(&into).map_or(0, Vec::len) - 1 + ins;
881    if after < 2 {
882        return false;
883    }
884    edges.get(&block).into_iter().flatten().any(|&(from, _)| {
885        let Some(term) = func.terminator(from) else { return false };
886        func.target_list(term).iter().count() >= 2
887    })
888}
889
890/// The runs of blocks that are one block written as several, head first.
891///
892/// Section 21.1's block merging, and the doc calls it a pure win for a reason worth stating: it
893/// does not delete an instruction or move one earlier, it takes a boundary out. Every analysis
894/// that is cheap inside a block and expensive across one gets more of the cheap kind, which is
895/// most of them, and the branch that stops being a branch is the smallest part of it.
896///
897/// A block goes into the one above it when the one above it ends in a jump and this is the only
898/// way in. Both halves are needed. One way in and a `br_if` above means the other arm would lose
899/// its terminator, and a jump above with two ways in means the second predecessor would arrive in
900/// the middle of a block.
901///
902/// The refusals are the entry block, which has to stay where control arrives even when one block
903/// jumps to it; a block that jumps to itself, whose one predecessor is itself; and a block whose
904/// address is taken, which is arrived at by an `indirect_br` the graph reads from the other end.
905///
906/// The answer is chains rather than pairs because a run of three is ordinary and the middle one
907/// stops existing partway through. Each block is the head of at most one of these and the tail of
908/// at most one, so what comes out is disjoint paths, and starting only from a head is what leaves
909/// a ring of blocks that all point at each other alone rather than walking it forever.
910fn chains(func: &Func, an: &mut Analyses) -> Vec<Vec<Block>> {
911    let cfg = an.cfg(func);
912    let Some(entry) = cfg.entry() else { return Vec::new() };
913    let addressed = addressed(func);
914    let mut below = HashMap::new();
915    let mut is_below = HashSet::new();
916    for block in func.blocks() {
917        let Some(term) = func.terminator(block) else { continue };
918        if func[term].opcode != Opcode::Jump {
919            continue;
920        }
921        let Some(call) = func.successors(term).next() else { continue };
922        let into = call.block;
923        let preds = cfg.predecessors(into);
924        if into == entry || into == block || addressed.contains(&into) {
925            continue;
926        }
927        if preds.len() != 1 || preds[0] != block {
928            continue;
929        }
930        below.insert(block, into);
931        is_below.insert(into);
932    }
933    let heads = func.blocks().filter(|it| below.contains_key(it) && !is_below.contains(it));
934    heads
935        .map(|head| {
936            let mut chain = vec![head];
937            let mut at = head;
938            while let Some(&next) = below.get(&at) {
939                chain.push(next);
940                at = next;
941            }
942            chain
943        })
944        .collect()
945}
946
947/// Every block some `block_addr` names.
948fn addressed(func: &Func) -> HashSet<Block> {
949    let mut taken = HashSet::new();
950    for block in func.blocks() {
951        for inst in func.insts(block) {
952            if func[inst].opcode != Opcode::BlockAddr {
953                continue;
954            }
955            for call in func.successors(inst) {
956                taken.insert(call.block);
957            }
958        }
959    }
960    taken
961}
962
963/// Moves everything in a block into the head of its chain and takes the block out of the function.
964///
965/// The jump is what is really being deleted, and the arguments it carried are what the merged
966/// block's parameters were going to be told. Binding each parameter to the argument in its place
967/// and pointing every reader at it is exactly what the jump was doing at run time, so the record
968/// goes in the map and the whole map is spent in one walk when the pass is done.
969fn merge(func: &mut Func, head: Block, block: Block, forward: &mut HashMap<Value, Value>) {
970    let term = func.terminator(head).expect("the head of a chain ends in a jump");
971    let call = func.successors(term).next().expect("a jump goes somewhere");
972    let args = func[call.args].to_vec();
973    let params = func[block].params.clone();
974    for (param, arg) in params.into_iter().zip(args) {
975        // Through whatever the merge above this one already decided, because a chain of three
976        // binds the middle block's parameter to something the head passed and then binds the last
977        // block's parameter to that same parameter.
978        let arg = uses::chase(forward, arg);
979        forward.insert(param, arg);
980    }
981    func.remove_inst(term);
982    for inst in func.insts(block).collect::<Vec<Inst>>() {
983        func.remove_inst(inst);
984        func.append_inst(head, inst);
985    }
986    func.remove_block(block);
987}
988
989/// Whether this condition is always true or always false, given what the edge binds.
990fn known(func: &Func, value: Value, subst: &Bindings) -> Option<bool> {
991    let value = resolve(subst, value);
992    if let Some((imm, _)) = constant(func, value) {
993        return Some(imm.unsigned() != 0);
994    }
995    compared(func, value, subst)
996}
997
998/// What a comparison of two constants comes out as.
999///
1000/// The comparison itself is never rewritten here, and it does not have to be. [`crate::fold`]
1001/// evaluates one whose operands are both already constants, so what is left for this is the case
1002/// folding cannot see: an operand that is a constant only along the edge being followed, which is
1003/// what `subst` carries. That is the whole reason this is still a question worth asking after
1004/// folding has run.
1005///
1006/// The arithmetic is [`crate::fold::compare`] rather than a copy of it, so the two places cannot
1007/// come to differ about what `slt` means.
1008fn compared(func: &Func, value: Value, subst: &Bindings) -> Option<bool> {
1009    let Def::Result { inst, .. } = func[value].def else { return None };
1010    let data = &func[inst];
1011    if data.opcode != Opcode::ICmp {
1012        return None;
1013    }
1014    let Extra::IntPred(pred) = data.extra else { return None };
1015    let args = &func[data.args];
1016    let (lhs, ty) = constant(func, resolve(subst, *args.first()?))?;
1017    let (rhs, _) = constant(func, resolve(subst, *args.get(1)?))?;
1018    Some(crate::fold::compare(pred, lhs, rhs, ty))
1019}
1020
1021#[cfg(test)]
1022mod tests {
1023    use rucc_base::Interner;
1024    use rucc_ir::{
1025        Block, Builder, Def, Flags, Func, Inst, IntPred, MemInfo, MemOrder, Module, Opcode,
1026        Restrict, Signature, Type, Value,
1027    };
1028    use rucc_target::{Arch, Env, Os, TargetInfo, Triple};
1029
1030    use super::SimplifyCfg;
1031    use crate::stats::Kind;
1032    use crate::testing::graph;
1033    use crate::{Fuel, Pass, Preserved, Stats};
1034
1035    /// Runs the pass with as much fuel as it wants.
1036    fn simplify(func: &mut Func) -> Stats {
1037        SimplifyCfg.run(func, &mut crate::machine::fixtures::analyses(), &mut Fuel::unlimited())
1038    }
1039
1040    /// The blocks the function still has, by number.
1041    fn blocks(func: &Func) -> Vec<usize> {
1042        func.blocks().map(Block::index).collect()
1043    }
1044
1045    /// The opcode of a block's terminator.
1046    fn terminator(func: &Func, block: usize) -> Opcode {
1047        let block = Block::from_usize(block);
1048        func[func.terminator(block).expect("every block here has one")].opcode
1049    }
1050
1051    /// Where a block's terminator goes, as block numbers.
1052    fn goes_to(func: &Func, block: usize) -> Vec<usize> {
1053        let block = Block::from_usize(block);
1054        let term = func.terminator(block).expect("every block here has one");
1055        func.successors(term).map(|call| call.block.index()).collect()
1056    }
1057
1058    /// The block the instruction that produced a value is in now, if it is in one.
1059    ///
1060    /// Which arm of a branch survived is a question about where its code ended up rather than
1061    /// about the shape of the graph, because the arm that survives is merged into the block above
1062    /// it in the same run and the two blocks stop being two.
1063    fn lives_in(func: &Func, value: Value) -> Option<usize> {
1064        let Def::Result { inst, .. } = func[value].def else { return None };
1065        func.block_of(inst).map(Block::index)
1066    }
1067
1068    /// A function with an entry, a `br_if` on `cond`, two arms and a join.
1069    ///
1070    /// The condition is built by the caller out of the builder it is handed, which is what lets
1071    /// one shape stand for a constant, a comparison and a value nothing knows anything about. Each
1072    /// arm holds one instruction that does nothing, which is there to be told apart from the one
1073    /// in the other arm, and the two of them come back with the function.
1074    fn diamond(cond: impl FnOnce(&mut Builder<'_>) -> Value) -> (Func, [Value; 2]) {
1075        let mut names = Interner::new();
1076        let mut func = Func::new(names.intern("f"), Signature::new());
1077        let entry = func.create_block();
1078        let then_block = func.create_block();
1079        let else_block = func.create_block();
1080        let join = func.create_block();
1081        let mut build = Builder::new(&mut func, entry);
1082        let cond = cond(&mut build);
1083        build.br_if(cond, then_block, &[], else_block, &[]);
1084        let mut marks = Vec::new();
1085        for (arm, mark) in [(then_block, 111), (else_block, 222)] {
1086            let mut build = Builder::new(&mut func, arm);
1087            marks.push(build.iconst(Type::int(32), mark));
1088            build.jump(join, &[]);
1089        }
1090        let mut build = Builder::new(&mut func, join);
1091        build.ret(&[]);
1092        (func, [marks[0], marks[1]])
1093    }
1094
1095    #[test]
1096    fn a_branch_on_a_true_constant_becomes_a_jump_to_the_first_arm() {
1097        let (mut func, [taken, other]) = diamond(|build| build.iconst(Type::int(1), 1));
1098        let stats = simplify(&mut func);
1099        assert!(stats.changed());
1100        assert_eq!(stats.count(Kind::Optimized, super::FOLDED), 1);
1101        // The arm it did not take is gone, because nothing else went there, and the arm it did
1102        // take had one way in and went into the entry along with the join below it.
1103        assert_eq!(stats.count(Kind::Optimized, super::REMOVED), 1);
1104        assert_eq!(stats.count(Kind::Optimized, super::MERGED), 2);
1105        assert_eq!(lives_in(&func, taken), Some(0));
1106        assert_eq!(lives_in(&func, other), None);
1107        assert_eq!(blocks(&func), [0]);
1108    }
1109
1110    #[test]
1111    fn a_branch_on_a_false_constant_becomes_a_jump_to_the_second_arm() {
1112        let (mut func, [other, taken]) = diamond(|build| build.iconst(Type::int(1), 0));
1113        assert!(simplify(&mut func).changed());
1114        assert_eq!(lives_in(&func, taken), Some(0));
1115        assert_eq!(lives_in(&func, other), None);
1116        assert_eq!(blocks(&func), [0]);
1117    }
1118
1119    #[test]
1120    fn folding_a_branch_and_merging_what_it_leaves_are_two_things_fuel_buys_apart() {
1121        // The same function as the test above, with fuel for the fold and nothing after it. The
1122        // jump is there to be seen, which is the shape the merge would otherwise take away.
1123        let (mut func, _) = diamond(|build| build.iconst(Type::int(1), 1));
1124        let stats =
1125            SimplifyCfg.run(&mut func, &mut crate::machine::fixtures::analyses(), &mut Fuel::of(1));
1126        assert_eq!(terminator(&func, 0), Opcode::Jump);
1127        assert_eq!(goes_to(&func, 0), [1]);
1128        assert_eq!(blocks(&func), [0, 1, 3]);
1129        assert_eq!(stats.count(Kind::Optimized, super::MERGED), 0);
1130        // Both blocks of the chain, because a block only reaches the head once the block between
1131        // them has, so running out before the first one means neither.
1132        assert_eq!(stats.count(Kind::Missed, super::NO_FUEL_MERGE), 2);
1133    }
1134
1135    #[test]
1136    fn a_branch_on_a_comparison_of_two_constants_is_read_without_folding_it() {
1137        // Both ways round on every predicate, which is where a sign error or an inverted
1138        // comparison would hide. A comparison the pass reads is left standing, because folding
1139        // it would produce an `i1` on its own and issue 352 says that does not lower.
1140        let cases: &[(IntPred, i128, i128, bool)] = &[
1141            (IntPred::Eq, 7, 7, true),
1142            (IntPred::Eq, 7, 8, false),
1143            (IntPred::Ne, 7, 8, true),
1144            (IntPred::Ne, 7, 7, false),
1145            (IntPred::Slt, -1, 1, true),
1146            (IntPred::Slt, 1, -1, false),
1147            (IntPred::Sle, -1, -1, true),
1148            (IntPred::Sle, 1, -1, false),
1149            (IntPred::Sgt, 1, -1, true),
1150            (IntPred::Sgt, -1, 1, false),
1151            (IntPred::Sge, -1, -1, true),
1152            (IntPred::Sge, -1, 1, false),
1153            (IntPred::Ult, 1, -1, true),
1154            (IntPred::Ult, -1, 1, false),
1155            (IntPred::Ule, -1, -1, true),
1156            (IntPred::Ule, -1, 1, false),
1157            (IntPred::Ugt, -1, 1, true),
1158            (IntPred::Ugt, 1, -1, false),
1159            (IntPred::Uge, -1, -1, true),
1160            (IntPred::Uge, 1, -1, false),
1161        ];
1162        for &(pred, lhs, rhs, taken) in cases {
1163            let (mut func, marks) = diamond(|build| {
1164                let lhs = build.iconst(Type::int(32), lhs);
1165                let rhs = build.iconst(Type::int(32), rhs);
1166                build.icmp(pred, lhs, rhs)
1167            });
1168            assert!(simplify(&mut func).changed(), "{pred:?} {lhs} {rhs}");
1169            let [went, gone] = if taken { [marks[0], marks[1]] } else { [marks[1], marks[0]] };
1170            assert_eq!(lives_in(&func, went), Some(0), "{pred:?} {lhs} {rhs}");
1171            assert_eq!(lives_in(&func, gone), None, "{pred:?} {lhs} {rhs}");
1172            let kept = func.insts(Block::from_usize(0)).any(|it| func[it].opcode == Opcode::ICmp);
1173            assert!(kept, "the comparison was folded away and issue 352 says it must not be");
1174        }
1175    }
1176
1177    #[test]
1178    fn a_branch_on_something_nobody_knows_is_left_alone() {
1179        let mut names = Interner::new();
1180        let mut func = Func::new(names.intern("f"), Signature::new().with_params(&[Type::int(1)]));
1181        let entry = func.create_block();
1182        let then_block = func.create_block();
1183        let else_block = func.create_block();
1184        let cond = func.append_param(entry, Type::int(1));
1185        let mut build = Builder::new(&mut func, entry);
1186        build.br_if(cond, then_block, &[], else_block, &[]);
1187        for arm in [then_block, else_block] {
1188            let mut build = Builder::new(&mut func, arm);
1189            build.ret(&[]);
1190        }
1191        let stats = simplify(&mut func);
1192        assert!(!stats.changed());
1193        assert!(stats.is_empty(), "a pass with nothing to say should say nothing");
1194        assert_eq!(terminator(&func, 0), Opcode::BrIf);
1195        assert_eq!(blocks(&func), [0, 1, 2]);
1196    }
1197
1198    /// A function whose entry switches on a constant, with a marker in the default and in each
1199    /// case, in that order.
1200    fn switched(on: i128, cases: &[i128]) -> (Func, Vec<Value>) {
1201        let mut names = Interner::new();
1202        let mut func = Func::new(names.intern("f"), Signature::new());
1203        let entry = func.create_block();
1204        let arms: Vec<Block> = (0..=cases.len()).map(|_| func.create_block()).collect();
1205        let mut build = Builder::new(&mut func, entry);
1206        let value = build.iconst(Type::int(32), on);
1207        let pairs: Vec<(i128, Block)> =
1208            cases.iter().enumerate().map(|(at, &case)| (case, arms[at + 1])).collect();
1209        build.switch(value, arms[0], &pairs);
1210        let mut marks = Vec::new();
1211        for (at, &arm) in arms.iter().enumerate() {
1212            let mut build = Builder::new(&mut func, arm);
1213            marks.push(build.iconst(Type::int(32), 100 + at as i128));
1214            build.ret(&[]);
1215        }
1216        (func, marks)
1217    }
1218
1219    #[test]
1220    fn a_switch_on_a_constant_takes_the_case_that_matches() {
1221        let (mut func, marks) = switched(5, &[4, 5]);
1222        assert!(simplify(&mut func).changed());
1223        assert_eq!(lives_in(&func, marks[2]), Some(0));
1224        assert_eq!(lives_in(&func, marks[0]), None);
1225        assert_eq!(lives_in(&func, marks[1]), None);
1226        assert_eq!(blocks(&func), [0]);
1227    }
1228
1229    #[test]
1230    fn a_switch_on_a_constant_no_case_names_takes_the_default() {
1231        let (mut func, marks) = switched(9, &[4]);
1232        assert!(simplify(&mut func).changed());
1233        assert_eq!(lives_in(&func, marks[0]), Some(0));
1234        assert_eq!(lives_in(&func, marks[1]), None);
1235        assert_eq!(blocks(&func), [0]);
1236    }
1237
1238    #[test]
1239    fn the_arguments_travel_with_the_edge_that_survives() {
1240        // The whole reason there are no phi nodes: the argument is in the branch beside the
1241        // block it goes to, so the surviving arm brings its own and the other one leaves with
1242        // the edge it was on. Both arms name the same block, so this is also the case branch
1243        // simplification has to leave alone: one block and two edges, because the two edges say
1244        // different things.
1245        let mut names = Interner::new();
1246        let mut func = Func::new(names.intern("f"), Signature::new());
1247        let entry = func.create_block();
1248        let join = func.create_block();
1249        let param = func.append_param(join, Type::int(32));
1250        let mut build = Builder::new(&mut func, entry);
1251        let cond = build.iconst(Type::int(1), 0);
1252        let taken = build.iconst(Type::int(32), 11);
1253        let other = build.iconst(Type::int(32), 22);
1254        build.br_if(cond, join, &[other], join, &[taken]);
1255        let mut build = Builder::new(&mut func, join);
1256        build.ret(&[param]);
1257        assert!(simplify(&mut func).changed());
1258        // The jump took the edge that survived, and then the block below it had one way in and
1259        // came up, which is where the parameter stopped being a parameter: whatever read it reads
1260        // the argument that edge was carrying.
1261        assert_eq!(blocks(&func), [0]);
1262        let term = func.terminator(entry).expect("the entry has one");
1263        assert_eq!(func[func[term].args], [taken]);
1264        assert_ne!(func[func[term].args], [param]);
1265    }
1266
1267    #[test]
1268    fn a_branch_whose_arms_are_the_same_edge_becomes_a_jump() {
1269        // Section 21.1's branch simplification, which is about the targets rather than about the
1270        // condition: nothing here knows what `cond` is and it does not matter, because both ways
1271        // out arrive at the same place carrying the same thing.
1272        let mut names = Interner::new();
1273        let signature = Signature::new().with_params(&[Type::int(1)]);
1274        let mut func = Func::new(names.intern("f"), signature);
1275        let entry = func.create_block();
1276        let join = func.create_block();
1277        let cond = func.append_param(entry, Type::int(1));
1278        let mut build = Builder::new(&mut func, entry);
1279        build.br_if(cond, join, &[], join, &[]);
1280        let mut build = Builder::new(&mut func, join);
1281        build.ret(&[]);
1282        let stats = simplify(&mut func);
1283        assert_eq!(stats.count(Kind::Optimized, super::FOLDED), 1);
1284        assert_eq!(stats.count(Kind::Optimized, super::MERGED), 1);
1285        assert_eq!(blocks(&func), [0]);
1286        assert_eq!(terminator(&func, 0), Opcode::Return);
1287    }
1288
1289    #[test]
1290    fn a_switch_whose_cases_all_go_to_one_place_becomes_a_jump() {
1291        let mut names = Interner::new();
1292        let signature = Signature::new().with_params(&[Type::int(32)]);
1293        let mut func = Func::new(names.intern("f"), signature);
1294        let entry = func.create_block();
1295        let join = func.create_block();
1296        let value = func.append_param(entry, Type::int(32));
1297        let mut build = Builder::new(&mut func, entry);
1298        build.switch(value, join, &[(4, join), (5, join)]);
1299        let mut build = Builder::new(&mut func, join);
1300        build.ret(&[]);
1301        let stats = simplify(&mut func);
1302        assert_eq!(stats.count(Kind::Optimized, super::FOLDED), 1);
1303        assert_eq!(blocks(&func), [0]);
1304    }
1305
1306    #[test]
1307    fn a_branch_to_one_block_by_two_edges_that_differ_is_left_alone() {
1308        // One block and two edges. Folding would have to pick one of the two arguments, and
1309        // whichever it picked would be the wrong one half the time.
1310        let mut names = Interner::new();
1311        let signature = Signature::new().with_params(&[Type::int(1)]);
1312        let mut func = Func::new(names.intern("f"), signature);
1313        let entry = func.create_block();
1314        let join = func.create_block();
1315        let cond = func.append_param(entry, Type::int(1));
1316        let param = func.append_param(join, Type::int(32));
1317        let mut build = Builder::new(&mut func, entry);
1318        let first = build.iconst(Type::int(32), 11);
1319        let second = build.iconst(Type::int(32), 22);
1320        build.br_if(cond, join, &[first], join, &[second]);
1321        let mut build = Builder::new(&mut func, join);
1322        // Returned rather than dropped, because a parameter nobody reads is one the step that
1323        // takes those out would take, and this test is about the branch above it.
1324        build.ret(&[param]);
1325        let stats = simplify(&mut func);
1326        assert!(!stats.changed());
1327        assert_eq!(terminator(&func, 0), Opcode::BrIf);
1328        assert_eq!(blocks(&func), [0, 1]);
1329    }
1330
1331    #[test]
1332    fn a_block_the_dead_arm_shared_with_a_live_one_stays() {
1333        // Issue 359 in the small. The block holding `bar` is inside the body of the dead `if`
1334        // and is a `case` of the switch as well, so the arm goes and the block does not.
1335        let mut names = Interner::new();
1336        let mut func = Func::new(names.intern("f"), Signature::new().with_params(&[Type::int(32)]));
1337        let entry = func.create_block();
1338        let dead = func.create_block();
1339        let shared = func.create_block();
1340        let exit = func.create_block();
1341        let x = func.append_param(entry, Type::int(32));
1342        let mut build = Builder::new(&mut func, entry);
1343        let never = build.iconst(Type::int(1), 0);
1344        build.switch(x, exit, &[(0, dead), (1, shared)]);
1345        // The `if (0)` inside the first case, whose body is where the second case's label sits.
1346        // The constant is the body of the arm that survives, and it is there so that the block is
1347        // a block with something in it rather than a forwarder that step three points past.
1348        let mut build = Builder::new(&mut func, dead);
1349        build.iconst(Type::int(32), 1);
1350        build.br_if(never, shared, &[], exit, &[]);
1351        for arm in [shared, exit] {
1352            let mut build = Builder::new(&mut func, arm);
1353            build.ret(&[]);
1354        }
1355        let stats = simplify(&mut func);
1356        assert!(stats.changed());
1357        // The switch is on a parameter, so it stays. The branch inside the dead arm folds to the
1358        // exit, and nothing is removed at all, because the shared block is still a case.
1359        assert_eq!(terminator(&func, 0), Opcode::Switch);
1360        assert_eq!(goes_to(&func, 1), [3]);
1361        assert_eq!(blocks(&func), [0, 1, 2, 3]);
1362        assert_eq!(stats.count(Kind::Optimized, super::REMOVED), 0);
1363    }
1364
1365    #[test]
1366    fn a_block_whose_address_is_taken_is_not_removed() {
1367        // Reachability here has to be the verifier's reachability. The graph does not carry the
1368        // edge from a `block_addr` to the block it names, and a pass that removed the block
1369        // under one would leave an instruction pointing at nothing.
1370        let mut names = Interner::new();
1371        let mut func = Func::new(names.intern("f"), Signature::new());
1372        let entry = func.create_block();
1373        let labelled = func.create_block();
1374        let arm = func.create_block();
1375        let mut build = Builder::new(&mut func, entry);
1376        let cond = build.iconst(Type::int(1), 1);
1377        let addr = build.block_addr(labelled);
1378        build.br_if(cond, arm, &[], labelled, &[]);
1379        let mut build = Builder::new(&mut func, arm);
1380        build.indirect_br(addr, &[labelled]);
1381        let mut build = Builder::new(&mut func, labelled);
1382        build.ret(&[]);
1383        assert!(simplify(&mut func).changed());
1384        assert!(blocks(&func).contains(&1), "the labelled block went with the arm");
1385        // The arm had one way in and came up into the entry, which is where the `indirect_br`
1386        // that reaches the labelled block is now.
1387        assert_eq!(blocks(&func), [0, 1]);
1388        assert_eq!(goes_to(&func, 0), [1]);
1389    }
1390
1391    #[test]
1392    fn a_block_only_an_unreachable_block_takes_the_address_of_goes_too() {
1393        // The other half of the same rule. Once the block holding the `block_addr` is gone, the
1394        // address is gone with it, and the block it named is reached by nothing.
1395        let mut names = Interner::new();
1396        let mut func = Func::new(names.intern("f"), Signature::new());
1397        let entry = func.create_block();
1398        let dead = func.create_block();
1399        let labelled = func.create_block();
1400        let mut build = Builder::new(&mut func, entry);
1401        let cond = build.iconst(Type::int(1), 1);
1402        build.br_if(cond, entry, &[], dead, &[]);
1403        let mut build = Builder::new(&mut func, dead);
1404        let addr = build.block_addr(labelled);
1405        build.indirect_br(addr, &[labelled]);
1406        let mut build = Builder::new(&mut func, labelled);
1407        build.ret(&[]);
1408        assert!(simplify(&mut func).changed());
1409        assert_eq!(blocks(&func), [0]);
1410    }
1411
1412    #[test]
1413    fn a_block_nothing_reaches_goes_even_when_no_branch_folded() {
1414        // Section 6.5 says this pass is the one that deletes them, and it says so about the
1415        // blocks the front end handed over as well as the ones a fold here stranded. Nothing in
1416        // this function folds, and the block still has to go, because every analysis below reads
1417        // the graph as though it is not there.
1418        let mut func = graph(&[&[], &[]]);
1419        let stats = simplify(&mut func);
1420        assert_eq!(stats.count(Kind::Optimized, super::FOLDED), 0);
1421        assert_eq!(stats.count(Kind::Optimized, super::REMOVED), 1);
1422        assert_eq!(blocks(&func), [0]);
1423    }
1424
1425    #[test]
1426    fn a_block_with_one_way_into_it_goes_into_the_block_above_it() {
1427        // Something in each of the first two blocks, so that this is three blocks for the merge
1428        // rather than two forwarders step three would point past before it got here.
1429        let mut names = Interner::new();
1430        let mut func = Func::new(names.intern("f"), Signature::new());
1431        let entry = func.create_block();
1432        let middle = func.create_block();
1433        let last = func.create_block();
1434        let mut build = Builder::new(&mut func, entry);
1435        build.iconst(Type::int(32), 1);
1436        build.jump(middle, &[]);
1437        let mut build = Builder::new(&mut func, middle);
1438        build.iconst(Type::int(32), 2);
1439        build.jump(last, &[]);
1440        let mut build = Builder::new(&mut func, last);
1441        build.ret(&[]);
1442        let stats = simplify(&mut func);
1443        // A run of three is one chain and not two rounds of one pair, because the block in the
1444        // middle stops being a block partway through.
1445        assert_eq!(stats.count(Kind::Optimized, super::MERGED), 2);
1446        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 0);
1447        assert_eq!(blocks(&func), [0]);
1448        assert_eq!(terminator(&func, 0), Opcode::Return);
1449    }
1450
1451    #[test]
1452    fn a_block_with_two_ways_into_it_stays_where_it_is() {
1453        // The join of a diamond nobody can fold. Merging it into either arm would leave the other
1454        // arm branching into the middle of a block.
1455        let mut names = Interner::new();
1456        let signature = Signature::new().with_params(&[Type::int(1)]);
1457        let mut func = Func::new(names.intern("f"), signature);
1458        let entry = func.create_block();
1459        let then_block = func.create_block();
1460        let else_block = func.create_block();
1461        let join = func.create_block();
1462        let cond = func.append_param(entry, Type::int(1));
1463        let mut build = Builder::new(&mut func, entry);
1464        build.br_if(cond, then_block, &[], else_block, &[]);
1465        for (arm, mark) in [(then_block, 111), (else_block, 222)] {
1466            // An arm with something in it, because an empty one is a forwarder and step three
1467            // would take it away before merging ever looked at the join.
1468            let mut build = Builder::new(&mut func, arm);
1469            build.iconst(Type::int(32), mark);
1470            build.jump(join, &[]);
1471        }
1472        let mut build = Builder::new(&mut func, join);
1473        build.ret(&[]);
1474        let stats = simplify(&mut func);
1475        assert!(!stats.changed());
1476        assert_eq!(blocks(&func), [0, 1, 2, 3]);
1477    }
1478
1479    #[test]
1480    fn a_block_above_one_that_does_not_end_in_a_jump_keeps_it() {
1481        // One way into the join, and the block above it is a branch. Merging would take the
1482        // terminator off the other arm.
1483        let mut names = Interner::new();
1484        let signature = Signature::new().with_params(&[Type::int(1)]);
1485        let mut func = Func::new(names.intern("f"), signature);
1486        let entry = func.create_block();
1487        let arm = func.create_block();
1488        let exit = func.create_block();
1489        let cond = func.append_param(entry, Type::int(1));
1490        let mut build = Builder::new(&mut func, entry);
1491        build.br_if(cond, arm, &[], exit, &[]);
1492        for block in [arm, exit] {
1493            let mut build = Builder::new(&mut func, block);
1494            build.ret(&[]);
1495        }
1496        let stats = simplify(&mut func);
1497        assert!(!stats.changed());
1498        assert_eq!(blocks(&func), [0, 1, 2]);
1499    }
1500
1501    #[test]
1502    fn the_entry_block_is_never_the_one_that_moves() {
1503        // A loop back to the entry, so the entry has one way in and the block above it ends in a
1504        // jump, which is every condition but the one that matters. Control arrives at the entry
1505        // and it has to still be there when it does.
1506        let mut names = Interner::new();
1507        let signature = Signature::new().with_params(&[Type::int(1)]);
1508        let mut func = Func::new(names.intern("f"), signature);
1509        let entry = func.create_block();
1510        let latch = func.create_block();
1511        let exit = func.create_block();
1512        let cond = func.append_param(entry, Type::int(1));
1513        let mut build = Builder::new(&mut func, entry);
1514        build.br_if(cond, latch, &[], exit, &[]);
1515        // The body of the loop, which is there so that the latch is a block and not a forwarder.
1516        let mut build = Builder::new(&mut func, latch);
1517        build.iconst(Type::int(32), 1);
1518        build.jump(entry, &[]);
1519        let mut build = Builder::new(&mut func, exit);
1520        build.ret(&[]);
1521        let stats = simplify(&mut func);
1522        assert!(!stats.changed());
1523        assert_eq!(blocks(&func), [0, 1, 2]);
1524    }
1525
1526    #[test]
1527    fn a_block_whose_address_is_taken_is_not_merged_away_either() {
1528        // The same rule as the one about deleting it. Merging it into the block above would take
1529        // the block out of the function, and the `block_addr` would name one that is not there.
1530        let mut names = Interner::new();
1531        let mut func = Func::new(names.intern("f"), Signature::new());
1532        let entry = func.create_block();
1533        let middle = func.create_block();
1534        let labelled = func.create_block();
1535        let mut build = Builder::new(&mut func, entry);
1536        build.block_addr(labelled);
1537        build.jump(middle, &[]);
1538        // Something in the middle block, so that this is a question about merging rather than one
1539        // about the forwarder removal that would otherwise get there first.
1540        let mut build = Builder::new(&mut func, middle);
1541        build.iconst(Type::int(32), 1);
1542        build.jump(labelled, &[]);
1543        let mut build = Builder::new(&mut func, labelled);
1544        build.ret(&[]);
1545        let stats = simplify(&mut func);
1546        // The middle block had one way in and no address, so it came up. The labelled block has
1547        // one way in too, and stayed.
1548        assert_eq!(stats.count(Kind::Optimized, super::MERGED), 1);
1549        assert_eq!(blocks(&func), [0, 2]);
1550    }
1551
1552    #[test]
1553    fn merging_binds_a_block_parameter_to_the_argument_the_jump_carried() {
1554        let mut names = Interner::new();
1555        let mut func = Func::new(names.intern("f"), Signature::new());
1556        let entry = func.create_block();
1557        let below = func.create_block();
1558        let param = func.append_param(below, Type::int(32));
1559        let mut build = Builder::new(&mut func, entry);
1560        let arg = build.iconst(Type::int(32), 7);
1561        build.jump(below, &[arg]);
1562        let mut build = Builder::new(&mut func, below);
1563        build.ret(&[param]);
1564        assert!(simplify(&mut func).changed());
1565        assert_eq!(blocks(&func), [0]);
1566        let term = func.terminator(entry).expect("the entry has one");
1567        assert_eq!(func[func[term].args], [arg]);
1568    }
1569
1570    #[test]
1571    fn a_chain_of_merges_follows_a_parameter_bound_to_a_parameter() {
1572        // The middle block passes its own parameter down, so the last block's parameter is bound
1573        // to something that is on its way to being the entry's constant. Following the map is
1574        // what makes the second merge worth as much as the first.
1575        let mut names = Interner::new();
1576        let mut func = Func::new(names.intern("f"), Signature::new());
1577        let entry = func.create_block();
1578        let middle = func.create_block();
1579        let last = func.create_block();
1580        let carried = func.append_param(middle, Type::int(32));
1581        let arrived = func.append_param(last, Type::int(32));
1582        let mut build = Builder::new(&mut func, entry);
1583        let arg = build.iconst(Type::int(32), 7);
1584        build.jump(middle, &[arg]);
1585        let mut build = Builder::new(&mut func, middle);
1586        build.jump(last, &[carried]);
1587        let mut build = Builder::new(&mut func, last);
1588        build.ret(&[arrived]);
1589        assert!(simplify(&mut func).changed());
1590        assert_eq!(blocks(&func), [0]);
1591        let term = func.terminator(entry).expect("the entry has one");
1592        assert_eq!(func[func[term].args], [arg]);
1593    }
1594
1595    /// A function whose entry branches on its own parameter into two arms that each hold one
1596    /// instruction and then jump where the caller says, head to tail.
1597    ///
1598    /// Two arms rather than one because almost every question about step three is a question about
1599    /// a block with more than one way in, and something in each arm because an empty arm is itself
1600    /// a forwarder and would answer a different question. The blocks are entry 0, the arms 1 and 2,
1601    /// and whatever the caller builds after that.
1602    fn arms(func: &mut Func) -> (Value, [Block; 2]) {
1603        let entry = func.create_block();
1604        let first = func.create_block();
1605        let second = func.create_block();
1606        let cond = func.append_param(entry, Type::int(1));
1607        let mut build = Builder::new(func, entry);
1608        let carried = build.iconst(Type::int(32), 7);
1609        build.br_if(cond, first, &[], second, &[]);
1610        for (arm, mark) in [(first, 111), (second, 222)] {
1611            let mut build = Builder::new(func, arm);
1612            build.iconst(Type::int(32), mark);
1613        }
1614        (carried, [first, second])
1615    }
1616
1617    /// A function with one `i1` parameter, which is what [`arms`] wants.
1618    fn taking_a_condition() -> Func {
1619        let mut names = Interner::new();
1620        let signature = Signature::new().with_params(&[Type::int(1)]);
1621        Func::new(names.intern("f"), signature)
1622    }
1623
1624    /// The arguments a block's terminator passes on the edge in that place.
1625    fn carries(func: &Func, block: usize, edge: usize) -> Vec<Value> {
1626        let block = Block::from_usize(block);
1627        let term = func.terminator(block).expect("every block here has one");
1628        let call = func.successors(term).nth(edge).expect("the edge is there");
1629        func[call.args].to_vec()
1630    }
1631
1632    #[test]
1633    fn a_block_that_does_nothing_but_jump_stops_being_in_the_way() {
1634        // Section 21.1's edge forwarding. Two arms arrive at a block that only jumps, so the two
1635        // of them go where it was going and it is not there any more.
1636        let mut func = taking_a_condition();
1637        let (_, arms) = arms(&mut func);
1638        let forwarder = func.create_block();
1639        let exit = func.create_block();
1640        for arm in arms {
1641            Builder::new(&mut func, arm).jump(forwarder, &[]);
1642        }
1643        Builder::new(&mut func, forwarder).jump(exit, &[]);
1644        Builder::new(&mut func, exit).ret(&[]);
1645        let stats = simplify(&mut func);
1646        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 1);
1647        assert_eq!(blocks(&func), [0, 1, 2, 4]);
1648        assert_eq!(goes_to(&func, 1), [4]);
1649        assert_eq!(goes_to(&func, 2), [4]);
1650    }
1651
1652    #[test]
1653    fn a_forwarder_hands_its_predecessors_the_arguments_it_was_passing() {
1654        // The forwarder was passing something, and taking it out means whoever ends up branching
1655        // to the block below has to pass it instead. Section 21.1's extra condition is about
1656        // exactly this, and a block with no parameters and no instructions cannot be where the
1657        // value came from, so there is nothing further to check. The one way in is through a block
1658        // that goes nowhere else, which keeps the edge off the list of ones that carry a move with
1659        // no block to put it in.
1660        let mut func = taking_a_condition();
1661        let (carried, [arm, above]) = arms(&mut func);
1662        let forwarder = func.create_block();
1663        let exit = func.create_block();
1664        let other = func.append_param(exit, Type::int(32));
1665        let mut build = Builder::new(&mut func, arm);
1666        let mine = build.iconst(Type::int(32), 9);
1667        build.jump(exit, &[mine]);
1668        Builder::new(&mut func, above).jump(forwarder, &[]);
1669        Builder::new(&mut func, forwarder).jump(exit, &[carried]);
1670        Builder::new(&mut func, exit).ret(&[other]);
1671        let stats = simplify(&mut func);
1672        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 1);
1673        assert_eq!(blocks(&func), [0, 1, 2, 4]);
1674        // The block above the forwarder is the edge that used to go through it, and it is carrying
1675        // what the forwarder was carrying.
1676        assert_eq!(carries(&func, 2, 0), [carried]);
1677        assert_eq!(carries(&func, 1, 0), [mine]);
1678        // Two edges saying different things, so the parameter is not redundant and stays.
1679        assert_eq!(stats.count(Kind::Optimized, super::SAME_EVERY_WAY), 0);
1680    }
1681
1682    #[test]
1683    fn a_forwarder_carrying_something_on_an_edge_out_of_a_branch_stays() {
1684        // Both ways out of the entry end up at the same block, and that block takes a parameter, so
1685        // the edge through the forwarder is one the back end would have to split again the moment
1686        // the forwarder stopped being there. The block is already the split, in the place the
1687        // layout wants it, so it is left where it is.
1688        let mut func = taking_a_condition();
1689        let (carried, [arm, forwarder]) = arms(&mut func);
1690        let exit = func.create_block();
1691        let other = func.append_param(exit, Type::int(32));
1692        // The second arm is emptied back out, which is what makes it a forwarder at all.
1693        for inst in func.insts(forwarder).collect::<Vec<Inst>>() {
1694            func.remove_inst(inst);
1695        }
1696        let mut build = Builder::new(&mut func, arm);
1697        let mine = build.iconst(Type::int(32), 9);
1698        build.jump(exit, &[mine]);
1699        Builder::new(&mut func, forwarder).jump(exit, &[carried]);
1700        Builder::new(&mut func, exit).ret(&[other]);
1701        let stats = simplify(&mut func);
1702        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 0);
1703        assert_eq!(blocks(&func), [0, 1, 2, 3]);
1704    }
1705
1706    #[test]
1707    fn a_forwarder_carrying_nothing_out_of_a_branch_goes_anyway() {
1708        // The same shape with nothing on the edge. There is no move to find a place for, so the
1709        // back end would leave the edge alone and the block is only in the way.
1710        let mut func = taking_a_condition();
1711        let (_, [arm, forwarder]) = arms(&mut func);
1712        let exit = func.create_block();
1713        for inst in func.insts(forwarder).collect::<Vec<Inst>>() {
1714            func.remove_inst(inst);
1715        }
1716        Builder::new(&mut func, arm).jump(exit, &[]);
1717        Builder::new(&mut func, forwarder).jump(exit, &[]);
1718        Builder::new(&mut func, exit).ret(&[]);
1719        let stats = simplify(&mut func);
1720        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 1);
1721        assert_eq!(blocks(&func), [0, 1, 3]);
1722    }
1723
1724    #[test]
1725    fn a_block_that_jumps_to_itself_is_not_a_forwarder() {
1726        // Section 21.1 says so in as many words, and the reason is that it does not forward
1727        // anywhere: pointing its predecessors past it would have to point them at it.
1728        let mut names = Interner::new();
1729        let mut func = Func::new(names.intern("f"), Signature::new());
1730        let entry = func.create_block();
1731        let spin = func.create_block();
1732        Builder::new(&mut func, entry).jump(spin, &[]);
1733        Builder::new(&mut func, spin).jump(spin, &[]);
1734        let stats = simplify(&mut func);
1735        assert!(!stats.changed());
1736        assert_eq!(blocks(&func), [0, 1]);
1737    }
1738
1739    #[test]
1740    fn the_entry_block_is_never_the_forwarder_that_goes() {
1741        // The entry doing nothing but jumping is every condition of a forwarder except the one
1742        // that matters. What happens instead is the block below coming up into it, which leaves
1743        // control arriving where it has to arrive.
1744        let mut names = Interner::new();
1745        let mut func = Func::new(names.intern("f"), Signature::new());
1746        let entry = func.create_block();
1747        let below = func.create_block();
1748        Builder::new(&mut func, entry).jump(below, &[]);
1749        let mut build = Builder::new(&mut func, below);
1750        build.iconst(Type::int(32), 1);
1751        build.ret(&[]);
1752        let stats = simplify(&mut func);
1753        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 0);
1754        assert_eq!(stats.count(Kind::Optimized, super::MERGED), 1);
1755        assert_eq!(blocks(&func), [0]);
1756    }
1757
1758    #[test]
1759    fn a_block_whose_address_is_taken_is_not_forwarded_past_either() {
1760        // The abnormal edge condition, which in this IR is the edge an `indirect_br` takes. The
1761        // block is arrived at from somewhere the graph reads from the other end, and pointing the
1762        // edges the graph does carry past it would not move that one.
1763        let mut names = Interner::new();
1764        let mut func = Func::new(names.intern("f"), Signature::new());
1765        let entry = func.create_block();
1766        let labelled = func.create_block();
1767        let exit = func.create_block();
1768        let mut build = Builder::new(&mut func, entry);
1769        let addr = build.block_addr(labelled);
1770        build.indirect_br(addr, &[labelled]);
1771        Builder::new(&mut func, labelled).jump(exit, &[]);
1772        let mut build = Builder::new(&mut func, exit);
1773        build.iconst(Type::int(32), 1);
1774        build.ret(&[]);
1775        let stats = simplify(&mut func);
1776        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 0);
1777        assert!(blocks(&func).contains(&1), "the labelled block was forwarded past");
1778    }
1779
1780    #[test]
1781    fn a_run_of_forwarders_comes_out_as_one_edge() {
1782        let mut func = taking_a_condition();
1783        let (_, arms) = arms(&mut func);
1784        let first = func.create_block();
1785        let second = func.create_block();
1786        let exit = func.create_block();
1787        for arm in arms {
1788            Builder::new(&mut func, arm).jump(first, &[]);
1789        }
1790        Builder::new(&mut func, first).jump(second, &[]);
1791        Builder::new(&mut func, second).jump(exit, &[]);
1792        Builder::new(&mut func, exit).ret(&[]);
1793        let stats = simplify(&mut func);
1794        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 2);
1795        assert_eq!(blocks(&func), [0, 1, 2, 5]);
1796        assert_eq!(goes_to(&func, 1), [5]);
1797        assert_eq!(goes_to(&func, 2), [5]);
1798    }
1799
1800    #[test]
1801    fn a_block_parameter_that_arrives_as_one_value_every_way_in_goes() {
1802        // Section 21.2. The parameter is not carrying anything, it is spelling the constant a
1803        // second way, and document 12 cannot see through the spelling.
1804        let mut func = taking_a_condition();
1805        let (carried, arms) = arms(&mut func);
1806        let join = func.create_block();
1807        let param = func.append_param(join, Type::int(32));
1808        for arm in arms {
1809            Builder::new(&mut func, arm).jump(join, &[carried]);
1810        }
1811        Builder::new(&mut func, join).ret(&[param]);
1812        let stats = simplify(&mut func);
1813        assert_eq!(stats.count(Kind::Optimized, super::SAME_EVERY_WAY), 1);
1814        assert!(func[Block::from_usize(3)].params.is_empty());
1815        // What read the parameter reads the value it was always going to be.
1816        let term = func.terminator(Block::from_usize(3)).expect("the join has one");
1817        assert_eq!(func[func[term].args], [carried]);
1818        // And the argument in its place is off both edges, because a branch that passes more
1819        // arguments than the block takes is one the verifier refuses.
1820        assert!(carries(&func, 1, 0).is_empty());
1821        assert!(carries(&func, 2, 0).is_empty());
1822    }
1823
1824    #[test]
1825    fn a_block_parameter_that_differs_on_one_way_in_stays() {
1826        let mut func = taking_a_condition();
1827        let (carried, arms) = arms(&mut func);
1828        let join = func.create_block();
1829        let param = func.append_param(join, Type::int(32));
1830        let mut build = Builder::new(&mut func, arms[0]);
1831        let mine = build.iconst(Type::int(32), 9);
1832        build.jump(join, &[mine]);
1833        Builder::new(&mut func, arms[1]).jump(join, &[carried]);
1834        Builder::new(&mut func, join).ret(&[param]);
1835        let stats = simplify(&mut func);
1836        assert!(!stats.changed());
1837        assert_eq!(func[Block::from_usize(3)].params, [param]);
1838    }
1839
1840    #[test]
1841    fn a_loop_header_parameter_whose_other_argument_is_itself_is_what_it_started_as() {
1842        // The subtlety section 21.2 spends its second paragraph on. The latch passes the
1843        // parameter back, so reading the arguments literally says two values and says leave it
1844        // alone. A value that can only ever be itself or the initial one was the initial one.
1845        let mut names = Interner::new();
1846        let signature = Signature::new().with_params(&[Type::int(1)]);
1847        let mut func = Func::new(names.intern("f"), signature);
1848        let entry = func.create_block();
1849        let header = func.create_block();
1850        let latch = func.create_block();
1851        let exit = func.create_block();
1852        let cond = func.append_param(entry, Type::int(1));
1853        let param = func.append_param(header, Type::int(32));
1854        let mut build = Builder::new(&mut func, entry);
1855        let init = build.iconst(Type::int(32), 7);
1856        build.jump(header, &[init]);
1857        Builder::new(&mut func, header).br_if(cond, latch, &[], exit, &[]);
1858        let mut build = Builder::new(&mut func, latch);
1859        build.iconst(Type::int(32), 1);
1860        build.jump(header, &[param]);
1861        Builder::new(&mut func, exit).ret(&[param]);
1862        let stats = simplify(&mut func);
1863        assert_eq!(stats.count(Kind::Optimized, super::SAME_EVERY_WAY), 1);
1864        assert!(func[Block::from_usize(1)].params.is_empty());
1865        let term = func.terminator(Block::from_usize(3)).expect("the exit has one");
1866        assert_eq!(func[func[term].args], [init]);
1867    }
1868
1869    #[test]
1870    fn the_entry_blocks_parameters_are_the_functions_and_stay() {
1871        // The entry's parameters arrive from the caller, which is a way in the graph has no edge
1872        // for. A branch back to the entry is one edge out of two, and reading it as though it
1873        // were the only one would replace an argument with whatever the loop happened to pass.
1874        let mut names = Interner::new();
1875        let signature = Signature::new().with_params(&[Type::int(1), Type::int(32)]);
1876        let mut func = Func::new(names.intern("f"), signature);
1877        let entry = func.create_block();
1878        let latch = func.create_block();
1879        let exit = func.create_block();
1880        let cond = func.append_param(entry, Type::int(1));
1881        let x = func.append_param(entry, Type::int(32));
1882        Builder::new(&mut func, entry).br_if(cond, latch, &[], exit, &[]);
1883        let mut build = Builder::new(&mut func, latch);
1884        let one = build.iconst(Type::int(1), 1);
1885        let seven = build.iconst(Type::int(32), 7);
1886        build.jump(entry, &[one, seven]);
1887        Builder::new(&mut func, exit).ret(&[x]);
1888        let stats = simplify(&mut func);
1889        assert!(!stats.changed());
1890        assert_eq!(func[Block::from_usize(0)].params, [cond, x]);
1891    }
1892
1893    #[test]
1894    fn taking_one_parameter_away_is_what_makes_the_next_one_redundant() {
1895        // Section 21.2's reason for a worklist. The last block's parameter arrives as the middle
1896        // block's parameter one way and as the constant the other way, which is two values until
1897        // the middle block's parameter turns out to be that same constant.
1898        let mut func = taking_a_condition();
1899        let (carried, arms) = arms(&mut func);
1900        let join = func.create_block();
1901        let inner = func.append_param(join, Type::int(32));
1902        let left = func.create_block();
1903        let right = func.create_block();
1904        let last = func.create_block();
1905        let outer = func.append_param(last, Type::int(32));
1906        for arm in arms {
1907            Builder::new(&mut func, arm).jump(join, &[carried]);
1908        }
1909        let cond = func[Block::from_usize(0)].params[0];
1910        Builder::new(&mut func, join).br_if(cond, left, &[], right, &[]);
1911        let mut build = Builder::new(&mut func, left);
1912        build.iconst(Type::int(32), 1);
1913        build.jump(last, &[inner]);
1914        let mut build = Builder::new(&mut func, right);
1915        build.iconst(Type::int(32), 2);
1916        build.jump(last, &[carried]);
1917        Builder::new(&mut func, last).ret(&[outer]);
1918        let stats = simplify(&mut func);
1919        assert_eq!(stats.count(Kind::Optimized, super::SAME_EVERY_WAY), 2);
1920        let term = func.terminator(Block::from_usize(6)).expect("the last block has one");
1921        assert_eq!(func[func[term].args], [carried]);
1922    }
1923
1924    #[test]
1925    fn a_forwarder_with_a_parameter_goes_once_the_parameter_does() {
1926        // The two halves of step three being one step. The block passes its own parameter on, so
1927        // it is not a forwarder while it has one, and the parameter is the same value both ways
1928        // in, so it does not have one for long.
1929        let mut func = taking_a_condition();
1930        let (carried, arms) = arms(&mut func);
1931        let forwarder = func.create_block();
1932        let param = func.append_param(forwarder, Type::int(32));
1933        let exit = func.create_block();
1934        let arrived = func.append_param(exit, Type::int(32));
1935        for arm in arms {
1936            Builder::new(&mut func, arm).jump(forwarder, &[carried]);
1937        }
1938        Builder::new(&mut func, forwarder).jump(exit, &[param]);
1939        Builder::new(&mut func, exit).ret(&[arrived]);
1940        let stats = simplify(&mut func);
1941        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 1);
1942        // Both of them: the forwarder's, which is what let it go, and the exit's, which arrives
1943        // as the same thing from both arms once the block between them is not there.
1944        assert_eq!(stats.count(Kind::Optimized, super::SAME_EVERY_WAY), 2);
1945        assert_eq!(blocks(&func), [0, 1, 2, 4]);
1946        let term = func.terminator(Block::from_usize(4)).expect("the exit has one");
1947        assert_eq!(func[func[term].args], [carried]);
1948    }
1949
1950    #[test]
1951    fn fuel_stops_step_three_the_same_way_it_stops_the_rest() {
1952        // One unit, and the first thing that asks for it is the parameter, because parameters go
1953        // first within a block. The forwarder then has nothing to spend and stays.
1954        let mut func = taking_a_condition();
1955        let (carried, arms) = arms(&mut func);
1956        let forwarder = func.create_block();
1957        let param = func.append_param(forwarder, Type::int(32));
1958        let exit = func.create_block();
1959        // Arriving at the exit and returned there, so that the forwarder's parameter is one
1960        // something reads and the step that takes the unread ones out leaves it alone.
1961        let arrived = func.append_param(exit, Type::int(32));
1962        for arm in arms {
1963            Builder::new(&mut func, arm).jump(forwarder, &[carried]);
1964        }
1965        Builder::new(&mut func, forwarder).jump(exit, &[param]);
1966        Builder::new(&mut func, exit).ret(&[arrived]);
1967        let stats =
1968            SimplifyCfg.run(&mut func, &mut crate::machine::fixtures::analyses(), &mut Fuel::of(1));
1969        assert_eq!(stats.count(Kind::Optimized, super::SAME_EVERY_WAY), 1);
1970        assert_eq!(stats.count(Kind::Optimized, super::FORWARDED), 0);
1971        assert_eq!(stats.count(Kind::Missed, super::NO_FUEL_FORWARD), 1);
1972        assert_eq!(blocks(&func), [0, 1, 2, 3, 4]);
1973    }
1974
1975    /// A loop that carries a counter and a pointer, and leaves when the pointer reaches `end`.
1976    ///
1977    /// The shape `crate::ivopts` produces once section 28.4 has moved the exit test off the
1978    /// counter: nothing asks the counter anything any more, and the only thing left reading it is
1979    /// the addition that produces what the latch hands back to it.
1980    ///
1981    /// `on_counter` puts the exit test back on the counter, which is the same loop with the
1982    /// counter live, and it is the negative half of every test below.
1983    fn walking_a_pointer(on_counter: bool) -> Func {
1984        let mut names = Interner::new();
1985        let signature = Signature::new().with_params(&[Type::int(64)]);
1986        let mut func = Func::new(names.intern("f"), signature);
1987        let entry = func.create_block();
1988        let head = func.create_block();
1989        let out = func.create_block();
1990        let end = func.append_param(entry, Type::int(64));
1991        let counter = func.append_param(head, Type::int(32));
1992        let pointer = func.append_param(head, Type::int(64));
1993        let mut build = Builder::new(&mut func, entry);
1994        let from_zero = build.iconst(Type::int(32), 0);
1995        let from_start = build.iconst(Type::int(64), 0);
1996        build.jump(head, &[from_zero, from_start]);
1997        let mut build = Builder::new(&mut func, head);
1998        let one = build.iconst(Type::int(32), 1);
1999        let eight = build.iconst(Type::int(64), 8);
2000        let next = build.binary(Opcode::Add, counter, one, Flags::NONE);
2001        let along = build.binary(Opcode::Add, pointer, eight, Flags::NONE);
2002        // The write the loop is there for, so that the pointer is read by something that happens
2003        // whichever way the exit test is written.
2004        let address = build.unary(Opcode::IntToPtr, pointer, Type::PTR);
2005        let info = MemInfo {
2006            size: 8,
2007            align: 8,
2008            order: MemOrder::NotAtomic,
2009            tbaa: None,
2010            owns: 0,
2011            restrict: Restrict::NONE,
2012        };
2013        build.store(eight, address, info, Flags::NONE);
2014        let going = if on_counter {
2015            let limit = build.iconst(Type::int(32), 10);
2016            build.icmp(IntPred::Ne, next, limit)
2017        } else {
2018            build.icmp(IntPred::Ne, along, end)
2019        };
2020        build.br_if(going, head, &[next, along], out, &[]);
2021        Builder::new(&mut func, out).ret(&[]);
2022        func
2023    }
2024
2025    #[test]
2026    fn a_counter_the_loop_stopped_asking_about_stops_going_round() {
2027        let mut func = walking_a_pointer(false);
2028        let stats = simplify(&mut func);
2029        assert_eq!(stats.count(Kind::Optimized, super::NOTHING_READS_IT), 1);
2030        // The pointer stays, because the test that decides whether to go round again reads it.
2031        assert_eq!(func[Block::from_usize(1)].params.len(), 1);
2032        // And the edge that was feeding the counter is carrying one value now instead of two.
2033        assert_eq!(carries(&func, 1, 0).len(), 1);
2034        assert_eq!(carries(&func, 0, 0).len(), 1);
2035    }
2036
2037    #[test]
2038    fn a_counter_the_loop_still_asks_about_goes_round_exactly_as_before() {
2039        let mut func = walking_a_pointer(true);
2040        let stats = simplify(&mut func);
2041        assert_eq!(stats.count(Kind::Optimized, super::NOTHING_READS_IT), 0);
2042        assert_eq!(func[Block::from_usize(1)].params.len(), 2);
2043    }
2044
2045    /// A counter nothing reads, with two instructions behind it rather than one.
2046    ///
2047    /// The addition reads the parameter and the doubling reads the addition, so taking the
2048    /// parameter out strands the first and taking the first out strands the second. The condition
2049    /// is asked about the function's own parameter so that nothing here folds and the loop stays a
2050    /// loop.
2051    fn counting_into_nothing() -> (Func, Value, Value) {
2052        let mut names = Interner::new();
2053        let signature = Signature::new().with_params(&[Type::int(32)]);
2054        let mut func = Func::new(names.intern("f"), signature);
2055        let entry = func.create_block();
2056        let head = func.create_block();
2057        let out = func.create_block();
2058        let limit = func.append_param(entry, Type::int(32));
2059        let counter = func.append_param(head, Type::int(32));
2060        let mut build = Builder::new(&mut func, entry);
2061        let zero = build.iconst(Type::int(32), 0);
2062        build.jump(head, &[zero]);
2063        let mut build = Builder::new(&mut func, head);
2064        let one = build.iconst(Type::int(32), 1);
2065        let next = build.binary(Opcode::Add, counter, one, Flags::NONE);
2066        let twice = build.binary(Opcode::Add, next, next, Flags::NONE);
2067        let going = build.icmp(IntPred::Ne, limit, one);
2068        build.br_if(going, head, &[next], out, &[]);
2069        Builder::new(&mut func, out).ret(&[]);
2070        (func, next, twice)
2071    }
2072
2073    #[test]
2074    fn what_was_reading_a_parameter_nothing_reads_goes_with_it() {
2075        let (mut func, next, twice) = counting_into_nothing();
2076        let stats = simplify(&mut func);
2077        assert_eq!(stats.count(Kind::Optimized, super::NOTHING_READS_IT), 1);
2078        // Both of them, and the second one is the point: it was never reading the parameter, it
2079        // was reading what did, so one walk that only took the direct readers would have left a
2080        // use of a value nothing defines. This is issue 1016.
2081        assert_eq!(lives_in(&func, next), None);
2082        assert_eq!(lives_in(&func, twice), None);
2083    }
2084
2085    #[test]
2086    fn the_functions_own_parameters_stay_whether_or_not_anything_reads_them() {
2087        // The entry's parameters are the signature. Nothing in this function reads the one it
2088        // has, and taking it out would be changing what the function is rather than what it does.
2089        let mut names = Interner::new();
2090        let signature = Signature::new().with_params(&[Type::int(32)]);
2091        let mut func = Func::new(names.intern("f"), signature);
2092        let entry = func.create_block();
2093        func.append_param(entry, Type::int(32));
2094        Builder::new(&mut func, entry).ret(&[]);
2095        let stats = simplify(&mut func);
2096        assert_eq!(stats.count(Kind::Optimized, super::NOTHING_READS_IT), 0);
2097        assert_eq!(func[entry].params.len(), 1);
2098    }
2099
2100    #[test]
2101    fn a_parameter_nothing_reads_costs_one_unit_of_fuel_and_stays_without_it() {
2102        let mut func = walking_a_pointer(false);
2103        let stats =
2104            SimplifyCfg.run(&mut func, &mut crate::machine::fixtures::analyses(), &mut Fuel::of(0));
2105        assert_eq!(stats.count(Kind::Optimized, super::NOTHING_READS_IT), 0);
2106        assert_eq!(stats.count(Kind::Missed, super::NO_FUEL_UNREAD), 1);
2107        assert_eq!(func[Block::from_usize(1)].params.len(), 2);
2108    }
2109
2110    #[test]
2111    fn the_counter_that_went_leaves_the_verifier_nothing_to_complain_about() {
2112        let target = TargetInfo::new(Triple::new(Arch::X86_64, Os::Linux, Env::Gnu));
2113        let mut names = Interner::new();
2114        let mut module = Module::new(names.intern("test.c"), &target);
2115        let mut func = walking_a_pointer(false);
2116        simplify(&mut func);
2117        module.add_func(func);
2118        rucc_ir::verify(&module, &names).expect("taking a parameter out left the function whole");
2119    }
2120
2121    #[test]
2122    fn step_three_leaves_the_verifier_nothing_to_complain_about() {
2123        // Section 21.6 names an argument list that stops matching its block's parameters as the
2124        // most common bug in this document, and both halves of step three change one.
2125        let target = TargetInfo::new(Triple::new(Arch::X86_64, Os::Linux, Env::Gnu));
2126        let mut names = Interner::new();
2127        let mut module = Module::new(names.intern("test.c"), &target);
2128        let mut func = taking_a_condition();
2129        let (carried, arms) = arms(&mut func);
2130        let forwarder = func.create_block();
2131        let param = func.append_param(forwarder, Type::int(32));
2132        let exit = func.create_block();
2133        let arrived = func.append_param(exit, Type::int(32));
2134        let mut build = Builder::new(&mut func, arms[0]);
2135        let mine = build.iconst(Type::int(32), 9);
2136        build.jump(exit, &[mine]);
2137        Builder::new(&mut func, arms[1]).jump(forwarder, &[carried]);
2138        Builder::new(&mut func, forwarder).jump(exit, &[param]);
2139        let mut build = Builder::new(&mut func, exit);
2140        // A reader of the parameter that is not the return, because this function returns nothing
2141        // and the point is that something downstream still has the value it was passed.
2142        build.icmp(IntPred::Eq, arrived, arrived);
2143        build.ret(&[]);
2144        simplify(&mut func);
2145        module.add_func(func);
2146        rucc_ir::verify(&module, &names).expect("step three left the function verifiable");
2147    }
2148
2149    #[test]
2150    fn out_of_fuel_leaves_the_function_exactly_as_it_was() {
2151        let (mut func, _) = diamond(|build| build.iconst(Type::int(1), 1));
2152        let before = blocks(&func);
2153        let stats =
2154            SimplifyCfg.run(&mut func, &mut crate::machine::fixtures::analyses(), &mut Fuel::of(0));
2155        assert!(!stats.changed());
2156        assert_eq!(stats.count(Kind::Missed, super::NO_FUEL), 1);
2157        assert_eq!(terminator(&func, 0), Opcode::BrIf);
2158        assert_eq!(blocks(&func), before);
2159    }
2160
2161    #[test]
2162    fn what_fuel_buys_is_one_whole_change_and_never_half_of_one() {
2163        // Two foldable branches and fuel for one. The half that removes the stranded blocks is
2164        // not charged for, because a limit that could stop between the two halves would leave a
2165        // block nothing reaches and the verifier would refuse the function.
2166        let mut func = graph(&[&[1, 2], &[3, 4], &[5], &[5], &[5], &[]]);
2167        let stats =
2168            SimplifyCfg.run(&mut func, &mut crate::machine::fixtures::analyses(), &mut Fuel::of(1));
2169        assert_eq!(stats.count(Kind::Optimized, super::FOLDED), 1);
2170        assert_eq!(stats.count(Kind::Missed, super::NO_FUEL), 1);
2171        // The entry folded to its first arm, so the second arm is stranded and goes, and the
2172        // block only it reached goes with it.
2173        assert_eq!(blocks(&func), [0, 1, 3, 4, 5]);
2174    }
2175
2176    #[test]
2177    fn the_pass_leaves_the_verifier_nothing_to_complain_about() {
2178        let target = TargetInfo::new(Triple::new(Arch::X86_64, Os::Linux, Env::Gnu));
2179        let mut names = Interner::new();
2180        let mut module = Module::new(names.intern("test.c"), &target);
2181        let mut func = graph(&[&[1, 2], &[3], &[3], &[4, 1], &[]]);
2182        simplify(&mut func);
2183        module.add_func(func);
2184        rucc_ir::verify(&module, &names).expect("the pass left the function verifiable");
2185    }
2186
2187    #[test]
2188    fn the_pass_says_it_preserves_nothing() {
2189        assert_eq!(SimplifyCfg.preserves(), Preserved::NONE);
2190    }
2191}