Enum TileOp 

pub enum TileOp {
    LoadShared {
        src: String,
        tile_size: (u32, u32),
    },
    Mma {
        a: String,
        b: String,
        c: String,
    },
    Elementwise {
        op: ElementwiseOp,
        operands: Vec<String>,
        output: Option<String>,
    },
    StoreShared {
        dst: String,
    },
    Barrier,
    Reduce {
        kind: ReduceKind,
        input: String,
        output: String,
    },
}

Tile operation in compute shader

Variants

LoadShared

Load tile from global to shared memory

Fields

src: String

Source tensor name

tile_size: (u32, u32)

Tile dimensions

Mma

Matrix multiply accumulate (tensor core pattern)

Fields

a: String

Input A tensor

b: String

Input B tensor

c: String

Output C tensor

Elementwise

Element-wise operation

Fields

op: ElementwiseOp

Operation type

operands: Vec<String>

Input operand names

output: Option<String>

Output name (defaults to first operand if None)

StoreShared

Store tile from shared to global memory

Fields

dst: String

Destination tensor name

Barrier

Synchronization barrier

Reduce

Reduction operation (sum, max, min)

Fields

kind: ReduceKind

Reduction type

input: String

Input tensor

output: String

Output scalar or reduced tensor

Trait Implementations

impl Clone for TileOp

fn clone(&self) -> TileOp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TileOp

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

Formats the value using the given formatter.