Struct FragmentShape 

pub struct FragmentShape {
    pub m: usize,
    pub n: usize,
    pub k: usize,
}

Fragment shape for WMMA operations. Maps to hardware-supported shapes like 16×16×16, 8×32×16, etc.

Fields

m: usize

n: usize

k: usize

Implementations

impl FragmentShape

pub const M16N16K16: Self

Standard 16×16×16 (SM 7.0+ Volta)

pub const M16N8K16: Self

Ampere 16×8×16

pub const M8N32K16: Self

INT8: 8×32×16

pub fn new(m: usize, n: usize, k: usize) -> Self

Custom shape

Trait Implementations

impl Clone for FragmentShape

fn clone(&self) -> FragmentShape

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for FragmentShape

impl Debug for FragmentShape

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

Formats the value using the given formatter.

impl PartialEq for FragmentShape

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