Mojo struct
PRegisterBuffer
struct PRegisterBuffer[accum_type: DType, dtype: DType, num_m_mmas: Int, num_n_mmas: Int, output_frag_size: Int]
Fields
- reg_tile (
LayoutTensor[accum_type, row_major((num_m_mmas * num_n_mmas), output_frag_size), MutableAnyOrigin, address_space=AddressSpace(5)]):
Implemented traits
AnyType,
UnknownDestructibility
Aliases
OutputTileType
alias OutputTileType = LayoutTensor[dtype, row_major(num_m_mmas, output_frag_size), MutableAnyOrigin, address_space=AddressSpace(5)]
RegisterTileType
alias RegisterTileType = LayoutTensor[accum_type, row_major((num_m_mmas * num_n_mmas), output_frag_size), MutableAnyOrigin, address_space=AddressSpace(5)]
Methods
__init__
__init__(out self)
interleave
interleave[tile_idx: Int](self) -> LayoutTensor[dtype, row_major(num_m_mmas, output_frag_size), MutableAnyOrigin, address_space=AddressSpace(5)]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!