Enum TileExpand 

pub enum TileExpand<N: Numeric, Sc: TileScope, IO: SliceVisibility> {
    SharedMemory(<SharedTile<N, IO> as CubeType>::ExpandType),
    Cmma(<CmmaTile<N> as CubeType>::ExpandType),
    Mma(<MmaTile<N> as CubeType>::ExpandType),
    Register(<RegisterTile<N> as CubeType>::ExpandType),
    PlaneVec(<PlaneVecTile<N> as CubeType>::ExpandType),
    Interleaved(<InterleavedTile<N> as CubeType>::ExpandType),
    Unit(<UnitTile<N> as CubeType>::ExpandType),
    WhiteboxFragment(<WhiteboxFragment<N> as CubeType>::ExpandType),
    Bounce(<BounceTile<N> as CubeType>::ExpandType),
    Broadcasted(<Value<N> as CubeType>::ExpandType),
    None,
    _Phantom(<ScopeMarker<Sc> as CubeType>::ExpandType),
}

Variants

SharedMemory(<SharedTile<N, IO> as CubeType>::ExpandType)

Cmma(<CmmaTile<N> as CubeType>::ExpandType)

Mma(<MmaTile<N> as CubeType>::ExpandType)

Register(<RegisterTile<N> as CubeType>::ExpandType)

PlaneVec(<PlaneVecTile<N> as CubeType>::ExpandType)

Interleaved(<InterleavedTile<N> as CubeType>::ExpandType)

Unit(<UnitTile<N> as CubeType>::ExpandType)

WhiteboxFragment(<WhiteboxFragment<N> as CubeType>::ExpandType)

Bounce(<BounceTile<N> as CubeType>::ExpandType)

Broadcasted(<Value<N> as CubeType>::ExpandType)

None

_Phantom(<ScopeMarker<Sc> as CubeType>::ExpandType)

Implementations

impl<N: Numeric, Sc: TileScope> TileExpand<N, Sc, ReadWrite>

