Mojo trait

RegisterBuffer

Implemented traits

comptime reg_dtype

comptime reg_tile_layout

static get_dtype() -> DType

Returns:

DType

zero(self: _Self)

get_reg_tile[stage: Int = 0](self: _Self) -> LayoutTensor[_Self.reg_dtype, _Self.reg_tile_layout, MutAnyOrigin, address_space=AddressSpace.LOCAL]

Returns:

LayoutTensor

View source

Was this page helpful?

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

Thank you for helping us improve!