Struct QueryReaderExpand 

pub struct QueryReaderExpand<AP: AttentionPrecision> { /* private fields */ }

Implementations

impl<AP: AttentionPrecision> QueryReaderExpand<AP>

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

impl<AP: AttentionPrecision> QueryReaderExpand<AP>

pub fn __expand_get_tile_method<P: AttentionPartitioner>( self, scope: &mut Scope, tile: <Coords2d as RudaType>::ExpandType, tile_size: AttentionTileSize, partition_seq_q: u32, partition_head_dim: u32, ) -> <StridedTile<QG<AP>, QGS<AP>> as RudaType>::ExpandType

pub fn __expand_head_dim_method( self, scope: &mut Scope, ) -> <u32 as RudaType>::ExpandType

pub fn __expand_get_staged_tile_method<P: AttentionPartitioner>( self, scope: &mut Scope, tile: <Coords2d as RudaType>::ExpandType, tile_size: AttentionTileSize, partition_seq_q: u32, plane_dim: u32, num_planes: u32, ) -> <StridedTile<QG<AP>, QGS<AP>> as RudaType>::ExpandType

pub fn __expand_new( scope: &mut Scope, stage_q_offset: <u32 as RudaType>::ExpandType, query: <View<Vector<QG<AP>, QGS<AP>>, Coords2d> as RudaType>::ExpandType, gmem_config: GlobalMemoryConfig, ) -> Self

pub fn __expand_get_tile<P: AttentionPartitioner>( scope: &mut Scope, this: Self, tile: <Coords2d as RudaType>::ExpandType, tile_size: AttentionTileSize, partition_seq_q: u32, partition_head_dim: u32, ) -> <StridedTile<QG<AP>, QGS<AP>> as RudaType>::ExpandType

pub fn __expand_head_dim( scope: &mut Scope, this: Self, ) -> <u32 as RudaType>::ExpandType

pub fn __expand_get_staged_tile<P: AttentionPartitioner>( scope: &mut Scope, this: Self, tile: <Coords2d as RudaType>::ExpandType, tile_size: AttentionTileSize, partition_seq_q: u32, plane_dim: u32, num_planes: u32, ) -> <StridedTile<QG<AP>, QGS<AP>> as RudaType>::ExpandType

Trait Implementations

impl<AP: AttentionPrecision> Clone for QueryReaderExpand<AP>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<AP: AttentionPrecision> IntoMut for QueryReaderExpand<AP>

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

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

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

fn expand_inactive_value(self, _scope: &mut Scope) -> Self

fn expand_reference(self, _scope: &mut Scope, _writable: bool) -> Self

fn expand_absent_reference(self, _scope: &mut Scope, _writable: bool) -> Self

fn expand_reborrow(self, _scope: &mut Scope, _writable: bool) -> Self

fn expand_return_reference(self, scope: &mut Scope, writable: bool) -> Self

fn expand_deref(self, _scope: &mut Scope) -> Self

impl<AP: AttentionPrecision> RudaDebug for QueryReaderExpand<AP>

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