Mojo function

zipped_divide

zipped_divide[LayoutType: TensorLayout, //, tile: Coord[tile.element_types]](layout: LayoutType) -> Layout[*?, *?]

Divides a layout into inner (tile) and outer (number-of-tiles) parts.

Given a layout and a tile shape, produces a hierarchical layout where the inner component has the tile shape with the original strides, and the outer component has shape = original_shape / tile with scaled strides.

Parameters:

• ​LayoutType (TensorLayout): The type of the input layout.
• ​tile (Coord): The tile shape to divide by.

Args:

• ​layout (LayoutType): The layout to divide.

Returns:

Layout: A ZippedDivideLayout with inner and outer components.

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