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 struct

Layout

struct Layout[shape_types: TypeList[shape_types.values], stride_types: TypeList[stride_types.values]]

A layout that supports mixed compile-time and runtime dimensions.

This layout provides a unified interface for layouts where some dimensions are known at compile time and others are determined at runtime. It enables more ergonomic layout definitions while maintaining performance.

A Layout's shape and strides must be non-negative.

Parameters​

Implemented traits​

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDeletable, Movable, RegisterPassable, TensorLayout, TrivialRegisterPassable, Writable

comptime members​

all_dims_known​

comptime all_dims_known = Layout[shape_types, stride_types].shape_known and Layout[shape_types, stride_types].stride_known

Whether all shape and stride dimensions are known at compile time.

flat_rank​

comptime flat_rank = TypeList[#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))].length

The number of dimensions after flattening nested coordinates.

rank​

comptime rank = TypeList[values].length

The number of dimensions in the layout.

shape_known​

comptime shape_known = Coord[shape_types].all_dims_known

Whether all shape dimensions are known at compile time.

static_cosize​

comptime static_cosize = #kgen.param_list.reduce(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] Int((add (mul #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[idx].static_value, #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[idx].static_value), (mul #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[idx].static_value, -1)))), base=Int(1), reducer=[PrevV: Int, VA: KGENParamList[Int], idx: __mlir_type.index] Int((add PrevV, #kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] Int((add (mul #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[idx].static_value, #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[idx].static_value), (mul #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[idx].static_value, -1))))[idx])))

The compile-time size of the memory region spanned by the layout.

static_product​

comptime static_product = Coord[*#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))].static_product

The compile-time product of all shape dimensions (handles nested Coords).

static_shape​

comptime static_shape[i: Int] = #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(i)].static_value

Returns the compile-time value of the i-th flattened shape dimension.

Parameters​

  • ​i (Int): The dimension index.

static_stride​

comptime static_stride[i: Int] = #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(#kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))), [idx: __mlir_type.index] #kgen.param_list.concat(#kgen.param_list.tabulate(len(values), [idx: __mlir_type.index] values[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(Int(idx))]._ParamListType))[SIMDLength(i)].static_value

Returns the compile-time value of the i-th flattened stride dimension.

Parameters​

  • ​i (Int): The dimension index.

stride_known​

comptime stride_known = Coord[stride_types].all_dims_known

Whether all stride dimensions are known at compile time.

Methods​

__init__​

def __init__() -> Self

Default-initialize a layout from its compile-time type parameters.

Each dimension is initialized to its default value: compile-time dimensions (ComptimeInt) get their static value, runtime dimensions (Scalar) get 0. This is useful for constructing a fully-static layout purely from its type, e.g. UpcastLayout[MyLayout, 2]().

def __init__(shape: Coord[shape_types], stride: Coord[stride_types]) -> Self

Initialize a layout with shape and stride.

Args:

__call__​

def __call__[index_type: CoordLike, *, linear_idx_type: DType = DType.int64](self, index: index_type) -> Scalar[linear_idx_type]

Maps a logical coordinate to a linear memory index.

Supports hierarchical indexing where the coordinate structure can differ from the shape structure. For a layout with shape (4, (3, 2)):

  • (1, (1, 1)): exact structure match, each element maps directly.
  • (1, 1): rank-matching, the scalar 1 is decomposed within the nested (3, 2) sub-dimension.
  • (1): scalar index decomposed across all dimensions.

Parameters:

  • ​index_type (CoordLike): The coordinate type.
  • ​linear_idx_type (DType): The data type for the returned linear index.

Args:

  • ​index (index_type): The logical coordinates to map.

Returns:

Scalar[linear_idx_type]: The linear memory index corresponding to the given coordinates.

idx2crd​

def idx2crd[*, out_dtype: DType = DType.int64](self, idx: Int) -> Coord[*#kgen.param_list.tabulate(len(shape_types.values), [idx: __mlir_type.index] Coord[*#kgen.param_list.tabulate(len(shape_types.values[SIMDLength(Int(idx))]._ParamListType), [idx: __mlir_type.index] Scalar[out_dtype])] if shape_types.values[SIMDLength(Int(idx))].is_tuple else Scalar[out_dtype])]

Maps a linear memory index back to logical coordinates.

This is the inverse of __call__ (crd2idx). Given a linear index, it computes the corresponding multi-dimensional coordinates using the per-element formula: coord[i] = (idx // stride[i]) % shape[i]. For hierarchical layouts (e.g. from zipped_divide), the result preserves the nested coordinate structure.

Examples: For a layout with shape (3, 4) and row-major strides:

  • layout.idx2crd(0) returns (0, 0).
  • layout.idx2crd(5) returns (1, 1).
  • layout.idx2crd(11) returns (2, 3).

Parameters:

  • ​out_dtype (DType): The data type for the output coordinate values.

Args:

  • ​idx (Int): The linear memory index to convert to coordinates.

Returns:

Coord[*#kgen.param_list.tabulate(len(shape_types.values), [idx: __mlir_type.index] Coord[*#kgen.param_list.tabulate(len(shape_types.values[SIMDLength(Int(idx))]._ParamListType), [idx: __mlir_type.index] Scalar[out_dtype])] if shape_types.values[SIMDLength(Int(idx))].is_tuple else Scalar[out_dtype])]: A Coord containing the logical coordinates corresponding to the linear index. For nested layouts, the result mirrors the shape nesting with Scalar leaves.

product​

def product(self) -> Int

Returns the total number of elements in the layout's domain.

For a layout with shape (m, n), this returns m * n, representing the total number of valid coordinates in the layout.

Returns:

Int: The total number of elements in the layout.

size​

def size(self) -> Int

Returns the total number of elements in the layout's domain.

Alias for product(). Compatible with the legacy Layout API.

Returns:

Int: The total number of elements in the layout.

cosize​

def cosize[linear_idx_type: DType = DType.int64](self) -> Scalar[linear_idx_type]

Returns the size of the memory region spanned by the layout.

For a layout with shape (m, n) and stride (r, s), this returns (m-1)*r + (n-1)*s + 1, representing the memory footprint.

Parameters:

  • ​linear_idx_type (DType): The data type for the returned size value.

Returns:

Scalar[linear_idx_type]: The size of the memory region required by the layout.

to_layout​

def to_layout(self) -> Layout

Converts this mixed layout to a legacy Layout using IntTuple.

Returns:

Layout: A legacy Layout with the same shape and stride.

to_legacy_layout​

static def to_legacy_layout() -> Layout

Converts this layout type to a legacy Layout via type-level extraction.

All dimensions must be known at compile time. Uses direct IntTuple construction (no append) for fast compile times.

Returns:

Layout: A legacy Layout with the same shape and stride.

reverse​

def reverse(self) -> Layout[*?, *?]

Reverse the order of dimensions in the layout.

Turns row-major into column-major ordering where the stride-1 dimension comes first, enabling coalesced scalar iteration.

Returns:

Layout[*?, *?]: A new Layout with shape and stride Coords reversed.

transpose​

def transpose(self) -> Layout[*?, *?]

Transposes the layout by reversing the order of dimensions.

For an n-dimensional layout, this reverses the order of both shapes and strides. For 2D layouts, this swaps rows and columns, converting row-major to column-major and vice versa.

Example:

from layout.tile_layout import row_major

var layout = row_major[3, 4]()  # shape (3,4), stride (4,1)
var transposed = layout.transpose()  # shape (4,3), stride (1,4)

Returns:

Layout[*?, *?]: A new Layout with transposed dimensions.

make_dynamic​

def make_dynamic[dtype: DType](self) -> Layout[*?, *?]

Convert all elements in shape and stride to Scalar[dtype].

Examples:

from layout.tile_layout import row_major
var layout = row_major[3, 4]()  # All compile-time
var dynamic = layout.make_dynamic[DType.int64]()
# dynamic has Int64 for all dimensions

Parameters:

  • ​dtype (DType): The data type for the resulting Scalar values.

Returns:

Layout[*?, *?]: A new Layout where all elements in shape and stride are converted to Scalar[dtype].

shape​

def shape[i: Int](self) -> shape_types.values[SIMDLength(i)]

Returns the i-th shape dimension.

Parameters:

  • ​i (Int): The dimension index.

Returns:

shape_types.values[SIMDLength(i)]: The shape value for dimension i.

stride​

def stride[i: Int](self) -> stride_types.values[SIMDLength(i)]

Returns the i-th stride dimension.

Parameters:

  • ​i (Int): The dimension index.

Returns:

stride_types.values[SIMDLength(i)]: The stride value for dimension i.

shape_coord​

def shape_coord(self) -> Coord[shape_types]

Returns the full shape as a Coord.

Returns:

Coord[shape_types]: A Coord containing all shape dimensions.

stride_coord​

def stride_coord(self) -> Coord[stride_types]

Returns the full stride as a Coord.

Returns:

Coord[stride_types]: A Coord containing all stride dimensions.

write_to​

def write_to(self, mut writer: T)

Writes the Layout representation to a Writer.

Args:

  • ​writer (T): The object to write to.

View source

View source

Was this page helpful?

Models

Cookbook

Puzzles

Blog

Community

Contact