Mojo struct
ComposedLayout
struct ComposedLayout[LayoutA: LayoutTrait, LayoutB: LayoutTrait, offset: OptionalReg[Int] = OptionalReg[Int]({:@stdlib::@builtin::@int::@Int {0}, 0})]
A layout composed of two layouts applied in sequence.
This struct combines two layouts where the output of the first layout is used as input to the second layout, with an optional offset in between.
Parameters
- LayoutA (
LayoutTrait): The first layout to apply. - LayoutB (
LayoutTrait): The second layout to apply. - offset (
OptionalReg[Int]): An optional offset to apply 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"): Indicates whether the layout has a valid shape. This is True if either layouts 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 the composed layout with two layouts.
Args:
- layout_a (
LayoutA): The first layout to apply. - layout_b (
LayoutB): The second layout to apply.
__copyinit__
__copyinit__(out self, other: Self)
Copy constructor for the composed layout.
Args:
- other (
Self): The composed layout to copy from.
__call__
__call__(self, idx: IntTuple[origin]) -> Int
Apply the composed layout to an index.
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 a runtime offset.
Args:
- idx (
IntTuple[origin]): The index to transform. - offset_val (
Int): The 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.
cosize
cosize(self) -> Int
Get the cosize of the composed layout.
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!