Mojo struct
ComposedLayout
struct ComposedLayout[LayoutA: LayoutTrait, LayoutB: LayoutTrait, offset: OptionalReg[Int] = OptionalReg[Int]({:@stdlib::@builtin::@int::@Int {0}, 0})]
Layout composed of two layouts applied sequentially.
Combines two layouts. Output of the first (LayoutA) is input to
the second (LayoutB), with optional offset in between.
Parameters
- LayoutA (
LayoutTrait): The first layout to apply. - LayoutB (
LayoutTrait): The second layout to apply. - offset (
OptionalReg[Int]): Optional offset between layouts (default: 0).
Aliases
has_shape = get_vtable_entry(:trait<@layout::@layout::@LayoutTrait> LayoutA, "has_shape") if get_vtable_entry(:trait<@layout::@layout::@LayoutTrait> LayoutA, "has_shape") else get_vtable_entry(:trait<@layout::@layout::@LayoutTrait> LayoutB, "has_shape"): True if either layout has a shape.
Fields
- layout_a (
LayoutA): The first layout to apply. - layout_b (
LayoutB): The second layout to apply.
Implemented traits
AnyType,
Copyable,
LayoutTrait,
UnknownDestructibility
Methods
__init__
__init__(out self, layout_a: LayoutA, layout_b: LayoutB)
Initialize ComposedLayout with two layouts.
Args:
- layout_a (
LayoutA): The first layout. - layout_b (
LayoutB): The second layout.
__copyinit__
__copyinit__(out self, other: Self)
Copy constructor for ComposedLayout.
Args:
- other (
Self): The ComposedLayout to copy from.
__call__
__call__(self, idx: IntTuple[origin]) -> Int
Apply composed layout to an index.
Applies LayoutA, then adds offset, then applies LayoutB.
Args:
- idx (
IntTuple[origin]): The index to transform.
Returns:
The transformed index.
__call__(self, idx: IntTuple[origin], offset_val: Int) -> Int
Apply composed layout with runtime offset.
Applies LayoutA, then adds runtime offset_val, then LayoutB.
Static offset must not be set when using runtime offset.
Args:
- idx (
IntTuple[origin]): The index to transform. - offset_val (
Int): Runtime offset to apply.
Returns:
The transformed index.
size
size(self) -> Int
Get the size of the composed layout.
Returns the size of the first layout (LayoutA).
Returns:
The size of the first layout.
cosize
cosize(self) -> Int
Get the cosize of the composed layout.
Returns the cosize of the second layout (LayoutB).
Returns:
The cosize of the second layout.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!