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โ
- โ
EpilogueWarpContext: Epilogue warp context - consumes TMEM data, signals completion. - โ
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!