Mojo struct
AccumTile
@register_passable(trivial)
struct AccumTile[dtype: DType, size: Int]
Accumulator tile holding upper and lower fragment data.
SM100 accumulators in TMEM are stored as two halves (upper 16 rows, lower 16 rows). This struct represents the complete tile being written.
This is the SM100 equivalent of SM90's RegTileType - the data being written by the tile writer.
Template Parameters: dtype: Data type of the fragments (typically epilogue_dtype). size: Number of elements per fragment.
Fields
- upper (
SIMD[dtype, size]): - lower (
SIMD[dtype, size]):
Implemented traits
AnyType,
Copyable,
ImplicitlyCopyable,
Movable,
UnknownDestructibility
comptime members
__copyinit__is_trivial
comptime __copyinit__is_trivial = True
__del__is_trivial
comptime __del__is_trivial = True
__moveinit__is_trivial
comptime __moveinit__is_trivial = True
Methods
__init__
__init__(upper: SIMD[dtype, size], lower: SIMD[dtype, size]) -> Self
Create an accumulator tile from upper and lower fragments.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!