Struct melior::dialect::ods::amdgpu::MFMAOperation

pub struct MFMAOperation<'c> { /* private fields */ }

A mfma operation. MLIR wrapper for CDNA mfma instructions.

The amdgpu.mfma op is an MLIR wrapper around intrinsics for various mfma instructions in the CDNA architecture, which perform multiple outer products in order to allow fast matrix multiplication.

The wrapper will select an appropriate mfma instruction, if one is available, based on the provided m, k, n, and nBlks attributes, along with the types of the source and destination arguments.

For information on the layouts of the input and output matrces (which are stored in sourceA, sourceB, destC, and destD), see the CDNA ISA documentation.

The cbsz, abid, and blgp parameters control how the lanes of the wave are permuted when matrix data is being loaded: blgp can be any number of fixed permutations, cbsz specifies the log_2 of the number of chunks the lanes holding sourceA are split into, and abid selects one of those chunks.

Note, this wrapper allows specifying vector<4Kxi8> arguments to MFMA intrinsics that take an integer type of width 4K. For example, one can provide a vector<4xi8> as an argument to an MFMA instruction that logically takes 4 i8s but whose intrinsics are specified to take an i32. In these cases, the bytes in the vector will be concatenated in little-endian order (that is, v[0] will go to arg[7:0], v[1] to arg[15:8] and so on).

The negateA, negateB, and negateC flags are only supported for double-precision operations on gfx940+.

Implementations

impl<'c> MFMAOperation<'c>

pub fn name() -> &'static str

Returns a name.

pub fn as_operation(&self) -> &Operation<'c>

Returns a generic operation.

pub fn builder( context: &'c Context, location: Location<'c> ) -> MFMAOperationBuilder<'c, Unset, Unset, Unset, Unset, Unset, Unset, Unset, Unset>

Creates a builder.

pub fn dest_d(&self) -> Result<OperationResult<'c, '_>, Error>

pub fn source_a(&self) -> Result<Value<'c, '_>, Error>

pub fn source_b(&self) -> Result<Value<'c, '_>, Error>

pub fn dest_c(&self) -> Result<Value<'c, '_>, Error>

pub fn m(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_m(&mut self, value: IntegerAttribute<'c>)

pub fn n(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_n(&mut self, value: IntegerAttribute<'c>)

pub fn k(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_k(&mut self, value: IntegerAttribute<'c>)

pub fn blocks(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_blocks(&mut self, value: IntegerAttribute<'c>)

pub fn cbsz(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_cbsz(&mut self, value: IntegerAttribute<'c>)

pub fn remove_cbsz(&mut self) -> Result<(), Error>

pub fn abid(&self) -> Result<IntegerAttribute<'c>, Error>

pub fn set_abid(&mut self, value: IntegerAttribute<'c>)

pub fn remove_abid(&mut self) -> Result<(), Error>

pub fn blgp(&self) -> Result<Attribute<'c>, Error>

pub fn set_blgp(&mut self, value: Attribute<'c>)

pub fn remove_blgp(&mut self) -> Result<(), Error>

pub fn reduce_precision(&self) -> Result<Attribute<'c>, Error>

pub fn set_reduce_precision(&mut self, value: Attribute<'c>)

pub fn remove_reduce_precision(&mut self) -> Result<(), Error>

pub fn negate_a(&self) -> Result<Attribute<'c>, Error>

pub fn set_negate_a(&mut self, value: Attribute<'c>)

pub fn remove_negate_a(&mut self) -> Result<(), Error>

pub fn negate_b(&self) -> Result<Attribute<'c>, Error>

pub fn set_negate_b(&mut self, value: Attribute<'c>)

pub fn remove_negate_b(&mut self) -> Result<(), Error>

pub fn negate_c(&self) -> Result<Attribute<'c>, Error>

pub fn set_negate_c(&mut self, value: Attribute<'c>)

pub fn remove_negate_c(&mut self) -> Result<(), Error>

Trait Implementations

impl<'c> From<MFMAOperation<'c>> for Operation<'c>

fn from(operation: MFMAOperation<'c>) -> Self

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for MFMAOperation<'c>

type Error = Error

The type returned in the event of a conversion error.

fn try_from(operation: Operation<'c>) -> Result<Self, Self::Error>

Performs the conversion.