Mojo module
warp_context
RAII warp context managers for SM100 matmul kernel.
MmaWarpContext: MMA warp - allocates TMEM, deallocates on exit EpilogueWarpContext: Epilogue warp - consumes TMEM, signals completion on exit
Structs
-
EpilogueWarp: Unified linear type for epilogue warp lifecycle. -
EpilogueWarpContext: Epilogue warp context - consumes TMEM data, signals completion. -
MmaWarp: Unified linear type for MMA warp TMEM lifecycle. -
MmaWarpContext: MMA warp context - owns TMEM lifecycle and output pipeline.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!