Function count_uses

crates/ir-to-asm/src/compiler/register_allocator.rs:112–130

Static use+def occurrence count per register; the allocation priority.

(func: &IrFunction)

// Static use+def occurrence count per register; the allocation priority.
fn count_uses(func: &IrFunction) -> HashMap<IrRegister, usize> {
    let mut counts: HashMap<IrRegister, usize> = HashMap::new();
    for block in &func.blocks {
        for inst in &block.instructions {
            for r in slot_coloring::inst_uses(inst) {
                *counts.entry(r).or_insert(0) += 1;
            }
            for r in slot_coloring::inst_defs(inst) {
                *counts.entry(r).or_insert(0) += 1;
            }
        }
        if let Some(term) = &block.terminator {
            for r in slot_coloring::term_uses(term) {
                *counts.entry(r).or_insert(0) += 1;
            }
        }
    }
    counts
}

#[cfg(test)]
mod tests {
    use super::*;