diff --git a/clang/include/clang/CIR/Dialect/IR/CIRAttrConstraints.td b/clang/include/clang/CIR/Dialect/IR/CIRAttrConstraints.td index a8c81dbd71a09..8f72ff4d754ad 100644 --- a/clang/include/clang/CIR/Dialect/IR/CIRAttrConstraints.td +++ b/clang/include/clang/CIR/Dialect/IR/CIRAttrConstraints.td @@ -38,6 +38,17 @@ def CIR_AnyIntOrFloatAttr : AnyAttrOf<[CIR_AnyIntAttr, CIR_AnyFPAttr], string cppType = "::mlir::TypedAttr"; } +//===----------------------------------------------------------------------===// +// GlobalViewAttr constraints +//===----------------------------------------------------------------------===// + +def CIR_AnyGlobalViewAttr : CIR_AttrConstraint<"::cir::GlobalViewAttr", "GlobalView attribute">; + +def CIR_AnyIntOrGlobalViewAttr : AnyAttrOf<[CIR_AnyIntAttr, CIR_AnyGlobalViewAttr], + "integer or global view attribute"> { + string cppType = "::mlir::TypedAttr"; +} + //===----------------------------------------------------------------------===// // ArrayAttr constraints //===----------------------------------------------------------------------===// @@ -45,4 +56,9 @@ def CIR_AnyIntOrFloatAttr : AnyAttrOf<[CIR_AnyIntAttr, CIR_AnyFPAttr], def CIR_IntArrayAttr : TypedArrayAttrBase; +def CIR_IntOrGlobalViewArrayAttr : TypedArrayAttrBase{ + string cppType = "::mlir::ArrayAttr"; +} + #endif // CLANG_CIR_DIALECT_IR_CIRATTRCONSTRAINTS_TD \ No newline at end of file diff --git a/clang/include/clang/CIR/Dialect/IR/CIRAttrs.td b/clang/include/clang/CIR/Dialect/IR/CIRAttrs.td index 16b818f851e1c..f8358de9a1eb9 100644 --- a/clang/include/clang/CIR/Dialect/IR/CIRAttrs.td +++ b/clang/include/clang/CIR/Dialect/IR/CIRAttrs.td @@ -779,4 +779,56 @@ def CIR_AddressPointAttr : CIR_Attr<"AddressPoint", "address_point"> { }]; } +//===----------------------------------------------------------------------===// +// TypeInfoAttr +//===----------------------------------------------------------------------===// + +def CIR_TypeInfoAttr : CIR_Attr<"TypeInfo", "typeinfo", [TypedAttrInterface]> { + let summary = "Represents a typeinfo used for RTTI"; + let description = [{ + The typeinfo data for a given class is stored into an ArrayAttr. The + layout is determined by the C++ ABI used (clang only implements + itanium on CIRGen). + + The verifier enforces that the output type is always a `!cir.record`, + and that the ArrayAttr element types match the equivalent member type + for the resulting record, i.e, a GlobalViewAttr for symbol reference or + an IntAttr for flags. + + Example: + + ``` + cir.global "private" external @_ZTVN10__cxxabiv120__si_class_type_infoE + : !cir.ptr + + !rec_anon_struct = !cir.record, !cir.ptr, + !cir.ptr}> + + cir.global constant external @type_info = #cir.typeinfo<{ + #cir.global_view<@_ZTVN10__cxxabiv120__si_class_type_infoE, [2 : i32]> + : !cir.ptr, #cir.global_view<@_ZTS1B> : !cir.ptr, + #cir.global_view<@_ZTI1A> : !cir.ptr}> : !rec_anon_struct + ``` + }]; + + let parameters = (ins + AttributeSelfTypeParameter<"">:$type, + CIR_IntOrGlobalViewArrayAttr:$data + ); + + let builders = [ + AttrBuilderWithInferredContext<(ins "mlir::Type":$type, + "mlir::ArrayAttr":$data), [{ + return $_get(type.getContext(), type, data); + }]> + ]; + + // Checks record element types should match the array for every equivalent + // element type. + let genVerifyDecl = 1; + let assemblyFormat = [{ + `<` custom($data) `>` + }]; +} + #endif // CLANG_CIR_DIALECT_IR_CIRATTRS_TD diff --git a/clang/lib/CIR/Dialect/IR/CIRDialect.cpp b/clang/lib/CIR/Dialect/IR/CIRDialect.cpp index 8918eb4cbb1ad..53126348c3bdc 100644 --- a/clang/lib/CIR/Dialect/IR/CIRDialect.cpp +++ b/clang/lib/CIR/Dialect/IR/CIRDialect.cpp @@ -26,8 +26,6 @@ #include "llvm/ADT/SmallSet.h" #include "llvm/Support/LogicalResult.h" -#include - using namespace mlir; using namespace cir; @@ -342,8 +340,8 @@ static LogicalResult checkConstantTypes(mlir::Operation *op, mlir::Type opType, if (mlir::isa( - attrType)) + cir::GlobalViewAttr, cir::PoisonAttr, cir::TypeInfoAttr, + cir::VTableAttr>(attrType)) return success(); assert(isa(attrType) && "What else could we be looking at here?"); @@ -2741,6 +2739,20 @@ LogicalResult cir::AtomicCmpXchg::verify() { return success(); } +//===----------------------------------------------------------------------===// +// TypeInfoAttr +//===----------------------------------------------------------------------===// + +LogicalResult cir::TypeInfoAttr::verify( + ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, + ::mlir::Type type, ::mlir::ArrayAttr typeInfoData) { + + if (cir::ConstRecordAttr::verify(emitError, type, typeInfoData).failed()) + return failure(); + + return success(); +} + //===----------------------------------------------------------------------===// // TableGen'd op method definitions //===----------------------------------------------------------------------===// diff --git a/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp b/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp index e18149a61abd0..8e0991e8c0708 100644 --- a/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp +++ b/clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp @@ -235,6 +235,7 @@ class CIRAttrToValue { mlir::Value visitCirAttr(cir::ConstRecordAttr attr); mlir::Value visitCirAttr(cir::ConstVectorAttr attr); mlir::Value visitCirAttr(cir::GlobalViewAttr attr); + mlir::Value visitCirAttr(cir::TypeInfoAttr attr); mlir::Value visitCirAttr(cir::VTableAttr attr); mlir::Value visitCirAttr(cir::ZeroAttr attr); @@ -521,6 +522,21 @@ mlir::Value CIRAttrToValue::visitCirAttr(cir::GlobalViewAttr globalAttr) { llvm_unreachable("Expecting pointer or integer type for GlobalViewAttr"); } +// TypeInfoAttr visitor. +mlir::Value CIRAttrToValue::visitCirAttr(cir::TypeInfoAttr typeInfoAttr) { + mlir::Type llvmTy = converter->convertType(typeInfoAttr.getType()); + mlir::Location loc = parentOp->getLoc(); + mlir::Value result = mlir::LLVM::UndefOp::create(rewriter, loc, llvmTy); + + for (auto [idx, elt] : llvm::enumerate(typeInfoAttr.getData())) { + mlir::Value init = visit(elt); + result = + mlir::LLVM::InsertValueOp::create(rewriter, loc, result, init, idx); + } + + return result; +} + // VTableAttr visitor. mlir::Value CIRAttrToValue::visitCirAttr(cir::VTableAttr vtableArr) { mlir::Type llvmTy = converter->convertType(vtableArr.getType()); diff --git a/clang/test/CIR/IR/invalid-type-info.cir b/clang/test/CIR/IR/invalid-type-info.cir new file mode 100644 index 0000000000000..4d4726bdac002 --- /dev/null +++ b/clang/test/CIR/IR/invalid-type-info.cir @@ -0,0 +1,17 @@ +// RUN: cir-opt %s -verify-diagnostics -split-input-file + +!u8i = !cir.int + +!rec_anon_struct = !cir.record, !cir.ptr, !cir.ptr}> + +// expected-error @below {{expected !cir.record type}} +cir.global constant external @type_info = #cir.typeinfo<{#cir.global_view<@_ZTVN10__cxxabiv120__si_class_type_infoE, [2 : i32]> : !cir.ptr, #cir.global_view<@_ZTS1B> : !cir.ptr, #cir.global_view<@_ZTI1A> : !cir.ptr}> : !u8i + +// ----- + +!u8i = !cir.int + +!rec_anon_struct = !cir.record + +// expected-error @below {{integer or global view array attribute}} +cir.global constant external @type_info = #cir.typeinfo<{ #cir.undef : !u8i, #cir.int<1> : !u8i, #cir.int<1> : !u8i}> : !rec_anon_struct