Struct SubgroupReduceOperation 

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

A subgroup_reduce operation. Reduce values among subgroup..

The subgroup_reduce op reduces the values of lanes (work items) across a subgroup.

The subgroup is divided into clusters starting at lane index 0. Within each cluster, there are size lanes, and the lane index advances by stride. A reduction is done for each cluster in parallel: every lane in the cluster is reduced, and the result is equal for all lanes in the cluster. If size is omitted, there is a single cluster covering the entire subgroup. If stride is omitted, the stride is 1 (the cluster’s lanes are contiguous).

When the reduced value is of a vector type, each vector element is reduced independently. Only 1-d vector types are allowed.

Example:

%1 = gpu.subgroup_reduce add %a : (f32) -> f32
%2 = gpu.subgroup_reduce add %b : (vector<4xf16>) -> vector<4xf16>
%3 = gpu.subgroup_reduce add %c cluster(size = 4) : (f32) -> f32
%3 = gpu.subgroup_reduce add %c cluster(size = 4, stride = 2) : (f32) -> f32

If uniform flag is set either none or all lanes of a subgroup need to execute this op in convergence.

The reduction operation must be one of:

• Integer types: add, mul, minui, minsi, maxui, maxsi, and, or, xor
• Floating point types: add, mul, minnumf, maxnumf, minimumf, maximumf

Implementations

impl<'c> SubgroupReduceOperation<'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>, ) -> SubgroupReduceOperationBuilder<'c, Unset, Unset>

Creates a builder.

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

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

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

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

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

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

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

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

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

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

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

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

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

Trait Implementations

impl<'c> Clone for SubgroupReduceOperation<'c>

fn clone(&self) -> SubgroupReduceOperation<'c>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<'c> Debug for SubgroupReduceOperation<'c>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'b> From<SubgroupReduceOperation<'b>> for GpuDialectOperation<'b>

fn from(op: SubgroupReduceOperation<'b>) -> Self

Converts to this type from the input type.

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

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

Converts to this type from the input type.

impl<'c> PartialEq for SubgroupReduceOperation<'c>

fn eq(&self, other: &SubgroupReduceOperation<'c>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

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

impl<'c> Eq for SubgroupReduceOperation<'c>

impl<'c> StructuralPartialEq for SubgroupReduceOperation<'c>