[Mlir-commits] [mlir] [MLIR][GPU] Add a pattern to rewrite gpu.subgroup_id (PR #137671)
Alan Li
llvmlistbot at llvm.org
Mon Apr 28 13:43:15 PDT 2025
================
@@ -88,6 +92,7 @@ inline void populateGpuRewritePatterns(RewritePatternSet &patterns) {
populateGpuAllReducePatterns(patterns);
populateGpuGlobalIdPatterns(patterns);
populateGpuShufflePatterns(patterns);
+ populateGpuSubgroupIdPatterns(patterns);
----------------
lialan wrote:
I see. SPV links `get_sub_group_id` function to get the subgroup id. So I think we should just remove this line inside `populateGpuRewritePatterns`.
side note: SPIRV uses the same calculation method to compute subgroup_id.
https://github.com/llvm/llvm-project/pull/137671
More information about the Mlir-commits
mailing list