Struct PartitionSchedulerExpand 

pub struct PartitionSchedulerExpand {
    pub m: <AxisScheduler as CubeType>::ExpandType,
    pub n: <AxisScheduler as CubeType>::ExpandType,
    pub k: <AxisScheduler as CubeType>::ExpandType,
}

Fields

m: <AxisScheduler as CubeType>::ExpandType

n: <AxisScheduler as CubeType>::ExpandType

k: <AxisScheduler as CubeType>::ExpandType

Implementations

impl PartitionSchedulerExpand

pub fn __expand_map_m_method( &self, scope: &Scope, i: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Maps a local M index to a global index.

pub fn __expand_map_n_method( &self, scope: &Scope, i: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Maps a local N index to a global index.

pub fn __expand_map_k_method( &self, scope: &Scope, i: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Maps a local K index to a global index.

pub fn __expand_new( scope: &Scope, partition_index_m: <u32 as CubeType>::ExpandType, partition_index_n: <u32 as CubeType>::ExpandType, partition_size: PartitionSize, partition_schedule_scheme: PartitionSchedulerScheme, ) -> <PartitionScheduler as CubeType>::ExpandType

Creates a PartitionScheduler for a partition at (partition_index_m, partition_index_n).

• Offset: rotates indices per axis to spread memory access and avoid conflicts.
• Naive: uses simple row-major order for partitions.

pub fn __expand_map_m( scope: &Scope, this: &Self, i: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Maps a local M index to a global index.

pub fn __expand_map_n( scope: &Scope, this: &Self, i: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Maps a local N index to a global index.

pub fn __expand_map_k( scope: &Scope, this: &Self, i: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Maps a local K index to a global index.

Trait Implementations

impl AsMutExpand for PartitionSchedulerExpand

fn __expand_ref_mut_method(&mut self, _: &Scope) -> &mut Self

fn __expand_as_mut_method(&mut self, scope: &Scope) -> &mut T

impl AsRefExpand for PartitionSchedulerExpand

fn __expand_ref_method(&self, _: &Scope) -> &Self

fn __expand_as_ref_method(&self, scope: &Scope) -> &T

impl CubeDebug for PartitionSchedulerExpand

fn set_debug_name(&self, scope: &Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl ExpandTypeClone for PartitionSchedulerExpand

fn clone_unchecked(&self) -> Self

Unchecked clone that only clones the conceptual runtime value. Should only be used in cases where each copy is used in a mutually exclusive branch (i.e. match, runtime enums). This is intentionally separated from Rust’s Clone semantics and should only be used for the conceptual expand values, never real data. Using two values in the same branch is undefined behaviour.

impl IntoExpand for PartitionSchedulerExpand

type Expand = PartitionSchedulerExpand

fn into_expand(self, _: &Scope) -> Self

impl IntoMut for PartitionSchedulerExpand

fn into_mut(self, scope: &Scope) -> Self

Convert the variable into a potentially new mutable variable in scope, copying if needed.