Mojo function
tma_multicast_tile_load
tma_multicast_tile_load[dtype: DType, tile_layout: Layout, desc_layout: Layout, //, cluster_size: Int, use_partitioned_multicast: Bool](tma_op: TMATensorTile[dtype, tile_layout, desc_layout], smem_tile: LayoutTensor[dtype, layout, origin, address_space=AddressSpace(3), element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=128], ref [3] mem_barrier: SharedMemBarrier, rank: UInt, coords: Tuple[UInt, UInt], multicast_mask: UInt16)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!