Mojo function
upcast
upcast[check: Bool = True](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.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!