Mojo struct
SourceTileStorage
struct SourceTileStorage[src_type: DType, src_dim0: Int, src_dim1: Int, num_epi_load_stages: Int]
Storage for source tensor C tiles (residual/skip connection input).
Used by the epilogue load warp to pre-fetch source tensor C via TMA, enabling overlap with MMA computation for residual operations like D = Conv(A,B) + betaC or D = MatMul(A,B) + betaC.
Parameters
- src_type (
DType): Data type for source tiles (same as output type). - src_dim0 (
Int): First dimension for source tiles (OutputM). - src_dim1 (
Int): Second dimension for source tiles (OutputN). - num_epi_load_stages (
Int): Number of epilogue load pipeline stages.
Fields
- src_tiles_storage (
SourceTileStorage[src_type, src_dim0, src_dim1, num_epi_load_stages].SrcTileArray.Storage):
Implemented traits
AnyType,
ImplicitlyDestructible
comptime members
__del__is_trivial
comptime __del__is_trivial = True
src_tile_layout
comptime src_tile_layout = Layout.row_major(src_dim0, src_dim1)
SrcTileArray
comptime SrcTileArray = SMemTileArray2DRowMajor[src_type, src_dim0, src_dim1, num_epi_load_stages]
Methods
src_tiles
src_tiles(ref[AddressSpace._value._mlir_value] self) -> SourceTileStorage[src_type, src_dim0, src_dim1, num_epi_load_stages].SrcTileArray
Get source tile array accessor (TileTensor-based).
Returns:
SourceTileStorage
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!