Struct Procedural 

pub struct Procedural { /* private fields */ }

Runtime state of a procedural source. origin tracks regions selected by Tile::at. Its recipe computes at its own element; a read names the element it wants.

Implementations

impl Procedural

pub fn tile<T: Numeric>(self) -> Tile

This source as a tile, placed alone.

pub fn tile_in<T: Numeric>(self, partitioning: &Partitioning) -> Tile

This source as a tile at the root of partitioning, windowed by its regions like an operand bound to it: its space must be the partitioning’s, or some of its axes.

pub fn __expand_tile<T: Numeric>( scope: &Scope, this: <Self as CubeType>::ExpandType, ) -> <Tile as CubeType>::ExpandType

This source as a tile, placed alone.

pub fn __expand_tile_in<T: Numeric>( scope: &Scope, this: <Self as CubeType>::ExpandType, partitioning: &<Partitioning as CubeType>::ExpandType, ) -> <Tile as CubeType>::ExpandType

This source as a tile at the root of partitioning, windowed by its regions like an operand bound to it: its space must be the partitioning’s, or some of its axes.

impl Procedural

pub fn new<T: Numeric, R: Recipe<T> + 'static>( _space: Space, _recipe: R, ) -> Self

A memory-free source over space, evaluated from recipe, which computes at T, where it is read.

pub fn __expand_new<T: Numeric, R: Recipe<T> + 'static>( scope: &Scope, space: Space, recipe: R::ExpandType, ) -> ProceduralExpand

pub fn separable<T: Numeric, R: Separable<T> + 'static>( _space: Space, _recipe: R, ) -> Self

where R::ExpandType: FactorReads,

new keeping the recipe’s factorization, one factor per contracted axis.

pub fn __expand_separable<T: Numeric, R: Separable<T> + 'static>( scope: &Scope, space: Space, recipe: R::ExpandType, ) -> ProceduralExpand

where R::ExpandType: SeparableExpand<T> + FactorReads,

pub fn normalized(self, _taps: TapSupport, _guard: DivGuard) -> Self

Normalize each factor’s taps where the gather contraction evaluates them. The recipe must state a factorization.

Trait Implementations

impl Clone for Procedural

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for Procedural

type ExpandType = ProceduralExpand

impl Vectorized for Procedural

fn vector_size(&self) -> usize

fn __expand_vector_size(scope: &Scope, this: &Self::ExpandType) -> usize

impl<T: Numeric, W: Size> ViewOperations<Vector<T, W>, Sequence<u32>> for Procedural

fn read(&self, pos: C) -> T

fn read_checked(&self, pos: C) -> T

fn read_masked(&self, pos: C, value: T) -> T

fn read_unchecked(&self, pos: C) -> T

fn as_linear_slice(&self, pos: C, size: C) -> &[T]

Create a slice starting from pos, with size. The layout handles translation into concrete indices.

fn tensor_map_load(&self, barrier: &Barrier, shared_memory: &mut [T], pos: C)

Execute a TMA load into shared memory, if the underlying storage supports it. Panics if it’s unsupported.

fn shape(&self) -> C

fn is_in_bounds(&self, pos: C) -> bool

fn __expand_read( scope: &Scope, this: &Self::ExpandType, pos: <C as CubeType>::ExpandType, ) -> <T as CubeType>::ExpandType

fn __expand_read_checked( scope: &Scope, this: &Self::ExpandType, pos: <C as CubeType>::ExpandType, ) -> <T as CubeType>::ExpandType

fn __expand_read_masked( scope: &Scope, this: &Self::ExpandType, pos: <C as CubeType>::ExpandType, value: <T as CubeType>::ExpandType, ) -> <T as CubeType>::ExpandType

fn __expand_read_unchecked( scope: &Scope, this: &Self::ExpandType, pos: <C as CubeType>::ExpandType, ) -> <T as CubeType>::ExpandType

fn __expand_as_linear_slice<'infer, 'scope>( scope: &'scope Scope, this: &'infer Self::ExpandType, pos: <C as CubeType>::ExpandType, size: <C as CubeType>::ExpandType, ) -> &'infer <[T] as CubeType>::ExpandType

fn __expand_tensor_map_load( scope: &Scope, this: &Self::ExpandType, barrier: &<Barrier as CubeType>::ExpandType, shared_memory: &mut <[T] as CubeType>::ExpandType, pos: <C as CubeType>::ExpandType, )

fn __expand_shape( scope: &Scope, this: &Self::ExpandType, ) -> <C as CubeType>::ExpandType

fn __expand_is_in_bounds( scope: &Scope, this: &Self::ExpandType, pos: <C as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType