Mojo function
downcast
downcast(layout: Layout, factor: Int) -> Layout
Splits elements in a layout to create a finer layout without changing the total number of elements so that the alignment is preserved.
This function is useful for converting between different data type granularities, such as from uint128 to bf16.
Args:
- layout (
Layout): The layout to downcast. - factor (
Int): The number of elements to split into.
Returns:
A new layout with adjusted shape and stride for the finer granularity.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!