Mojo function

get_accumulator_layout

get_accumulator_layout[*, c_smem_dim1: Int, block_tile_shape: IndexList[3], mma_shape: IndexList[3], cta_group: Int]() -> Layout

Compute the register accumulator layout for blockwise FP8.

Returns a 2D layout (num_stages, num_elements) for the register tiles.

Returns:

Layout

View source

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!