Enum WmmaInstruction

pub enum WmmaInstruction<D: Dialect> {
    Fill {
        frag: Variable<D>,
        value: Variable<D>,
    },
    Load {
        frag: Variable<D>,
        value: Variable<D>,
        stride: Variable<D>,
        layout: Option<FragmentLayout<D>>,
    },
    Execute {
        frag_a: Variable<D>,
        frag_b: Variable<D>,
        frag_c: Variable<D>,
        frag_d: Variable<D>,
        warp_size: u32,
    },
    Store {
        output: Variable<D>,
        frag: Variable<D>,
        stride: Variable<D>,
        layout: FragmentLayout<D>,
    },
    Cast {
        input: Variable<D>,
        output: Variable<D>,
    },
}

Warp Matrix-Multiply and Accumulate Instruction.

Variants

Fill

Fill the fragment with the value.

Fields

frag: Variable<D>

value: Variable<D>

Load

Load the value into the fragment given the stride.

Fields

frag: Variable<D>

value: Variable<D>

stride: Variable<D>

layout: Option<FragmentLayout<D>>

Execute

Executes D=A*B+C;

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

Fields

frag_a: Variable<D>

frag_b: Variable<D>

frag_c: Variable<D>

frag_d: Variable<D>

warp_size: u32

Store

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

Fields

output: Variable<D>

frag: Variable<D>

stride: Variable<D>

layout: FragmentLayout<D>

Cast

Cast

Fields

input: Variable<D>

output: Variable<D>

Trait Implementations

impl<D: Clone + Dialect> Clone for WmmaInstruction<D>

fn clone(&self) -> WmmaInstruction<D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<D: Debug + Dialect> Debug for WmmaInstruction<D>

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

Formats the value using the given formatter.

impl<D: Dialect> Display for WmmaInstruction<D>

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

Formats the value using the given formatter.

impl<D: Copy + Dialect> Copy for WmmaInstruction<D>