Mojo function
is_valid_mma_tiler_and_cluster_shape
is_valid_mma_tiler_and_cluster_shape(mma_tiler_m: Int, mma_tiler_n: Int, cluster_m: Int, cluster_n: Int) -> Bool
Check if MMA tiler and cluster shape are valid.
Constraints (from NVIDIA CuTe DSL):
- MMA tiler M: 128 or 256
- MMA tiler N: 128 or 256
- Cluster M must be multiple of 2 if MMA tiler M is 256
- Cluster M/N: Power of 2, <=4 per axis (for SF multicast)
- Total cluster size: <=16
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!