Mojo trait
RegisterMMABuffer
Implemented traits
comptime members
mma_dtype
comptime mma_dtype
mma_tile_layout
comptime mma_tile_layout
reg_dtype
comptime reg_dtype
reg_tile_layout
comptime 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, MutAnyOrigin, address_space=AddressSpace.LOCAL]
Returns:
get_dtype
zero
zero(self: _Self)
get_reg_tile
get_reg_tile[stage: Int = 0](self: _Self) -> LayoutTensor[_Self.reg_dtype, _Self.reg_tile_layout, MutAnyOrigin, address_space=AddressSpace.LOCAL]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!