[Mlir-commits] [mlir] [mlir][spirv] Add basic support for SPV_EXT_replicated_composites (PR #147067)

[Jakub Kuderski]

================
@@ -765,6 +765,57 @@ void mlir::spirv::AddressOfOp::getAsmResultNames(
   setNameFn(getResult(), specialName.str());
 }
 
+//===----------------------------------------------------------------------===//
+// spirv.EXTConstantCompositeReplicate
+//===----------------------------------------------------------------------===//
+
+ParseResult
+spirv::EXTConstantCompositeReplicateOp::parse(OpAsmParser &parser,
+                                              OperationState &result) {
+
+  Attribute value;
+  StringRef valueAttrName =
+      spirv::EXTConstantCompositeReplicateOp::getValueAttrName(result.name);
+  Type resultType;
+
+  if (parser.parseLSquare() ||
+      parser.parseAttribute(value, valueAttrName, result.attributes) ||
+      parser.parseRSquare() || parser.parseColonType(resultType))
+    return failure();
+
+  if (isa<NoneType, TensorType>(resultType))
+    if (parser.parseColonType(resultType))
+      return failure();
+
+  if (isa<TensorArmType>(resultType))
+    if (parser.parseOptionalColon().succeeded())
----------------
kuhar wrote:

nit: use free functions for success / failure checks

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