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

store_z

def store_z[row_count: Int, dtype: DType](src: Pointer[Scalar[dtype], _safe=False], start_index: Int)

Loads row_count rows from memory at src into AMX Z accumulator registers starting at start_index.

Parameters:

row_count (Int): Number of Z accumulator rows to load; must be 1 or 2.
dtype (DType): Element type of the Z accumulator rows.

Args:

src (Pointer[Scalar[dtype], _safe=False]): Source memory address to load rows from.
start_index (Int): Starting Z accumulator row index to load into.