Mojo module
tile_tensor
TileTensor type for structured memory access with compile-time layout information.
comptime values
LTToTTLayout
comptime LTToTTLayout[lt_layout: Layout] = Layout[*?, *?]
Derive a TileTensor Layout from a legacy Layout.
Known dimensions become ComptimeInt, UNKNOWN_VALUE dimensions become
RuntimeInt. Hierarchical layouts (e.g. from tile_to_shape) are
collapsed via product_each so each mode becomes a single value.
Parameters
- lt_layout (
Layout): The legacy Layout to convert.
Structs
-
NullableTileTensor: A TileTensor variant whose pointer may be absent (null). -
TileTensor: A tensor type with trait-based layouts supporting nested and hierarchical indexing.
Functions
-
flatten_leading: Merge the first two dimensions of a rank-3 TileTensor: (A, B, C) -> (A*B, C). -
lt_to_tt: Convert a LayoutTensor to a TileTensor. -
stack_allocation: Allocate a TileTensor on the stack with the given layout.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!