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

load_z

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

Stores row_count rows from AMX Z accumulator registers starting at start_index to memory at src.

Parameters:

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

Args:

src (Pointer[Scalar[dtype], _safe=False]): Destination memory address for the stored rows.
start_index (Int): Starting Z accumulator row index to store from.