IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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

ElementwiseFusionTile

The TileTensor variant of ElementwiseFusion: a tile-based pure elementwise fusion struct emitted by the graph compiler.

compute fills the driver-provided dst tile (which carries the tile layout) and returns it.

Implemented traitsโ€‹

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

Required methodsโ€‹

computeโ€‹

def compute[dtype: DType, rank: Int, LayoutType: TensorLayout, Copier: TileCopier](self, tile_coords: IndexList[rank], copier: Copier, dst: TileTensor[dtype, LayoutType, MutAnyOrigin]) -> TileTensor[dtype, LayoutType, MutAnyOrigin]

Returns:

TileTensor[dtype, LayoutType, MutAnyOrigin]

Was this page helpful?