[all-commits] [llvm/llvm-project] ac65b2: [MLIR][GPU] Add a pattern to rewrite gpu.subgroup_...

Alan Li via All-commits all-commits at lists.llvm.org
Tue Apr 29 07:55:11 PDT 2025


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: ac65b2c327d791a4731360c52625d11b8d233449
      https://github.com/llvm/llvm-project/commit/ac65b2c327d791a4731360c52625d11b8d233449
  Author: Alan Li <me at alanli.org>
  Date:   2025-04-29 (Tue, 29 Apr 2025)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/Transforms/Passes.h
    M mlir/lib/Dialect/GPU/CMakeLists.txt
    M mlir/lib/Dialect/GPU/Transforms/GlobalIdRewriter.cpp
    A mlir/lib/Dialect/GPU/Transforms/SubgroupIdRewriter.cpp
    A mlir/test/Dialect/GPU/subgroupId-rewrite.mlir
    M mlir/test/lib/Dialect/GPU/TestGpuRewrite.cpp

  Log Message:
  -----------
  [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)
```



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications


More information about the All-commits mailing list