For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo function
compute_input_consumer_count
compute_input_consumer_count[CLUSTER_M: Int, CLUSTER_N: Int, cta_group: Int, CLUSTER_SIZE: Int = 0, epilogue_threads: Int = 0]() -> Int
Compute input pipeline barrier consumer count.
For standard kernels, consumers are the MMA warps across the cluster. For blockwise FP8 kernels, epilogue warps also consume input tiles (A-scales), so pass CLUSTER_SIZE and epilogue_threads to include them.
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!