Mojo function
compute_total_tiles
compute_total_tiles[tile_m: Int, tile_n: Int, max_groups: Int](problem_sizes: LayoutTensor[DType.int32, Layout.row_major(max_groups, 4), MutAnyOrigin], num_groups: Int) -> Int
Compute total number of tiles across all groups.
Args:
- problem_sizes (
LayoutTensor): (num_groups, 4) tensor with [M, N, K, L] per group. - num_groups (
Int): Number of GEMM problems.
Returns:
Int: Total tile count.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!