Mojo module
kernel_common
Shared kernel components for SM100 warp-specialized matmul kernels.
This module contains common components used by all SM100 matmul kernel variants:
- WarpRole: Warp specialization roles (MMA, Load, Scheduler, Epilogue)
- KernelContext: Common kernel state (election vars, CTA coords, masks)
- consumer_main_loop: Legacy MMA consumer loop (deprecated but kept for compatibility)
Structs
-
KernelContext: Shared kernel state: election vars, CTA coords, multicast masks, pipeline states. -
WarpRole: Warp role identifiers for SM100 warp-specialized kernel.
Functions
-
consumer_main_loop: DEPRECATED: Legacy MMA consumer loop for external callers.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!