rucc_codegen/slots.rs
1//! One stack slot allocator: every byte a function asks for itself, placed together.
2//!
3//! Design: `spec/optimizer/36-lowering-and-isel.md` section 36.7.
4//!
5//! A frame holds two kinds of thing the function asked for. Locals are what an `alloca` becomes and
6//! the lowering knows about them before anything else runs. Spill slots are what the allocator
7//! gives a value it ran out of registers for, and nothing knows how many of those there are until
8//! it has finished. Placed apart, the frame is the sum of the two areas. Placed together it is the
9//! most either of them needs at any one moment, because two things that are never both wanted can
10//! be the same bytes. That is the same answer the allocator gives about registers and it is the
11//! same reason.
12//!
13//! This runs after allocation and reads the allocator's own liveness rather than working one out.
14//! Running before it would mean guessing which values are spilled, and a guess has to be either
15//! conservative or wrong. Asking again afterwards would mean two answers about one function that
16//! are free to disagree, and the one the machine runs is the allocator's.
17//!
18//! # What a cell is
19//!
20//! A [`Cell`] is a run of bytes in the frame, as wide and as aligned as the widest and strictest
21//! thing in it. [`Slots`] says which cell every local and every spill slot went in, and
22//! [`crate::frame`] is what turns cells into offsets. Nothing else changes: an instruction reading
23//! a local still asks the frame where that local is and gets back an offset, and two locals sharing
24//! a cell get the same one.
25//!
26//! # What may share
27//!
28//! A spill slot holds one value, so where the slot is wanted is where that value is live, and the
29//! allocator has already said where that is.
30//!
31//! A local is harder, because what a local is wanted over is not the live range of anything. The
32//! bytes are reached through an address, the address is a value like any other, and the bytes go on
33//! meaning something for exactly as long as anything can still come by that address. So the
34//! question asked here is where the address gets to, and the answer has to be the whole of it or
35//! the local does not share at all. [`reach`] asks it. An address read as the base of a load or a
36//! store is a read of the local at that instruction and goes no further. An address read by another
37//! address computation is the same local under a second name and is followed. An address read any
38//! other way is one this pass cannot follow to the end, and the local it belongs to is left out.
39//!
40//! Left out is therefore the answer for every local whose address is handed to a call, stored into
41//! memory, or carried between blocks as an argument. That is what section 36.7 means by an address
42//! taken local: not one the program wrote an `&` in front of, which is a question the types
43//! answered and the types are gone by here, but one whose bytes something can reach at a moment
44//! liveness does not know about.
45//!
46//! # Where a local is wanted is not where its address is live
47//!
48//! Knowing which instructions reach a local is only half of it. The address that reaches it is a
49//! value and the object is not, so an address register that dies right after the store through it
50//! says nothing about how long those bytes have to go on holding what was stored. A local written
51//! at one point and read at another has to hold its contents through everything in between, however
52//! little of what is in between mentions the local at all.
53//!
54//! So the area of a local is worked out as its own question over the control flow graph: its bytes
55//! matter at every point that has a touch behind it and a touch in front of it. A point with
56//! nothing in front is one where the object is finished with, and a point with nothing behind is
57//! one where it holds nothing anybody may read, since the contents of a local nothing has written
58//! yet are not contents. The two halves of that question are reachability over the graph rather
59//! than over the line the function was laid out in. Over the line would be wrong for a loop: a
60//! local written at the bottom of a body and read at the top of the next turn is one whose bytes
61//! matter across the header too, and the header is laid out before either of the two touches.
62//!
63//! # The moves count too
64//!
65//! Where a spilled value is live is not quite everywhere its slot is touched. The store that fills
66//! the slot goes after the instruction that wrote the value, the reload that empties it goes before
67//! the instruction that reads it, and the moves an edge turns into go at the end of a block or the
68//! start of one, none of which is a point the value is live at. The edge moves are the ones that
69//! matter: the sequencer put them in an order that works because it was told every place in them
70//! was a different place, and two slots it was told apart are two this pass must not put together
71//! behind its back.
72//!
73//! So the moves are read as well as the liveness. Every edit that names a slot puts a point either
74//! side of where it stands into that slot's area, which is the gap between two points the edit
75//! really sits in, and after that the question is the same question everywhere else in this file.
76//!
77//! # A spill slot is not a variable
78//!
79//! The two kinds are asked about separately, because the reason a frame ever lays two things out
80//! apart that could share is the debugger, and that reason covers one kind and not the other. A
81//! local is a variable somebody wrote down and can ask the value of, so two locals sharing bytes
82//! means a variable that is out of scope reads as whatever took its place, which is what `-O0`
83//! exists not to do and what `-fstack-reuse=none` turns off at every level. A spill slot holds a
84//! value the allocator ran out of registers for, it has no name, nothing can ask for it, and the
85//! only thing that ever reads it is the instruction the allocator wrote. Laying those out one
86//! each buys a debugger nothing and costs a frame everything, since most frames are mostly spill
87//! slots.
88//!
89//! So spill slots share at every level and locals share only where the level says they may. What
90//! says which is whether this pass is handed a [`Reach`]: with one, the locals it followed join
91//! in, and without one every local gets bytes of its own and the spill slots are fitted around
92//! them.
93//!
94//! # How big it is allowed to get
95//!
96//! Fitting each thing into the first cell it does not clash with compares it against the cells so
97//! far, so a function whose things mostly cannot share costs the square of how many there are.
98//! What bounds that is a budget of comparisons rather than a count of things: the fit spends
99//! [`BUDGET`] of them and lays out whatever is left one cell each. A function whose things do
100//! share never comes near it, because what each one is compared against is the cells and not the
101//! things, and the whole point of sharing is that there are far fewer cells than things. lua's
102//! interpreter, which is 2802 slots fitted into 144 cells and the largest function in the corpus,
103//! spends an eighth of the budget and adds a seventh of a second to the file it is in. A function
104//! with that many slots that are all live at once would spend the lot, and it gets the layout it
105//! would have got anyway.
106
107use std::cmp::Reverse;
108use std::collections::{BinaryHeap, HashMap, HashSet};
109
110use rucc_base::Interner;
111use rucc_mir::{Func, Inst, Opcode, Reg};
112use rucc_regalloc::Allocation;
113use rucc_regalloc::assign::Place;
114use rucc_regalloc::live::{Area, Live, Range};
115use rucc_regalloc::order::Order;
116use rucc_regalloc::rewrite::At;
117use rucc_target::FrameInsts;
118
119use crate::frame::Local;
120
121/// How many cells the fit may look at before it stops pairing things up and gives everything left
122/// a cell of its own.
123///
124/// See the note on how big it is allowed to get in the module documentation. One unit is one thing
125/// compared against one cell, which is what costs. The largest function in the corpus spends an
126/// eighth of this, so the budget is a guard against a generated file rather than something the
127/// ordinary path meets.
128pub const BUDGET: usize = 1 << 20;
129
130/// One run of bytes in the frame, holding one local, one spill slot, or several of each.
131#[derive(Debug, Clone, Copy, PartialEq, Eq)]
132pub struct Cell {
133 /// How many bytes of it there are, which is as many as the largest thing in it needs.
134 pub size: u32,
135 /// What its address has to be a multiple of, which is the strictest thing in it.
136 pub align: u32,
137}
138
139/// Which cell of the frame every local and every spill slot of a function is in.
140#[derive(Debug, Clone, Default, PartialEq, Eq)]
141pub struct Slots {
142 cells: Vec<Cell>,
143 locals: Vec<usize>,
144 slots: Vec<usize>,
145 /// Where each local that went in beside something else is wanted, and `None` for the rest.
146 shared: Vec<Option<Vec<Range>>>,
147}
148
149impl Slots {
150 /// The frame with nothing sharing anything: a cell of its own for every local and every spill
151 /// slot, in the order the two lists are in.
152 ///
153 /// This is the layout there was before this pass, and it is what a frame gets when nothing has
154 /// asked for sharing and what it falls back to on a function with too many slots to pair up
155 /// cheaply.
156 #[must_use]
157 pub fn apart(locals: &[Local], widths: &[u32]) -> Self {
158 let mut cells = Vec::with_capacity(locals.len() + widths.len());
159 for &Local { size, align } in locals {
160 cells.push(Cell { size, align });
161 }
162 for &width in widths {
163 cells.push(Cell { size: width, align: width });
164 }
165 Self {
166 locals: (0..locals.len()).collect(),
167 slots: (locals.len()..cells.len()).collect(),
168 shared: vec![None; locals.len()],
169 cells,
170 }
171 }
172
173 /// The frame with everything that can share sharing, worked out from the allocator's liveness.
174 ///
175 /// `reach` is what [`reach`] said about this function before the allocator ran, or `None` for a
176 /// build whose locals keep bytes of their own, which is `-O0` and `-fstack-reuse=none`. The
177 /// spill slots share either way, for the reason in the module documentation. `widths` is how
178 /// many bytes a slot of each of the allocation's spill slots takes, and `locals` is the
179 /// function's own objects in the order the lowering recorded them. `func` is the function the
180 /// allocator has finished with, which is asked for the shape of its control flow and nothing
181 /// else: the rewrite took the values away but it left every block and every edge where it was.
182 #[must_use]
183 pub fn share(
184 func: &Func,
185 reach: Option<&Reach>,
186 allocation: &Allocation,
187 locals: &[Local],
188 widths: &[u32],
189 ) -> Self {
190 let mut wants = Vec::with_capacity(locals.len() + widths.len());
191 let mut reached = reach
192 .map(|reach| areas(func, reach, &allocation.live, &allocation.order))
193 .unwrap_or_default();
194 for (local, &Local { size, align }) in locals.iter().enumerate() {
195 let area = reached.get_mut(local).and_then(Option::take);
196 wants.push(Want { what: What::Local(local), size, align, area });
197 }
198 let held = spilled(allocation, widths.len());
199 let moved = moved(allocation, widths.len());
200 for (slot, &width) in widths.iter().enumerate() {
201 let area = held[slot]
202 .and_then(|reg| allocation.live.area(reg))
203 .map(|live| merged(live.pieces().chain(moved[slot].iter().copied())));
204 wants.push(Want { what: What::Slot(slot), size: width, align: width, area });
205 }
206 fit(wants, locals.len(), widths.len(), BUDGET)
207 }
208
209 /// The cells the frame is made of, which is what [`crate::frame`] places.
210 #[must_use]
211 pub fn cells(&self) -> &[Cell] {
212 &self.cells
213 }
214
215 /// Which cell a local is in.
216 #[must_use]
217 pub fn local(&self, local: usize) -> Option<usize> {
218 self.locals.get(local).copied()
219 }
220
221 /// Which cell a spill slot is in.
222 #[must_use]
223 pub fn slot(&self, slot: u32) -> Option<usize> {
224 self.slots.get(usize::try_from(slot).ok()?).copied()
225 }
226
227 /// Where a local is wanted, in the allocator's points, if its cell holds something else too,
228 /// and `None` for a local whose bytes are its own.
229 ///
230 /// The bytes of a local that shares are only its over this area. Outside it they hold
231 /// whatever else went in the cell, which is why the debugging information asks: a place given
232 /// for the whole function would have a debugger print the other thing under this one's name.
233 #[must_use]
234 pub fn shared(&self, local: usize) -> Option<&[Range]> {
235 self.shared.get(local)?.as_deref()
236 }
237
238 /// How many cells were saved by sharing, which is how many things went in beside something
239 /// else.
240 ///
241 /// This is a count rather than a number of bytes, because how many bytes it saved is the
242 /// difference between two frames and a frame is not worked out here.
243 #[must_use]
244 pub fn saved(&self) -> usize {
245 self.locals.len() + self.slots.len() - self.cells.len()
246 }
247}
248
249/// One thing that wants bytes in the frame, and everywhere it wants them.
250#[derive(Debug)]
251struct Want {
252 what: What,
253 size: u32,
254 align: u32,
255 /// Where it is wanted, or `None` for one this pass could not follow, which shares with nothing.
256 area: Option<Vec<Range>>,
257}
258
259/// Which of the two lists a want came off.
260#[derive(Debug, Clone, Copy)]
261enum What {
262 Local(usize),
263 Slot(usize),
264}
265
266/// Fits every want into the fewest cells, largest and strictest first.
267///
268/// Largest first because a cell only ever grows to hold what goes in it, and starting with the
269/// small ones means growing a cell to several times the size of the thing that opened it, which
270/// leaves the same bytes taken and a worse chance for everything after. The order is settled
271/// entirely by the want rather than partly by which came first, so the same function lays out the
272/// same way every time.
273///
274/// What `budget` is is comparisons of one thing against one cell, which is what costs. Past that
275/// everything left opens a cell of its own, which is the layout a frame had before this pass
276/// existed, and the wants are in a settled order so which ones those are is settled too.
277fn fit(mut wants: Vec<Want>, locals: usize, slots: usize, mut budget: usize) -> Slots {
278 let mut order: Vec<usize> = (0..wants.len()).collect();
279 order.sort_by_key(|&want| {
280 let Want { size, align, .. } = wants[want];
281 (Reverse(align), Reverse(size), want)
282 });
283
284 let mut cells: Vec<Cell> = Vec::new();
285 // `None` is a cell nothing else may go in, which is what a thing this pass could not follow
286 // opens. A cell with an area is one anything that does not clash with that area may join.
287 let mut busy: Vec<Option<Vec<Range>>> = Vec::new();
288 let mut of_local = vec![0; locals];
289 let mut of_slot = vec![0; slots];
290 let mut areas = vec![None; locals];
291 let mut held = Vec::new();
292 for want in order {
293 let Want { what, size, align, area } = std::mem::replace(
294 &mut wants[want],
295 Want { what: What::Local(0), size: 0, align: 0, area: None },
296 );
297 let mut into = None;
298 if let Some(area) = &area {
299 for (cell, held) in busy.iter().enumerate() {
300 if budget == 0 {
301 break;
302 }
303 budget -= 1;
304 if held.as_ref().is_some_and(|held| !clashes(held, area)) {
305 into = Some(cell);
306 break;
307 }
308 }
309 }
310 // Kept for a local as well as handed to the cell, since whether it shared is only known once
311 // everything has been fitted, and one that did is asked about again. See [`Slots::shared`].
312 let mine = if let What::Local(_) = what { area.clone() } else { None };
313 let cell = match into {
314 Some(cell) => {
315 cells[cell].size = cells[cell].size.max(size);
316 cells[cell].align = cells[cell].align.max(align);
317 let held = busy[cell].take().unwrap_or_default();
318 busy[cell] = Some(merged(held.into_iter().chain(area.into_iter().flatten())));
319 cell
320 }
321 None => {
322 cells.push(Cell { size, align });
323 busy.push(area);
324 cells.len() - 1
325 }
326 };
327 match what {
328 What::Local(local) => {
329 of_local[local] = cell;
330 areas[local] = mine;
331 }
332 What::Slot(slot) => of_slot[slot] = cell,
333 }
334 if held.len() <= cell {
335 held.resize(cell + 1, 0);
336 }
337 held[cell] += 1;
338 }
339 let shared = areas
340 .into_iter()
341 .zip(&of_local)
342 .map(|(area, &cell)| area.filter(|_| held[cell] > 1))
343 .collect();
344 Slots { cells, locals: of_local, slots: of_slot, shared }
345}
346
347/// Which value the allocator put in each spill slot, by slot number.
348///
349/// A slot holds one value, because the allocator takes a fresh one every time it spills, so this is
350/// the assignment read the other way round.
351fn spilled(allocation: &Allocation, slots: usize) -> Vec<Option<Reg>> {
352 let mut held = vec![None; slots];
353 for (reg, place) in allocation.assignment.placed() {
354 if let Place::Slot(slot) = place {
355 if let Some(at) = usize::try_from(slot).ok().and_then(|slot| held.get_mut(slot)) {
356 *at = Some(reg);
357 }
358 }
359 }
360 held
361}
362
363/// What carries the address of each of a function's locals, or `None` for one whose address gets
364/// away somewhere this pass cannot follow.
365///
366/// Worked out before allocation, because it is a question about values and a value is written once
367/// only until the allocator's rewrite has been through. Read after it, because that is when the
368/// liveness these names are looked up in exists.
369#[derive(Debug, Clone, Default)]
370pub struct Reach {
371 through: Vec<Option<Carried>>,
372}
373
374impl Reach {
375 /// Every point one local is touched at, which is where its address is live and where an
376 /// instruction that swallowed the address stands.
377 fn touches(&self, local: usize, live: &Live, order: &Order) -> Option<Vec<Range>> {
378 let held = self.through.get(local)?.as_ref()?;
379 let mut spots: Vec<Range> = Vec::new();
380 for ® in &held.regs {
381 spots.extend(live.area(reg).into_iter().flat_map(Area::pieces));
382 }
383 for &inst in &held.at {
384 spots.push(Range { start: order.early(inst), end: order.late(inst) });
385 }
386 Some(spots)
387 }
388
389 /// Whether a local may share its bytes with anything, which is what the tests ask.
390 #[must_use]
391 pub fn shares(&self, local: usize) -> bool {
392 self.through.get(local).is_some_and(Option::is_some)
393 }
394}
395
396/// Everywhere the bytes of each local have to go on holding what was put in them.
397///
398/// A point counts if a touch of that local can have happened before it and another can still
399/// happen after it. Before is reachability forward through the graph from the blocks that touch
400/// the local, after is the same walk backwards, and the bytes matter where the two meet. See the
401/// note in the module documentation on why this is asked over the graph and not over the line the
402/// function was laid out in.
403///
404/// A local this pass could not follow the address of comes back `None`, which is the answer that
405/// shares with nothing.
406fn areas(func: &Func, reach: &Reach, live: &Live, order: &Order) -> Vec<Option<Vec<Range>>> {
407 let blocks = order.blocks();
408 let count = reach.through.len();
409 let words = count.div_ceil(64);
410
411 // Where each block starts, which is ascending, so the block a point is in is a search.
412 let starts: Vec<u32> = blocks.iter().map(|&block| order.start(block)).collect();
413 let holding = |point: u32| starts.partition_point(|&start| start <= point).saturating_sub(1);
414
415 // Which locals each block touches, as bits for the walk and as a range for the answer. A touch
416 // that runs through whole blocks between the two it starts and stops in covers those blocks
417 // top to bottom whatever the walk says, so they are one piece of the answer straight away and
418 // only the two ends are left for the blocks to decide.
419 let mut touched = vec![vec![0u64; words]; blocks.len()];
420 let mut inside: Vec<Vec<(usize, Range)>> = vec![Vec::new(); blocks.len()];
421 let mut through: Vec<Vec<Range>> = vec![Vec::new(); count];
422 for local in 0..count {
423 let Some(spots) = reach.touches(local, live, order) else { continue };
424 for spot in spots {
425 let (first, last) = (holding(spot.start), holding(spot.end));
426 for row in &mut touched[first..=last] {
427 row[local / 64] |= 1 << (local % 64);
428 }
429 let mut clip = |at: usize| {
430 let block = blocks[at];
431 let start = spot.start.max(order.start(block));
432 let end = spot.end.min(order.end(block));
433 inside[at].push((local, Range { start, end }));
434 };
435 clip(first);
436 if last != first {
437 clip(last);
438 }
439 if last > first + 1 {
440 let start = order.start(blocks[first + 1]);
441 through[local].push(Range { start, end: order.end(blocks[last - 1]) });
442 }
443 }
444 }
445
446 // One range per block per local, from the first touch in the block to the last. A block runs
447 // top to bottom, so whatever sits between two touches of the same local is between them in the
448 // run as well, and the bytes have to have held what they hold all the way through it.
449 for spots in inside.iter_mut() {
450 spots.sort_unstable_by_key(|&(local, Range { start, .. })| (local, start));
451 let mut kept = 0;
452 for at in 1..spots.len() {
453 if spots[at].0 == spots[kept].0 {
454 spots[kept].1.end = spots[kept].1.end.max(spots[at].1.end);
455 } else {
456 kept += 1;
457 spots[kept] = spots[at];
458 }
459 }
460 spots.truncate(spots.len().min(kept + 1));
461 }
462
463 // The graph, by position in the line rather than by block, because everything else here is.
464 let mut place = vec![0usize; func.block_count()];
465 for (at, &block) in blocks.iter().enumerate() {
466 place[block.index()] = at;
467 }
468 let mut ahead: Vec<Vec<usize>> = vec![Vec::new(); blocks.len()];
469 let mut behind: Vec<Vec<usize>> = vec![Vec::new(); blocks.len()];
470 for (at, &block) in blocks.iter().enumerate() {
471 for call in &func[block].succs {
472 let to = place[call.block.index()];
473 ahead[at].push(to);
474 behind[to].push(at);
475 }
476 }
477
478 let written = spread(&behind, &touched, words, true);
479 let read = spread(&ahead, &touched, words, false);
480
481 let mut out = vec![None; count];
482 for (local, pieces) in out.iter_mut().enumerate() {
483 if reach.shares(local) {
484 *pieces = Some(std::mem::take(&mut through[local]));
485 }
486 }
487 for (at, &block) in blocks.iter().enumerate() {
488 let whole = Range { start: order.start(block), end: order.end(block) };
489 for word in 0..words {
490 let mut bits = written[at][word] & read[at][word];
491 while bits != 0 {
492 let local = word * 64 + bits.trailing_zeros() as usize;
493 bits &= bits - 1;
494 if let Some(pieces) = out[local].as_mut() {
495 joined(pieces, whole);
496 }
497 }
498 }
499 // A block that touches the local is covered from the touch, or from the top of the block
500 // if something above already wrote it, and to the touch, or to the bottom if something
501 // below still reads it.
502 for &(local, spot) in &inside[at] {
503 let held = |bits: &[Vec<u64>]| bits[at][local / 64] & (1 << (local % 64)) != 0;
504 let start = if held(&written) { whole.start } else { spot.start };
505 let end = if held(&read) { whole.end } else { spot.end };
506 if let Some(pieces) = out[local].as_mut() {
507 joined(pieces, Range { start, end });
508 }
509 }
510 }
511 for pieces in out.iter_mut().flatten() {
512 *pieces = merged(std::mem::take(pieces));
513 }
514 out
515}
516
517/// Adds a piece to a local's area, stretching the last one instead when the piece starts on the
518/// point right after it.
519///
520/// Blocks are taken in the line's order and a block starts one point after the one before it ends,
521/// so a local wanted all the way through a run of blocks is one piece for the run rather than one
522/// piece per block. On jtckdint the answer used to be a piece per block for each of the locals
523/// whose address stays live across its 16000 blocks, which with the one piece per block pushed for
524/// the touches above was 3% of the instructions of an optimized build. The join only ever happens
525/// where one block ends and the next starts, so no point is added or lost, and a stretch of debug
526/// info is still found block by block from the same points.
527fn joined(pieces: &mut Vec<Range>, piece: Range) {
528 match pieces.last_mut() {
529 Some(last) if last.end.checked_add(1) == Some(piece.start) => last.end = piece.end,
530 _ => pieces.push(piece),
531 }
532}
533
534/// Which locals a touch of can reach the start of each block, following the given edges.
535///
536/// One walk stands for both directions. Handed the edges into each block it says which locals were
537/// touched somewhere above, and handed the edges out of each block it says which are touched
538/// somewhere below. Blocks are taken in [`settling`]'s order, so a block with no loop around it is
539/// looked at after everything it reads from is final and only once, and a block is only looked at
540/// again when a block it reads from changed.
541///
542/// It used to be rounds over every block in the line's order until one changed nothing, which
543/// settles a straight stretch in one round only when the line runs the way the edges do. It does
544/// not have to: jtckdint's main has a chain a thousand blocks long laid out against its edges, and
545/// the rounds over its 16000 blocks took a thousand passes and five seconds to move the answer down
546/// it one block at a time.
547///
548/// A block is allowed to be its own neighbour, which is what a loop of one block is, and the row it
549/// is working on is a copy for that reason. Reading a block's own answer back is a no change either
550/// way, since the answer being built is the one being read, but what a block touches does come back
551/// to itself around a back edge and that is the half that has to arrive. tamnd/rucc#1207.
552fn spread(
553 edges: &[Vec<usize>],
554 touched: &[Vec<u64>],
555 words: usize,
556 forward: bool,
557) -> Vec<Vec<u64>> {
558 let count = edges.len();
559 let mut readers: Vec<Vec<usize>> = vec![Vec::new(); count];
560 for (at, froms) in edges.iter().enumerate() {
561 for &from in froms {
562 readers[from].push(at);
563 }
564 }
565 let order = settling(&readers, forward);
566 let mut rank = vec![0; count];
567 for (place, &at) in order.iter().enumerate() {
568 rank[at] = place;
569 }
570 let mut out = vec![vec![0u64; words]; count];
571 let mut waiting: BinaryHeap<Reverse<usize>> = (0..count).map(Reverse).collect();
572 let mut queued = vec![true; count];
573 let mut row = vec![0u64; words];
574 while let Some(Reverse(place)) = waiting.pop() {
575 let at = order[place];
576 queued[at] = false;
577 row.copy_from_slice(&out[at]);
578 let mut grew = false;
579 for &from in &edges[at] {
580 for word in 0..words {
581 let had = row[word];
582 row[word] |= out[from][word] | touched[from][word];
583 grew |= row[word] != had;
584 }
585 }
586 if grew {
587 out[at].copy_from_slice(&row);
588 for &reader in &readers[at] {
589 if !queued[reader] {
590 queued[reader] = true;
591 waiting.push(Reverse(rank[reader]));
592 }
593 }
594 }
595 }
596 out
597}
598
599/// The blocks in an order where, loops aside, every block comes after the blocks it reads from.
600///
601/// That is reverse postorder of a walk along the way the answer flows, from every block in turn so
602/// that one nothing reaches is still in it. Any walk's reverse postorder puts a block after all its
603/// predecessors once the back edges are left out, whichever block it starts from.
604fn settling(readers: &[Vec<usize>], forward: bool) -> Vec<usize> {
605 let count = readers.len();
606 let mut seen = vec![false; count];
607 let mut post = Vec::with_capacity(count);
608 let mut stack: Vec<(usize, usize)> = Vec::new();
609 let roots: Vec<usize> = if forward { (0..count).collect() } else { (0..count).rev().collect() };
610 for root in roots {
611 if seen[root] {
612 continue;
613 }
614 seen[root] = true;
615 stack.push((root, 0));
616 while let Some((at, next)) = stack.last_mut() {
617 if let Some(&to) = readers[*at].get(*next) {
618 *next += 1;
619 if !seen[to] {
620 seen[to] = true;
621 stack.push((to, 0));
622 }
623 } else {
624 post.push(*at);
625 stack.pop();
626 }
627 }
628 }
629 post.reverse();
630 post
631}
632
633/// Everywhere one local is reached from.
634#[derive(Debug, Clone, Default)]
635struct Carried {
636 /// The values that hold its address.
637 regs: Vec<Reg>,
638 /// The instructions that reach it with no value in between, which is what an address folded
639 /// into its reader leaves behind.
640 at: Vec<Inst>,
641}
642
643/// Follows the address of every local of a function as far as it goes.
644///
645/// `addresses` is the list [`crate::lower`] built and [`crate::fold`] rewrote, which says which
646/// instruction carries the address of which local. `count` is how many locals there are, since a
647/// local nothing on that list names is one this has no account of rather than one nothing touches.
648///
649/// Run after the fold and before allocation. After the fold because an address that ended up inside
650/// its reader is an address no value holds and this has to see it that way. Before allocation
651/// because every answer here is about a virtual register, and the rewrite the allocator ends with
652/// is what stops there being one.
653#[must_use]
654pub fn reach(
655 func: &Func,
656 addresses: &[(Inst, usize)],
657 count: usize,
658 insts: &FrameInsts,
659 names: &mut Interner,
660) -> Reach {
661 let lea = Opcode::new(names.intern(&format!("{}{}", insts.prefix, insts.lea)));
662 let mut through: Vec<Option<Carried>> = vec![None; count];
663 for &(inst, local) in addresses {
664 let Some(held) = through.get_mut(local) else { continue };
665 let held = held.get_or_insert_with(Carried::default);
666 // Either the `lea` the lowering wrote, whose result is the address and goes on from here,
667 // or a reader the fold put the address inside, which touches the local where it stands and
668 // hands nothing on. The opcode is the whole of the difference: a reader that is itself a
669 // `lea` really does hand an address on, and this reads it as one.
670 if func[inst].opcode == lea {
671 match def(func, inst) {
672 Some(reg) => held.regs.push(reg),
673 None => {
674 through[local] = None;
675 continue;
676 }
677 }
678 }
679 // On the list either way, so that a local whose address nothing reads is still wanted where
680 // the address of it was taken rather than nowhere at all.
681 held.at.push(inst);
682 }
683
684 let readers = readers(func);
685 let crossing = crossing(func);
686 for held in &mut through {
687 if let Some(carried) = held.take() {
688 *held = follow(func, lea, &readers, &crossing, carried);
689 }
690 }
691 Reach { through }
692}
693
694/// Follows every address a local is reached through to every value that address becomes.
695///
696/// Gives back nothing for a local whose address is read some way this cannot account for, which is
697/// any way but as the base or the index of a memory operand. A call argument is one of those, a
698/// value stored into memory is another, and so is a value carried into a block as an argument,
699/// which is the one that is not an operand at all.
700fn follow(
701 func: &Func,
702 lea: Opcode,
703 readers: &HashMap<Reg, Vec<Inst>>,
704 crossing: &HashSet<Reg>,
705 mut held: Carried,
706) -> Option<Carried> {
707 let mut seen: HashSet<Reg> = held.regs.iter().copied().collect();
708 let mut queue = held.regs.clone();
709 while let Some(reg) = queue.pop() {
710 if crossing.contains(®) {
711 return None;
712 }
713 for &inst in readers.get(®).map(Vec::as_slice).unwrap_or_default() {
714 if !addressed(func, inst, reg) {
715 return None;
716 }
717 if func[inst].opcode == lea {
718 let next = def(func, inst)?;
719 if seen.insert(next) {
720 held.regs.push(next);
721 queue.push(next);
722 }
723 }
724 }
725 }
726 Some(held)
727}
728
729/// Whether every read of a value by an instruction is as part of the address it works on.
730///
731/// Anything else is a read this pass cannot follow: the value has gone somewhere that is not an
732/// address into this frame any more, and where its bytes are reached from afterwards is no longer a
733/// question about liveness.
734fn addressed(func: &Func, inst: Inst, reg: Reg) -> bool {
735 let data = &func[inst];
736 let Some(mem) = data.mem else { return false };
737 let amode = func[mem];
738 func[data.operands].iter().enumerate().all(|(at, operand)| {
739 if operand.reg != reg || operand.role.is_def() {
740 return true;
741 }
742 let at = u8::try_from(at).ok();
743 at.is_some() && (amode.base == at || amode.index == at)
744 })
745}
746
747/// The one virtual register an instruction writes, or nothing when it writes none or several.
748fn def(func: &Func, inst: Inst) -> Option<Reg> {
749 let mut found = None;
750 for operand in &func[func[inst].operands] {
751 if !operand.role.is_def() {
752 continue;
753 }
754 if operand.reg.number().is_none() || found.is_some() {
755 return None;
756 }
757 found = Some(operand.reg);
758 }
759 found
760}
761
762/// Which instructions read each virtual register.
763fn readers(func: &Func) -> HashMap<Reg, Vec<Inst>> {
764 let mut readers: HashMap<Reg, Vec<Inst>> = HashMap::new();
765 for block in func.blocks() {
766 for inst in func.insts(block) {
767 for operand in &func[func[inst].operands] {
768 if operand.role.is_def() || operand.reg.number().is_none() {
769 continue;
770 }
771 let at = readers.entry(operand.reg).or_default();
772 if at.last() != Some(&inst) {
773 at.push(inst);
774 }
775 }
776 }
777 }
778 readers
779}
780
781/// Every virtual register that goes between blocks, as an argument an edge carries or as a
782/// parameter one arrives in.
783///
784/// These are the reads that are not operands, so the walk above would not see them, and an address
785/// that goes round a loop this way is one whose local is left out rather than one followed into a
786/// second name.
787fn crossing(func: &Func) -> HashSet<Reg> {
788 let mut crossing = HashSet::new();
789 for block in func.blocks() {
790 crossing.extend(func[block].params.iter().map(|param| param.reg));
791 for call in &func[block].succs {
792 crossing.extend(call.args.iter().copied());
793 }
794 }
795 crossing
796}
797
798/// Where the moves the allocator handed back touch each slot of the frame.
799///
800/// A point either side of where each of them stands, which is the gap between two points the move
801/// really goes in. See the note on the moves in the module documentation.
802fn moved(allocation: &Allocation, slots: usize) -> Vec<Vec<Range>> {
803 let order = &allocation.order;
804 let mut moved = vec![Vec::new(); slots];
805 for edit in &allocation.edits {
806 let at = match edit.at {
807 At::Before(inst) => order.early(inst),
808 At::After(inst) => order.late(inst),
809 At::StartOf(block) => order.start(block),
810 At::EndOf(block) => order.end(block),
811 };
812 let around =
813 Range { start: at.saturating_sub(1), end: at.saturating_add(1).min(order.points()) };
814 for place in [edit.mov.to, edit.mov.from] {
815 if let Place::Slot(slot) = place {
816 if let Some(at) = usize::try_from(slot).ok().and_then(|slot| moved.get_mut(slot)) {
817 at.push(around);
818 }
819 }
820 }
821 }
822 moved
823}
824
825/// The same stretches of the function, in order, with everything that touches joined up.
826fn merged(pieces: impl IntoIterator<Item = Range>) -> Vec<Range> {
827 let mut pieces: Vec<Range> = pieces.into_iter().collect();
828 pieces.sort_by_key(|piece| (piece.start, piece.end));
829 let mut merged: Vec<Range> = Vec::with_capacity(pieces.len());
830 for piece in pieces {
831 match merged.last_mut() {
832 Some(last) if piece.start <= last.end => last.end = last.end.max(piece.end),
833 _ => merged.push(piece),
834 }
835 }
836 merged
837}
838
839/// Whether two stretches of a function are both wanted anywhere, which is what stops two things
840/// sharing a cell.
841///
842/// Both lists are in order and neither is long, so this walks them together and stops at the first
843/// pair that touches rather than comparing every piece with every other.
844fn clashes(one: &[Range], two: &[Range]) -> bool {
845 let (mut mine, mut theirs) = (0, 0);
846 while mine < one.len() && theirs < two.len() {
847 if one[mine].overlaps(two[theirs]) {
848 return true;
849 }
850 if one[mine].end < two[theirs].end {
851 mine += 1;
852 } else {
853 theirs += 1;
854 }
855 }
856 false
857}
858
859#[cfg(test)]
860mod tests {
861 use rucc_base::Interner;
862 use rucc_mir::{Block, BlockCall, Mem, Operand};
863 use rucc_regalloc::assign::Env;
864 use rucc_regalloc::order::Point;
865 use rucc_target::x86_64::{FRAME, GPR, REGS, SYSV};
866
867 use super::*;
868 use crate::frame::{Frame, Layout};
869
870 /// A function being built, with the names and the opcodes a test needs to hand.
871 struct Building {
872 names: Interner,
873 func: Func,
874 lea: Opcode,
875 nop: Opcode,
876 addresses: Vec<(Inst, usize)>,
877 }
878
879 impl Building {
880 /// An empty function of one block.
881 fn new() -> (Self, Block) {
882 let mut names = Interner::new();
883 let func = Func::new(names.intern("f"));
884 let lea = Opcode::new(names.intern(&format!("{}{}", FRAME.prefix, FRAME.lea)));
885 let nop = Opcode::new(names.intern("x64.nop"));
886 let mut building = Self { names, func, lea, nop, addresses: Vec::new() };
887 let block = building.func.create_block();
888 (building, block)
889 }
890
891 /// The address of a local, taken the way the lowering takes one: a `lea` off the stack
892 /// pointer with nothing in its displacement yet.
893 fn local(&mut self, block: Block, which: usize) -> Reg {
894 let sp = Operand::read(Reg::physical(SYSV.stack_pointer), GPR);
895 let reg = self.func.new_vreg(GPR);
896 let inst = self.func.build(block, self.lea).def(reg, GPR).mem(Mem::at(sp)).finish();
897 self.addresses.push((inst, which));
898 reg
899 }
900
901 /// An instruction that reads a local through its address, which is every ordinary use of
902 /// one.
903 fn through(&mut self, block: Block, addr: Reg) {
904 let at = Operand::read(addr, GPR);
905 self.func.build(block, self.nop).mem(Mem::at(at)).finish();
906 }
907
908 /// An instruction that reads a value as a value, which is what handing an address to a
909 /// call looks like from here.
910 fn held(&mut self, block: Block, reg: Reg) {
911 self.func.build(block, self.nop).uses(reg, GPR).finish();
912 }
913
914 /// A value written and then read, which is one more thing wanting a register in between.
915 fn value(&mut self, block: Block) -> Reg {
916 let reg = self.func.new_vreg(GPR);
917 self.func.build(block, self.nop).def(reg, GPR).finish();
918 reg
919 }
920
921 /// What this pass says about the function, and then what the allocator says, in that
922 /// order because the first question is about values and the second takes them away.
923 fn allocate(&mut self, locals: usize, registers: usize) -> (Reach, Allocation) {
924 let reach = reach(&self.func, &self.addresses, locals, &FRAME, &mut self.names);
925 let env =
926 Env::new().with(GPR, &SYSV.int_order[..registers], &SYSV.int_order[registers..]);
927 let allocation = rucc_regalloc::run(&mut self.func, &env, "test", true);
928 (reach, allocation)
929 }
930 }
931
932 /// A local of one word, which is what most of them are.
933 const WORD: Local = Local { size: 8, align: 8 };
934
935 #[test]
936 fn two_locals_that_are_never_both_wanted_are_the_same_bytes() {
937 let (mut building, block) = Building::new();
938 let first = building.local(block, 0);
939 building.through(block, first);
940 let second = building.local(block, 1);
941 building.through(block, second);
942 let (reach, allocation) = building.allocate(2, 4);
943
944 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
945 assert_eq!(plan.cells().len(), 1, "one run of bytes for the two of them");
946 assert_eq!(plan.local(0), plan.local(1));
947 assert_eq!(plan.saved(), 1);
948 }
949
950 #[test]
951 fn a_local_that_went_in_beside_another_says_where_it_is_wanted_and_one_alone_does_not() {
952 let (mut building, block) = Building::new();
953 let first = building.local(block, 0);
954 building.through(block, first);
955 let second = building.local(block, 1);
956 building.through(block, second);
957 let (reach, allocation) = building.allocate(2, 4);
958
959 // Each of the two is wanted over a stretch the other is not, and it is those stretches the
960 // debugging information gives each of them a place over rather than the whole function.
961 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
962 let (one, two) = (plan.shared(0).expect("shares"), plan.shared(1).expect("shares"));
963 assert!(!one.is_empty() && !two.is_empty());
964 assert!(!clashes(one, two), "wanted apart: {one:?} and {two:?}");
965
966 // The same function with nothing allowed to share, where each local's bytes are its own
967 // over the whole of it.
968 let plan = Slots::share(&building.func, None, &allocation, &[WORD, WORD], &[]);
969 assert_eq!((plan.shared(0), plan.shared(1)), (None, None));
970 }
971
972 #[test]
973 fn two_locals_that_are_both_wanted_at_once_are_not() {
974 let (mut building, block) = Building::new();
975 let first = building.local(block, 0);
976 let second = building.local(block, 1);
977 // Both addresses are live at this point, which is the whole of the difference from the
978 // test above.
979 building.through(block, first);
980 building.through(block, second);
981 let (reach, allocation) = building.allocate(2, 4);
982
983 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
984 assert_eq!(plan.cells().len(), 2);
985 assert_ne!(plan.local(0), plan.local(1));
986 assert_eq!(plan.saved(), 0);
987 }
988
989 #[test]
990 fn a_local_and_a_spilled_value_that_do_not_meet_share_one_run_of_bytes() {
991 let (mut building, block) = Building::new();
992 let addr = building.local(block, 0);
993 building.through(block, addr);
994 // Three values wanted at once with two registers to hand out, after the local is finished
995 // with, so what spills is spilled over a stretch the local is not wanted over.
996 let values: Vec<Reg> = (0..3).map(|_| building.value(block)).collect();
997 for ® in &values {
998 building.held(block, reg);
999 }
1000 let (reach, allocation) = building.allocate(1, 2);
1001
1002 assert_eq!(allocation.assignment.spilled(), 1, "one value went to the stack");
1003 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD], &[8]);
1004 assert_eq!(plan.cells().len(), 1);
1005 assert_eq!(plan.local(0), plan.slot(0));
1006 }
1007
1008 #[test]
1009 fn a_local_whose_address_is_handed_to_something_shares_with_nothing() {
1010 let (mut building, block) = Building::new();
1011 let first = building.local(block, 0);
1012 // Read as a value rather than as an address, which is what a call argument is and is the
1013 // point past which this pass cannot say where the bytes are reached from.
1014 building.held(block, first);
1015 let second = building.local(block, 1);
1016 building.through(block, second);
1017 let (reach, allocation) = building.allocate(2, 4);
1018
1019 assert!(!reach.shares(0), "an address that got away");
1020 assert!(reach.shares(1));
1021 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
1022 assert_eq!(plan.cells().len(), 2);
1023 assert_ne!(plan.local(0), plan.local(1));
1024 }
1025
1026 #[test]
1027 fn a_local_whose_address_is_carried_into_a_block_shares_with_nothing() {
1028 let (mut building, block) = Building::new();
1029 let addr = building.local(block, 0);
1030 let next = building.func.create_block();
1031 let param = building.func.append_param(next, GPR);
1032 building.func.build(block, building.nop).finish();
1033 building.func.succs_mut(block).push(BlockCall::with(next, vec![addr]));
1034 building.through(next, param);
1035 let (reach, _) = building.allocate(1, 4);
1036
1037 assert!(!reach.shares(0), "an address that goes between blocks");
1038 }
1039
1040 #[test]
1041 fn a_local_touched_again_later_keeps_its_bytes_over_everything_in_between() {
1042 let (mut building, block) = Building::new();
1043 let first = building.local(block, 0);
1044 building.through(block, first);
1045 // Another local in the stretch between the two touches of the first one. Nothing mentions
1046 // the first local in here, which is exactly the case: it is not being read, but what it
1047 // holds is still wanted below, so these cannot be the same bytes.
1048 let second = building.local(block, 1);
1049 building.through(block, second);
1050 // The first local again, reached through an address worked out a second time.
1051 let again = building.local(block, 0);
1052 building.through(block, again);
1053 let (reach, allocation) = building.allocate(2, 4);
1054
1055 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
1056 assert_ne!(plan.local(0), plan.local(1));
1057 assert_eq!(plan.saved(), 0);
1058 }
1059
1060 #[test]
1061 fn a_local_touched_in_a_loop_keeps_its_bytes_over_the_rest_of_the_loop() {
1062 let (mut building, block) = Building::new();
1063 let header = building.func.create_block();
1064 let body = building.func.create_block();
1065 building.func.build(block, building.nop).finish();
1066 building.func.succs_mut(block).push(BlockCall::to(header));
1067
1068 // The header is laid out before the body and touches a local of its own.
1069 let held = building.local(header, 1);
1070 building.through(header, held);
1071 building.func.build(header, building.nop).finish();
1072 building.func.succs_mut(header).push(BlockCall::to(body));
1073
1074 // The body touches the other one, every turn of the loop, and the header runs between one
1075 // turn and the next. So the body's local is wanted over the header as well, which is a
1076 // thing only the edges say: in the line the function is laid out in, the header is above
1077 // the only touch there is.
1078 let addr = building.local(body, 0);
1079 building.through(body, addr);
1080 building.func.build(body, building.nop).finish();
1081 building.func.succs_mut(body).push(BlockCall::to(header));
1082 let (reach, allocation) = building.allocate(2, 4);
1083
1084 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
1085 assert_ne!(plan.local(0), plan.local(1));
1086 }
1087
1088 #[test]
1089 fn a_local_wanted_across_a_run_of_blocks_is_one_piece_for_the_run() {
1090 let (mut building, block) = Building::new();
1091 let first = building.local(block, 0);
1092 building.through(block, first);
1093 let mut last = block;
1094 for _ in 0..3 {
1095 let next = building.func.create_block();
1096 building.func.succs_mut(last).push(BlockCall::to(next));
1097 building.func.build(next, building.nop).finish();
1098 last = next;
1099 }
1100 let again = building.local(last, 0);
1101 building.through(last, again);
1102 let (reach, allocation) = building.allocate(1, 4);
1103
1104 let areas = areas(&building.func, &reach, &allocation.live, &allocation.order);
1105 let pieces = areas[0].as_ref().expect("shares");
1106 assert_eq!(pieces.len(), 1, "one piece from the first touch to the last: {pieces:?}");
1107 }
1108
1109 /// A loop of one block, which is a block that is its own predecessor and its own successor.
1110 /// The walk over the graph has to take that rather than fall over it, and what comes back is
1111 /// the same answer the two block loop above gets: the body runs again, so a local touched at
1112 /// the bottom of it is wanted at the top. tamnd/rucc#1207.
1113 #[test]
1114 fn a_block_that_is_its_own_neighbour_is_a_loop_like_any_other() {
1115 let (mut building, block) = Building::new();
1116 let loops = building.func.create_block();
1117 building.func.build(block, building.nop).finish();
1118 building.func.succs_mut(block).push(BlockCall::to(loops));
1119
1120 // One local touched at the top of the block and the other at the bottom. The edge back to
1121 // the top is what puts the second one over the first.
1122 let held = building.local(loops, 1);
1123 building.through(loops, held);
1124 let addr = building.local(loops, 0);
1125 building.through(loops, addr);
1126 building.func.build(loops, building.nop).finish();
1127 building.func.succs_mut(loops).push(BlockCall::to(loops));
1128 let (reach, allocation) = building.allocate(2, 4);
1129
1130 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
1131 assert_ne!(plan.local(0), plan.local(1));
1132 }
1133
1134 #[test]
1135 fn an_address_a_second_address_computation_reads_is_the_same_local_followed_on() {
1136 let (mut building, block) = Building::new();
1137 let first = building.local(block, 0);
1138 // `lea` off a `lea`, which is what the address of a field of a local is. The local is
1139 // wanted wherever the second address is, not only where the first one is.
1140 let derived = building.func.new_vreg(GPR);
1141 let at = Operand::read(first, GPR);
1142 building.func.build(block, building.lea).def(derived, GPR).mem(Mem::at(at)).finish();
1143 let second = building.local(block, 1);
1144 building.through(block, second);
1145 building.through(block, derived);
1146 let (reach, allocation) = building.allocate(2, 4);
1147
1148 assert!(reach.shares(0), "a derived address is still an address into this frame");
1149 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
1150 assert_eq!(plan.cells().len(), 2, "the two locals are wanted at once after all");
1151 }
1152
1153 #[test]
1154 fn a_cell_two_things_share_is_as_wide_and_as_strict_as_both_of_them() {
1155 let (mut building, block) = Building::new();
1156 let first = building.local(block, 0);
1157 building.through(block, first);
1158 let second = building.local(block, 1);
1159 building.through(block, second);
1160 let (reach, allocation) = building.allocate(2, 4);
1161
1162 let narrow = Local { size: 4, align: 4 };
1163 let wide = Local { size: 16, align: 16 };
1164 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[narrow, wide], &[]);
1165 assert_eq!(plan.cells(), [Cell { size: 16, align: 16 }]);
1166 assert_eq!(plan.local(0), plan.local(1));
1167 }
1168
1169 #[test]
1170 fn a_local_nothing_on_the_address_list_names_shares_with_nothing() {
1171 let (mut building, block) = Building::new();
1172 let addr = building.local(block, 0);
1173 building.through(block, addr);
1174 let (reach, allocation) = building.allocate(2, 4);
1175
1176 // A list with nothing on it for a local is this pass having no account of it rather than
1177 // a local nothing touches, so it keeps bytes of its own.
1178 assert!(!reach.shares(1));
1179 let plan = Slots::share(&building.func, Some(&reach), &allocation, &[WORD, WORD], &[]);
1180 assert_eq!(plan.cells().len(), 2);
1181 }
1182
1183 #[test]
1184 fn the_frame_with_nothing_sharing_gives_every_local_and_every_slot_a_run_of_its_own() {
1185 let plan = Slots::apart(&[WORD, Local { size: 4, align: 4 }], &[8, 16]);
1186
1187 assert_eq!(plan.cells().len(), 4);
1188 assert_eq!(plan.saved(), 0);
1189 assert_eq!((plan.local(0), plan.local(1)), (Some(0), Some(1)));
1190 assert_eq!((plan.slot(0), plan.slot(1)), (Some(2), Some(3)));
1191 assert_eq!(plan.cells()[3], Cell { size: 16, align: 16 });
1192 }
1193
1194 #[test]
1195 fn a_frame_whose_locals_share_is_smaller_and_puts_them_at_the_same_offset() {
1196 let (mut building, block) = Building::new();
1197 let first = building.local(block, 0);
1198 building.through(block, first);
1199 let second = building.local(block, 1);
1200 building.through(block, second);
1201 let (reach, allocation) = building.allocate(2, 4);
1202
1203 // Not a leaf, so the frame is taken rather than kept in the red zone and its size is a
1204 // number rather than nothing, and big enough that the convention's alignment does not
1205 // round the difference away.
1206 let locals = [Local { size: 64, align: 8 }; 2];
1207 let base = Layout { leaf: false, locals: &locals, ..Layout::new(&SYSV, REGS) };
1208 let apart = Frame::of(&building.func, &allocation, &base);
1209 let plan = Slots::share(&building.func, Some(&reach), &allocation, &locals, &[]);
1210 let layout = Layout { share: Some(&plan), ..base };
1211 let together = Frame::of(&building.func, &allocation, &layout);
1212
1213 assert_ne!(apart.local(0), apart.local(1));
1214 assert_eq!(together.local(0), together.local(1));
1215 // Sixty four bytes of frame gone, and eight more in each of them for the word that lands
1216 // the stack pointer back where a call wants it.
1217 assert_eq!((apart.size(), together.size()), (136, 72));
1218 }
1219
1220 #[test]
1221 fn a_run_of_bytes_that_ends_part_way_through_its_alignment_costs_the_frame_nothing() {
1222 let (mut building, block) = Building::new();
1223 let addr = building.local(block, 0);
1224 building.through(block, addr);
1225 let (_, allocation) = building.allocate(1, 4);
1226
1227 // Twenty four bytes asking for sixteen is what a cell shared by a wide thing and a strict
1228 // one looks like, and it ends eight bytes into an alignment. Which way round the two are
1229 // given is not allowed to matter, because the order they are placed in is this pass's
1230 // business and the order they were declared in is not.
1231 let ragged = Local { size: 24, align: 16 };
1232 let whole = Local { size: 32, align: 16 };
1233 let size = |locals: &[Local]| {
1234 let layout = Layout { leaf: false, locals, ..Layout::new(&SYSV, REGS) };
1235 Frame::of(&building.func, &allocation, &layout).size()
1236 };
1237
1238 assert_eq!(size(&[ragged, whole]), size(&[whole, ragged]));
1239 // The two of them end to end with no hole between, which with the return address on top
1240 // of it is already where a call wants the stack pointer, so nothing is added for that.
1241 assert_eq!(size(&[ragged, whole]), 56);
1242 }
1243
1244 #[test]
1245 fn a_build_whose_locals_keep_their_own_bytes_still_shares_the_spill_slots() {
1246 let (mut building, block) = Building::new();
1247 let first = building.local(block, 0);
1248 building.through(block, first);
1249 let second = building.local(block, 1);
1250 building.through(block, second);
1251 // Two stretches of three values with two registers to hand out, one after the other, so
1252 // what goes to the stack in the first is finished with before the second starts.
1253 for _ in 0..2 {
1254 let values: Vec<Reg> = (0..3).map(|_| building.value(block)).collect();
1255 for ® in &values {
1256 building.held(block, reg);
1257 }
1258 }
1259 let (_, allocation) = building.allocate(2, 2);
1260
1261 let widths = vec![8; allocation.assignment.spilled()];
1262 let plan = Slots::share(&building.func, None, &allocation, &[WORD, WORD], &widths);
1263 assert_ne!(plan.local(0), plan.local(1), "a variable somebody can ask for keeps its bytes");
1264 assert_eq!(plan.slot(0), plan.slot(1), "and two spilled values that never meet share");
1265 }
1266
1267 /// A spill slot wanting bytes over one stretch of the line.
1268 fn slot(number: usize, start: Point, end: Point) -> Want {
1269 Want { what: What::Slot(number), size: 8, align: 8, area: Some(vec![Range { start, end }]) }
1270 }
1271
1272 #[test]
1273 fn what_is_left_when_the_budget_runs_out_gets_bytes_of_its_own() {
1274 // Three that are never both wanted, which is one run of bytes for the three of them when
1275 // there is anything to spend on finding that out.
1276 let three = || vec![slot(0, 0, 10), slot(1, 20, 30), slot(2, 40, 50)];
1277 assert_eq!(fit(three(), 0, 3, BUDGET).cells().len(), 1);
1278 // One comparison puts the second beside the first and leaves nothing for the third.
1279 assert_eq!(fit(three(), 0, 3, 1).cells().len(), 2);
1280 assert_eq!(fit(three(), 0, 3, 0).cells().len(), 3);
1281 }
1282}