Enum CubeCountPlanSelection

pub enum CubeCountPlanSelection {
    FromProblem,
    Sm {
        cubes_first: bool,
        num_sms: u32,
        sm_usage: SmAllocation,
    },
    Flattened,
    Spread,
}

Front-facing configuration when crafting a MatmulSelection Allows choosing a strategy before knowing actual values

Variants

FromProblem

X: num cubes in m, Y: num cubes in n, Z: num cubes in batch

Sm

If not cubes_first: X: num SMs, Y: num cubes per SM If cubes_first: X: num cubes per SM, Y: num SMs

Fields

cubes_first: bool

num_sms: u32

sm_usage: SmAllocation

Flattened

X: total cubes flattened (num SMs * num cubes per SM)

Spread

Heuristically find a balance for X, Y, Z that respects hardware limits

Trait Implementations

impl Clone for CubeCountPlanSelection

fn clone(&self) -> CubeCountPlanSelection

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CubeCountPlanSelection

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

Formats the value using the given formatter.

impl Default for CubeCountPlanSelection

fn default() -> CubeCountPlanSelection

Returns the “default value” for a type.

impl Hash for CubeCountPlanSelection

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 CubeCountPlanSelection

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

impl Eq for CubeCountPlanSelection

impl StructuralPartialEq for CubeCountPlanSelection