Enum cubecl_core::ir::Operation

pub enum Operation {
    Operator(Operator),
    Procedure(Procedure),
    Metadata(Metadata),
    Branch(Branch),
    Synchronization(Synchronization),
    Subcube(Subcube),
    CoopMma(CoopMma),
}

All operations that can be used in a GPU compute shader.

Notes:

Operator and Procedure can be vectorized, but Metadata and Branch can’t. Therefore, during tracing, only operators and procedures can be registered.

Procedure expansions can safely use all operation variants.

Variants

Operator(Operator)

Procedure(Procedure)

Metadata(Metadata)

Branch(Branch)

Synchronization(Synchronization)

Subcube(Subcube)

CoopMma(CoopMma)

Trait Implementations

impl Clone for Operation

fn clone(&self) -> Operation

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Operation

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Operation

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl From<Branch> for Operation

fn from(value: Branch) -> Self

Converts to this type from the input type.

impl From<Metadata> for Operation

fn from(val: Metadata) -> Self

Converts to this type from the input type.

impl From<Operator> for Operation

fn from(val: Operator) -> Self

Converts to this type from the input type.

impl From<Procedure> for Operation

fn from(val: Procedure) -> Self

Converts to this type from the input type.

impl From<Synchronization> for Operation

fn from(value: Synchronization) -> Self

Converts to this type from the input type.

impl PartialEq for Operation

fn eq(&self, other: &Operation) -> 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 Serialize for Operation

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for Operation