(grid)
| 415 | |
| 416 | |
| 417 | def metadata_fn( |
| 418 | grid: tuple, |
| 419 | metadata: NamedTuple, |
| 420 | args: dict, |
| 421 | ): |
| 422 | """Copied from https://github.com/triton-lang/triton/blob/main/third_party/proton/tutorials/matmul.py""" |
| 423 | grid_x, grid_y, grid_z = unpack_grid(grid) |
| 424 | num_warps = metadata.num_warps |
| 425 | num_stages = metadata.num_stages |
| 426 | cluster_x, cluster_y, cluster_z = unpack_grid((metadata.num_ctas,)) |