Struct CubeMmaConfig 

pub struct CubeMmaConfig {
    pub a_type: StorageType,
    pub b_type: StorageType,
    pub cd_type: StorageType,
    pub m_granularity: u32,
    pub m_max: u32,
    pub n_granularity: u32,
    pub n_max: u32,
    pub k_granularity: u32,
    pub k_max: u32,
    pub units_per_block: Option<u32>,
}

Shape and element types of a valid flexible MMA configuration Only Vulkan for now, but this should also be usable for wgmma/xmma on datacenter CUDA. Actual matrix size must be multiple of granularity and <= max.

Fields

a_type: StorageType

Element of the A matrix

b_type: StorageType

Element of the B matrix

cd_type: StorageType

Element of the C/D matrices

m_granularity: u32

The granularity of the matrix on the m dimension

m_max: u32

The maximum value for m

n_granularity: u32

The size of the matrix on the n dimension

n_max: u32

The maximum value for n

k_granularity: u32

The size of the matrix on the k dimension

k_max: u32

The maximum value for k

units_per_block: Option<u32>

The number of units that must be in the cube for this configuration to be valid. None means it’s always valid (but might still have an optimal value).

Trait Implementations

impl Clone for CubeMmaConfig

fn clone(&self) -> CubeMmaConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for CubeMmaConfig

impl Debug for CubeMmaConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for CubeMmaConfig

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for CubeMmaConfig

impl Hash for CubeMmaConfig

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 Ord for CubeMmaConfig

fn cmp(&self, other: &CubeMmaConfig) -> 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 CubeMmaConfig

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

Equality operator ==.

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

Inequality operator !=.

impl PartialOrd for CubeMmaConfig

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

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 CubeMmaConfig