Struct MmaDefinitionExpand 

pub struct MmaDefinitionExpand<A: CubeType, B: CubeType, CD: CubeType> {
    pub m: u32,
    pub n: u32,
    pub k: u32,
    pub a_type: StorageType,
    pub b_type: StorageType,
    pub cd_type: StorageType,
    pub scales_factor: Option<u32>,
    pub scales_type: Option<StorageType>,
    /* private fields */
}

Expand type of MmaDefinition.

Fields

m: u32

n: u32

k: u32

a_type: StorageType

b_type: StorageType

cd_type: StorageType

scales_factor: Option<u32>

scales_type: Option<StorageType>

Implementations

impl<A: CubePrimitive, B: CubePrimitive, CD: CubePrimitive> MmaDefinitionExpand<A, B, CD>

pub fn __expand_num_elems_method( self, scope: &mut Scope, ident: MatrixIdent, ) -> u32

pub fn __expand_elems_per_lane_method( self, scope: &mut Scope, ident: MatrixIdent, ) -> u32

pub fn __expand_lines_per_lane_method( self, scope: &mut Scope, ident: MatrixIdent, ) -> u32

pub fn __expand_line_layout_method( self, scope: &mut Scope, ident: MatrixIdent, ) -> MatrixLayout

pub fn __expand_line_size_method( self, scope: &mut Scope, ident: MatrixIdent, ) -> u32

pub fn __expand_position_of_nth_method( self, scope: &mut Scope, lane_id: <u32 as CubeType>::ExpandType, elem_idx: <u32 as CubeType>::ExpandType, ident: MatrixIdent, ) -> <(u32, u32) as CubeType>::ExpandType

pub fn __expand_scales_index_method( self, scope: &mut Scope, lane_id: <u32 as CubeType>::ExpandType, ident: MatrixIdent, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_scales_count_method(self, scope: &mut Scope) -> u32

pub fn __expand_scales_line_size_method(self, scope: &mut Scope) -> u32

pub fn __expand_execute_method( self, scope: &mut Scope, registers_a: <Sequence<Line<A>> as CubeType>::ExpandType, registers_b: <Sequence<Line<B>> as CubeType>::ExpandType, registers_c: <Sequence<Line<CD>> as CubeType>::ExpandType, ) -> <Array<Line<CD>> as CubeType>::ExpandType

pub fn __expand_execute_scaled_method<S: CubePrimitive>( self, scope: &mut Scope, registers_a: <Sequence<Line<A>> as CubeType>::ExpandType, registers_b: <Sequence<Line<B>> as CubeType>::ExpandType, registers_c: <Sequence<Line<CD>> as CubeType>::ExpandType, scales_a: <Line<S> as CubeType>::ExpandType, scales_b: <Line<S> as CubeType>::ExpandType, ) -> <Array<Line<CD>> as CubeType>::ExpandType

pub fn __expand_new(scope: &mut Scope, m: u32, n: u32, k: u32) -> Self

pub fn __expand_new_scaled<S: CubePrimitive>( scope: &mut Scope, m: u32, n: u32, k: u32, scale_factor: u32, ) -> Self

pub fn __expand_num_elems( scope: &mut Scope, this: Self, ident: MatrixIdent, ) -> u32

pub fn __expand_elems_per_lane( scope: &mut Scope, this: Self, ident: MatrixIdent, ) -> u32

pub fn __expand_lines_per_lane( scope: &mut Scope, this: Self, ident: MatrixIdent, ) -> u32

pub fn __expand_line_layout( scope: &mut Scope, this: Self, ident: MatrixIdent, ) -> MatrixLayout

pub fn __expand_line_size( scope: &mut Scope, this: Self, ident: MatrixIdent, ) -> u32

pub fn __expand_position_of_nth( scope: &mut Scope, this: Self, lane_id: <u32 as CubeType>::ExpandType, elem_idx: <u32 as CubeType>::ExpandType, ident: MatrixIdent, ) -> <(u32, u32) as CubeType>::ExpandType

pub fn __expand_scales_index( scope: &mut Scope, this: Self, lane_id: <u32 as CubeType>::ExpandType, ident: MatrixIdent, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_scales_count(scope: &mut Scope, this: Self) -> u32

pub fn __expand_scales_line_size(scope: &mut Scope, this: Self) -> u32

pub fn __expand_execute( scope: &mut Scope, this: Self, registers_a: <Sequence<Line<A>> as CubeType>::ExpandType, registers_b: <Sequence<Line<B>> as CubeType>::ExpandType, registers_c: <Sequence<Line<CD>> as CubeType>::ExpandType, ) -> <Array<Line<CD>> as CubeType>::ExpandType

pub fn __expand_execute_scaled<S: CubePrimitive>( scope: &mut Scope, this: Self, registers_a: <Sequence<Line<A>> as CubeType>::ExpandType, registers_b: <Sequence<Line<B>> as CubeType>::ExpandType, registers_c: <Sequence<Line<CD>> as CubeType>::ExpandType, scales_a: <Line<S> as CubeType>::ExpandType, scales_b: <Line<S> as CubeType>::ExpandType, ) -> <Array<Line<CD>> as CubeType>::ExpandType

Trait Implementations

impl<A: CubeType, B: CubeType, CD: CubeType> Clone for MmaDefinitionExpand<A, B, CD>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<A: CubeType, B: CubeType, CD: CubeType> CubeDebug for MmaDefinitionExpand<A, B, CD>

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<A: Debug + CubeType, B: Debug + CubeType, CD: Debug + CubeType> Debug for MmaDefinitionExpand<A, B, CD>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<A: CubeType, B: CubeType, CD: CubeType> IntoMut for MmaDefinitionExpand<A, B, CD>

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