For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

rows_owned

def rows_owned[config: MLA_SM100_Decode_Config](block_x: Int) -> Int

Returns the number of output rows the CTA at grid x-index block_x owns.

Every head group is out_rows tall except the last, which holds the remainder of num_q_heads.

Parameters:

• ​config (MLA_SM100_Decode_Config): Decode config supplying the head-group geometry.

Args:

• ​block_x (Int): The CTA's grid x-index.

Returns:

Int: The number of output rows the CTA owns, at most out_rows.