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

Mojo function

pipeline_layout

def pipeline_layout[layout: Layout, pipeline_stages: Int]() -> Layout

Builds a 2D layout extended with a pipeline-stage dimension.

Combines the given 2D layout with a row-major stage dimension of length pipeline_stages so that each stage owns a full copy of the tile layout.

Parameters:

layout (Layout): The base 2D tile layout to replicate per stage.
pipeline_stages (Int): Number of pipeline stages in the buffer.

Returns:

Layout: A layout suitable for staging multiple copies of the tile in shared memory.