Enum Feature

pub enum Feature {
    Plane,
    Cmma {
        a: Elem,
        b: Elem,
        c: Elem,
        m: u8,
        k: u8,
        n: u8,
    },
    CmmaWarpSize(i32),
    Type(Elem),
    AtomicFloat(AtomicFeature),
    Pipeline,
    Barrier,
    Tma(TmaFeature),
    CubeCluster,
    DynamicLineSize,
}

Every feature that can be supported by a cube runtime.

Variants

Plane

The plane feature enables all basic warp/subgroup operations.

Cmma

The cmma feature enables cooperative matrix-multiply and accumulate operations.

Fields

a: Elem

b: Elem

c: Elem

m: u8

k: u8

n: u8

CmmaWarpSize(i32)

Type(Elem)

AtomicFloat(AtomicFeature)

Features supported for floating point atomics. For integers, all methods are supported as long as the type is.

Pipeline

The pipeline feature enables pipelined (async) operations

Barrier

The barrier feature enables barrier (async) operations

Tma(TmaFeature)

Tensor Memory Accelerator features. Minimum H100/RTX 5000 series for base set

CubeCluster

Clustered launches and intra-cluster operations like cluster shared memory

DynamicLineSize

Enables to change the line size of containers during kernel execution.

Trait Implementations

impl Clone for Feature

fn clone(&self) -> Feature

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Feature

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

Formats the value using the given formatter.

impl Ord for Feature

fn cmp(&self, other: &Feature) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized,

Restrict a value to a certain interval.

impl PartialEq for Feature

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

fn partial_cmp(&self, other: &Feature) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

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

Tests less than (for self and other) and is used by the < operator.

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

Tests less than or equal to (for self and other) and is used by the <= operator.

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

Tests greater than (for self and other) and is used by the > operator.

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

Tests greater than or equal to (for self and other) and is used by the >= operator.

impl Copy for Feature

impl Eq for Feature

impl StructuralPartialEq for Feature