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