zhc_ir 0.3.0

Graph-based intermediate representation framework with dialect support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use zhc_utils::{SafeAs, iter::CollectInVec};

use crate::{Dialect, IR, OpId, OpIdRaw, OpMap, ValId, ValMap, traversal::OpWalkerVerifier};

pub fn reschedule<'a, D: Dialect>(
    ir: &'a IR<D>,
    walker: impl Iterator<Item = OpId>,
) -> (IR<D>, OpMap<OpId>, ValMap<ValId>) {
    let mut output = IR::empty();
    let mut valmap = ir.empty_valmap();
    let walker = walker.covec();
    assert_eq!(
        walker.len().sas::<OpIdRaw>(),
        ir.n_ops(),
        "Tried to schedule with a walker that does not visit all operations."
    );
    assert!(
        walker.iter().copied().is_topo_sorted(ir),
        "Tried to schedule with a walker that is not topo sorted."
    );
    let mut opmap = ir.empty_opmap();
    for op in ir.walk_ops_with(walker.into_iter()) {
        let new_args = op
            .get_arg_valids()
            .iter()
            .map(|a| *valmap.get(a).unwrap())
            .collect();
        let (new_opid, new_rets) = output.add_op(op.get_instruction().clone(), new_args);
        assert!(opmap.insert(op.get_id(), new_opid).is_none());
        op.get_return_valids()
            .iter()
            .copied()
            .zip(new_rets.into_iter())
            .for_each(|(old, new)| assert!(valmap.insert(old, new).is_none()));
    }
    (output, opmap, valmap)
}