Struct StageSize

pub struct StageSize {
    pub m: u8,
    pub n: u8,
    pub k: u8,
}

Fields

m: u8

n: u8

k: u8

Implementations

impl StageSize

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

pub fn get(&self, dim: MatmulDim) -> u32

pub fn m(&self) -> u32

pub fn n(&self) -> u32

pub fn k(&self) -> u32

pub fn mn(&self) -> u32

pub fn mk(&self) -> u32

pub fn nk(&self) -> u32

pub fn mnk(&self) -> u32

Trait Implementations

impl Clone for StageSize

fn clone(&self) -> StageSize

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for StageSize

type ExpandType = StageSizeExpand

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl Debug for StageSize

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

Formats the value using the given formatter.

impl From<(i32, i32, i32)> for StageSize

fn from(value: (i32, i32, i32)) -> Self

Converts to this type from the input type.

impl From<(u16, u16, u16)> for StageSize

fn from(value: (u16, u16, u16)) -> Self

Converts to this type from the input type.

impl From<(u32, u32, u32)> for StageSize

fn from(value: (u32, u32, u32)) -> Self

Converts to this type from the input type.

impl From<(u8, u8, u8)> for StageSize

fn from(value: (u8, u8, u8)) -> Self

Converts to this type from the input type.

impl From<(usize, usize, usize)> for StageSize

fn from(value: (usize, usize, usize)) -> Self

Converts to this type from the input type.

impl From<StageSize> for (i32, i32, i32)

fn from(value: StageSize) -> Self

Converts to this type from the input type.

impl From<StageSize> for (u16, u16, u16)

fn from(value: StageSize) -> Self

Converts to this type from the input type.

impl From<StageSize> for (u32, u32, u32)

fn from(value: StageSize) -> Self

Converts to this type from the input type.

impl From<StageSize> for (u8, u8, u8)

fn from(value: StageSize) -> Self

Converts to this type from the input type.

impl From<StageSize> for (usize, usize, usize)

fn from(value: StageSize) -> Self

Converts to this type from the input type.

impl Hash for StageSize

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 StageSize

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

impl Eq for StageSize

impl StructuralPartialEq for StageSize