celox-sir-opt 0.4.0

Backend-independent SIR optimization policy and passes for Celox
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

//! Public adapters for targeted SIR optimization operations.

use super::passes::control_flow::pass_guarded_region_sinking;
use super::passes::dataflow::pass_vectorize_concat;
use super::passes::memory::{
    fused_comb_dse, pass_dead_store_elimination, pass_global_store_load_forwarding,
    pass_identity_store_bypass,
};
use super::pipeline::pass_manager;
use super::*;

pub fn promote_eval_apply_working_round_trips(
    eu: &mut ExecutionUnit<RegionedAbsoluteAddr>,
) -> bool {
    pass_global_store_load_forwarding::promote_eval_apply_working_round_trips(eu)
}

pub fn remove_dead_sir_definitions(eu: &mut ExecutionUnit<RegionedAbsoluteAddr>) {
    pass_vectorize_concat::remove_dead_definitions(eu);
}

pub fn eliminate_unobserved_comb_state_stores(
    eu: &mut ExecutionUnit<RegionedAbsoluteAddr>,
    provenance: &crate::ir::SirMergeProvenance,
    first_ff_unit: usize,
) -> Result<usize, crate::OptimizationError> {
    fused_comb_dse::eliminate(eu, provenance, first_ff_unit)
}

pub fn eliminate_shared_comb_state_stores(
    eu: &mut ExecutionUnit<RegionedAbsoluteAddr>,
    direct_ff_writes: &[crate::ir::VarAtomBase<RegionedAbsoluteAddr>],
) -> Result<usize, crate::OptimizationError> {
    fused_comb_dse::eliminate_shared(eu, direct_ff_writes)
}

pub fn promote_fused_comb_static_slots(
    eu: &mut ExecutionUnit<RegionedAbsoluteAddr>,
) -> Result<bool, crate::OptimizationError> {
    pass_global_store_load_forwarding::promote_fused_comb_static_slots(eu)
}

pub fn retain_final_identity_aliases(program: &mut OptimizationContext<'_>, four_state: bool) {
    pass_identity_store_bypass::retain_final_identity_aliases(program, four_state);
}

pub fn remove_final_identity_alias_stores(
    program: &mut OptimizationContext<'_>,
    validated_aliases: &crate::HashMap<AbsoluteAddr, AbsoluteAddr>,
    four_state: bool,
) {
    pass_identity_store_bypass::remove_final_identity_alias_stores(
        program,
        validated_aliases,
        four_state,
    );
}

pub fn optimize_rooted_comb_memory(
    program: &mut OptimizationContext<'_>,
    externally_live: &crate::HashSet<AbsoluteAddr>,
    four_state: bool,
) {
    pass_dead_store_elimination::eliminate_dead_stores(program, externally_live);
    let options = PassOptions {
        four_state,
        ..PassOptions::default()
    };
    for eu in &mut program.sir.eval_comb {
        pass_vectorize_concat::remove_dead_definitions(eu);
        pass_guarded_region_sinking::eliminate_dead_control_regions(eu);
        pass_manager::ExecutionUnitPass::run(&ControlFlowSimplifyPass, eu, &options);
        pass_vectorize_concat::remove_dead_definitions(eu);
        pass_guarded_region_sinking::sink_pure_values_with_predicate_repair(eu);
        pass_guarded_region_sinking::eliminate_dead_control_regions(eu);
        pass_manager::ExecutionUnitPass::run(&ControlFlowSimplifyPass, eu, &options);
        pass_vectorize_concat::remove_dead_definitions(eu);
    }
}