Enum StridedLayout

pub enum StridedLayout {
    Pitched(FastDivmod),
    None,
}

Layout for tensor that may or may not be strided on the last dimension. Efficiently translates the absolute index to strided index.

Variants

Pitched(FastDivmod)

None

Implementations

impl StridedLayout

pub fn new_Pitched(_0: FastDivmod) -> Self

pub fn __expand_new_Pitched( _: &mut Scope, _0: <FastDivmod as CubeType>::ExpandType, ) -> StridedLayoutExpand

pub fn new_None() -> Self

pub fn __expand_new_None(_: &mut Scope) -> StridedLayoutExpand

impl StridedLayout

pub fn index<T: CubePrimitive>( &self, tensor: &Tensor<Line<T>>, index: u32, ) -> u32

Translates absolute index to strided index if applicable

pub fn __expand_index<T: CubePrimitive>( context: &mut Scope, this: <Self as CubeType>::ExpandType, tensor: <Tensor<Line<T>> as CubeType>::ExpandType, index: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Trait Implementations

impl CubeDebug for StridedLayout

fn set_debug_name(&self, scope: &mut 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 CubeType for StridedLayout

type ExpandType = StridedLayoutExpand

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

Wrapper around the init method, necessary to type inference.

impl LaunchArg for StridedLayout

type RuntimeArg<'a, R: Runtime> = StridedLayoutArgs<'a, R>

The runtime argument for the kernel.

fn compilation_arg<'a, R: Runtime>( runtime_arg: &Self::RuntimeArg<'a, R>, ) -> Self::CompilationArg

impl LaunchArgExpand for StridedLayout

type CompilationArg = StridedLayoutCompilationArg

Compilation argument.

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an output variable during compilation that fill the KernelBuilder.