Mojo function
apple_mma_store
apple_mma_store[dtype: DType](ptr: UnsafePointer[Scalar[dtype], ptr.origin, address_space=ptr.address_space], row_stride: Int, frag: SIMD[dtype, 8], col_stride: Int = 1)
Stores a 16x16 matrix fragment from the current simdgroup thread.
Parameters:
- dtype (
DType): Element type of the matrix.
Args:
- ptr (
UnsafePointer): Pointer to the top-left corner of the 16x16 tile. - row_stride (
Int): Distance between consecutive rows in the buffer. - frag (
SIMD): SIMD vector of 8 elements to store. - col_stride (
Int): Distance between consecutive columns within a row.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!