[CIR] Fix parsing of a generic TBAAAttr

Author: Jezurko

clang/include/clang/CIR/Dialect/IR/CIRAttrs.h

@@ -15,6 +15,7 @@
 
 #include "mlir/IR/Attributes.h"
 #include "mlir/IR/BuiltinAttributeInterfaces.h"
+#include "mlir/IR/OpImplementation.h"
 
 #include "clang/CIR/Dialect/IR/CIROpsEnums.h"
 
@@ -50,4 +51,24 @@ class VectorType;
 #define GET_ATTRDEF_CLASSES
 #include "clang/CIR/Dialect/IR/CIROpsAttributes.h.inc"
 
+template <> struct ::mlir::FieldParser<cir::TBAAAttr, cir::TBAAAttr> {
+  static mlir::FailureOr<cir::TBAAAttr> parse(mlir::AsmParser &parser) {
+    mlir::Attribute attribute;
+    if (parser.parseAttribute(attribute))
+      return mlir::failure();
+    if (auto omnipotentChar =
+            mlir::dyn_cast<cir::TBAAOmnipotentCharAttr>(attribute))
+      return omnipotentChar;
+    if (auto vtablePtr = mlir::dyn_cast<cir::TBAAVTablePointerAttr>(attribute))
+      return vtablePtr;
+    if (auto scalar = mlir::dyn_cast<cir::TBAAScalarAttr>(attribute))
+      return scalar;
+    if (auto tag = mlir::dyn_cast<cir::TBAATagAttr>(attribute))
+      return tag;
+    if (auto structAttr = mlir::dyn_cast<cir::TBAAStructAttr>(attribute))
+      return structAttr;
+    return parser.emitError(parser.getCurrentLocation(), "Expected TBAAAttr");
+  }
+};
+
 #endif // CLANG_CIR_DIALECT_IR_CIRATTRS_H

clang/test/CIR/CodeGen/tbaa-parse.cpp

@@ -0,0 +1,12 @@
+// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -fclangir -emit-cir -o %t -O1
+// RUN: cir-opt %t
+
+struct S {
+    short i;
+};
+
+struct S glob;
+int main(void) {
+    glob.i = 0;
+    return glob.i;
+}