Mojo struct
NVIDIASharedMemoryManager
struct NVIDIASharedMemoryManager[name: StringSlice[StaticConstantOrigin] = "extern_ptr_syml", alignment: Int = 128, memory_alignment: Int = 8]
Fields
- base_ptr (
UnsafePointer[Int8, address_space=AddressSpace(3)]): - offset (
Int):
Implemented traits
AnyType,
UnknownDestructibility
Aliases
__del__is_trivial
alias __del__is_trivial = Int.__del__is_trivial if UnsafePointer[Int8, address_space=AddressSpace(3)].__del__is_trivial else UnsafePointer[Int8, address_space=AddressSpace(3)].__del__is_trivial
Array
alias Array[type: AnyType, num_tiles: Int] = SMemArray[type, num_tiles]
Parameters
Tile
alias Tile[dtype: DType, layout: Layout] = SMemTile[dtype, layout, alignment]
Parameters
TileIter
alias TileIter[dtype: DType, layout: Layout, num_tiles: Int] = SMemTileIter[dtype, layout, alignment, num_tiles]
Parameters
Methods
__init__
__init__(out self)
tile_iter
tile_iter[dtype: DType, layout: Layout, num_tiles: Int](mut self) -> LayoutTensorIter[dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), alignment=alignment]
Returns:
tile
tile[dtype: DType, layout: Layout](mut self) -> LayoutTensor[dtype, layout, MutableAnyOrigin, address_space=AddressSpace(3), alignment=alignment]
Returns:
array
array[type: AnyType, size: Int](mut self) -> UnsafePointer[type, address_space=AddressSpace(3)]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!