pub fn __expand_mma_method<L: Numeric, R: Numeric>( self, scope: &mut Scope, lhs: <Tile<L, Sc, ReadWrite> as CubeType>::ExpandType, rhs: <Tile<R, Sc, ReadWrite> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_mma<L: Numeric, R: Numeric>( scope: &mut Scope, this: Self, lhs: <Tile<L, Sc, ReadWrite> as CubeType>::ExpandType, rhs: <Tile<R, Sc, ReadWrite> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

impl<E: Float> TileExpand<E, Plane, ReadWrite>

pub fn __expand_row_max_method( self, scope: &mut Scope, acc: <RowWise<E> as CubeType>::ExpandType, base: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_row_sum_method( self, scope: &mut Scope, acc: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_exp_diff_method( self, scope: &mut Scope, rowwise: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_rowwise_scale_method( self, scope: &mut Scope, scale: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_row_max( scope: &mut Scope, this: Self, acc: <RowWise<E> as CubeType>::ExpandType, base: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_row_sum( scope: &mut Scope, this: Self, acc: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_exp_diff( scope: &mut Scope, this: Self, rowwise: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_rowwise_scale( scope: &mut Scope, this: Self, scale: <RowWise<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

impl<N: Numeric, Sc: TileScope> TileExpand<N, Sc, ReadWrite>

pub fn __expand_copy_from_method<SE: Numeric, SS: Size, L: Numeric, R: Numeric, A: Numeric, SIO: SliceVisibility>( self, scope: &mut Scope, source: <Tile<SE, Sc, SIO> as CubeType>::ExpandType, ident: StageIdent, ) -> <() as CubeType>::ExpandType

pub fn __expand_copy_from<SE: Numeric, SS: Size, L: Numeric, R: Numeric, A: Numeric, SIO: SliceVisibility>( scope: &mut Scope, this: Self, source: <Tile<SE, Sc, SIO> as CubeType>::ExpandType, ident: StageIdent, ) -> <() as CubeType>::ExpandType

impl<E: Float> TileExpand<E, Plane, ReadWrite>

pub fn __expand_scale_and_mask_method<M: Mask>( self, scope: &mut Scope, scale: <E as CubeType>::ExpandType, mask: <M as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_fill_zero_method( self, scope: &mut Scope, ) -> <() as CubeType>::ExpandType

pub fn __expand_scale_and_mask<M: Mask>( scope: &mut Scope, this: Self, scale: <E as CubeType>::ExpandType, mask: <M as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

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

impl<N: Numeric, Sc: TileScope> TileExpand<N, Sc, ReadWrite>

pub fn __expand_load_mask_from_strided_tile_method<E: Numeric, ES: Size>( self, scope: &mut Scope, tile: <StridedTile<E, ES> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_load_mask_from_strided_tile<E: Numeric, ES: Size>( scope: &mut Scope, this: Self, tile: <StridedTile<E, ES> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

impl<Acc: Float> TileExpand<Acc, Plane, ReadWrite>

pub fn __expand_softmax_method<Lhs: Float, M: Mask>( self, scope: &mut Scope, mask: <M as CubeType>::ExpandType, softmaxed_tile: <Tile<Lhs, Plane, ReadWrite> as CubeType>::ExpandType, state: <(RowWise<Acc>, RowWise<Acc>) as CubeType>::ExpandType, head_dim_factor: <Acc as CubeType>::ExpandType, ) -> <RowWise<Acc> as CubeType>::ExpandType

pub fn __expand_scale_mul_method<SM: Float>( self, scope: &mut Scope, scale: <RowWise<SM> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_scale_div_method<SM: Float>( self, scope: &mut Scope, running_state_l: <RowWise<SM> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_write_results_method<DE: Float, DS: Size>( self, scope: &mut Scope, dest: <Tile<DE, Plane, ReadWrite> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_softmax<Lhs: Float, M: Mask>( scope: &mut Scope, this: Self, mask: <M as CubeType>::ExpandType, softmaxed_tile: <Tile<Lhs, Plane, ReadWrite> as CubeType>::ExpandType, state: <(RowWise<Acc>, RowWise<Acc>) as CubeType>::ExpandType, head_dim_factor: <Acc as CubeType>::ExpandType, ) -> <RowWise<Acc> as CubeType>::ExpandType

pub fn __expand_scale_mul<SM: Float>( scope: &mut Scope, this: Self, scale: <RowWise<SM> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_scale_div<SM: Float>( scope: &mut Scope, this: Self, running_state_l: <RowWise<SM> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_write_results<DE: Float, DS: Size>( scope: &mut Scope, this: Self, dest: <Tile<DE, Plane, ReadWrite> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

Trait Implementations

impl<N: Numeric, Sc: TileScope, IO: SliceVisibility> Clone for TileExpand<N, Sc, IO>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<N: Numeric, Sc: TileScope, IO: SliceVisibility> CubeDebug for TileExpand<N, Sc, IO>

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<N: Numeric, Sc: TileScope, IO: SliceVisibility> CubeEnum for TileExpand<N, Sc, IO>

type RuntimeValue = ()

fn discriminant(&self) -> NativeExpand<i32>

fn runtime_value(self) -> Self::RuntimeValue

Return the runtime value of this enum, if only one variant has a value. Should return () for all other cases.

fn discriminant_of(variant_name: &'static str) -> i32

fn discriminant_of_value(&self, variant_name: &'static str) -> i32

impl<N: Numeric, Sc: TileScope, IO: SliceVisibility> IntoMut for TileExpand<N, Sc, IO>

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

Convert the variable into a potentially new mutable variable in scope, copying if needed.

impl<E: Numeric, Sc: TileScope, IO: SliceVisibility> MaskExpand for TileExpand<E, Sc, IO>

fn __expand_should_mask_method( self, scope: &mut Scope, local_pos: <Coords2d as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType