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 function

upcast

def upcast[check: Bool = True](var layout: Layout, factor: Int) -> Layout

Fuses consecutive elements in a layout to create a coarser layout.

This function is useful for converting between different data type granularities, such as from bytes to larger data types like bfloat16 or tf32.

Parameters:

• ​check (Bool): Whether to check for incompatible factors.

Args:

• ​layout (Layout): The layout to upcast.
• ​factor (Int): The number of consecutive elements to fuse into one.

Returns:

Layout: A new layout with adjusted shape and stride for the coarser granularity.