cubecl_convolution::loader::weight_tma

Type Alias TmaWeightReader

Source 
pub type TmaWeightReader<MP> = FullStageToTileReader<<MP as MatmulPrecision>::ES, TmaWeightTiling>;

Aliased Type§

pub struct TmaWeightReader<MP> {
    pub stage_memory: StageMemory<<MP as MatmulPrecision>::ES, ContiguousTilingLayout<RowMajorTilingOrder>>,
    pub input_ident: InputIdent,
}

Fields§

§stage_memory: StageMemory<<MP as MatmulPrecision>::ES, ContiguousTilingLayout<RowMajorTilingOrder>>

Stage memory from which to read

§input_ident: InputIdent

Ident of the stage memory