Skip to main content

Mojo function

stsm_helper

stsm_helper[swizzle: Swizzle, stageN: Int, vec_dtype: DType, vec_size: Int, DstLayout: TensorLayout, transpose_c: Bool = False, swizzle_mode: TensorMapSwizzle = TensorMapSwizzle.SWIZZLE_128B](vec: InlineArray[Scalar[vec_dtype], vec_size], dst: TileTensor[dst.dtype, DstLayout, MutAnyOrigin, address_space=AddressSpace.SHARED], warp_offset: UInt32 = 0)

Was this page helpful?