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

flat_scale_window

def flat_scale_window[dtype: DType, TILE: Int]() -> Int

Scales a flat scale TMA box STAGES for a TILE-key tile.

One alignment unit wider than the tile, because a caller whose tile base is not 16-byte aligned rounds it down and skips the residual. One unit suffices -- the residual is strictly less than one.

Parameters:

• ​dtype (DType): Scale element type.
• ​TILE (Int): Keys per tile.

Returns:

Int: Scales staged per box.