Mojo module
structuring
Aliases
eval
alias eval[T: AnyType, //, val: T] = val
Helper alias to force evaluation of expressions at compile time.
Parameters
- T (
AnyType): - val (
T):
RegTileType
alias RegTileType[_dtype: DType, layout: Layout, alignment: Int = align_of[SIMD[_dtype, simd_width_of[_dtype]()]]()] = LayoutTensor[_dtype, layout, MutableAnyOrigin, address_space=AddressSpace(5), alignment=alignment]
Type alias for register (local memory) tile tensors.
Parameters
SMemBarrier
alias SMemBarrier = UnsafePointer[SharedMemBarrier, address_space=AddressSpace(3)]
Type alias for shared memory barrier pointer.
SMemTileType
alias SMemTileType[_dtype: DType, layout: Layout, alignment: Int = align_of[SIMD[_dtype, simd_width_of[_dtype]()]]()] = LayoutTensor[_dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), alignment=alignment]
Type alias for shared memory tile tensors.
Parameters
SMemWarpTileType
alias SMemWarpTileType[_dtype: DType, layout: Layout, warp_rows: Int, warp_cols: Int] = LayoutTensor[_dtype, LayoutTensor._compute_tile_layout[True, _dtype, layout, MutableAnyOrigin, AddressSpace(3), Layout(IntTuple(1), IntTuple(1)), _get_layout_type(layout, AddressSpace(3)), _get_index_type(layout, AddressSpace(3)), False, align_of[SIMD[_dtype, simd_width_of[_dtype]()]](), warp_rows, warp_cols]()[0], MutableAnyOrigin, address_space=AddressSpace(3), layout_int_type=_get_layout_type(layout, AddressSpace(3)), linear_idx_type=_get_index_type(layout, AddressSpace(3)), masked=_tile_is_masked[layout, warp_rows, warp_cols](), alignment=align_of[SIMD[_dtype, simd_width_of[_dtype]()]]()]
Type alias for warp-level shared memory tiles with specified dimensions.
Parameters
Structs
-
IteratorScatterGatherAmd: Iterator-based AMD data movement delegate for scatter-gather operations. -
NVIDIASharedMemoryManager: Manager for allocating and organizing shared memory on NVIDIA GPUs. -
ScatterGatherAmd: Tile-based AMD data movement delegate for scatter-gather operations. -
SMemArray: Helper struct for allocating shared memory arrays. -
SMemTile: Helper struct for allocating shared memory tiles. -
SMemTileArray: Helper struct for allocating arrays of shared memory tiles. -
SMemTileArrayType: Shared memory tile array for managing multiple tiles in shared memory.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!