Mojo struct
SMemTileIter
struct SMemTileIter[dtype: DType, layout: Layout, alignment: Int, num_tiles: Int]
Implemented traits
AnyType,
UnknownDestructibility
Aliases
__del__is_trivial
alias __del__is_trivial = True
storage_size
alias storage_size = ((layout.size() * size_of[dtype]()) * num_tiles)
T
alias T = LayoutTensorIter[dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), alignment=alignment]
tile_storage_size
alias tile_storage_size = (layout.size() * size_of[dtype]())
Methods
build
static build(mut smem_mgr: NVIDIASharedMemoryManager[name, alignment, memory_alignment]) -> LayoutTensorIter[dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), alignment=alignment]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!