Mojo trait
KVBuffer
Implemented traits
comptime members
mma_tile_layout
comptime mma_tile_layout
Required methods
get_dtype
load_from_dram
load_from_dram(mut self: _Self)
get_mma_tile
get_mma_tile(self: _Self) -> LayoutTensor[_Self._dtype, _Self.mma_tile_layout, MutAnyOrigin, address_space=AddressSpace.LOCAL]
Returns:
copy_to_shared
copy_to_shared[tile_id: Int = 0](self: _Self)
load_from_shared
load_from_shared[k_mma: Int](self: _Self)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!