Mojo struct
SMemTileArray
struct SMemTileArray[dtype: DType, layout: Layout, alignment: Int, num_tiles: Int]
Helper struct for allocating arrays of shared memory tiles.
Parameters
- dtype (
DType): Data type of the tile elements. - layout (
Layout): Layout configuration for each tile. - alignment (
Int): Memory alignment requirement. - num_tiles (
Int): Number of tiles in the array.
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 = SMemTileArrayType[dtype, layout, num_tiles, alignment]
Methods
build
static build(mut smem_mgr: NVIDIASharedMemoryManager[name, alignment, memory_alignment]) -> SMemTileArrayType[dtype, layout, num_tiles, alignment]
Build a shared memory tile array using the memory manager.
Args:
- smem_mgr (
NVIDIASharedMemoryManager): The NVIDIA shared memory manager.
Returns:
SMemTileArrayType: The allocated shared memory tile array.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!