Mojo function
write_bf16x2_row_to_smem_fast
write_bf16x2_row_to_smem_fast[layout: Layout, *, out_dtype: DType, in_dtype: DType, config: MLA_SM100_Decode_Config, local_tile_size: Int](shared_mem: UnsafePointer[Scalar[out_dtype], MutAnyOrigin, address_space=AddressSpace.SHARED], local_mem: LayoutTensor[in_dtype, layout, MutAnyOrigin, address_space=AddressSpace.LOCAL], col_start: Int, row_start: Int)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!