1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306 |
- #pragma once
- #ifdef __GNUC__
- #pragma GCC diagnostic push
- #pragma GCC diagnostic ignored "-Wunused-parameter"
- #endif
- //===- llvm/Attributes.h - Container for Attributes -------------*- C++ -*-===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- //
- /// \file
- /// This file contains the simple types necessary to represent the
- /// attributes associated with functions and their calls.
- //
- //===----------------------------------------------------------------------===//
- #ifndef LLVM_IR_ATTRIBUTES_H
- #define LLVM_IR_ATTRIBUTES_H
- #include "llvm-c/Types.h"
- #include "llvm/ADT/ArrayRef.h"
- #include "llvm/ADT/BitmaskEnum.h"
- #include "llvm/ADT/SmallString.h"
- #include "llvm/ADT/StringRef.h"
- #include "llvm/Config/llvm-config.h"
- #include "llvm/Support/Alignment.h"
- #include "llvm/Support/CodeGen.h"
- #include "llvm/Support/PointerLikeTypeTraits.h"
- #include <bitset>
- #include <cassert>
- #include <cstdint>
- #include <optional>
- #include <set>
- #include <string>
- #include <utility>
- namespace llvm {
- class AttrBuilder;
- class AttributeMask;
- class AttributeImpl;
- class AttributeListImpl;
- class AttributeSetNode;
- class FoldingSetNodeID;
- class Function;
- class LLVMContext;
- class MemoryEffects;
- class Type;
- class raw_ostream;
- enum class AllocFnKind : uint64_t {
- Unknown = 0,
- Alloc = 1 << 0, // Allocator function returns a new allocation
- Realloc = 1 << 1, // Allocator function resizes the `allocptr` argument
- Free = 1 << 2, // Allocator function frees the `allocptr` argument
- Uninitialized = 1 << 3, // Allocator function returns uninitialized memory
- Zeroed = 1 << 4, // Allocator function returns zeroed memory
- Aligned = 1 << 5, // Allocator function aligns allocations per the
- // `allocalign` argument
- LLVM_MARK_AS_BITMASK_ENUM(/* LargestValue = */ Aligned)
- };
- //===----------------------------------------------------------------------===//
- /// \class
- /// Functions, function parameters, and return types can have attributes
- /// to indicate how they should be treated by optimizations and code
- /// generation. This class represents one of those attributes. It's light-weight
- /// and should be passed around by-value.
- class Attribute {
- public:
- /// This enumeration lists the attributes that can be associated with
- /// parameters, function results, or the function itself.
- ///
- /// Note: The `uwtable' attribute is about the ABI or the user mandating an
- /// entry in the unwind table. The `nounwind' attribute is about an exception
- /// passing by the function.
- ///
- /// In a theoretical system that uses tables for profiling and SjLj for
- /// exceptions, they would be fully independent. In a normal system that uses
- /// tables for both, the semantics are:
- ///
- /// nil = Needs an entry because an exception might pass by.
- /// nounwind = No need for an entry
- /// uwtable = Needs an entry because the ABI says so and because
- /// an exception might pass by.
- /// uwtable + nounwind = Needs an entry because the ABI says so.
- enum AttrKind {
- // IR-Level Attributes
- None, ///< No attributes have been set
- #define GET_ATTR_ENUM
- #include "llvm/IR/Attributes.inc"
- EndAttrKinds, ///< Sentinal value useful for loops
- EmptyKey, ///< Use as Empty key for DenseMap of AttrKind
- TombstoneKey, ///< Use as Tombstone key for DenseMap of AttrKind
- };
- static const unsigned NumIntAttrKinds = LastIntAttr - FirstIntAttr + 1;
- static const unsigned NumTypeAttrKinds = LastTypeAttr - FirstTypeAttr + 1;
- static bool isEnumAttrKind(AttrKind Kind) {
- return Kind >= FirstEnumAttr && Kind <= LastEnumAttr;
- }
- static bool isIntAttrKind(AttrKind Kind) {
- return Kind >= FirstIntAttr && Kind <= LastIntAttr;
- }
- static bool isTypeAttrKind(AttrKind Kind) {
- return Kind >= FirstTypeAttr && Kind <= LastTypeAttr;
- }
- static bool canUseAsFnAttr(AttrKind Kind);
- static bool canUseAsParamAttr(AttrKind Kind);
- static bool canUseAsRetAttr(AttrKind Kind);
- private:
- AttributeImpl *pImpl = nullptr;
- Attribute(AttributeImpl *A) : pImpl(A) {}
- public:
- Attribute() = default;
- //===--------------------------------------------------------------------===//
- // Attribute Construction
- //===--------------------------------------------------------------------===//
- /// Return a uniquified Attribute object.
- static Attribute get(LLVMContext &Context, AttrKind Kind, uint64_t Val = 0);
- static Attribute get(LLVMContext &Context, StringRef Kind,
- StringRef Val = StringRef());
- static Attribute get(LLVMContext &Context, AttrKind Kind, Type *Ty);
- /// Return a uniquified Attribute object that has the specific
- /// alignment set.
- static Attribute getWithAlignment(LLVMContext &Context, Align Alignment);
- static Attribute getWithStackAlignment(LLVMContext &Context, Align Alignment);
- static Attribute getWithDereferenceableBytes(LLVMContext &Context,
- uint64_t Bytes);
- static Attribute getWithDereferenceableOrNullBytes(LLVMContext &Context,
- uint64_t Bytes);
- static Attribute getWithAllocSizeArgs(
- LLVMContext &Context, unsigned ElemSizeArg,
- const std::optional<unsigned> &NumElemsArg);
- static Attribute getWithVScaleRangeArgs(LLVMContext &Context,
- unsigned MinValue, unsigned MaxValue);
- static Attribute getWithByValType(LLVMContext &Context, Type *Ty);
- static Attribute getWithStructRetType(LLVMContext &Context, Type *Ty);
- static Attribute getWithByRefType(LLVMContext &Context, Type *Ty);
- static Attribute getWithPreallocatedType(LLVMContext &Context, Type *Ty);
- static Attribute getWithInAllocaType(LLVMContext &Context, Type *Ty);
- static Attribute getWithUWTableKind(LLVMContext &Context, UWTableKind Kind);
- static Attribute getWithMemoryEffects(LLVMContext &Context, MemoryEffects ME);
- /// For a typed attribute, return the equivalent attribute with the type
- /// changed to \p ReplacementTy.
- Attribute getWithNewType(LLVMContext &Context, Type *ReplacementTy) {
- assert(isTypeAttribute() && "this requires a typed attribute");
- return get(Context, getKindAsEnum(), ReplacementTy);
- }
- static Attribute::AttrKind getAttrKindFromName(StringRef AttrName);
- static StringRef getNameFromAttrKind(Attribute::AttrKind AttrKind);
- /// Return true if the provided string matches the IR name of an attribute.
- /// example: "noalias" return true but not "NoAlias"
- static bool isExistingAttribute(StringRef Name);
- //===--------------------------------------------------------------------===//
- // Attribute Accessors
- //===--------------------------------------------------------------------===//
- /// Return true if the attribute is an Attribute::AttrKind type.
- bool isEnumAttribute() const;
- /// Return true if the attribute is an integer attribute.
- bool isIntAttribute() const;
- /// Return true if the attribute is a string (target-dependent)
- /// attribute.
- bool isStringAttribute() const;
- /// Return true if the attribute is a type attribute.
- bool isTypeAttribute() const;
- /// Return true if the attribute is any kind of attribute.
- bool isValid() const { return pImpl; }
- /// Return true if the attribute is present.
- bool hasAttribute(AttrKind Val) const;
- /// Return true if the target-dependent attribute is present.
- bool hasAttribute(StringRef Val) const;
- /// Return the attribute's kind as an enum (Attribute::AttrKind). This
- /// requires the attribute to be an enum, integer, or type attribute.
- Attribute::AttrKind getKindAsEnum() const;
- /// Return the attribute's value as an integer. This requires that the
- /// attribute be an integer attribute.
- uint64_t getValueAsInt() const;
- /// Return the attribute's value as a boolean. This requires that the
- /// attribute be a string attribute.
- bool getValueAsBool() const;
- /// Return the attribute's kind as a string. This requires the
- /// attribute to be a string attribute.
- StringRef getKindAsString() const;
- /// Return the attribute's value as a string. This requires the
- /// attribute to be a string attribute.
- StringRef getValueAsString() const;
- /// Return the attribute's value as a Type. This requires the attribute to be
- /// a type attribute.
- Type *getValueAsType() const;
- /// Returns the alignment field of an attribute as a byte alignment
- /// value.
- MaybeAlign getAlignment() const;
- /// Returns the stack alignment field of an attribute as a byte
- /// alignment value.
- MaybeAlign getStackAlignment() const;
- /// Returns the number of dereferenceable bytes from the
- /// dereferenceable attribute.
- uint64_t getDereferenceableBytes() const;
- /// Returns the number of dereferenceable_or_null bytes from the
- /// dereferenceable_or_null attribute.
- uint64_t getDereferenceableOrNullBytes() const;
- /// Returns the argument numbers for the allocsize attribute.
- std::pair<unsigned, std::optional<unsigned>> getAllocSizeArgs() const;
- /// Returns the minimum value for the vscale_range attribute.
- unsigned getVScaleRangeMin() const;
- /// Returns the maximum value for the vscale_range attribute or std::nullopt
- /// when unknown.
- std::optional<unsigned> getVScaleRangeMax() const;
- // Returns the unwind table kind.
- UWTableKind getUWTableKind() const;
- // Returns the allocator function kind.
- AllocFnKind getAllocKind() const;
- /// Returns memory effects.
- MemoryEffects getMemoryEffects() const;
- /// The Attribute is converted to a string of equivalent mnemonic. This
- /// is, presumably, for writing out the mnemonics for the assembly writer.
- std::string getAsString(bool InAttrGrp = false) const;
- /// Return true if this attribute belongs to the LLVMContext.
- bool hasParentContext(LLVMContext &C) const;
- /// Equality and non-equality operators.
- bool operator==(Attribute A) const { return pImpl == A.pImpl; }
- bool operator!=(Attribute A) const { return pImpl != A.pImpl; }
- /// Less-than operator. Useful for sorting the attributes list.
- bool operator<(Attribute A) const;
- void Profile(FoldingSetNodeID &ID) const;
- /// Return a raw pointer that uniquely identifies this attribute.
- void *getRawPointer() const {
- return pImpl;
- }
- /// Get an attribute from a raw pointer created by getRawPointer.
- static Attribute fromRawPointer(void *RawPtr) {
- return Attribute(reinterpret_cast<AttributeImpl*>(RawPtr));
- }
- };
- // Specialized opaque value conversions.
- inline LLVMAttributeRef wrap(Attribute Attr) {
- return reinterpret_cast<LLVMAttributeRef>(Attr.getRawPointer());
- }
- // Specialized opaque value conversions.
- inline Attribute unwrap(LLVMAttributeRef Attr) {
- return Attribute::fromRawPointer(Attr);
- }
- //===----------------------------------------------------------------------===//
- /// \class
- /// This class holds the attributes for a particular argument, parameter,
- /// function, or return value. It is an immutable value type that is cheap to
- /// copy. Adding and removing enum attributes is intended to be fast, but adding
- /// and removing string or integer attributes involves a FoldingSet lookup.
- class AttributeSet {
- friend AttributeListImpl;
- template <typename Ty, typename Enable> friend struct DenseMapInfo;
- // TODO: Extract AvailableAttrs from AttributeSetNode and store them here.
- // This will allow an efficient implementation of addAttribute and
- // removeAttribute for enum attrs.
- /// Private implementation pointer.
- AttributeSetNode *SetNode = nullptr;
- private:
- explicit AttributeSet(AttributeSetNode *ASN) : SetNode(ASN) {}
- public:
- /// AttributeSet is a trivially copyable value type.
- AttributeSet() = default;
- AttributeSet(const AttributeSet &) = default;
- ~AttributeSet() = default;
- static AttributeSet get(LLVMContext &C, const AttrBuilder &B);
- static AttributeSet get(LLVMContext &C, ArrayRef<Attribute> Attrs);
- bool operator==(const AttributeSet &O) const { return SetNode == O.SetNode; }
- bool operator!=(const AttributeSet &O) const { return !(*this == O); }
- /// Add an argument attribute. Returns a new set because attribute sets are
- /// immutable.
- [[nodiscard]] AttributeSet addAttribute(LLVMContext &C,
- Attribute::AttrKind Kind) const;
- /// Add a target-dependent attribute. Returns a new set because attribute sets
- /// are immutable.
- [[nodiscard]] AttributeSet addAttribute(LLVMContext &C, StringRef Kind,
- StringRef Value = StringRef()) const;
- /// Add attributes to the attribute set. Returns a new set because attribute
- /// sets are immutable.
- [[nodiscard]] AttributeSet addAttributes(LLVMContext &C,
- AttributeSet AS) const;
- /// Remove the specified attribute from this set. Returns a new set because
- /// attribute sets are immutable.
- [[nodiscard]] AttributeSet removeAttribute(LLVMContext &C,
- Attribute::AttrKind Kind) const;
- /// Remove the specified attribute from this set. Returns a new set because
- /// attribute sets are immutable.
- [[nodiscard]] AttributeSet removeAttribute(LLVMContext &C,
- StringRef Kind) const;
- /// Remove the specified attributes from this set. Returns a new set because
- /// attribute sets are immutable.
- [[nodiscard]] AttributeSet
- removeAttributes(LLVMContext &C, const AttributeMask &AttrsToRemove) const;
- /// Return the number of attributes in this set.
- unsigned getNumAttributes() const;
- /// Return true if attributes exists in this set.
- bool hasAttributes() const { return SetNode != nullptr; }
- /// Return true if the attribute exists in this set.
- bool hasAttribute(Attribute::AttrKind Kind) const;
- /// Return true if the attribute exists in this set.
- bool hasAttribute(StringRef Kind) const;
- /// Return the attribute object.
- Attribute getAttribute(Attribute::AttrKind Kind) const;
- /// Return the target-dependent attribute object.
- Attribute getAttribute(StringRef Kind) const;
- MaybeAlign getAlignment() const;
- MaybeAlign getStackAlignment() const;
- uint64_t getDereferenceableBytes() const;
- uint64_t getDereferenceableOrNullBytes() const;
- Type *getByValType() const;
- Type *getStructRetType() const;
- Type *getByRefType() const;
- Type *getPreallocatedType() const;
- Type *getInAllocaType() const;
- Type *getElementType() const;
- std::optional<std::pair<unsigned, std::optional<unsigned>>> getAllocSizeArgs()
- const;
- unsigned getVScaleRangeMin() const;
- std::optional<unsigned> getVScaleRangeMax() const;
- UWTableKind getUWTableKind() const;
- AllocFnKind getAllocKind() const;
- MemoryEffects getMemoryEffects() const;
- std::string getAsString(bool InAttrGrp = false) const;
- /// Return true if this attribute set belongs to the LLVMContext.
- bool hasParentContext(LLVMContext &C) const;
- using iterator = const Attribute *;
- iterator begin() const;
- iterator end() const;
- #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
- void dump() const;
- #endif
- };
- //===----------------------------------------------------------------------===//
- /// \class
- /// Provide DenseMapInfo for AttributeSet.
- template <> struct DenseMapInfo<AttributeSet, void> {
- static AttributeSet getEmptyKey() {
- auto Val = static_cast<uintptr_t>(-1);
- Val <<= PointerLikeTypeTraits<void *>::NumLowBitsAvailable;
- return AttributeSet(reinterpret_cast<AttributeSetNode *>(Val));
- }
- static AttributeSet getTombstoneKey() {
- auto Val = static_cast<uintptr_t>(-2);
- Val <<= PointerLikeTypeTraits<void *>::NumLowBitsAvailable;
- return AttributeSet(reinterpret_cast<AttributeSetNode *>(Val));
- }
- static unsigned getHashValue(AttributeSet AS) {
- return (unsigned((uintptr_t)AS.SetNode) >> 4) ^
- (unsigned((uintptr_t)AS.SetNode) >> 9);
- }
- static bool isEqual(AttributeSet LHS, AttributeSet RHS) { return LHS == RHS; }
- };
- //===----------------------------------------------------------------------===//
- /// \class
- /// This class holds the attributes for a function, its return value, and
- /// its parameters. You access the attributes for each of them via an index into
- /// the AttributeList object. The function attributes are at index
- /// `AttributeList::FunctionIndex', the return value is at index
- /// `AttributeList::ReturnIndex', and the attributes for the parameters start at
- /// index `AttributeList::FirstArgIndex'.
- class AttributeList {
- public:
- enum AttrIndex : unsigned {
- ReturnIndex = 0U,
- FunctionIndex = ~0U,
- FirstArgIndex = 1,
- };
- private:
- friend class AttrBuilder;
- friend class AttributeListImpl;
- friend class AttributeSet;
- friend class AttributeSetNode;
- template <typename Ty, typename Enable> friend struct DenseMapInfo;
- /// The attributes that we are managing. This can be null to represent
- /// the empty attributes list.
- AttributeListImpl *pImpl = nullptr;
- public:
- /// Create an AttributeList with the specified parameters in it.
- static AttributeList get(LLVMContext &C,
- ArrayRef<std::pair<unsigned, Attribute>> Attrs);
- static AttributeList get(LLVMContext &C,
- ArrayRef<std::pair<unsigned, AttributeSet>> Attrs);
- /// Create an AttributeList from attribute sets for a function, its
- /// return value, and all of its arguments.
- static AttributeList get(LLVMContext &C, AttributeSet FnAttrs,
- AttributeSet RetAttrs,
- ArrayRef<AttributeSet> ArgAttrs);
- private:
- explicit AttributeList(AttributeListImpl *LI) : pImpl(LI) {}
- static AttributeList getImpl(LLVMContext &C, ArrayRef<AttributeSet> AttrSets);
- AttributeList setAttributesAtIndex(LLVMContext &C, unsigned Index,
- AttributeSet Attrs) const;
- public:
- AttributeList() = default;
- //===--------------------------------------------------------------------===//
- // AttributeList Construction and Mutation
- //===--------------------------------------------------------------------===//
- /// Return an AttributeList with the specified parameters in it.
- static AttributeList get(LLVMContext &C, ArrayRef<AttributeList> Attrs);
- static AttributeList get(LLVMContext &C, unsigned Index,
- ArrayRef<Attribute::AttrKind> Kinds);
- static AttributeList get(LLVMContext &C, unsigned Index,
- ArrayRef<Attribute::AttrKind> Kinds,
- ArrayRef<uint64_t> Values);
- static AttributeList get(LLVMContext &C, unsigned Index,
- ArrayRef<StringRef> Kind);
- static AttributeList get(LLVMContext &C, unsigned Index,
- AttributeSet Attrs);
- static AttributeList get(LLVMContext &C, unsigned Index,
- const AttrBuilder &B);
- // TODO: remove non-AtIndex versions of these methods.
- /// Add an attribute to the attribute set at the given index.
- /// Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- addAttributeAtIndex(LLVMContext &C, unsigned Index,
- Attribute::AttrKind Kind) const;
- /// Add an attribute to the attribute set at the given index.
- /// Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- addAttributeAtIndex(LLVMContext &C, unsigned Index, StringRef Kind,
- StringRef Value = StringRef()) const;
- /// Add an attribute to the attribute set at the given index.
- /// Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- addAttributeAtIndex(LLVMContext &C, unsigned Index, Attribute A) const;
- /// Add attributes to the attribute set at the given index.
- /// Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList addAttributesAtIndex(LLVMContext &C,
- unsigned Index,
- const AttrBuilder &B) const;
- /// Add a function attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addFnAttribute(LLVMContext &C,
- Attribute::AttrKind Kind) const {
- return addAttributeAtIndex(C, FunctionIndex, Kind);
- }
- /// Add a function attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addFnAttribute(LLVMContext &C,
- Attribute Attr) const {
- return addAttributeAtIndex(C, FunctionIndex, Attr);
- }
- /// Add a function attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList
- addFnAttribute(LLVMContext &C, StringRef Kind,
- StringRef Value = StringRef()) const {
- return addAttributeAtIndex(C, FunctionIndex, Kind, Value);
- }
- /// Add function attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addFnAttributes(LLVMContext &C,
- const AttrBuilder &B) const {
- return addAttributesAtIndex(C, FunctionIndex, B);
- }
- /// Add a return value attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addRetAttribute(LLVMContext &C,
- Attribute::AttrKind Kind) const {
- return addAttributeAtIndex(C, ReturnIndex, Kind);
- }
- /// Add a return value attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addRetAttribute(LLVMContext &C,
- Attribute Attr) const {
- return addAttributeAtIndex(C, ReturnIndex, Attr);
- }
- /// Add a return value attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addRetAttributes(LLVMContext &C,
- const AttrBuilder &B) const {
- return addAttributesAtIndex(C, ReturnIndex, B);
- }
- /// Add an argument attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList
- addParamAttribute(LLVMContext &C, unsigned ArgNo,
- Attribute::AttrKind Kind) const {
- return addAttributeAtIndex(C, ArgNo + FirstArgIndex, Kind);
- }
- /// Add an argument attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList
- addParamAttribute(LLVMContext &C, unsigned ArgNo, StringRef Kind,
- StringRef Value = StringRef()) const {
- return addAttributeAtIndex(C, ArgNo + FirstArgIndex, Kind, Value);
- }
- /// Add an attribute to the attribute list at the given arg indices. Returns a
- /// new list because attribute lists are immutable.
- [[nodiscard]] AttributeList addParamAttribute(LLVMContext &C,
- ArrayRef<unsigned> ArgNos,
- Attribute A) const;
- /// Add an argument attribute to the list. Returns a new list because
- /// attribute lists are immutable.
- [[nodiscard]] AttributeList addParamAttributes(LLVMContext &C, unsigned ArgNo,
- const AttrBuilder &B) const {
- return addAttributesAtIndex(C, ArgNo + FirstArgIndex, B);
- }
- /// Remove the specified attribute at the specified index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeAttributeAtIndex(LLVMContext &C, unsigned Index,
- Attribute::AttrKind Kind) const;
- /// Remove the specified attribute at the specified index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeAttributeAtIndex(LLVMContext &C, unsigned Index, StringRef Kind) const;
- [[nodiscard]] AttributeList removeAttribute(LLVMContext &C, unsigned Index,
- StringRef Kind) const {
- return removeAttributeAtIndex(C, Index, Kind);
- }
- /// Remove the specified attributes at the specified index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeAttributesAtIndex(LLVMContext &C, unsigned Index,
- const AttributeMask &AttrsToRemove) const;
- /// Remove all attributes at the specified index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList removeAttributesAtIndex(LLVMContext &C,
- unsigned Index) const;
- /// Remove the specified attribute at the function index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeFnAttribute(LLVMContext &C, Attribute::AttrKind Kind) const {
- return removeAttributeAtIndex(C, FunctionIndex, Kind);
- }
- /// Remove the specified attribute at the function index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList removeFnAttribute(LLVMContext &C,
- StringRef Kind) const {
- return removeAttributeAtIndex(C, FunctionIndex, Kind);
- }
- /// Remove the specified attribute at the function index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeFnAttributes(LLVMContext &C, const AttributeMask &AttrsToRemove) const {
- return removeAttributesAtIndex(C, FunctionIndex, AttrsToRemove);
- }
- /// Remove the attributes at the function index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList removeFnAttributes(LLVMContext &C) const {
- return removeAttributesAtIndex(C, FunctionIndex);
- }
- /// Remove the specified attribute at the return value index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeRetAttribute(LLVMContext &C, Attribute::AttrKind Kind) const {
- return removeAttributeAtIndex(C, ReturnIndex, Kind);
- }
- /// Remove the specified attribute at the return value index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList removeRetAttribute(LLVMContext &C,
- StringRef Kind) const {
- return removeAttributeAtIndex(C, ReturnIndex, Kind);
- }
- /// Remove the specified attribute at the return value index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeRetAttributes(LLVMContext &C,
- const AttributeMask &AttrsToRemove) const {
- return removeAttributesAtIndex(C, ReturnIndex, AttrsToRemove);
- }
- /// Remove the specified attribute at the specified arg index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeParamAttribute(LLVMContext &C, unsigned ArgNo,
- Attribute::AttrKind Kind) const {
- return removeAttributeAtIndex(C, ArgNo + FirstArgIndex, Kind);
- }
- /// Remove the specified attribute at the specified arg index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeParamAttribute(LLVMContext &C, unsigned ArgNo, StringRef Kind) const {
- return removeAttributeAtIndex(C, ArgNo + FirstArgIndex, Kind);
- }
- /// Remove the specified attribute at the specified arg index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- removeParamAttributes(LLVMContext &C, unsigned ArgNo,
- const AttributeMask &AttrsToRemove) const {
- return removeAttributesAtIndex(C, ArgNo + FirstArgIndex, AttrsToRemove);
- }
- /// Remove all attributes at the specified arg index from this
- /// attribute list. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList removeParamAttributes(LLVMContext &C,
- unsigned ArgNo) const {
- return removeAttributesAtIndex(C, ArgNo + FirstArgIndex);
- }
- /// Replace the type contained by attribute \p AttrKind at index \p ArgNo wih
- /// \p ReplacementTy, preserving all other attributes.
- [[nodiscard]] AttributeList
- replaceAttributeTypeAtIndex(LLVMContext &C, unsigned ArgNo,
- Attribute::AttrKind Kind,
- Type *ReplacementTy) const {
- Attribute Attr = getAttributeAtIndex(ArgNo, Kind);
- auto Attrs = removeAttributeAtIndex(C, ArgNo, Kind);
- return Attrs.addAttributeAtIndex(C, ArgNo,
- Attr.getWithNewType(C, ReplacementTy));
- }
- /// \brief Add the dereferenceable attribute to the attribute set at the given
- /// index. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList addDereferenceableRetAttr(LLVMContext &C,
- uint64_t Bytes) const;
- /// \brief Add the dereferenceable attribute to the attribute set at the given
- /// arg index. Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList addDereferenceableParamAttr(LLVMContext &C,
- unsigned ArgNo,
- uint64_t Bytes) const;
- /// Add the dereferenceable_or_null attribute to the attribute set at
- /// the given arg index. Returns a new list because attribute lists are
- /// immutable.
- [[nodiscard]] AttributeList
- addDereferenceableOrNullParamAttr(LLVMContext &C, unsigned ArgNo,
- uint64_t Bytes) const;
- /// Add the allocsize attribute to the attribute set at the given arg index.
- /// Returns a new list because attribute lists are immutable.
- [[nodiscard]] AttributeList
- addAllocSizeParamAttr(LLVMContext &C, unsigned ArgNo, unsigned ElemSizeArg,
- const std::optional<unsigned> &NumElemsArg);
- //===--------------------------------------------------------------------===//
- // AttributeList Accessors
- //===--------------------------------------------------------------------===//
- /// The attributes for the specified index are returned.
- AttributeSet getAttributes(unsigned Index) const;
- /// The attributes for the argument or parameter at the given index are
- /// returned.
- AttributeSet getParamAttrs(unsigned ArgNo) const;
- /// The attributes for the ret value are returned.
- AttributeSet getRetAttrs() const;
- /// The function attributes are returned.
- AttributeSet getFnAttrs() const;
- /// Return true if the attribute exists at the given index.
- bool hasAttributeAtIndex(unsigned Index, Attribute::AttrKind Kind) const;
- /// Return true if the attribute exists at the given index.
- bool hasAttributeAtIndex(unsigned Index, StringRef Kind) const;
- /// Return true if attribute exists at the given index.
- bool hasAttributesAtIndex(unsigned Index) const;
- /// Return true if the attribute exists for the given argument
- bool hasParamAttr(unsigned ArgNo, Attribute::AttrKind Kind) const {
- return hasAttributeAtIndex(ArgNo + FirstArgIndex, Kind);
- }
- /// Return true if the attribute exists for the given argument
- bool hasParamAttr(unsigned ArgNo, StringRef Kind) const {
- return hasAttributeAtIndex(ArgNo + FirstArgIndex, Kind);
- }
- /// Return true if attributes exists for the given argument
- bool hasParamAttrs(unsigned ArgNo) const {
- return hasAttributesAtIndex(ArgNo + FirstArgIndex);
- }
- /// Return true if the attribute exists for the return value.
- bool hasRetAttr(Attribute::AttrKind Kind) const {
- return hasAttributeAtIndex(ReturnIndex, Kind);
- }
- /// Return true if the attribute exists for the return value.
- bool hasRetAttr(StringRef Kind) const {
- return hasAttributeAtIndex(ReturnIndex, Kind);
- }
- /// Return true if attributes exist for the return value.
- bool hasRetAttrs() const { return hasAttributesAtIndex(ReturnIndex); }
- /// Return true if the attribute exists for the function.
- bool hasFnAttr(Attribute::AttrKind Kind) const;
- /// Return true if the attribute exists for the function.
- bool hasFnAttr(StringRef Kind) const;
- /// Return true the attributes exist for the function.
- bool hasFnAttrs() const { return hasAttributesAtIndex(FunctionIndex); }
- /// Return true if the specified attribute is set for at least one
- /// parameter or for the return value. If Index is not nullptr, the index
- /// of a parameter with the specified attribute is provided.
- bool hasAttrSomewhere(Attribute::AttrKind Kind,
- unsigned *Index = nullptr) const;
- /// Return the attribute object that exists at the given index.
- Attribute getAttributeAtIndex(unsigned Index, Attribute::AttrKind Kind) const;
- /// Return the attribute object that exists at the given index.
- Attribute getAttributeAtIndex(unsigned Index, StringRef Kind) const;
- /// Return the attribute object that exists at the arg index.
- Attribute getParamAttr(unsigned ArgNo, Attribute::AttrKind Kind) const {
- return getAttributeAtIndex(ArgNo + FirstArgIndex, Kind);
- }
- /// Return the attribute object that exists at the given index.
- Attribute getParamAttr(unsigned ArgNo, StringRef Kind) const {
- return getAttributeAtIndex(ArgNo + FirstArgIndex, Kind);
- }
- /// Return the attribute object that exists for the function.
- Attribute getFnAttr(Attribute::AttrKind Kind) const {
- return getAttributeAtIndex(FunctionIndex, Kind);
- }
- /// Return the attribute object that exists for the function.
- Attribute getFnAttr(StringRef Kind) const {
- return getAttributeAtIndex(FunctionIndex, Kind);
- }
- /// Return the alignment of the return value.
- MaybeAlign getRetAlignment() const;
- /// Return the alignment for the specified function parameter.
- MaybeAlign getParamAlignment(unsigned ArgNo) const;
- /// Return the stack alignment for the specified function parameter.
- MaybeAlign getParamStackAlignment(unsigned ArgNo) const;
- /// Return the byval type for the specified function parameter.
- Type *getParamByValType(unsigned ArgNo) const;
- /// Return the sret type for the specified function parameter.
- Type *getParamStructRetType(unsigned ArgNo) const;
- /// Return the byref type for the specified function parameter.
- Type *getParamByRefType(unsigned ArgNo) const;
- /// Return the preallocated type for the specified function parameter.
- Type *getParamPreallocatedType(unsigned ArgNo) const;
- /// Return the inalloca type for the specified function parameter.
- Type *getParamInAllocaType(unsigned ArgNo) const;
- /// Return the elementtype type for the specified function parameter.
- Type *getParamElementType(unsigned ArgNo) const;
- /// Get the stack alignment of the function.
- MaybeAlign getFnStackAlignment() const;
- /// Get the stack alignment of the return value.
- MaybeAlign getRetStackAlignment() const;
- /// Get the number of dereferenceable bytes (or zero if unknown) of the return
- /// value.
- uint64_t getRetDereferenceableBytes() const;
- /// Get the number of dereferenceable bytes (or zero if unknown) of an arg.
- uint64_t getParamDereferenceableBytes(unsigned Index) const;
- /// Get the number of dereferenceable_or_null bytes (or zero if unknown) of
- /// the return value.
- uint64_t getRetDereferenceableOrNullBytes() const;
- /// Get the number of dereferenceable_or_null bytes (or zero if unknown) of an
- /// arg.
- uint64_t getParamDereferenceableOrNullBytes(unsigned ArgNo) const;
- /// Get the unwind table kind requested for the function.
- UWTableKind getUWTableKind() const;
- AllocFnKind getAllocKind() const;
- /// Returns memory effects of the function.
- MemoryEffects getMemoryEffects() const;
- /// Return the attributes at the index as a string.
- std::string getAsString(unsigned Index, bool InAttrGrp = false) const;
- /// Return true if this attribute list belongs to the LLVMContext.
- bool hasParentContext(LLVMContext &C) const;
- //===--------------------------------------------------------------------===//
- // AttributeList Introspection
- //===--------------------------------------------------------------------===//
- using iterator = const AttributeSet *;
- iterator begin() const;
- iterator end() const;
- unsigned getNumAttrSets() const;
- // Implementation of indexes(). Produces iterators that wrap an index. Mostly
- // to hide the awkwardness of unsigned wrapping when iterating over valid
- // indexes.
- struct index_iterator {
- unsigned NumAttrSets;
- index_iterator(int NumAttrSets) : NumAttrSets(NumAttrSets) {}
- struct int_wrapper {
- int_wrapper(unsigned i) : i(i) {}
- unsigned i;
- unsigned operator*() { return i; }
- bool operator!=(const int_wrapper &Other) { return i != Other.i; }
- int_wrapper &operator++() {
- // This is expected to undergo unsigned wrapping since FunctionIndex is
- // ~0 and that's where we start.
- ++i;
- return *this;
- }
- };
- int_wrapper begin() { return int_wrapper(AttributeList::FunctionIndex); }
- int_wrapper end() { return int_wrapper(NumAttrSets - 1); }
- };
- /// Use this to iterate over the valid attribute indexes.
- index_iterator indexes() const { return index_iterator(getNumAttrSets()); }
- /// operator==/!= - Provide equality predicates.
- bool operator==(const AttributeList &RHS) const { return pImpl == RHS.pImpl; }
- bool operator!=(const AttributeList &RHS) const { return pImpl != RHS.pImpl; }
- /// Return a raw pointer that uniquely identifies this attribute list.
- void *getRawPointer() const {
- return pImpl;
- }
- /// Return true if there are no attributes.
- bool isEmpty() const { return pImpl == nullptr; }
- void print(raw_ostream &O) const;
- void dump() const;
- };
- //===----------------------------------------------------------------------===//
- /// \class
- /// Provide DenseMapInfo for AttributeList.
- template <> struct DenseMapInfo<AttributeList, void> {
- static AttributeList getEmptyKey() {
- auto Val = static_cast<uintptr_t>(-1);
- Val <<= PointerLikeTypeTraits<void*>::NumLowBitsAvailable;
- return AttributeList(reinterpret_cast<AttributeListImpl *>(Val));
- }
- static AttributeList getTombstoneKey() {
- auto Val = static_cast<uintptr_t>(-2);
- Val <<= PointerLikeTypeTraits<void*>::NumLowBitsAvailable;
- return AttributeList(reinterpret_cast<AttributeListImpl *>(Val));
- }
- static unsigned getHashValue(AttributeList AS) {
- return (unsigned((uintptr_t)AS.pImpl) >> 4) ^
- (unsigned((uintptr_t)AS.pImpl) >> 9);
- }
- static bool isEqual(AttributeList LHS, AttributeList RHS) {
- return LHS == RHS;
- }
- };
- //===----------------------------------------------------------------------===//
- /// \class
- /// This class stores enough information to efficiently remove some attributes
- /// from an existing AttrBuilder, AttributeSet or AttributeList.
- class AttributeMask {
- std::bitset<Attribute::EndAttrKinds> Attrs;
- std::set<SmallString<32>, std::less<>> TargetDepAttrs;
- public:
- AttributeMask() = default;
- AttributeMask(const AttributeMask &) = delete;
- AttributeMask(AttributeMask &&) = default;
- AttributeMask(AttributeSet AS) {
- for (Attribute A : AS)
- addAttribute(A);
- }
- /// Add an attribute to the mask.
- AttributeMask &addAttribute(Attribute::AttrKind Val) {
- assert((unsigned)Val < Attribute::EndAttrKinds &&
- "Attribute out of range!");
- Attrs[Val] = true;
- return *this;
- }
- /// Add the Attribute object to the builder.
- AttributeMask &addAttribute(Attribute A) {
- if (A.isStringAttribute())
- addAttribute(A.getKindAsString());
- else
- addAttribute(A.getKindAsEnum());
- return *this;
- }
- /// Add the target-dependent attribute to the builder.
- AttributeMask &addAttribute(StringRef A) {
- TargetDepAttrs.insert(A);
- return *this;
- }
- /// Return true if the builder has the specified attribute.
- bool contains(Attribute::AttrKind A) const {
- assert((unsigned)A < Attribute::EndAttrKinds && "Attribute out of range!");
- return Attrs[A];
- }
- /// Return true if the builder has the specified target-dependent
- /// attribute.
- bool contains(StringRef A) const { return TargetDepAttrs.count(A); }
- /// Return true if the mask contains the specified attribute.
- bool contains(Attribute A) const {
- if (A.isStringAttribute())
- return contains(A.getKindAsString());
- return contains(A.getKindAsEnum());
- }
- };
- //===----------------------------------------------------------------------===//
- /// \class
- /// This class is used in conjunction with the Attribute::get method to
- /// create an Attribute object. The object itself is uniquified. The Builder's
- /// value, however, is not. So this can be used as a quick way to test for
- /// equality, presence of attributes, etc.
- class AttrBuilder {
- LLVMContext &Ctx;
- SmallVector<Attribute, 8> Attrs;
- public:
- AttrBuilder(LLVMContext &Ctx) : Ctx(Ctx) {}
- AttrBuilder(const AttrBuilder &) = delete;
- AttrBuilder(AttrBuilder &&) = default;
- AttrBuilder(LLVMContext &Ctx, const Attribute &A) : Ctx(Ctx) {
- addAttribute(A);
- }
- AttrBuilder(LLVMContext &Ctx, AttributeSet AS);
- void clear();
- /// Add an attribute to the builder.
- AttrBuilder &addAttribute(Attribute::AttrKind Val);
- /// Add the Attribute object to the builder.
- AttrBuilder &addAttribute(Attribute A);
- /// Add the target-dependent attribute to the builder.
- AttrBuilder &addAttribute(StringRef A, StringRef V = StringRef());
- /// Remove an attribute from the builder.
- AttrBuilder &removeAttribute(Attribute::AttrKind Val);
- /// Remove the target-dependent attribute from the builder.
- AttrBuilder &removeAttribute(StringRef A);
- /// Remove the target-dependent attribute from the builder.
- AttrBuilder &removeAttribute(Attribute A) {
- if (A.isStringAttribute())
- return removeAttribute(A.getKindAsString());
- else
- return removeAttribute(A.getKindAsEnum());
- }
- /// Add the attributes from the builder. Attributes in the passed builder
- /// overwrite attributes in this builder if they have the same key.
- AttrBuilder &merge(const AttrBuilder &B);
- /// Remove the attributes from the builder.
- AttrBuilder &remove(const AttributeMask &AM);
- /// Return true if the builder has any attribute that's in the
- /// specified builder.
- bool overlaps(const AttributeMask &AM) const;
- /// Return true if the builder has the specified attribute.
- bool contains(Attribute::AttrKind A) const;
- /// Return true if the builder has the specified target-dependent
- /// attribute.
- bool contains(StringRef A) const;
- /// Return true if the builder has IR-level attributes.
- bool hasAttributes() const { return !Attrs.empty(); }
- /// Return Attribute with the given Kind. The returned attribute will be
- /// invalid if the Kind is not present in the builder.
- Attribute getAttribute(Attribute::AttrKind Kind) const;
- /// Return Attribute with the given Kind. The returned attribute will be
- /// invalid if the Kind is not present in the builder.
- Attribute getAttribute(StringRef Kind) const;
- /// Return raw (possibly packed/encoded) value of integer attribute or
- /// std::nullopt if not set.
- std::optional<uint64_t> getRawIntAttr(Attribute::AttrKind Kind) const;
- /// Retrieve the alignment attribute, if it exists.
- MaybeAlign getAlignment() const {
- return MaybeAlign(getRawIntAttr(Attribute::Alignment).value_or(0));
- }
- /// Retrieve the stack alignment attribute, if it exists.
- MaybeAlign getStackAlignment() const {
- return MaybeAlign(getRawIntAttr(Attribute::StackAlignment).value_or(0));
- }
- /// Retrieve the number of dereferenceable bytes, if the
- /// dereferenceable attribute exists (zero is returned otherwise).
- uint64_t getDereferenceableBytes() const {
- return getRawIntAttr(Attribute::Dereferenceable).value_or(0);
- }
- /// Retrieve the number of dereferenceable_or_null bytes, if the
- /// dereferenceable_or_null attribute exists (zero is returned otherwise).
- uint64_t getDereferenceableOrNullBytes() const {
- return getRawIntAttr(Attribute::DereferenceableOrNull).value_or(0);
- }
- /// Retrieve type for the given type attribute.
- Type *getTypeAttr(Attribute::AttrKind Kind) const;
- /// Retrieve the byval type.
- Type *getByValType() const { return getTypeAttr(Attribute::ByVal); }
- /// Retrieve the sret type.
- Type *getStructRetType() const { return getTypeAttr(Attribute::StructRet); }
- /// Retrieve the byref type.
- Type *getByRefType() const { return getTypeAttr(Attribute::ByRef); }
- /// Retrieve the preallocated type.
- Type *getPreallocatedType() const {
- return getTypeAttr(Attribute::Preallocated);
- }
- /// Retrieve the inalloca type.
- Type *getInAllocaType() const { return getTypeAttr(Attribute::InAlloca); }
- /// Retrieve the allocsize args, or std::nullopt if the attribute does not
- /// exist.
- std::optional<std::pair<unsigned, std::optional<unsigned>>> getAllocSizeArgs()
- const;
- /// Add integer attribute with raw value (packed/encoded if necessary).
- AttrBuilder &addRawIntAttr(Attribute::AttrKind Kind, uint64_t Value);
- /// This turns an alignment into the form used internally in Attribute.
- /// This call has no effect if Align is not set.
- AttrBuilder &addAlignmentAttr(MaybeAlign Align);
- /// This turns an int alignment (which must be a power of 2) into the
- /// form used internally in Attribute.
- /// This call has no effect if Align is 0.
- /// Deprecated, use the version using a MaybeAlign.
- inline AttrBuilder &addAlignmentAttr(unsigned Align) {
- return addAlignmentAttr(MaybeAlign(Align));
- }
- /// This turns a stack alignment into the form used internally in Attribute.
- /// This call has no effect if Align is not set.
- AttrBuilder &addStackAlignmentAttr(MaybeAlign Align);
- /// This turns an int stack alignment (which must be a power of 2) into
- /// the form used internally in Attribute.
- /// This call has no effect if Align is 0.
- /// Deprecated, use the version using a MaybeAlign.
- inline AttrBuilder &addStackAlignmentAttr(unsigned Align) {
- return addStackAlignmentAttr(MaybeAlign(Align));
- }
- /// This turns the number of dereferenceable bytes into the form used
- /// internally in Attribute.
- AttrBuilder &addDereferenceableAttr(uint64_t Bytes);
- /// This turns the number of dereferenceable_or_null bytes into the
- /// form used internally in Attribute.
- AttrBuilder &addDereferenceableOrNullAttr(uint64_t Bytes);
- /// This turns one (or two) ints into the form used internally in Attribute.
- AttrBuilder &addAllocSizeAttr(unsigned ElemSizeArg,
- const std::optional<unsigned> &NumElemsArg);
- /// This turns two ints into the form used internally in Attribute.
- AttrBuilder &addVScaleRangeAttr(unsigned MinValue,
- std::optional<unsigned> MaxValue);
- /// Add a type attribute with the given type.
- AttrBuilder &addTypeAttr(Attribute::AttrKind Kind, Type *Ty);
- /// This turns a byval type into the form used internally in Attribute.
- AttrBuilder &addByValAttr(Type *Ty);
- /// This turns a sret type into the form used internally in Attribute.
- AttrBuilder &addStructRetAttr(Type *Ty);
- /// This turns a byref type into the form used internally in Attribute.
- AttrBuilder &addByRefAttr(Type *Ty);
- /// This turns a preallocated type into the form used internally in Attribute.
- AttrBuilder &addPreallocatedAttr(Type *Ty);
- /// This turns an inalloca type into the form used internally in Attribute.
- AttrBuilder &addInAllocaAttr(Type *Ty);
- /// Add an allocsize attribute, using the representation returned by
- /// Attribute.getIntValue().
- AttrBuilder &addAllocSizeAttrFromRawRepr(uint64_t RawAllocSizeRepr);
- /// Add a vscale_range attribute, using the representation returned by
- /// Attribute.getIntValue().
- AttrBuilder &addVScaleRangeAttrFromRawRepr(uint64_t RawVScaleRangeRepr);
- /// This turns the unwind table kind into the form used internally in
- /// Attribute.
- AttrBuilder &addUWTableAttr(UWTableKind Kind);
- // This turns the allocator kind into the form used internally in Attribute.
- AttrBuilder &addAllocKindAttr(AllocFnKind Kind);
- /// Add memory effect attribute.
- AttrBuilder &addMemoryAttr(MemoryEffects ME);
- ArrayRef<Attribute> attrs() const { return Attrs; }
- bool operator==(const AttrBuilder &B) const;
- bool operator!=(const AttrBuilder &B) const { return !(*this == B); }
- };
- namespace AttributeFuncs {
- enum AttributeSafetyKind : uint8_t {
- ASK_SAFE_TO_DROP = 1,
- ASK_UNSAFE_TO_DROP = 2,
- ASK_ALL = ASK_SAFE_TO_DROP | ASK_UNSAFE_TO_DROP,
- };
- /// Which attributes cannot be applied to a type. The argument \p ASK indicates,
- /// if only attributes that are known to be safely droppable are contained in
- /// the mask; only attributes that might be unsafe to drop (e.g., ABI-related
- /// attributes) are in the mask; or both.
- AttributeMask typeIncompatible(Type *Ty, AttributeSafetyKind ASK = ASK_ALL);
- /// Get param/return attributes which imply immediate undefined behavior if an
- /// invalid value is passed. For example, this includes noundef (where undef
- /// implies UB), but not nonnull (where null implies poison). It also does not
- /// include attributes like nocapture, which constrain the function
- /// implementation rather than the passed value.
- AttributeMask getUBImplyingAttributes();
- /// \returns Return true if the two functions have compatible target-independent
- /// attributes for inlining purposes.
- bool areInlineCompatible(const Function &Caller, const Function &Callee);
- /// Checks if there are any incompatible function attributes between
- /// \p A and \p B.
- ///
- /// \param [in] A - The first function to be compared with.
- /// \param [in] B - The second function to be compared with.
- /// \returns true if the functions have compatible attributes.
- bool areOutlineCompatible(const Function &A, const Function &B);
- /// Merge caller's and callee's attributes.
- void mergeAttributesForInlining(Function &Caller, const Function &Callee);
- /// Merges the functions attributes from \p ToMerge into function \p Base.
- ///
- /// \param [in,out] Base - The function being merged into.
- /// \param [in] ToMerge - The function to merge attributes from.
- void mergeAttributesForOutlining(Function &Base, const Function &ToMerge);
- /// Update min-legal-vector-width if it is in Attribute and less than Width.
- void updateMinLegalVectorWidthAttr(Function &Fn, uint64_t Width);
- } // end namespace AttributeFuncs
- } // end namespace llvm
- #endif // LLVM_IR_ATTRIBUTES_H
- #ifdef __GNUC__
- #pragma GCC diagnostic pop
- #endif
|