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):
NVIDIASharedMemoryManager
alias NVIDIASharedMemoryManager = SharedMemoryManager[NVIDIASharedMemoryBasePtr]
RegTileType
alias RegTileType[_dtype: DType, layout: Layout, /, *, element_layout: Layout = Layout(IntTuple(1), IntTuple(1)), layout_int_type: DType = _get_layout_type(layout, AddressSpace(5)), linear_idx_type: DType = _get_index_type(layout, AddressSpace(5)), masked: Bool = False, alignment: Int = align_of[_dtype]()] = LayoutTensor[_dtype, layout, MutableAnyOrigin, address_space=AddressSpace(5), element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, 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, /, *, element_layout: Layout = Layout(IntTuple(1), IntTuple(1)), layout_int_type: DType = _get_layout_type(layout, AddressSpace(3)), linear_idx_type: DType = _get_index_type(layout, AddressSpace(3)), masked: Bool = False, alignment: Int = align_of[_dtype]()] = LayoutTensor[_dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment]
Type alias for shared memory tile tensors.
Parameters
Structs
-
IteratorScatterGatherAmd: Iterator-based AMD scatter-gather for DRAM-register data movement. -
NVIDIASharedMemoryBasePtr: -
ScatterGatherAmd: AMD tile-based scatter-gather for DRAM-register data movement. -
SharedMemoryManager: -
SMemArrayType: Shared memory array of fixed size. -
SMemTileArrayType: Array of tiles in shared memory.
Traits
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!