Enum OpKind 

pub enum OpKind {
    MatMul,
    MatMulId,
    Norm,
    Rope,
    Elementwise,
    Copy,
    Gather,
    Sdpa,
    Softmax,
    MoeGate,
    Other,
}

Classification of a compute operation for reorder safety analysis.

Operations marked as reorderable can be freely reordered by the graph optimizer (Phase 4e.3) as long as their data dependencies allow it. Non-reorderable operations have side effects or dependencies that require them to stay in their original sequential position.

Variants

MatMul

Matrix multiplication (reorderable).

MatMulId

Expert-routed matrix multiplication (reorderable).

Norm

Normalization — RMS norm, layer norm (reorderable).

Rope

Rotary position embedding (reorderable).

Elementwise

Elementwise ops — add, mul, scale, gelu, softcap, etc. (reorderable).

Copy

Memory copy — KV cache copy, embedding gather (reorderable).

Gather

Gather/scatter (reorderable).

Sdpa

Scaled dot-product attention (NOT reorderable).

Softmax

Softmax (NOT reorderable).

MoeGate

MoE gate with CPU readback dependency (NOT reorderable).

Other

Anything else (NOT reorderable).

Implementations

impl OpKind

pub fn is_reorderable(&self) -> bool

Whether this op kind is safe to reorder in the graph optimizer.

Trait Implementations

impl Clone for OpKind

fn clone(&self) -> OpKind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OpKind

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

Formats the value using the given formatter.

impl PartialEq for OpKind

fn eq(&self, other: &OpKind) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for OpKind

impl Eq for OpKind

impl StructuralPartialEq for OpKind