Mojo function
wgmma_output_layout
wgmma_output_layout[mma_n: Int, C: Layout]() -> Layout
Returns the output layout component for WGMMA C matrix.
Generates the second mode of the WGMMA C layout, which maps output vector coordinates to linearized indices in the output matrix.
Parameters:
- mma_n (
Int): The N dimension of the WGMMA instruction. - C (
Layout): The layout of the C matrix.
Returns:
Layout - A layout mapping output vector coordinates to linearized indices.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!