Skip to main content

/

Docs

Nightly

v25.1

Log in

Mojo function

create_tma_tile

create_tma_tile[*tile_sizes: Int, *, swizzle_mode: TensorMapSwizzle = TensorMapSwizzle(__init__[__mlir_type.!kgen.int_literal](0))](ctx: DeviceContext, tensor: LayoutTensor[dtype, layout, mut=mut, origin=origin, address_space=address_space, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment]) -> TMATensorTile[dtype, row_major[::Origin[::Bool(_to_int_tuple[*::Int]())]

create_tma_tile[type: DType, rank: Int, tile_shape: Index[$1], /, is_k_major: Bool = True, swizzle_mode: TensorMapSwizzle = TensorMapSwizzle(__init__[__mlir_type.!kgen.int_literal](0)), *, __tile_layout: Layout = row_major($2.__getitem__[::Indexer](0), $2.__getitem__[::Indexer](1)), __desc_layout: Layout = _tma_desc_tile_layout[::DType,::Int,stdlib::utils::index::IndexList[$1, stdlib::sys::info::bitwidthof[AnyTrivialRegType,__mlir_type.!kgen.target]()](ctx: DeviceContext, tensor: LayoutTensor[type, layout, mut=mut, origin=origin, address_space=address_space, element_layout=element_layout, layout_bitwidth=layout_bitwidth, masked=masked, alignment=alignment]) -> TMATensorTile[type, __tile_layout, __desc_layout]