Struct melior::dialect::ods::vector::MatmulOperation

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

A matrix_multiply operation. Vector matrix multiplication op that operates on flattened 1-D MLIR vectors.

This is the counterpart of llvm.matrix.multiply in MLIR. It serves the purposes of more progressive lowering and localized type conversion. Higher levels typically lower matrix multiplications into ‘vector.contract’ operations. Subsequent rewriting rule progressively lower these operations into ‘vector.matrix_multiply’ operations to bring the operations closer to the hardware ISA.

The ‘vector.matrix_multiply’ op treats lhs as matrix with <lhs_rows> rows and <lhs_columns> columns, rhs as matrix with <lhs_columns> rows and <rhs_columns> and multiplies them. The result matrix is returned embedded in the result vector.

Also see:

http://llvm.org/docs/LangRef.html#llvm-matrix-multiply-intrinsic

Example:

%C = vector.matrix_multiply %A, %B
  { lhs_rows = 4: i32, lhs_columns = 16: i32 , rhs_columns = 3: i32 } :
  (vector<64xf64>, vector<48xf64>) -> vector<12xf64>

Implementations

impl<'c> MatmulOperation<'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> ) -> MatmulOperationBuilder<'c, Unset, Unset, Unset, Unset, Unset, Unset>

Creates a builder.

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

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

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

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

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

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

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

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

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

Trait Implementations

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

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

Converts to this type from the input type.

impl<'c> TryFrom<Operation<'c>> for MatmulOperation<'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.