Skip to main content

/

Docs

Nightly

v24.6

Log in

Mojo function

create_tma_tile

create_tma_tile[*tile_sizes: Int, *, swizzle_mode: TensorMapSwizzle = TensorMapSwizzle(SIMD(0))](ctx: DeviceContext, tensor: LayoutTensor[dtype, layout, rank, address_space=address_space, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment]) -> TMATensorTile[dtype, row_major[::Origin[{False}]](_to_int_tuple[*::Int]())]

create_tma_tile[type: DType, rank: Int, tile_shape: IndexList[$1], /, is_k_major: Bool = True, swizzle_mode: TensorMapSwizzle = TensorMapSwizzle(SIMD(0)), *, __tile_layout: Layout = row_major($2.__getitem__[::Indexer](0), $2.__getitem__[::Indexer](1)), __desc_layout: Layout = _tma_desc_tile_layout[::DType,layout::layout::Layout,::Bool,gpu::host::_nvidia_cuda::TensorMapSwizzle]()](ctx: DeviceContext, tensor: LayoutTensor[type, layout, rank, address_space=address_space, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment]) -> TMATensorTile[type, __tile_layout, __desc_layout]