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

apple_mma_store_8x8

def apple_mma_store_8x8[dtype: DType](ptr: Pointer[Scalar[dtype], address_space=ptr.address_space, _safe=True], row_stride: Int, frag: SIMD[dtype, SIMDLength(2)], col_stride: Int = Int(1))

Stores an 8x8 matrix fragment from the current simdgroup thread.

Parameters:

dtype (DType): Element type of the matrix.

Args:

ptr (Pointer[Scalar[dtype], address_space=ptr.address_space, _safe=True]): Pointer to the top-left corner of the 8x8 tile.
row_stride (Int): Distance between consecutive rows in the buffer.
frag (SIMD[dtype, SIMDLength(2)]): SIMD vector of 2 elements to store.
col_stride (Int): Distance between consecutive columns within a row.