Mojo function

make_layout

make_layout[l1: Layout, l2: Layout, /, *, linear_idx_type: DType = uint64](a: RuntimeLayout[l1, bitwidth=bitwidth, linear_idx_type=linear_idx_type], b: RuntimeLayout[l2, bitwidth=bitwidth, linear_idx_type=linear_idx_type]) -> RuntimeLayout[make_layout(l1, l2), bitwidth=bitwidth, linear_idx_type=linear_idx_type]

Combine two runtime layouts into a single composite layout.

This creates a new layout by concatenating the dimensions and strides of the input layouts.

Parameters:

• ​l1 (Layout): The static layout type of a.
• ​l2 (Layout): The static layout type of b.
• ​linear_idx_type (DType): The integer type of the all index calculated by the returned runtime layout.

Args:

• ​a (RuntimeLayout[l1, bitwidth=bitwidth, linear_idx_type=linear_idx_type]): The first RuntimeLayout to combine.
• ​b (RuntimeLayout[l2, bitwidth=bitwidth, linear_idx_type=linear_idx_type]): The second RuntimeLayout to combine.

Returns:

A new RuntimeLayout with dimensions from both input layouts.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit