Struct MatmulFeatures 

pub struct MatmulFeatures {
    pub cmma: BTreeSet<MmaConfig>,
    pub cube_mma: BTreeSet<CubeMmaConfig>,
    pub mma: BTreeSet<MmaConfig>,
    pub scaled_mma: BTreeSet<ScaledMmaConfig>,
    pub ldmatrix: BTreeSet<StorageType>,
    pub stmatrix: BTreeSet<StorageType>,
    pub cmma_tensor_addressing: bool,
}

Matrix multiplication-related features

Fields

cmma: BTreeSet<MmaConfig>

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

cube_mma: BTreeSet<CubeMmaConfig>

Cube MMA is like cmma but at the cube level, rather than the plane level. Loading may be staged in shared memory by the driver on Vulkan - check cube_mma_reserved_shared_memory to take this into account when generating a matmul config.

mma: BTreeSet<MmaConfig>

The manual MMA feature enables cooperative matrix-multiply with manually managed data movement

scaled_mma: BTreeSet<ScaledMmaConfig>

Scaled MMA allows combining matrix multiplication with unscaling quantized values into a single instruction. Scales must fit a specific layout and block size.

ldmatrix: BTreeSet<StorageType>

Types supported for ldmatrix, if any

stmatrix: BTreeSet<StorageType>

Types supported by stmatrix, if any

cmma_tensor_addressing: bool

Whether tensor addressing is supported for CMMA load/store

Trait Implementations

impl Clone for MatmulFeatures

fn clone(&self) -> MatmulFeatures

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MatmulFeatures

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

Formats the value using the given formatter.

impl Default for MatmulFeatures

fn default() -> MatmulFeatures

Returns the “default value” for a type.

impl Eq for MatmulFeatures

impl Hash for MatmulFeatures

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for MatmulFeatures

fn eq(&self, other: &MatmulFeatures) -> bool

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for MatmulFeatures