For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo trait

OutputFusionTile

Trait for tile-based output fusion structs that provide custom store behavior for a TileTensor.

The tile analog of OutputFusion: store is terminal (owns the store and returns nothing), where ComputeOutputFusionTile.compute transforms a tile and returns it for the primary kernel to store.

Takes two copiers, both chosen by the calling kernel: copier loads any aux epilogue inputs (e.g. a broadcast bias) into local, and store_copier writes the transformed tile out to the output. Passing the store copier (rather than having the graph compiler synthesize one) keeps the copier choice with the kernel author.

Implemented traits

AnyType, Copyable, Deinitable, ImplicitlyCopyable, Movable, RegisterPassable, TrivialRegisterPassable

Required methods

store

def store[dtype: DType, rank: Int, LayoutType: TensorLayout, Copier: TileCopier, StoreCopier: TileCopier](self, tile_coords: IndexList[rank], copier: Copier, store_copier: StoreCopier, val: TileTensor[dtype, LayoutType, MutAnyOrigin])