Enum MatmulTransformAction 

pub enum MatmulTransformAction {
    Keep,
    MergeBatches {
        rows: usize,
    },
}

The action to take for a batched matmul with a broadcast rhs.

Variants

Keep

Launch the matmul as-is.

MergeBatches

Fold every lhs/out batch dim into the rows: [.., b, m, k] @ [.., 1, k, n] runs as one [b*m, k] @ [k, n] matmul against the shared rhs, instead of b matmuls that each re-read it.

Fields

rows: usize

The merged row count (batches × rows).

Implementations

impl MatmulTransformAction

pub fn apply(&self, meta: &mut Metadata)

Apply the action to one merged operand (lhs or out): fold the batch dims into the rows, in place. A pure metadata rewrite — the buffer is shared.

Requires batch-contiguous rows (see MatmulTransformAnalysis); the merge keeps the rank, with every batch dim set to 1.

Trait Implementations

impl Clone for MatmulTransformAction

fn clone(&self) -> MatmulTransformAction

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for MatmulTransformAction

impl Debug for MatmulTransformAction

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

Formats the value using the given formatter.

impl Eq for MatmulTransformAction

impl PartialEq for MatmulTransformAction

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for MatmulTransformAction