[Mlir-commits] [mlir] [MLIR][GPU] Support grid constant, byval, byref on gpu.func (PR #172037)

[Asher Mancinelli]

================
@@ -359,6 +359,14 @@ GPUFuncOpLowering::matchAndRewrite(gpu::GPUFuncOp gpuFuncOp, OpAdaptor adaptor,
     if (argAttr.empty())
       continue;
 
+    const bool isArgTypeUnchanged =
+        remapping->size == 1 &&
+        llvmFuncOp.getArgument(remapping->inputNo).getType() == argTy;
+    if (isArgTypeUnchanged) {
+      llvmFuncOp.setArgAttrs(remapping->inputNo, argAttr);
+      continue;
+    }
+
----------------
ashermancinelli wrote:

How does this look?

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