Struct melior::dialect::ods::gpu::SubgroupMmaElementwiseOperation

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

A subgroup_mma_elementwise operation. GPU warp elementwise operation on a matrix.

The gpu.subgroup_mma_elementwise takes !gpu.mma_matrix inputs and compute a new !gpu.mma_matrix by applying an elementwise operation to each element.

Since the operation is elementwise and the matrix type must match, the matrix elements are processed independently of the matrix layout.

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

Example:

 %0 =  %A, %B { opType = "ADD" } :
  (!gpu.mma_matrix<16x16xf16, "COp">, !gpu.mma_matrix<16x16xf16, "COp">)
  -> !gpu.mma_matrix<16x16xf16, "COp">

Implementations

impl<'c> SubgroupMmaElementwiseOperation<'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> ) -> SubgroupMmaElementwiseOperationBuilder<'c, Unset, Unset, Unset>

Creates a builder.

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

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

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

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

Trait Implementations

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

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

Converts to this type from the input type.

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