Mojo function
calculate_output_tile_bounds
calculate_output_tile_bounds[c_layout: Layout, //, BM: Int, BN: Int](c: LayoutTensor[dtype, c_layout, MutableAnyOrigin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment], block_y: Int, block_x: Int) -> Tuple[UInt32, UInt32]
Calculate the output bounds for the current thread block.
Returns the maximum valid row and column indices for this block's output tile, accounting for edge cases where the tile extends beyond the matrix dimensions.
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!