Struct CubeMapping 

pub struct CubeMapping {
    pub can_yield_extra_cubes: bool,
    /* private fields */
}

Runtime-side counterpart of CubeCountPlan: given the cube position, resolves the conceptual (x, y, z) coordinates in problem space.

Each operation is responsible for mapping the returned generic (x, y, z) tuple to its own domain axes (e.g. matmul interprets them as (m, n, batch), gemv as (matrix_axis, _, batch), attention as (seq_q, batch_heads, _)).

Fields

can_yield_extra_cubes: bool

Implementations

impl CubeMapping

pub fn num_valid_cubes(&self) -> usize

Returns the number of valid cubes (problem-space volume).

pub fn cube_pos_to_xyz(&self) -> (u32, u32, u32)

Given a cube position, returns the generic problem-space coordinates (x, y, z).

Consumers assign meaning to x/y/z (matmul: m/n/batch, gemv: matrix/_/batch, etc.).

pub fn __expand_num_valid_cubes( scope: &Scope, this: &<Self as CubeType>::ExpandType, ) -> <usize as CubeType>::ExpandType

Returns the number of valid cubes (problem-space volume).

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

Given a cube position, returns the generic problem-space coordinates (x, y, z).

Consumers assign meaning to x/y/z (matmul: m/n/batch, gemv: matrix/_/batch, etc.).

Trait Implementations

impl CubeType for CubeMapping

type ExpandType = CubeMappingExpand

impl LaunchArg for CubeMapping

type RuntimeArg<R: Runtime> = CubeMappingLaunch<R>

The runtime argument for the kernel.

type CompilationArg = CubeMappingCompilationArg

Compilation argument.

fn register<R: Runtime>( arg: Self::RuntimeArg<R>, launcher: &mut KernelLauncher<R>, ) -> Self::CompilationArg

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

Register a variable during compilation that fill the KernelBuilder.