Mojo struct
OutputTileStorage
struct OutputTileStorage[c_type: DType, c_dim0: Int, c_dim1: Int, num_output_stages: Int]
Storage for output tiles (C matrix).
Single source of truth for output tile array and storage. Separate from input tiles since output has different stage count.
All tiles use TileTensor natively. C tiles also store as TileTensor but provide a LayoutTensor accessor for epilogue_components.mojo compatibility with .reshape[] and .tile[] methods.
Parameters
- c_type (
DType): Data type for C matrix tiles. - c_dim0 (
Int): First dimension for C tiles (OutputM). - c_dim1 (
Int): Second dimension for C tiles (OutputN). - num_output_stages (
Int): Number of output pipeline stages.
Fields
- c_tiles_storage (
OutputTileStorage[c_type, c_dim0, c_dim1, num_output_stages].CTileArray.Storage):
Implemented traits
AnyType,
ImplicitlyDestructible
comptime members
__del__is_trivial
comptime __del__is_trivial = True
c_tile_layout
comptime c_tile_layout = Layout.row_major(c_dim0, c_dim1)
CTileArray
comptime CTileArray = SMemTileArray2DRowMajor[c_type, c_dim0, c_dim1, num_output_stages]
CTileArrayLT
comptime CTileArrayLT = SMemTileArray[c_type, OutputTileStorage[c_type, c_dim0, c_dim1, num_output_stages].c_tile_layout, num_output_stages, 128]
Methods
c_tiles
c_tiles(ref[AddressSpace._value._mlir_value] self) -> OutputTileStorage[c_type, c_dim0, c_dim1, num_output_stages].CTileArrayLT
Get C tile array accessor (LayoutTensor-based for backward compat).
Returns LayoutTensor view for compatibility with tile_writer.mojo which uses .reshape[] and .tile[] methods.
Returns:
OutputTileStorage
c_tiles_tt
c_tiles_tt(ref[AddressSpace._value._mlir_value] self) -> OutputTileStorage[c_type, c_dim0, c_dim1, num_output_stages].CTileArray
Get C tile array accessor (TileTensor-based).
Returns native TileTensor for future TileTensor-native code paths.
Returns:
OutputTileStorage
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!