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]

PipelineBarrier

alias PipelineBarrier[num_pipeline_stages: Int] = SMemArrayType[SharedMemBarrier, num_pipeline_stages]

Type alias for shared memory pipeline barrier array.

Parameters

• ​num_pipeline_stages (Int):

RegTileType

alias RegTileType[_dtype: DType, layout: Layout, /, *, element_layout: Layout = Layout(IntTuple(1), IntTuple(1)), layout_int_type: DType = _get_layout_type(layout, AddressSpace.LOCAL), linear_idx_type: DType = _get_index_type(layout, AddressSpace.LOCAL), masked: Bool = False, alignment: Int = align_of[_dtype]()] = LayoutTensor[_dtype, layout, MutAnyOrigin, address_space=AddressSpace.LOCAL, 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

• ​_dtype (DType):
• ​layout (Layout):
• ​element_layout (Layout):
• ​layout_int_type (DType):
• ​linear_idx_type (DType):
• ​masked (Bool):
• ​alignment (Int):

SMemBarrier

alias SMemBarrier = LegacyUnsafePointer[SharedMemBarrier, address_space=AddressSpace.SHARED]

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.SHARED), linear_idx_type: DType = _get_index_type(layout, AddressSpace.SHARED), masked: Bool = False, alignment: Int = align_of[_dtype]()] = LayoutTensor[_dtype, layout, MutAnyOrigin, address_space=AddressSpace.SHARED, 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

• ​_dtype (DType):
• ​layout (Layout):
• ​element_layout (Layout):
• ​layout_int_type (DType):
• ​linear_idx_type (DType):
• ​masked (Bool):
• ​alignment (Int):

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

• ​SharedMemoryBasePtr:

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit