Struct StridedTilingLayout

pub struct StridedTilingLayout {}

Implementations

impl StridedTilingLayout

pub fn nth_slice<ES: Numeric, S: StageConfig>( stage: &mut Stage<ES, Self>, nth: u32, ident: Ident, config: S, ) -> SliceMut<Line<ES>>

Returns the nth slice of the stage

pub fn __expand_nth_slice<ES: Numeric, S: StageConfig>( context: &mut Scope, stage: <Stage<ES, Self> as CubeType>::ExpandType, nth: <u32 as CubeType>::ExpandType, ident: Ident, config: S, ) -> <SliceMut<Line<ES>> as CubeType>::ExpandType

Trait Implementations

impl<MP: MatmulPrecision> AsyncLoadingJob<MP, StridedTilingLayout> for Job

fn execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( this: &mut Self, task_id: u32, tensor_reader: &TensorReader<MP::EI>, stage: &mut Stage<MP::ES, StridedTilingLayout>, mechanism: &CM, config: G, )

fn task_count(this: &Self) -> u32

fn __expand_execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, tensor_reader: <TensorReader<MP::EI> as CubeType>::ExpandType, stage: <Stage<MP::ES, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl<MP: MatmulPrecision> AsyncLoadingJob<MP, StridedTilingLayout> for Job

fn execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( this: &mut Self, task_id: u32, tensor_reader: &TensorReader<MP::EI>, stage: &mut Stage<MP::ES, StridedTilingLayout>, mechanism: &CM, config: G, )

fn task_count(this: &Self) -> u32

fn __expand_execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, tensor_reader: <TensorReader<MP::EI> as CubeType>::ExpandType, stage: <Stage<MP::ES, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl<MP: MatmulPrecision> AsyncLoadingJob<MP, StridedTilingLayout> for Job

fn execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( this: &mut Self, task_id: u32, tensor_reader: &TensorReader<MP::EI>, stage: &mut Stage<MP::ES, StridedTilingLayout>, mechanism: &CM, config: G, )

fn task_count(this: &Self) -> u32

fn __expand_execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, tensor_reader: <TensorReader<MP::EI> as CubeType>::ExpandType, stage: <Stage<MP::ES, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl<MP: MatmulPrecision> AsyncLoadingJob<MP, StridedTilingLayout> for Job

fn execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( this: &mut Self, _task_id: u32, tensor_reader: &TensorReader<MP::EI>, stage: &mut Stage<MP::ES, StridedTilingLayout>, mechanism: &CM, config: G, )

fn task_count(_this: &Self) -> u32

fn __expand_execute_task<CM: CopyMechanism<MP::ES>, G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, _task_id: <u32 as CubeType>::ExpandType, tensor_reader: <TensorReader<MP::EI> as CubeType>::ExpandType, stage: <Stage<MP::ES, StridedTilingLayout> as CubeType>::ExpandType, mechanism: <CM as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( context: &mut Scope, _this: <Self as CubeType>::ExpandType, ) -> u32

impl Clone for StridedTilingLayout

fn clone(&self) -> StridedTilingLayout

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<MP: MatmulPrecision> LoadingJob<MP, StridedTilingLayout> for Job

fn execute_task<G: GlobalConfig>( this: &mut Self, task_id: u32, tensor_reader: &TensorReader<MP::EI>, stage: &mut Stage<MP::ES, StridedTilingLayout>, quantization: &CubeOption<Quantization<MP>>, config: G, )

fn task_count(this: &Self) -> u32

fn __expand_execute_task<G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, task_id: <u32 as CubeType>::ExpandType, tensor_reader: <TensorReader<MP::EI> as CubeType>::ExpandType, stage: <Stage<MP::ES, StridedTilingLayout> as CubeType>::ExpandType, quantization: <CubeOption<Quantization<MP>> as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

fn __expand_task_count( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> u32

impl TilingLayout for StridedTilingLayout

fn get_tile<ES: Numeric, S: StageConfig>( stage: &Stage<ES, Self>, x: u32, y: u32, ident: Ident, config: S, ) -> Tile<ES>

fn __expand_get_tile<ES: Numeric, S: StageConfig>( context: &mut Scope, stage: <Stage<ES, Self> as CubeType>::ExpandType, x: <u32 as CubeType>::ExpandType, y: <u32 as CubeType>::ExpandType, ident: Ident, config: S, ) -> <Tile<ES> as CubeType>::ExpandType

impl Copy for StridedTilingLayout