Struct BlockScaledLayout 

pub struct BlockScaledLayout { /* private fields */ }

Workaround for enums not supporting comptime, should fix that in the future

Implementations

impl BlockScaledLayout

pub fn new( shape: Coords2d, scales_layout: BatchedGlobalLayout, block_size: Coords2d, ) -> Self

pub fn __expand_new( scope: &mut Scope, shape: <Coords2d as CubeType>::ExpandType, scales_layout: <BatchedGlobalLayout as CubeType>::ExpandType, block_size: Coords2d, ) -> <Self as CubeType>::ExpandType

Trait Implementations

impl CubeType for BlockScaledLayout

type ExpandType = BlockScaledLayoutExpand

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

Wrapper around the init method, necessary to type inference.

impl LaunchArg for BlockScaledLayout

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

The runtime argument for the kernel.

type CompilationArg = BlockScaledLayoutCompilationArg

Compilation argument.

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

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.