Mojo module
blockwise_fp8
comptime values
logger
comptime logger = Logger(stdout, "", False)
smem_layout_3D
comptime smem_layout_3D[layout: Layout] = Layout(IntTuple(IntTuple(1), layout.shape[0].owned_copy(), layout.shape[1].owned_copy(), Tuple()), IntTuple(IntTuple(0), layout.stride[0].owned_copy(), layout.stride[1].owned_copy(), Tuple()))
Parameters
- layout (
Layout):
Functions
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!