For the complete Mojo documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /docs/manual/basics.md).
apple_mma_store_8x8
def apple_mma_store_8x8[dtype: DType](ptr: UnsafePointer[Scalar[dtype], address_space=ptr.address_space], row_stride: Int, frag: SIMD[dtype, SIMDSize(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 (
UnsafePointer[Scalar[dtype], address_space=ptr.address_space]): Pointer to the top-left corner of the 8x8 tile. - row_stride (
Int): Distance between consecutive rows in the buffer. - frag (
SIMD[dtype, SIMDSize(2)]): SIMD vector of 2 elements to store. - col_stride (
Int): Distance between consecutive columns within a row.