Mojo function
tma_tile_qo
tma_tile_qo[dtype: DType, //, swizzle_mode: TensorMapSwizzle, *, BM: Int, BK: Int, depth: Int](ctx: DeviceContext, ptr: UnsafePointer[Scalar[dtype], ptr.origin], rows: Int, out res: TMATensorTile[dtype, 2, IndexList(VariadicList(BM, BK), Tuple()), _default_desc_shape[2, dtype, IndexList(VariadicList(BM, BK), Tuple()), swizzle_mode]()])
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!