Enum cubecl_cuda::compiler::WmmaInstruction

pub enum WmmaInstruction {
    Fill {
        frag: Variable,
        value: Variable,
    },
    Load {
        frag: Variable,
        value: Variable,
        stride: Variable,
    },
    Execute {
        frag_a: Variable,
        frag_b: Variable,
        frag_c: Variable,
        frag_d: Variable,
    },
    Store {
        output: Variable,
        frag: Variable,
        stride: Variable,
        layout: FragmentLayout,
    },
}

Warp Matrix-Multiply and Accumulate Instruction.

Variants

Fill

Fill the fragment with the value.

Fields

frag: Variable

value: Variable

Load

Load the value into the fragment given the stride.

Fields

frag: Variable

value: Variable

stride: Variable

Execute

Executes D=A*B+C;

For implementing a matmul, D=C : C+=A*B

Fields

frag_a: Variable

frag_b: Variable

frag_c: Variable

frag_d: Variable

Store

Store the fragment in an output variable following the stride and the layout.

Fields

output: Variable

frag: Variable

stride: Variable

layout: FragmentLayout

Trait Implementations

impl Clone for WmmaInstruction

fn clone(&self) -> WmmaInstruction

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for WmmaInstruction

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

Formats the value using the given formatter.

impl Display for WmmaInstruction

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

Formats the value using the given formatter.

impl Copy for WmmaInstruction