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