(tile_id, num_pid_in_group, num_pid_m, GROUP_SIZE_M, NUM_SMS)
| 284 | |
| 285 | @triton.jit |
| 286 | def _compute_pid(tile_id, num_pid_in_group, num_pid_m, GROUP_SIZE_M, NUM_SMS): |
| 287 | group_id = tile_id // num_pid_in_group |
| 288 | first_pid_m = group_id * GROUP_SIZE_M |
| 289 | group_size_m = min(num_pid_m - first_pid_m, GROUP_SIZE_M) |
| 290 | pid_m = first_pid_m + (tile_id % group_size_m) |
| 291 | pid_n = (tile_id % num_pid_in_group) // group_size_m |
| 292 | return pid_m, pid_n |
| 293 | |
| 294 | |
| 295 | @triton.autotune( |
no outgoing calls
no test coverage detected