Struct melior::dialect::ods::gpu::SubgroupMmaLoadMatrixOperation

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

A subgroup_mma_load_matrix operation. GPU warp synchronous matrix load.

The gpu.subgroup_mma_load_matrix operation loads a matrix collectively using all the threads in a subgroup.

This operation takes a memref as its first operand: it is the source matrix from which data is to be loaded. The op returns a !gpu.mma_matrix. The source memref can be in global memory or shared memory. The load address is determined using indices. The matrix being loaded into is the result. The leadDimension attribute specifies the leading dimension size of the source matrix which eventually allows the lowering to determine the size of each row. If the transpose attribute is present then the op does a transposed load.

For integer types, the resulting !gpu.mma_matrix type needs to specify the signedness of the data if the matrix type is an A or B operand for gpu.subgroup_mma_compute.

This op is often meant to be used along with gpu.subgroup_mma_store_matrix and gpu.subgroup_mma_compute.

Example:

 %0 = gpu.subgroup_mma_load_matrix src[%i,%j] : {leadDimension = 32 : i32}
      : memref<32x32xf16, 3>, !gpu.mma_matrix<16x16xf16, "AOp">

Implementations

impl<'c> SubgroupMmaLoadMatrixOperation<'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> ) -> SubgroupMmaLoadMatrixOperationBuilder<'c, Unset, Unset, Unset, Unset>

Creates a builder.

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

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

pub fn indices(&self) -> impl Iterator<Item = Value<'c, '_>>

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

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

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

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

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

Trait Implementations

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

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

Converts to this type from the input type.

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