Enum CoopMma

pub enum CoopMma {
    Fill {
        mat: Variable,
        value: Variable,
    },
    Load {
        mat: Variable,
        value: Variable,
        stride: Variable,
        layout: Option<MatrixLayout>,
    },
    Execute {
        mat_a: Variable,
        mat_b: Variable,
        mat_c: Variable,
        mat_d: Variable,
    },
    Store {
        output: Variable,
        mat: Variable,
        stride: Variable,
        layout: MatrixLayout,
    },
}

Cooperative Matrix-Multiply and Accumulate Instruction.

Variants

Fill

Fill the matrix with the value.

Fields

mat: Variable

value: Variable

Load

Load the value into the matrix given the stride.

Fields

mat: Variable

value: Variable

stride: Variable

layout: Option<MatrixLayout>

Execute

Executes D=A*B+C;

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

Fields

mat_a: Variable

mat_b: Variable

mat_c: Variable

mat_d: Variable

Store

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

Fields

output: Variable

mat: Variable

stride: Variable

layout: MatrixLayout

Trait Implementations

impl Clone for CoopMma

fn clone(&self) -> CoopMma

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for CoopMma

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for CoopMma

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for CoopMma

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

Formats the value using the given formatter.

impl PartialEq for CoopMma

fn eq(&self, other: &CoopMma) -> 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 Serialize for CoopMma

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for CoopMma

impl StructuralPartialEq for CoopMma