9#ifndef LLVM_CLANG_LIB_CIR_CODEGEN_CIRGENBUILDER_H
10#define LLVM_CLANG_LIB_CIR_CODEGEN_CIRGENBUILDER_H
15#include "mlir/IR/Attributes.h"
16#include "mlir/IR/BuiltinAttributes.h"
17#include "mlir/Support/LLVM.h"
23#include "llvm/ADT/APFloat.h"
24#include "llvm/ADT/STLExtras.h"
30 llvm::StringMap<unsigned> recordNames;
31 llvm::StringMap<unsigned> globalsVersioning;
40 mlir::Attribute
getString(llvm::StringRef str, mlir::Type eltTy,
41 std::optional<size_t> size) {
42 size_t finalSize = size.value_or(str.size());
44 size_t lastNonZeroPos = str.find_last_not_of(
'\0');
47 if (lastNonZeroPos == llvm::StringRef::npos) {
48 auto arrayTy = cir::ArrayType::get(eltTy, finalSize);
49 return cir::ZeroAttr::get(arrayTy);
52 size_t trailingZerosNum = 0;
53 if (finalSize > lastNonZeroPos + 2)
54 trailingZerosNum = finalSize - lastNonZeroPos - 1;
55 auto truncatedArrayTy =
56 cir::ArrayType::get(eltTy, finalSize - trailingZerosNum);
57 auto fullArrayTy = cir::ArrayType::get(eltTy, finalSize);
58 return cir::ConstArrayAttr::get(
60 mlir::StringAttr::get(str.drop_back(trailingZerosNum),
66 cir::ArrayType arrayTy)
const {
67 return cir::ConstArrayAttr::get(arrayTy, attrs);
73 mlir::Type
type = {});
80 for (
auto &f : arrayAttr) {
81 auto ta = mlir::cast<mlir::TypedAttr>(f);
82 members.push_back(ta.getType());
88 auto sTy = mlir::cast<cir::RecordType>(ty);
89 return cir::ConstRecordAttr::get(sTy, arrayAttr);
94 return cir::TypeInfoAttr::get(anonRecord.getType(), fieldsAttr);
100 auto it = recordNames.find(baseName);
101 if (it == recordNames.end()) {
102 recordNames[baseName] = 0;
106 return baseName +
"." + std::to_string(recordNames[baseName]++);
110 if (&
format == &llvm::APFloat::IEEEdouble())
111 return cir::LongDoubleType::get(getContext(), typeCache.DoubleTy);
112 if (&
format == &llvm::APFloat::x87DoubleExtended())
113 return cir::LongDoubleType::get(getContext(), typeCache.FP80Ty);
114 if (&
format == &llvm::APFloat::IEEEquad())
115 return cir::LongDoubleType::get(getContext(), typeCache.FP128Ty);
116 if (&
format == &llvm::APFloat::PPCDoubleDouble())
117 llvm_unreachable(
"NYI: PPC double-double format for long double");
118 llvm_unreachable(
"Unsupported format for long double");
126 bool isVarArg =
false) {
127 return cir::FuncType::get(params, retTy, isVarArg);
134 return cir::RecordType::Class;
136 return cir::RecordType::Struct;
138 return cir::RecordType::Union;
140 llvm_unreachable(
"interface records are NYI");
142 llvm_unreachable(
"enums are not records");
144 llvm_unreachable(
"Unsupported record kind");
152 bool packed,
bool padded,
153 llvm::StringRef name) {
154 const auto nameAttr = getStringAttr(name);
155 auto kind = cir::RecordType::RecordKind::Struct;
164 assert(!
type.isIncomplete() ||
165 (
type.getMembers() == members &&
type.getPacked() == packed &&
166 type.getPadded() == padded));
170 type.complete(members, packed, padded);
178 llvm::StringRef name =
"");
185 const mlir::StringAttr nameAttr = getStringAttr(name);
186 cir::RecordType::RecordKind kind = cir::RecordType::RecordKind::Struct;
195 if (mlir::isa<cir::ZeroAttr>(
attr))
198 if (
const auto ptrVal = mlir::dyn_cast<cir::ConstPtrAttr>(
attr))
199 return ptrVal.isNullValue();
201 if (
const auto intVal = mlir::dyn_cast<cir::IntAttr>(
attr))
202 return intVal.isNullValue();
204 if (
const auto boolVal = mlir::dyn_cast<cir::BoolAttr>(
attr))
205 return !boolVal.getValue();
207 if (
auto fpAttr = mlir::dyn_cast<cir::FPAttr>(
attr)) {
208 auto fpVal = fpAttr.getValue();
210 llvm::APFloat fv(+0.0);
211 fv.convert(fpVal.getSemantics(), llvm::APFloat::rmNearestTiesToEven,
213 return fv.bitwiseIsEqual(fpVal);
216 if (
const auto arrayVal = mlir::dyn_cast<cir::ConstArrayAttr>(
attr)) {
217 if (mlir::isa<mlir::StringAttr>(arrayVal.getElts()))
221 mlir::cast<mlir::ArrayAttr>(arrayVal.getElts()),
222 [&](
const mlir::Attribute &elt) { return isNullValue(elt); });
242 return cir::IntType::get(getContext(), n,
false);
257 return cir::IntType::get(getContext(), n,
true);
273 cir::ConstantOp
getConstInt(mlir::Location loc, llvm::APSInt intVal);
275 cir::ConstantOp
getConstInt(mlir::Location loc, llvm::APInt intVal);
277 cir::ConstantOp
getConstInt(mlir::Location loc, mlir::Type t, uint64_t
c);
279 cir::ConstantOp
getConstFP(mlir::Location loc, mlir::Type t,
280 llvm::APFloat fpVal);
283 return i == typeCache.UInt8Ty || i == typeCache.SInt8Ty;
286 return i == typeCache.UInt16Ty || i == typeCache.SInt16Ty;
289 return i == typeCache.UInt32Ty || i == typeCache.SInt32Ty;
292 return i == typeCache.UInt64Ty || i == typeCache.SInt64Ty;
294 bool isInt(mlir::Type i) {
return mlir::isa<cir::IntType>(i); }
301 bool packed =
false,
bool padded =
false) {
303 auto kind = cir::RecordType::RecordKind::Struct;
319 return cir::ConstantOp::create(*
this, loc, cir::IntAttr::get(sInt64Ty,
c));
324 if (
auto intTy = mlir::dyn_cast<cir::IntType>(value.getType())) {
326 if (intTy.isUnsigned())
328 return cir::UnaryOp::create(*
this, value.getLoc(), value.getType(),
329 cir::UnaryOpKind::Minus, value);
332 llvm_unreachable(
"negation for the given type is NYI");
340 return cir::CastOp::create(*
this, v.getLoc(), destType,
341 cir::CastKind::floating, v);
344 mlir::Value
createFSub(mlir::Location loc, mlir::Value lhs, mlir::Value rhs) {
349 return cir::BinOp::create(*
this, loc, cir::BinOpKind::Sub, lhs, rhs);
352 mlir::Value
createFAdd(mlir::Location loc, mlir::Value lhs, mlir::Value rhs) {
357 return cir::BinOp::create(*
this, loc, cir::BinOpKind::Add, lhs, rhs);
359 mlir::Value
createFMul(mlir::Location loc, mlir::Value lhs, mlir::Value rhs) {
364 return cir::BinOp::create(*
this, loc, cir::BinOpKind::Mul, lhs, rhs);
366 mlir::Value
createFDiv(mlir::Location loc, mlir::Value lhs, mlir::Value rhs) {
371 return cir::BinOp::create(*
this, loc, cir::BinOpKind::Div, lhs, rhs);
375 cir::PointerType destType,
bool isRefCast,
376 cir::DynamicCastInfoAttr info) {
378 isRefCast ? cir::DynamicCastKind::Ref : cir::DynamicCastKind::Ptr;
379 return cir::DynamicCastOp::create(*
this, loc, destType, castKind, src, info,
384 bool vtableUseRelativeLayout) {
388 return cir::DynamicCastOp::create(
389 *
this, loc, destTy, cir::DynamicCastKind::Ptr, src,
390 cir::DynamicCastInfoAttr{}, vtableUseRelativeLayout);
394 mlir::Type destType,
unsigned offset,
395 bool assumeNotNull) {
401 cir::BaseClassAddrOp::create(*
this, loc, ptrTy, addr.
getPointer(),
402 mlir::APInt(64, offset), assumeNotNull);
407 mlir::Value addr, uint64_t offset) {
408 return cir::VTTAddrPointOp::create(*
this, loc, retTy,
409 mlir::FlatSymbolRefAttr{}, addr, offset);
413 mlir::FlatSymbolRefAttr sym, uint64_t offset) {
414 return cir::VTTAddrPointOp::create(*
this, loc, retTy, sym, mlir::Value{},
421 mlir::Type destType) {
431 bool isVolatile =
false) {
433 return cir::LoadOp::create(*
this, loc, addr.
getPointer(),
false,
435 cir::MemOrderAttr{});
439 mlir::Value ptr, llvm::MaybeAlign align) {
440 if (ty != mlir::cast<cir::PointerType>(ptr.getType()).getPointee())
442 uint64_t alignment = align ? align->value() : 0;
444 return cir::LoadOp::create(*
this, loc, ptr,
false,
446 cir::MemOrderAttr{});
456 bool isVolatile =
false,
457 mlir::IntegerAttr align = {},
458 cir::MemOrderAttr order = {}) {
461 return CIRBaseBuilderTy::createStore(loc, val, dst.getPointer(), isVolatile,
468 auto srcPtrTy = mlir::cast<cir::PointerType>(value.getType());
469 auto srcComplexTy = mlir::cast<cir::ComplexType>(srcPtrTy.getPointee());
470 return cir::ComplexRealPtrOp::create(
471 *
this, loc,
getPointerTo(srcComplexTy.getElementType()), value);
483 auto srcPtrTy = mlir::cast<cir::PointerType>(value.getType());
484 auto srcComplexTy = mlir::cast<cir::ComplexType>(srcPtrTy.getPointee());
485 return cir::ComplexImagPtrOp::create(
486 *
this, loc,
getPointerTo(srcComplexTy.getElementType()), value);
498 mlir::Location arrayLocEnd, mlir::Value arrayPtr,
499 mlir::Type eltTy, mlir::Value idx,
518 [[nodiscard]] cir::GlobalOp
520 mlir::StringRef name, mlir::Type
type,
bool isConstant,
521 cir::GlobalLinkageKind linkage) {
523 std::string uniqueName;
524 if (
unsigned version = globalsVersioning[name.str()]++)
525 uniqueName = name.str() +
"." + std::to_string(version);
527 uniqueName = name.str();
533 return cir::StackSaveOp::create(*
this, loc, ty);
537 return cir::StackRestoreOp::create(*
this, loc, v);
541 Address dstAddr, mlir::Type storageType,
543 bool isLvalueVolatile,
bool useVolatile) {
549 useVolatile ? cir::IntType::get(storageType.getContext(),
552 return cir::SetBitfieldOp::create(
553 *
this, loc, resultType, dstAddr.
getPointer(), storageType, src,
559 Address addr, mlir::Type storageType,
561 bool isLvalueVolatile,
bool useVolatile) {
567 useVolatile ? cir::IntType::get(storageType.getContext(),
570 return cir::GetBitfieldOp::create(*
this, loc, resultType, addr.
getPointer(),
571 storageType, info.
name, info.
size, offset,
__device__ __2f16 float c
cir::PointerType getPointerTo(mlir::Type ty)
mlir::Value createPtrBitcast(mlir::Value src, mlir::Type newPointeeTy)
mlir::Value createIntCast(mlir::Value src, mlir::Type newTy)
mlir::Value createBitcast(mlir::Value src, mlir::Type newTy)
CIRBaseBuilderTy(mlir::MLIRContext &mlirContext)
cir::GlobalOp createGlobal(mlir::ModuleOp mlirModule, mlir::Location loc, mlir::StringRef name, mlir::Type type, bool isConstant, cir::GlobalLinkageKind linkage)
mlir::IntegerAttr getAlignmentAttr(clang::CharUnits alignment)
cir::ConstantOp getConstantInt(mlir::Location loc, mlir::Type ty, int64_t value)
cir::PointerType getVoidPtrTy(clang::LangAS langAS=clang::LangAS::Default)
mlir::Value getPointer() const
mlir::Type getElementType() const
clang::CharUnits getAlignment() const
cir::RecordType getCompleteNamedRecordType(llvm::ArrayRef< mlir::Type > members, bool packed, bool padded, llvm::StringRef name)
Get a CIR named record type.
cir::StackSaveOp createStackSave(mlir::Location loc, mlir::Type ty)
cir::IntType getUInt64Ty()
cir::IntType getUInt8Ty()
cir::TypeInfoAttr getTypeInfo(mlir::ArrayAttr fieldsAttr)
mlir::Value createComplexRealPtr(mlir::Location loc, mlir::Value value)
Create a cir.complex.real_ptr operation that derives a pointer to the real part of the complex value ...
cir::RecordType::RecordKind getRecordKind(const clang::TagTypeKind kind)
Get a CIR record kind from a AST declaration tag.
cir::IntType getSIntNTy(int n)
cir::ConstRecordAttr getAnonConstRecord(mlir::ArrayAttr arrayAttr, bool packed=false, bool padded=false, mlir::Type ty={})
cir::ConstantOp getSInt64(uint64_t c, mlir::Location loc)
cir::RecordType getIncompleteRecordTy(llvm::StringRef name, const clang::RecordDecl *rd)
Get an incomplete CIR struct type.
cir::ConstantOp getUInt32(uint32_t c, mlir::Location loc)
cir::GlobalOp createVersionedGlobal(mlir::ModuleOp module, mlir::Location loc, mlir::StringRef name, mlir::Type type, bool isConstant, cir::GlobalLinkageKind linkage)
Creates a versioned global variable.
cir::PointerType getUInt8PtrTy()
std::string getUniqueRecordName(const std::string &baseName)
mlir::Attribute getConstRecordOrZeroAttr(mlir::ArrayAttr arrayAttr, bool packed=false, bool padded=false, mlir::Type type={})
cir::RecordType getAnonRecordTy(llvm::ArrayRef< mlir::Type > members, bool packed=false, bool padded=false)
Get a CIR anonymous record type.
mlir::Value createVTTAddrPoint(mlir::Location loc, mlir::Type retTy, mlir::FlatSymbolRefAttr sym, uint64_t offset)
Address createBaseClassAddr(mlir::Location loc, Address addr, mlir::Type destType, unsigned offset, bool assumeNotNull)
mlir::Value createComplexImagPtr(mlir::Location loc, mlir::Value value)
Create a cir.complex.imag_ptr operation that derives a pointer to the imaginary part of the complex v...
mlir::Value maybeBuildArrayDecay(mlir::Location loc, mlir::Value arrayPtr, mlir::Type eltTy)
Returns a decayed pointer to the first element of the array pointed to by arrayPtr.
mlir::Attribute getString(llvm::StringRef str, mlir::Type eltTy, std::optional< size_t > size)
Get a cir::ConstArrayAttr for a string literal.
cir::LoadOp createAlignedLoad(mlir::Location loc, mlir::Type ty, mlir::Value ptr, llvm::MaybeAlign align)
cir::ConstantOp getConstFP(mlir::Location loc, mlir::Type t, llvm::APFloat fpVal)
mlir::Value createFloatingCast(mlir::Value v, mlir::Type destType)
cir::FuncType getFuncType(llvm::ArrayRef< mlir::Type > params, mlir::Type retTy, bool isVarArg=false)
mlir::Value createFDiv(mlir::Location loc, mlir::Value lhs, mlir::Value rhs)
std::string getUniqueAnonRecordName()
Address createElementBitCast(mlir::Location loc, Address addr, mlir::Type destType)
Cast the element type of the given address to a different type, preserving information like the align...
cir::IntType getSInt8Ty()
bool isInt16Ty(mlir::Type i)
bool isInt32Ty(mlir::Type i)
mlir::Value createDynCastToVoid(mlir::Location loc, mlir::Value src, bool vtableUseRelativeLayout)
cir::IntType getUInt16Ty()
mlir::Value createDynCast(mlir::Location loc, mlir::Value src, cir::PointerType destType, bool isRefCast, cir::DynamicCastInfoAttr info)
mlir::Value createGetBitfield(mlir::Location loc, mlir::Type resultType, Address addr, mlir::Type storageType, const CIRGenBitFieldInfo &info, bool isLvalueVolatile, bool useVolatile)
bool isNullValue(mlir::Attribute attr) const
bool isInt64Ty(mlir::Type i)
cir::StackRestoreOp createStackRestore(mlir::Location loc, mlir::Value v)
mlir::Value createSetBitfield(mlir::Location loc, mlir::Type resultType, Address dstAddr, mlir::Type storageType, mlir::Value src, const CIRGenBitFieldInfo &info, bool isLvalueVolatile, bool useVolatile)
Address createComplexRealPtr(mlir::Location loc, Address addr)
mlir::Value createFAdd(mlir::Location loc, mlir::Value lhs, mlir::Value rhs)
CIRGenBuilderTy(mlir::MLIRContext &mlirContext, const CIRGenTypeCache &tc)
cir::RecordType getCompleteRecordType(mlir::ArrayAttr fields, bool packed=false, bool padded=false, llvm::StringRef name="")
cir::ConstantOp getConstInt(mlir::Location loc, llvm::APSInt intVal)
cir::VoidType getVoidTy()
void computeGlobalViewIndicesFromFlatOffset(int64_t offset, mlir::Type ty, cir::CIRDataLayout layout, llvm::SmallVectorImpl< int64_t > &indices)
cir::IntType getUInt32Ty()
mlir::Type getPtrToVPtrType()
cir::IntType getSInt32Ty()
Address createComplexImagPtr(mlir::Location loc, Address addr)
mlir::Value createFMul(mlir::Location loc, mlir::Value lhs, mlir::Value rhs)
cir::IntType getSInt64Ty()
cir::ConstantOp getSInt32(int32_t c, mlir::Location loc)
mlir::Value createVTTAddrPoint(mlir::Location loc, mlir::Type retTy, mlir::Value addr, uint64_t offset)
cir::LoadOp createLoad(mlir::Location loc, Address addr, bool isVolatile=false)
cir::LongDoubleType getLongDoubleTy(const llvm::fltSemantics &format) const
cir::IntType getSInt16Ty()
cir::ConstArrayAttr getConstArray(mlir::Attribute attrs, cir::ArrayType arrayTy) const
mlir::Value createFSub(mlir::Location loc, mlir::Value lhs, mlir::Value rhs)
cir::IntType getUIntNTy(int n)
mlir::Value getArrayElement(mlir::Location arrayLocBegin, mlir::Location arrayLocEnd, mlir::Value arrayPtr, mlir::Type eltTy, mlir::Value idx, bool shouldDecay)
Create a cir.ptr_stride operation to get access to an array element.
cir::StoreOp createStore(mlir::Location loc, mlir::Value val, Address dst, bool isVolatile=false, mlir::IntegerAttr align={}, cir::MemOrderAttr order={})
bool isInt8Ty(mlir::Type i)
mlir::Value createNeg(mlir::Value value)
cir::LoadOp createAlignedLoad(mlir::Location loc, mlir::Type ty, mlir::Value ptr, clang::CharUnits align=clang::CharUnits::One())
CharUnits - This is an opaque type for sizes expressed in character units.
llvm::Align getAsAlign() const
getAsAlign - Returns Quantity as a valid llvm::Align, Beware llvm::Align assumes power of two 8-bit b...
static CharUnits One()
One - Construct a CharUnits quantity of one.
Represents a struct/union/class.
TagKind getTagKind() const
const internal::VariadicAllOfMatcher< Type > type
Matches Types in the clang AST.
TagTypeKind
The kind of a tag type.
@ Interface
The "__interface" keyword.
@ Struct
The "struct" keyword.
@ Class
The "class" keyword.
@ Union
The "union" keyword.
@ Enum
The "enum" keyword.
static bool metaDataNode()
static bool addressSpace()
static bool fpConstraints()
static bool astRecordDeclAttr()
static bool fastMathFlags()
Record with information about how a bitfield should be accessed.
unsigned offset
The offset within a contiguous run of bitfields that are represented as a single "field" within the c...
unsigned volatileStorageSize
The storage size in bits which should be used when accessing this bitfield.
unsigned size
The total size of the bit-field, in bits.
unsigned isSigned
Whether the bit-field is signed.
unsigned volatileOffset
The offset within a contiguous run of bitfields that are represented as a single "field" within the c...
llvm::StringRef name
The name of a bitfield.
This structure provides a set of types that are commonly used during IR emission.