[Mlir-commits] [mlir] [mlir][SPIRV] Support Alignment and FuncParamAttr on parameters (PR #231069)

[Alexander Shaposhnikov]

================
@@ -485,13 +508,19 @@ LogicalResult spirv::Deserializer::processMemberName(ArrayRef<uint32_t> words) {
 
 LogicalResult spirv::Deserializer::setFunctionArgAttrs(
     uint32_t argID, SmallVectorImpl<Attribute> &argAttrs, size_t argIndex) {
-  if (!decorations.contains(argID)) {
-    argAttrs[argIndex] = DictionaryAttr::get(context, {});
-    return success();
-  }
-
   spirv::DecorationAttr foundDecorationAttr;
+  NamedAttrList attributes;
+  if (auto it = functionParameterAttrs.find(argID);
+      it != functionParameterAttrs.end()) {
+    attributes.set("spirv.func_param_attr", it->second);
+    functionParameterAttrs.erase(it);
+  }
   for (NamedAttribute decAttr : decorations[argID]) {
+    if (decAttr.getName() == "alignment") {
----------------
alexander-shaposhnikov wrote:

Done

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