Struct melior::dialect::ods::gpu::SubgroupMmaStoreMatrixOperation

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

A subgroup_mma_store_matrix operation. GPU warp synchronous matrix store.

The gpu.subgroup_mma_store_matrix operation stores a matrix collectively using all the threads in a subgroup.

This operation takes a !gpu.mma_matrix and a memref as operands. !gpu.mma_matrix is the source value containing the data to be stored into the destination memref which can be in global or shared memory. The store address is determined using the indices provided. The leadDimension attribute specifies the leading dimension of the destination matrix. If the transpose attribute is present then the op does a transposed store.

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

Example:

gpu.subgroup_mma_store_matrix %D, %sg[%i,%j] : { leadDimension = 32 : i32}
                : !gpu.mma_matrix<16x16xf16, "COp">, memref<32x32xf16, 3>

Implementations

impl<'c> SubgroupMmaStoreMatrixOperation<'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> ) -> SubgroupMmaStoreMatrixOperationBuilder<'c, Unset, Unset, Unset, Unset>

Creates a builder.

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

pub fn dst_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<SubgroupMmaStoreMatrixOperation<'c>> for Operation<'c>

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

Converts to this type from the input type.

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