Mojo function
upcast
upcast(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.
Args:
- layout (
Layout): The layout to upcast. - factor (
Int): The number of consecutive elements to fuse into one.
Returns:
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!