Skip to main content

Mojo trait

RegisterMMABuffer

Implemented traits

AnyType, RegisterBuffer, UnknownDestructibility

Aliases

__del__is_trivial

alias __del__is_trivial

A flag (often compiler generated) to indicate whether the implementation of __del__ is trivial.

The implementation of __del__ is considered to be trivial if:

  • The struct has a compiler-generated trivial destructor and all its fields have a trivial __del__ method.

In practice, it means that the __del__ can be considered as no-op.

mma_dtype

alias mma_dtype

mma_tile_layout

alias mma_tile_layout

reg_dtype

alias reg_dtype

reg_tile_layout

alias reg_tile_layout

Required methods

get_mma_tile

get_mma_tile[tile_idx: Int, k_idx: Int](self: _Self) -> LayoutTensor[_Self.mma_dtype, _Self.mma_tile_layout, MutableAnyOrigin, address_space=AddressSpace(5)]

Returns:

LayoutTensor

get_dtype

static get_dtype() -> DType

Returns:

DType

zero

zero(self: _Self)

get_reg_tile

get_reg_tile(self: _Self) -> LayoutTensor[_Self.reg_dtype, _Self.reg_tile_layout, MutableAnyOrigin, address_space=AddressSpace(5)]

Returns:

LayoutTensor

Was this page helpful?