Alan Li ac65b2c327 [MLIR][GPU] Add a pattern to rewrite gpu.subgroup_id (#137671) ...

This patch impelemnts a rewrite pattern for transforming
`gpu.subgroup_id` to:
```
subgroup_id = linearized_thread_id / gpu.subgroup_size
```

where:
```
linearized_thread_id = thread_id.x + block_dim.x * (thread_id.y + block_dim.y * thread_id.z)
```

2025-04-29 10:54:48 -04:00

1.7 KiB

Raw Blame History

View Raw