Struct FusionPattern 

pub struct FusionPattern { /* private fields */ }

A fusion rule: an op-type sequence rewritten to a single replacement op.

Implementations

impl FusionPattern

pub fn new(name: &str, ops: &[&str], replacement: &str) -> Self

A new structural pattern matching ops in sequence, replaced by replacement. The fused node’s inputs are the matched region’s external inputs in first-seen order.

pub fn layernorm() -> Self

The schema-aware LayerNorm pattern: the canonical 9-op decomposition (ReduceMean, Sub, Pow, ReduceMean, Add, Sqrt, Div, Mul, Add) rewritten to a com.microsoft::LayerNormalization node with inputs [X, Scale, B] and synthesized axis/epsilon attributes.

pub fn kind(&self) -> RewriteKind

This pattern’s rewrite kind.

pub fn attention() -> Self

The schema-aware SDPA-core pattern, rewritten to a com.microsoft::FusedAttention node with inputs [Q, K, V] (+ optional [mask]) and synthesized scale/k_transposed attributes. Anchored on the Softmax (see Self::try_match_attention).

pub fn gelu() -> Self

The schema-aware exact-GELU pattern: the Erf decomposition 0.5·X · (1 + Erf(X / √2)) rewritten to a com.microsoft::Gelu node with the single input [X] and no attributes. Anchored on the Erf (see Self::try_match_gelu).

pub fn pattern_name(&self) -> &str

This pattern’s name.

pub fn find_match(&self, graph: &Graph) -> Option<PatternMatch>

Find the next occurrence of this pattern, scanning nodes in id order.

[RewriteKind::LayerNorm] uses a dedicated DAG-aware matcher (Self::try_match_layernorm) because a real LayerNorm decomposition is a diamond DAG whose mean feeds two branches (variance + numerator) and may even use two distinct Sub(x, mean) nodes; the linear successor-walk used by the structural patterns can’t express that. All structural patterns (MatMul+Add, MatMul+Add+Relu) keep the linear-chain matcher.

pub fn apply_fusion(&self, graph: &mut Graph, m: &PatternMatch) -> Result<()>

Apply a match: remove the matched nodes and insert the replacement, reusing m.output so downstream consumers and graph outputs stay wired.

Trait Implementations

impl Clone for FusionPattern

fn clone(&self) -> FusionPattern

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for FusionPattern

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.