For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo module
output_writer
TileWriter for SM100 matmul output pipeline.
Writes accumulated results from TMEM → Registers → SMEM → GMEM (via TMA).
Usage: var writer = TileWriterconfig=..., ... writer.write(smem.c_tiles(), stage, coord, shape, elect)
comptime values
P3_MAX_RANKS
comptime P3_MAX_RANKS = 8
P5_SEND_MBX_INTS
comptime P5_SEND_MBX_INTS = 32
P5_SEND_MBX_STRIDE
comptime P5_SEND_MBX_STRIDE = 4
P5SendMailboxPtr
comptime P5SendMailboxPtr = Pointer[Int32, MutUntrackedOrigin, address_space=AddressSpace.SHARED]
Structs
-
NullPeerSink: No-op sink: the epilogue keeps its local store. -
P3PeerSendConfig: Runtime state for the in-epilogue EP-combine peer scatter-send. -
StandardOutputWriter: DefaultOutputWriterpolicy: local TMA store viaTileWriter. -
TileWriter: Output tile writer for SM100 matmul epilogue.
Traits
-
EpiloguePeerSink: A second destination for the epilogue's SMEM output tile.