Enum WmmaInstruction 

pub enum WmmaInstruction<D: Dialect> {
    Fill {
        frag: Variable<D>,
        value: Variable<D>,
    },
    Load {
        frag: Variable<D>,
        value: Variable<D>,
        offset: 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,
    },
    ExecuteManual {
        shape: MmaShape<D>,
        frag_a: Variable<D>,
        frag_b: Variable<D>,
        frag_c: Variable<D>,
        frag_d: Variable<D>,
    },
    ExecuteScaled {
        shape: MmaShape<D>,
        frag_a: Variable<D>,
        frag_b: Variable<D>,
        frag_c: Variable<D>,
        frag_d: Variable<D>,
        scales_a: Variable<D>,
        scales_b: Variable<D>,
        scales_factor: u32,
    },
    Store {
        output: Variable<D>,
        frag: Variable<D>,
        stride: Variable<D>,
        offset: Variable<D>,
        layout: FragmentLayout<D>,
    },
    LdMatrix {
        output: Variable<D>,
        buffer: Variable<D>,
        offset: Variable<D>,
        line_size: Option<u32>,
        factor: u32,
        transpose: bool,
    },
    StMatrix {
        registers: Variable<D>,
        buffer: Variable<D>,
        offset: Variable<D>,
        line_size: Option<u32>,
        factor: u32,
        transpose: bool,
    },
    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>

offset: 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

ExecuteManual

Executes D=A*B+C using manually managed registers;

For implementing a matmul, D=C : C+=A*B Takes a sequence of registers for the inputs, and returns an array of registers for the output. PTX requires output registers to be non-overlapping, so we use array to ensure that and handle potentially destructuring it internally.

Fields

shape: MmaShape<D>

frag_a: Variable<D>

frag_b: Variable<D>

frag_c: Variable<D>

frag_d: Variable<D>

ExecuteScaled

Executes D=A*B+C using manually managed registers;

For implementing a matmul, D=C : C+=A*B Takes a sequence of registers for the inputs, and returns an array of registers for the output. PTX requires output registers to be non-overlapping, so we use array to ensure that and handle potentially destructuring it internally.

Fields

shape: MmaShape<D>

frag_a: Variable<D>

frag_b: Variable<D>

frag_c: Variable<D>

frag_d: Variable<D>

scales_a: Variable<D>

scales_b: Variable<D>

scales_factor: 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>

offset: Variable<D>

layout: FragmentLayout<D>

LdMatrix

Load a part of a fragment into registers, either 1, 2, or 4 at once.

Fields

output: Variable<D>

buffer: Variable<D>

offset: Variable<D>

line_size: Option<u32>

factor: u32

transpose: bool

StMatrix

Store a part of a fragment into smem, either 1, 2, or 4 at once.

Fields

registers: Variable<D>

buffer: Variable<D>

offset: Variable<D>

line_size: Option<u32>

factor: u32

transpose: bool

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 duplicate 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: PartialEq + Dialect> PartialEq for WmmaInstruction<D>

fn eq(&self, other: &WmmaInstruction<D>) -> 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<D: Dialect> StructuralPartialEq for WmmaInstruction<D>