Skip to main content

lift_opt/
lib.rs

1//! LIFT Opt: Optimisation passes for the LIFT compiler framework.
2//!
3//! Provides 11 passes: canonicalisation, constant folding, dead code
4//! elimination, tensor fusion, common subexpression elimination,
5//! flash attention replacement, quantisation annotation, gate cancellation,
6//! rotation merge, noise-aware scheduling, and layout mapping.
7
8pub mod canonicalize;
9pub mod common_subexpr;
10pub mod constant_fold;
11pub mod dce;
12pub mod flash_attention;
13pub mod gate_cancel;
14pub mod gate_decompose;
15pub mod layout_mapping;
16pub mod noise_aware_schedule;
17pub mod quantisation_pass;
18pub mod real_routing;
19pub mod rotation_merge;
20pub mod tensor_fusion;
21
22pub use canonicalize::Canonicalize;
23pub use common_subexpr::CommonSubexprElimination;
24pub use constant_fold::ConstantFolding;
25pub use dce::DeadCodeElimination;
26pub use flash_attention::FlashAttentionPass;
27pub use gate_cancel::GateCancellation;
28pub use gate_decompose::GateDecomposition;
29pub use layout_mapping::LayoutMapping;
30pub use noise_aware_schedule::NoiseAwareSchedule;
31pub use quantisation_pass::QuantisationPass;
32pub use real_routing::RealRouting;
33pub use rotation_merge::RotationMerge;
34pub use tensor_fusion::TensorFusion;