Enum SwizzleMode 

pub enum SwizzleMode {
    None,
    B32,
    B64,
    B128,
}

Swizzling mode of the shared memory. Default None. Matches the base TMA functionality, alternative chunk sizes or more complex patterns don’t really apply to matmul.

Variants

None

No swizzling

B32

Swizzle 16B chunks within 32B span Swizzle<1,4,3>

B64

Swizzle 16B chunks within 64B span Swizzle<2,4,3>

B128

Swizzle 16B chunks within 128B span Swizzle<3,4,3>

Implementations

impl SwizzleMode

pub fn atom_size(&self) -> usize

pub fn span_size(&self) -> usize

Trait Implementations

impl Clone for SwizzleMode

fn clone(&self) -> SwizzleMode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SwizzleMode

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

Formats the value using the given formatter.

impl Default for SwizzleMode

fn default() -> SwizzleMode

Returns the “default value” for a type.

impl From<SwizzleMode> for TensorMapSwizzle

fn from(value: SwizzleMode) -> Self

Converts to this type from the input type.

impl Hash for SwizzleMode

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 SwizzleMode

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

impl Eq for SwizzleMode

impl StructuralPartialEq for SwizzleMode