[Mlir-commits] [mlir] [MLIR][ROCDL] Lower `gpu.subgroup_id` to `wavefrontsize` (PR #137360)

[Krzysztof Drewniak]

================
@@ -113,6 +132,20 @@ struct GPULaneIdOpToROCDL : ConvertOpToLLVMPattern<gpu::LaneIdOp> {
   }
 };
 
+struct GPUSubgroupSizeOpToROCDL : ConvertOpToLLVMPattern<gpu::SubgroupSizeOp> {
+  using ConvertOpToLLVMPattern::ConvertOpToLLVMPattern;
+  LogicalResult
+  matchAndRewrite(gpu::SubgroupSizeOp op, gpu::SubgroupSizeOp::Adaptor adaptor,
+                  ConversionPatternRewriter &rewriter) const override {
+    Value wavefrontOp = rewriter.create<ROCDL::WavefrontSizeOp>(
----------------
krzysz00 wrote:

Can you plumb through `upper_bound` here? 

https://github.com/llvm/llvm-project/pull/137360