10b57cec5SDimitry Andric //===- Function.cpp - Implement the Global object classes -----------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // This file implements the Function class for the IR library. 100b57cec5SDimitry Andric // 110b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 120b57cec5SDimitry Andric 130b57cec5SDimitry Andric #include "llvm/IR/Function.h" 140b57cec5SDimitry Andric #include "SymbolTableListTraitsImpl.h" 150b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h" 160b57cec5SDimitry Andric #include "llvm/ADT/DenseSet.h" 170b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 180b57cec5SDimitry Andric #include "llvm/ADT/SmallString.h" 190b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h" 200b57cec5SDimitry Andric #include "llvm/ADT/StringExtras.h" 210b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 225ffd83dbSDimitry Andric #include "llvm/IR/AbstractCallSite.h" 230b57cec5SDimitry Andric #include "llvm/IR/Argument.h" 240b57cec5SDimitry Andric #include "llvm/IR/Attributes.h" 250b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h" 260b57cec5SDimitry Andric #include "llvm/IR/Constant.h" 27*0fca6ea1SDimitry Andric #include "llvm/IR/ConstantRange.h" 280b57cec5SDimitry Andric #include "llvm/IR/Constants.h" 290b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h" 300b57cec5SDimitry Andric #include "llvm/IR/GlobalValue.h" 310b57cec5SDimitry Andric #include "llvm/IR/InstIterator.h" 320b57cec5SDimitry Andric #include "llvm/IR/Instruction.h" 33fe6060f1SDimitry Andric #include "llvm/IR/IntrinsicInst.h" 340b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h" 35480093f4SDimitry Andric #include "llvm/IR/IntrinsicsAArch64.h" 36480093f4SDimitry Andric #include "llvm/IR/IntrinsicsAMDGPU.h" 37480093f4SDimitry Andric #include "llvm/IR/IntrinsicsARM.h" 38480093f4SDimitry Andric #include "llvm/IR/IntrinsicsBPF.h" 3981ad6265SDimitry Andric #include "llvm/IR/IntrinsicsDirectX.h" 40480093f4SDimitry Andric #include "llvm/IR/IntrinsicsHexagon.h" 415f757f3fSDimitry Andric #include "llvm/IR/IntrinsicsLoongArch.h" 42480093f4SDimitry Andric #include "llvm/IR/IntrinsicsMips.h" 43480093f4SDimitry Andric #include "llvm/IR/IntrinsicsNVPTX.h" 44480093f4SDimitry Andric #include "llvm/IR/IntrinsicsPowerPC.h" 45480093f4SDimitry Andric #include "llvm/IR/IntrinsicsR600.h" 46480093f4SDimitry Andric #include "llvm/IR/IntrinsicsRISCV.h" 47480093f4SDimitry Andric #include "llvm/IR/IntrinsicsS390.h" 48*0fca6ea1SDimitry Andric #include "llvm/IR/IntrinsicsSPIRV.h" 49e8d8bef9SDimitry Andric #include "llvm/IR/IntrinsicsVE.h" 50480093f4SDimitry Andric #include "llvm/IR/IntrinsicsWebAssembly.h" 51480093f4SDimitry Andric #include "llvm/IR/IntrinsicsX86.h" 52480093f4SDimitry Andric #include "llvm/IR/IntrinsicsXCore.h" 530b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h" 540b57cec5SDimitry Andric #include "llvm/IR/MDBuilder.h" 550b57cec5SDimitry Andric #include "llvm/IR/Metadata.h" 560b57cec5SDimitry Andric #include "llvm/IR/Module.h" 57fe6060f1SDimitry Andric #include "llvm/IR/Operator.h" 580b57cec5SDimitry Andric #include "llvm/IR/SymbolTableListTraits.h" 590b57cec5SDimitry Andric #include "llvm/IR/Type.h" 600b57cec5SDimitry Andric #include "llvm/IR/Use.h" 610b57cec5SDimitry Andric #include "llvm/IR/User.h" 620b57cec5SDimitry Andric #include "llvm/IR/Value.h" 630b57cec5SDimitry Andric #include "llvm/IR/ValueSymbolTable.h" 640b57cec5SDimitry Andric #include "llvm/Support/Casting.h" 65fe6060f1SDimitry Andric #include "llvm/Support/CommandLine.h" 660b57cec5SDimitry Andric #include "llvm/Support/Compiler.h" 670b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h" 68bdd1243dSDimitry Andric #include "llvm/Support/ModRef.h" 690b57cec5SDimitry Andric #include <cassert> 700b57cec5SDimitry Andric #include <cstddef> 710b57cec5SDimitry Andric #include <cstdint> 720b57cec5SDimitry Andric #include <cstring> 730b57cec5SDimitry Andric #include <string> 740b57cec5SDimitry Andric 750b57cec5SDimitry Andric using namespace llvm; 760b57cec5SDimitry Andric using ProfileCount = Function::ProfileCount; 770b57cec5SDimitry Andric 780b57cec5SDimitry Andric // Explicit instantiations of SymbolTableListTraits since some of the methods 790b57cec5SDimitry Andric // are not in the public header file... 800b57cec5SDimitry Andric template class llvm::SymbolTableListTraits<BasicBlock>; 810b57cec5SDimitry Andric 82*0fca6ea1SDimitry Andric static cl::opt<int> NonGlobalValueMaxNameSize( 83fe6060f1SDimitry Andric "non-global-value-max-name-size", cl::Hidden, cl::init(1024), 84fe6060f1SDimitry Andric cl::desc("Maximum size for the name of non-global values.")); 85fe6060f1SDimitry Andric 86*0fca6ea1SDimitry Andric extern cl::opt<bool> UseNewDbgInfoFormat; 87*0fca6ea1SDimitry Andric 885f757f3fSDimitry Andric void Function::convertToNewDbgValues() { 895f757f3fSDimitry Andric IsNewDbgInfoFormat = true; 905f757f3fSDimitry Andric for (auto &BB : *this) { 915f757f3fSDimitry Andric BB.convertToNewDbgValues(); 925f757f3fSDimitry Andric } 935f757f3fSDimitry Andric } 945f757f3fSDimitry Andric 955f757f3fSDimitry Andric void Function::convertFromNewDbgValues() { 965f757f3fSDimitry Andric IsNewDbgInfoFormat = false; 975f757f3fSDimitry Andric for (auto &BB : *this) { 985f757f3fSDimitry Andric BB.convertFromNewDbgValues(); 995f757f3fSDimitry Andric } 1005f757f3fSDimitry Andric } 1015f757f3fSDimitry Andric 1025f757f3fSDimitry Andric void Function::setIsNewDbgInfoFormat(bool NewFlag) { 1035f757f3fSDimitry Andric if (NewFlag && !IsNewDbgInfoFormat) 1045f757f3fSDimitry Andric convertToNewDbgValues(); 1055f757f3fSDimitry Andric else if (!NewFlag && IsNewDbgInfoFormat) 1065f757f3fSDimitry Andric convertFromNewDbgValues(); 1075f757f3fSDimitry Andric } 108*0fca6ea1SDimitry Andric void Function::setNewDbgInfoFormatFlag(bool NewFlag) { 109*0fca6ea1SDimitry Andric for (auto &BB : *this) { 110*0fca6ea1SDimitry Andric BB.setNewDbgInfoFormatFlag(NewFlag); 111*0fca6ea1SDimitry Andric } 112*0fca6ea1SDimitry Andric IsNewDbgInfoFormat = NewFlag; 113*0fca6ea1SDimitry Andric } 1145f757f3fSDimitry Andric 1150b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1160b57cec5SDimitry Andric // Argument Implementation 1170b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 1180b57cec5SDimitry Andric 1190b57cec5SDimitry Andric Argument::Argument(Type *Ty, const Twine &Name, Function *Par, unsigned ArgNo) 1200b57cec5SDimitry Andric : Value(Ty, Value::ArgumentVal), Parent(Par), ArgNo(ArgNo) { 1210b57cec5SDimitry Andric setName(Name); 1220b57cec5SDimitry Andric } 1230b57cec5SDimitry Andric 1240b57cec5SDimitry Andric void Argument::setParent(Function *parent) { 1250b57cec5SDimitry Andric Parent = parent; 1260b57cec5SDimitry Andric } 1270b57cec5SDimitry Andric 128e8d8bef9SDimitry Andric bool Argument::hasNonNullAttr(bool AllowUndefOrPoison) const { 1290b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 130e8d8bef9SDimitry Andric if (getParent()->hasParamAttribute(getArgNo(), Attribute::NonNull) && 131e8d8bef9SDimitry Andric (AllowUndefOrPoison || 132e8d8bef9SDimitry Andric getParent()->hasParamAttribute(getArgNo(), Attribute::NoUndef))) 1330b57cec5SDimitry Andric return true; 1340b57cec5SDimitry Andric else if (getDereferenceableBytes() > 0 && 1350b57cec5SDimitry Andric !NullPointerIsDefined(getParent(), 1360b57cec5SDimitry Andric getType()->getPointerAddressSpace())) 1370b57cec5SDimitry Andric return true; 1380b57cec5SDimitry Andric return false; 1390b57cec5SDimitry Andric } 1400b57cec5SDimitry Andric 1410b57cec5SDimitry Andric bool Argument::hasByValAttr() const { 1420b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 1430b57cec5SDimitry Andric return hasAttribute(Attribute::ByVal); 1440b57cec5SDimitry Andric } 1450b57cec5SDimitry Andric 146e8d8bef9SDimitry Andric bool Argument::hasByRefAttr() const { 147e8d8bef9SDimitry Andric if (!getType()->isPointerTy()) 148e8d8bef9SDimitry Andric return false; 149e8d8bef9SDimitry Andric return hasAttribute(Attribute::ByRef); 150e8d8bef9SDimitry Andric } 151e8d8bef9SDimitry Andric 1520b57cec5SDimitry Andric bool Argument::hasSwiftSelfAttr() const { 1530b57cec5SDimitry Andric return getParent()->hasParamAttribute(getArgNo(), Attribute::SwiftSelf); 1540b57cec5SDimitry Andric } 1550b57cec5SDimitry Andric 1560b57cec5SDimitry Andric bool Argument::hasSwiftErrorAttr() const { 1570b57cec5SDimitry Andric return getParent()->hasParamAttribute(getArgNo(), Attribute::SwiftError); 1580b57cec5SDimitry Andric } 1590b57cec5SDimitry Andric 1600b57cec5SDimitry Andric bool Argument::hasInAllocaAttr() const { 1610b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 1620b57cec5SDimitry Andric return hasAttribute(Attribute::InAlloca); 1630b57cec5SDimitry Andric } 1640b57cec5SDimitry Andric 1655ffd83dbSDimitry Andric bool Argument::hasPreallocatedAttr() const { 1665ffd83dbSDimitry Andric if (!getType()->isPointerTy()) 1675ffd83dbSDimitry Andric return false; 1685ffd83dbSDimitry Andric return hasAttribute(Attribute::Preallocated); 1695ffd83dbSDimitry Andric } 1705ffd83dbSDimitry Andric 171e8d8bef9SDimitry Andric bool Argument::hasPassPointeeByValueCopyAttr() const { 1720b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 1730b57cec5SDimitry Andric AttributeList Attrs = getParent()->getAttributes(); 174349cc55cSDimitry Andric return Attrs.hasParamAttr(getArgNo(), Attribute::ByVal) || 175349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::InAlloca) || 176349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::Preallocated); 1775ffd83dbSDimitry Andric } 1785ffd83dbSDimitry Andric 179e8d8bef9SDimitry Andric bool Argument::hasPointeeInMemoryValueAttr() const { 180e8d8bef9SDimitry Andric if (!getType()->isPointerTy()) 181e8d8bef9SDimitry Andric return false; 182e8d8bef9SDimitry Andric AttributeList Attrs = getParent()->getAttributes(); 183349cc55cSDimitry Andric return Attrs.hasParamAttr(getArgNo(), Attribute::ByVal) || 184349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::StructRet) || 185349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::InAlloca) || 186349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::Preallocated) || 187349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::ByRef); 188e8d8bef9SDimitry Andric } 1895ffd83dbSDimitry Andric 190e8d8bef9SDimitry Andric /// For a byval, sret, inalloca, or preallocated parameter, get the in-memory 191e8d8bef9SDimitry Andric /// parameter type. 192349cc55cSDimitry Andric static Type *getMemoryParamAllocType(AttributeSet ParamAttrs) { 1935ffd83dbSDimitry Andric // FIXME: All the type carrying attributes are mutually exclusive, so there 1945ffd83dbSDimitry Andric // should be a single query to get the stored type that handles any of them. 1955ffd83dbSDimitry Andric if (Type *ByValTy = ParamAttrs.getByValType()) 196e8d8bef9SDimitry Andric return ByValTy; 197e8d8bef9SDimitry Andric if (Type *ByRefTy = ParamAttrs.getByRefType()) 198e8d8bef9SDimitry Andric return ByRefTy; 1995ffd83dbSDimitry Andric if (Type *PreAllocTy = ParamAttrs.getPreallocatedType()) 200e8d8bef9SDimitry Andric return PreAllocTy; 201fe6060f1SDimitry Andric if (Type *InAllocaTy = ParamAttrs.getInAllocaType()) 202fe6060f1SDimitry Andric return InAllocaTy; 203fe6060f1SDimitry Andric if (Type *SRetTy = ParamAttrs.getStructRetType()) 204fe6060f1SDimitry Andric return SRetTy; 2055ffd83dbSDimitry Andric 206e8d8bef9SDimitry Andric return nullptr; 207e8d8bef9SDimitry Andric } 208e8d8bef9SDimitry Andric 209e8d8bef9SDimitry Andric uint64_t Argument::getPassPointeeByValueCopySize(const DataLayout &DL) const { 210e8d8bef9SDimitry Andric AttributeSet ParamAttrs = 211349cc55cSDimitry Andric getParent()->getAttributes().getParamAttrs(getArgNo()); 212349cc55cSDimitry Andric if (Type *MemTy = getMemoryParamAllocType(ParamAttrs)) 213e8d8bef9SDimitry Andric return DL.getTypeAllocSize(MemTy); 2145ffd83dbSDimitry Andric return 0; 2150b57cec5SDimitry Andric } 2160b57cec5SDimitry Andric 217e8d8bef9SDimitry Andric Type *Argument::getPointeeInMemoryValueType() const { 218e8d8bef9SDimitry Andric AttributeSet ParamAttrs = 219349cc55cSDimitry Andric getParent()->getAttributes().getParamAttrs(getArgNo()); 220349cc55cSDimitry Andric return getMemoryParamAllocType(ParamAttrs); 221e8d8bef9SDimitry Andric } 222e8d8bef9SDimitry Andric 223480093f4SDimitry Andric MaybeAlign Argument::getParamAlign() const { 224480093f4SDimitry Andric assert(getType()->isPointerTy() && "Only pointers have alignments"); 225480093f4SDimitry Andric return getParent()->getParamAlign(getArgNo()); 226480093f4SDimitry Andric } 227480093f4SDimitry Andric 228fe6060f1SDimitry Andric MaybeAlign Argument::getParamStackAlign() const { 229fe6060f1SDimitry Andric return getParent()->getParamStackAlign(getArgNo()); 230fe6060f1SDimitry Andric } 231fe6060f1SDimitry Andric 2320b57cec5SDimitry Andric Type *Argument::getParamByValType() const { 2330b57cec5SDimitry Andric assert(getType()->isPointerTy() && "Only pointers have byval types"); 2340b57cec5SDimitry Andric return getParent()->getParamByValType(getArgNo()); 2350b57cec5SDimitry Andric } 2360b57cec5SDimitry Andric 237e8d8bef9SDimitry Andric Type *Argument::getParamStructRetType() const { 238e8d8bef9SDimitry Andric assert(getType()->isPointerTy() && "Only pointers have sret types"); 239e8d8bef9SDimitry Andric return getParent()->getParamStructRetType(getArgNo()); 240e8d8bef9SDimitry Andric } 241e8d8bef9SDimitry Andric 242e8d8bef9SDimitry Andric Type *Argument::getParamByRefType() const { 243fe6060f1SDimitry Andric assert(getType()->isPointerTy() && "Only pointers have byref types"); 244e8d8bef9SDimitry Andric return getParent()->getParamByRefType(getArgNo()); 245e8d8bef9SDimitry Andric } 246e8d8bef9SDimitry Andric 247fe6060f1SDimitry Andric Type *Argument::getParamInAllocaType() const { 248fe6060f1SDimitry Andric assert(getType()->isPointerTy() && "Only pointers have inalloca types"); 249fe6060f1SDimitry Andric return getParent()->getParamInAllocaType(getArgNo()); 250fe6060f1SDimitry Andric } 251fe6060f1SDimitry Andric 2520b57cec5SDimitry Andric uint64_t Argument::getDereferenceableBytes() const { 2530b57cec5SDimitry Andric assert(getType()->isPointerTy() && 2540b57cec5SDimitry Andric "Only pointers have dereferenceable bytes"); 2550b57cec5SDimitry Andric return getParent()->getParamDereferenceableBytes(getArgNo()); 2560b57cec5SDimitry Andric } 2570b57cec5SDimitry Andric 2580b57cec5SDimitry Andric uint64_t Argument::getDereferenceableOrNullBytes() const { 2590b57cec5SDimitry Andric assert(getType()->isPointerTy() && 2600b57cec5SDimitry Andric "Only pointers have dereferenceable bytes"); 2610b57cec5SDimitry Andric return getParent()->getParamDereferenceableOrNullBytes(getArgNo()); 2620b57cec5SDimitry Andric } 2630b57cec5SDimitry Andric 26406c3fb27SDimitry Andric FPClassTest Argument::getNoFPClass() const { 26506c3fb27SDimitry Andric return getParent()->getParamNoFPClass(getArgNo()); 26606c3fb27SDimitry Andric } 26706c3fb27SDimitry Andric 268*0fca6ea1SDimitry Andric std::optional<ConstantRange> Argument::getRange() const { 269*0fca6ea1SDimitry Andric const Attribute RangeAttr = getAttribute(llvm::Attribute::Range); 270*0fca6ea1SDimitry Andric if (RangeAttr.isValid()) 271*0fca6ea1SDimitry Andric return RangeAttr.getRange(); 272*0fca6ea1SDimitry Andric return std::nullopt; 273*0fca6ea1SDimitry Andric } 274*0fca6ea1SDimitry Andric 2750b57cec5SDimitry Andric bool Argument::hasNestAttr() const { 2760b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 2770b57cec5SDimitry Andric return hasAttribute(Attribute::Nest); 2780b57cec5SDimitry Andric } 2790b57cec5SDimitry Andric 2800b57cec5SDimitry Andric bool Argument::hasNoAliasAttr() const { 2810b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 2820b57cec5SDimitry Andric return hasAttribute(Attribute::NoAlias); 2830b57cec5SDimitry Andric } 2840b57cec5SDimitry Andric 2850b57cec5SDimitry Andric bool Argument::hasNoCaptureAttr() const { 2860b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 2870b57cec5SDimitry Andric return hasAttribute(Attribute::NoCapture); 2880b57cec5SDimitry Andric } 2890b57cec5SDimitry Andric 290fe6060f1SDimitry Andric bool Argument::hasNoFreeAttr() const { 291fe6060f1SDimitry Andric if (!getType()->isPointerTy()) return false; 292fe6060f1SDimitry Andric return hasAttribute(Attribute::NoFree); 293fe6060f1SDimitry Andric } 294fe6060f1SDimitry Andric 2950b57cec5SDimitry Andric bool Argument::hasStructRetAttr() const { 2960b57cec5SDimitry Andric if (!getType()->isPointerTy()) return false; 2970b57cec5SDimitry Andric return hasAttribute(Attribute::StructRet); 2980b57cec5SDimitry Andric } 2990b57cec5SDimitry Andric 3000b57cec5SDimitry Andric bool Argument::hasInRegAttr() const { 3010b57cec5SDimitry Andric return hasAttribute(Attribute::InReg); 3020b57cec5SDimitry Andric } 3030b57cec5SDimitry Andric 3040b57cec5SDimitry Andric bool Argument::hasReturnedAttr() const { 3050b57cec5SDimitry Andric return hasAttribute(Attribute::Returned); 3060b57cec5SDimitry Andric } 3070b57cec5SDimitry Andric 3080b57cec5SDimitry Andric bool Argument::hasZExtAttr() const { 3090b57cec5SDimitry Andric return hasAttribute(Attribute::ZExt); 3100b57cec5SDimitry Andric } 3110b57cec5SDimitry Andric 3120b57cec5SDimitry Andric bool Argument::hasSExtAttr() const { 3130b57cec5SDimitry Andric return hasAttribute(Attribute::SExt); 3140b57cec5SDimitry Andric } 3150b57cec5SDimitry Andric 3160b57cec5SDimitry Andric bool Argument::onlyReadsMemory() const { 3170b57cec5SDimitry Andric AttributeList Attrs = getParent()->getAttributes(); 318349cc55cSDimitry Andric return Attrs.hasParamAttr(getArgNo(), Attribute::ReadOnly) || 319349cc55cSDimitry Andric Attrs.hasParamAttr(getArgNo(), Attribute::ReadNone); 3200b57cec5SDimitry Andric } 3210b57cec5SDimitry Andric 3220b57cec5SDimitry Andric void Argument::addAttrs(AttrBuilder &B) { 3230b57cec5SDimitry Andric AttributeList AL = getParent()->getAttributes(); 3240b57cec5SDimitry Andric AL = AL.addParamAttributes(Parent->getContext(), getArgNo(), B); 3250b57cec5SDimitry Andric getParent()->setAttributes(AL); 3260b57cec5SDimitry Andric } 3270b57cec5SDimitry Andric 3280b57cec5SDimitry Andric void Argument::addAttr(Attribute::AttrKind Kind) { 3290b57cec5SDimitry Andric getParent()->addParamAttr(getArgNo(), Kind); 3300b57cec5SDimitry Andric } 3310b57cec5SDimitry Andric 3320b57cec5SDimitry Andric void Argument::addAttr(Attribute Attr) { 3330b57cec5SDimitry Andric getParent()->addParamAttr(getArgNo(), Attr); 3340b57cec5SDimitry Andric } 3350b57cec5SDimitry Andric 3360b57cec5SDimitry Andric void Argument::removeAttr(Attribute::AttrKind Kind) { 3370b57cec5SDimitry Andric getParent()->removeParamAttr(getArgNo(), Kind); 3380b57cec5SDimitry Andric } 3390b57cec5SDimitry Andric 34004eeddc0SDimitry Andric void Argument::removeAttrs(const AttributeMask &AM) { 341fe6060f1SDimitry Andric AttributeList AL = getParent()->getAttributes(); 34204eeddc0SDimitry Andric AL = AL.removeParamAttributes(Parent->getContext(), getArgNo(), AM); 343fe6060f1SDimitry Andric getParent()->setAttributes(AL); 344fe6060f1SDimitry Andric } 345fe6060f1SDimitry Andric 3460b57cec5SDimitry Andric bool Argument::hasAttribute(Attribute::AttrKind Kind) const { 3470b57cec5SDimitry Andric return getParent()->hasParamAttribute(getArgNo(), Kind); 3480b57cec5SDimitry Andric } 3490b57cec5SDimitry Andric 3500b57cec5SDimitry Andric Attribute Argument::getAttribute(Attribute::AttrKind Kind) const { 3510b57cec5SDimitry Andric return getParent()->getParamAttribute(getArgNo(), Kind); 3520b57cec5SDimitry Andric } 3530b57cec5SDimitry Andric 3540b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 3550b57cec5SDimitry Andric // Helper Methods in Function 3560b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 3570b57cec5SDimitry Andric 3580b57cec5SDimitry Andric LLVMContext &Function::getContext() const { 3590b57cec5SDimitry Andric return getType()->getContext(); 3600b57cec5SDimitry Andric } 3610b57cec5SDimitry Andric 362*0fca6ea1SDimitry Andric const DataLayout &Function::getDataLayout() const { 363*0fca6ea1SDimitry Andric return getParent()->getDataLayout(); 364*0fca6ea1SDimitry Andric } 365*0fca6ea1SDimitry Andric 3660b57cec5SDimitry Andric unsigned Function::getInstructionCount() const { 3670b57cec5SDimitry Andric unsigned NumInstrs = 0; 3680b57cec5SDimitry Andric for (const BasicBlock &BB : BasicBlocks) 3690b57cec5SDimitry Andric NumInstrs += std::distance(BB.instructionsWithoutDebug().begin(), 3700b57cec5SDimitry Andric BB.instructionsWithoutDebug().end()); 3710b57cec5SDimitry Andric return NumInstrs; 3720b57cec5SDimitry Andric } 3730b57cec5SDimitry Andric 3740b57cec5SDimitry Andric Function *Function::Create(FunctionType *Ty, LinkageTypes Linkage, 3750b57cec5SDimitry Andric const Twine &N, Module &M) { 3760b57cec5SDimitry Andric return Create(Ty, Linkage, M.getDataLayout().getProgramAddressSpace(), N, &M); 3770b57cec5SDimitry Andric } 3780b57cec5SDimitry Andric 379fe6060f1SDimitry Andric Function *Function::createWithDefaultAttr(FunctionType *Ty, 380fe6060f1SDimitry Andric LinkageTypes Linkage, 381fe6060f1SDimitry Andric unsigned AddrSpace, const Twine &N, 382fe6060f1SDimitry Andric Module *M) { 383fe6060f1SDimitry Andric auto *F = new Function(Ty, Linkage, AddrSpace, N, M); 38404eeddc0SDimitry Andric AttrBuilder B(F->getContext()); 38581ad6265SDimitry Andric UWTableKind UWTable = M->getUwtable(); 38681ad6265SDimitry Andric if (UWTable != UWTableKind::None) 38781ad6265SDimitry Andric B.addUWTableAttr(UWTable); 388fe6060f1SDimitry Andric switch (M->getFramePointer()) { 389fe6060f1SDimitry Andric case FramePointerKind::None: 390fe6060f1SDimitry Andric // 0 ("none") is the default. 391fe6060f1SDimitry Andric break; 392*0fca6ea1SDimitry Andric case FramePointerKind::Reserved: 393*0fca6ea1SDimitry Andric B.addAttribute("frame-pointer", "reserved"); 394*0fca6ea1SDimitry Andric break; 395fe6060f1SDimitry Andric case FramePointerKind::NonLeaf: 396fe6060f1SDimitry Andric B.addAttribute("frame-pointer", "non-leaf"); 397fe6060f1SDimitry Andric break; 398fe6060f1SDimitry Andric case FramePointerKind::All: 399fe6060f1SDimitry Andric B.addAttribute("frame-pointer", "all"); 400fe6060f1SDimitry Andric break; 401fe6060f1SDimitry Andric } 402fcaf7f86SDimitry Andric if (M->getModuleFlag("function_return_thunk_extern")) 403fcaf7f86SDimitry Andric B.addAttribute(Attribute::FnRetThunkExtern); 404*0fca6ea1SDimitry Andric StringRef DefaultCPU = F->getContext().getDefaultTargetCPU(); 405*0fca6ea1SDimitry Andric if (!DefaultCPU.empty()) 406*0fca6ea1SDimitry Andric B.addAttribute("target-cpu", DefaultCPU); 407*0fca6ea1SDimitry Andric StringRef DefaultFeatures = F->getContext().getDefaultTargetFeatures(); 408*0fca6ea1SDimitry Andric if (!DefaultFeatures.empty()) 409*0fca6ea1SDimitry Andric B.addAttribute("target-features", DefaultFeatures); 410*0fca6ea1SDimitry Andric 411*0fca6ea1SDimitry Andric // Check if the module attribute is present and not zero. 412*0fca6ea1SDimitry Andric auto isModuleAttributeSet = [&](const StringRef &ModAttr) -> bool { 413*0fca6ea1SDimitry Andric const auto *Attr = 414*0fca6ea1SDimitry Andric mdconst::extract_or_null<ConstantInt>(M->getModuleFlag(ModAttr)); 415*0fca6ea1SDimitry Andric return Attr && !Attr->isZero(); 416*0fca6ea1SDimitry Andric }; 417*0fca6ea1SDimitry Andric 418*0fca6ea1SDimitry Andric auto AddAttributeIfSet = [&](const StringRef &ModAttr) { 419*0fca6ea1SDimitry Andric if (isModuleAttributeSet(ModAttr)) 420*0fca6ea1SDimitry Andric B.addAttribute(ModAttr); 421*0fca6ea1SDimitry Andric }; 422*0fca6ea1SDimitry Andric 423*0fca6ea1SDimitry Andric StringRef SignType = "none"; 424*0fca6ea1SDimitry Andric if (isModuleAttributeSet("sign-return-address")) 425*0fca6ea1SDimitry Andric SignType = "non-leaf"; 426*0fca6ea1SDimitry Andric if (isModuleAttributeSet("sign-return-address-all")) 427*0fca6ea1SDimitry Andric SignType = "all"; 428*0fca6ea1SDimitry Andric if (SignType != "none") { 429*0fca6ea1SDimitry Andric B.addAttribute("sign-return-address", SignType); 430*0fca6ea1SDimitry Andric B.addAttribute("sign-return-address-key", 431*0fca6ea1SDimitry Andric isModuleAttributeSet("sign-return-address-with-bkey") 432*0fca6ea1SDimitry Andric ? "b_key" 433*0fca6ea1SDimitry Andric : "a_key"); 434*0fca6ea1SDimitry Andric } 435*0fca6ea1SDimitry Andric AddAttributeIfSet("branch-target-enforcement"); 436*0fca6ea1SDimitry Andric AddAttributeIfSet("branch-protection-pauth-lr"); 437*0fca6ea1SDimitry Andric AddAttributeIfSet("guarded-control-stack"); 438*0fca6ea1SDimitry Andric 439349cc55cSDimitry Andric F->addFnAttrs(B); 440fe6060f1SDimitry Andric return F; 441fe6060f1SDimitry Andric } 442fe6060f1SDimitry Andric 4430b57cec5SDimitry Andric void Function::removeFromParent() { 4440b57cec5SDimitry Andric getParent()->getFunctionList().remove(getIterator()); 4450b57cec5SDimitry Andric } 4460b57cec5SDimitry Andric 4470b57cec5SDimitry Andric void Function::eraseFromParent() { 4480b57cec5SDimitry Andric getParent()->getFunctionList().erase(getIterator()); 4490b57cec5SDimitry Andric } 4500b57cec5SDimitry Andric 451bdd1243dSDimitry Andric void Function::splice(Function::iterator ToIt, Function *FromF, 452bdd1243dSDimitry Andric Function::iterator FromBeginIt, 453bdd1243dSDimitry Andric Function::iterator FromEndIt) { 454bdd1243dSDimitry Andric #ifdef EXPENSIVE_CHECKS 455bdd1243dSDimitry Andric // Check that FromBeginIt is before FromEndIt. 456bdd1243dSDimitry Andric auto FromFEnd = FromF->end(); 457bdd1243dSDimitry Andric for (auto It = FromBeginIt; It != FromEndIt; ++It) 458bdd1243dSDimitry Andric assert(It != FromFEnd && "FromBeginIt not before FromEndIt!"); 459bdd1243dSDimitry Andric #endif // EXPENSIVE_CHECKS 460bdd1243dSDimitry Andric BasicBlocks.splice(ToIt, FromF->BasicBlocks, FromBeginIt, FromEndIt); 461bdd1243dSDimitry Andric } 462bdd1243dSDimitry Andric 463bdd1243dSDimitry Andric Function::iterator Function::erase(Function::iterator FromIt, 464bdd1243dSDimitry Andric Function::iterator ToIt) { 465bdd1243dSDimitry Andric return BasicBlocks.erase(FromIt, ToIt); 466bdd1243dSDimitry Andric } 467bdd1243dSDimitry Andric 4680b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 4690b57cec5SDimitry Andric // Function Implementation 4700b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 4710b57cec5SDimitry Andric 4720b57cec5SDimitry Andric static unsigned computeAddrSpace(unsigned AddrSpace, Module *M) { 4730b57cec5SDimitry Andric // If AS == -1 and we are passed a valid module pointer we place the function 4740b57cec5SDimitry Andric // in the program address space. Otherwise we default to AS0. 4750b57cec5SDimitry Andric if (AddrSpace == static_cast<unsigned>(-1)) 4760b57cec5SDimitry Andric return M ? M->getDataLayout().getProgramAddressSpace() : 0; 4770b57cec5SDimitry Andric return AddrSpace; 4780b57cec5SDimitry Andric } 4790b57cec5SDimitry Andric 4800b57cec5SDimitry Andric Function::Function(FunctionType *Ty, LinkageTypes Linkage, unsigned AddrSpace, 4810b57cec5SDimitry Andric const Twine &name, Module *ParentModule) 4820b57cec5SDimitry Andric : GlobalObject(Ty, Value::FunctionVal, 4830b57cec5SDimitry Andric OperandTraits<Function>::op_begin(this), 0, Linkage, name, 4840b57cec5SDimitry Andric computeAddrSpace(AddrSpace, ParentModule)), 485*0fca6ea1SDimitry Andric NumArgs(Ty->getNumParams()), IsNewDbgInfoFormat(UseNewDbgInfoFormat) { 4860b57cec5SDimitry Andric assert(FunctionType::isValidReturnType(getReturnType()) && 4870b57cec5SDimitry Andric "invalid return type"); 4880b57cec5SDimitry Andric setGlobalObjectSubClassData(0); 4890b57cec5SDimitry Andric 4900b57cec5SDimitry Andric // We only need a symbol table for a function if the context keeps value names 4910b57cec5SDimitry Andric if (!getContext().shouldDiscardValueNames()) 492fe6060f1SDimitry Andric SymTab = std::make_unique<ValueSymbolTable>(NonGlobalValueMaxNameSize); 4930b57cec5SDimitry Andric 4940b57cec5SDimitry Andric // If the function has arguments, mark them as lazily built. 4950b57cec5SDimitry Andric if (Ty->getNumParams()) 4960b57cec5SDimitry Andric setValueSubclassData(1); // Set the "has lazy arguments" bit. 4970b57cec5SDimitry Andric 498*0fca6ea1SDimitry Andric if (ParentModule) { 4990b57cec5SDimitry Andric ParentModule->getFunctionList().push_back(this); 500*0fca6ea1SDimitry Andric IsNewDbgInfoFormat = ParentModule->IsNewDbgInfoFormat; 501*0fca6ea1SDimitry Andric } 5020b57cec5SDimitry Andric 5035f757f3fSDimitry Andric HasLLVMReservedName = getName().starts_with("llvm."); 5040b57cec5SDimitry Andric // Ensure intrinsics have the right parameter attributes. 5050b57cec5SDimitry Andric // Note, the IntID field will have been set in Value::setName if this function 5060b57cec5SDimitry Andric // name is a valid intrinsic ID. 5070b57cec5SDimitry Andric if (IntID) 5080b57cec5SDimitry Andric setAttributes(Intrinsic::getAttributes(getContext(), IntID)); 5090b57cec5SDimitry Andric } 5100b57cec5SDimitry Andric 5110b57cec5SDimitry Andric Function::~Function() { 5120b57cec5SDimitry Andric dropAllReferences(); // After this it is safe to delete instructions. 5130b57cec5SDimitry Andric 5140b57cec5SDimitry Andric // Delete all of the method arguments and unlink from symbol table... 5150b57cec5SDimitry Andric if (Arguments) 5160b57cec5SDimitry Andric clearArguments(); 5170b57cec5SDimitry Andric 5180b57cec5SDimitry Andric // Remove the function from the on-the-side GC table. 5190b57cec5SDimitry Andric clearGC(); 5200b57cec5SDimitry Andric } 5210b57cec5SDimitry Andric 5220b57cec5SDimitry Andric void Function::BuildLazyArguments() const { 5230b57cec5SDimitry Andric // Create the arguments vector, all arguments start out unnamed. 5240b57cec5SDimitry Andric auto *FT = getFunctionType(); 5250b57cec5SDimitry Andric if (NumArgs > 0) { 5260b57cec5SDimitry Andric Arguments = std::allocator<Argument>().allocate(NumArgs); 5270b57cec5SDimitry Andric for (unsigned i = 0, e = NumArgs; i != e; ++i) { 5280b57cec5SDimitry Andric Type *ArgTy = FT->getParamType(i); 5290b57cec5SDimitry Andric assert(!ArgTy->isVoidTy() && "Cannot have void typed arguments!"); 5300b57cec5SDimitry Andric new (Arguments + i) Argument(ArgTy, "", const_cast<Function *>(this), i); 5310b57cec5SDimitry Andric } 5320b57cec5SDimitry Andric } 5330b57cec5SDimitry Andric 5340b57cec5SDimitry Andric // Clear the lazy arguments bit. 5350b57cec5SDimitry Andric unsigned SDC = getSubclassDataFromValue(); 5368bcb0991SDimitry Andric SDC &= ~(1 << 0); 5378bcb0991SDimitry Andric const_cast<Function*>(this)->setValueSubclassData(SDC); 5380b57cec5SDimitry Andric assert(!hasLazyArguments()); 5390b57cec5SDimitry Andric } 5400b57cec5SDimitry Andric 5410b57cec5SDimitry Andric static MutableArrayRef<Argument> makeArgArray(Argument *Args, size_t Count) { 5420b57cec5SDimitry Andric return MutableArrayRef<Argument>(Args, Count); 5430b57cec5SDimitry Andric } 5440b57cec5SDimitry Andric 5455ffd83dbSDimitry Andric bool Function::isConstrainedFPIntrinsic() const { 546*0fca6ea1SDimitry Andric return Intrinsic::isConstrainedFPIntrinsic(getIntrinsicID()); 5475ffd83dbSDimitry Andric } 5485ffd83dbSDimitry Andric 5490b57cec5SDimitry Andric void Function::clearArguments() { 5500b57cec5SDimitry Andric for (Argument &A : makeArgArray(Arguments, NumArgs)) { 5510b57cec5SDimitry Andric A.setName(""); 5520b57cec5SDimitry Andric A.~Argument(); 5530b57cec5SDimitry Andric } 5540b57cec5SDimitry Andric std::allocator<Argument>().deallocate(Arguments, NumArgs); 5550b57cec5SDimitry Andric Arguments = nullptr; 5560b57cec5SDimitry Andric } 5570b57cec5SDimitry Andric 5580b57cec5SDimitry Andric void Function::stealArgumentListFrom(Function &Src) { 5590b57cec5SDimitry Andric assert(isDeclaration() && "Expected no references to current arguments"); 5600b57cec5SDimitry Andric 5610b57cec5SDimitry Andric // Drop the current arguments, if any, and set the lazy argument bit. 5620b57cec5SDimitry Andric if (!hasLazyArguments()) { 5630b57cec5SDimitry Andric assert(llvm::all_of(makeArgArray(Arguments, NumArgs), 5640b57cec5SDimitry Andric [](const Argument &A) { return A.use_empty(); }) && 5650b57cec5SDimitry Andric "Expected arguments to be unused in declaration"); 5660b57cec5SDimitry Andric clearArguments(); 5670b57cec5SDimitry Andric setValueSubclassData(getSubclassDataFromValue() | (1 << 0)); 5680b57cec5SDimitry Andric } 5690b57cec5SDimitry Andric 5700b57cec5SDimitry Andric // Nothing to steal if Src has lazy arguments. 5710b57cec5SDimitry Andric if (Src.hasLazyArguments()) 5720b57cec5SDimitry Andric return; 5730b57cec5SDimitry Andric 5740b57cec5SDimitry Andric // Steal arguments from Src, and fix the lazy argument bits. 5750b57cec5SDimitry Andric assert(arg_size() == Src.arg_size()); 5760b57cec5SDimitry Andric Arguments = Src.Arguments; 5770b57cec5SDimitry Andric Src.Arguments = nullptr; 5780b57cec5SDimitry Andric for (Argument &A : makeArgArray(Arguments, NumArgs)) { 5790b57cec5SDimitry Andric // FIXME: This does the work of transferNodesFromList inefficiently. 5800b57cec5SDimitry Andric SmallString<128> Name; 5810b57cec5SDimitry Andric if (A.hasName()) 5820b57cec5SDimitry Andric Name = A.getName(); 5830b57cec5SDimitry Andric if (!Name.empty()) 5840b57cec5SDimitry Andric A.setName(""); 5850b57cec5SDimitry Andric A.setParent(this); 5860b57cec5SDimitry Andric if (!Name.empty()) 5870b57cec5SDimitry Andric A.setName(Name); 5880b57cec5SDimitry Andric } 5890b57cec5SDimitry Andric 5900b57cec5SDimitry Andric setValueSubclassData(getSubclassDataFromValue() & ~(1 << 0)); 5910b57cec5SDimitry Andric assert(!hasLazyArguments()); 5920b57cec5SDimitry Andric Src.setValueSubclassData(Src.getSubclassDataFromValue() | (1 << 0)); 5930b57cec5SDimitry Andric } 5940b57cec5SDimitry Andric 5955f757f3fSDimitry Andric void Function::deleteBodyImpl(bool ShouldDrop) { 5960b57cec5SDimitry Andric setIsMaterializable(false); 5970b57cec5SDimitry Andric 5980b57cec5SDimitry Andric for (BasicBlock &BB : *this) 5990b57cec5SDimitry Andric BB.dropAllReferences(); 6000b57cec5SDimitry Andric 6010b57cec5SDimitry Andric // Delete all basic blocks. They are now unused, except possibly by 6020b57cec5SDimitry Andric // blockaddresses, but BasicBlock's destructor takes care of those. 6030b57cec5SDimitry Andric while (!BasicBlocks.empty()) 6040b57cec5SDimitry Andric BasicBlocks.begin()->eraseFromParent(); 6050b57cec5SDimitry Andric 6060b57cec5SDimitry Andric if (getNumOperands()) { 6075f757f3fSDimitry Andric if (ShouldDrop) { 6085f757f3fSDimitry Andric // Drop uses of any optional data (real or placeholder). 6090b57cec5SDimitry Andric User::dropAllReferences(); 6100b57cec5SDimitry Andric setNumHungOffUseOperands(0); 6115f757f3fSDimitry Andric } else { 6125f757f3fSDimitry Andric // The code needs to match Function::allocHungoffUselist(). 6135f757f3fSDimitry Andric auto *CPN = ConstantPointerNull::get(PointerType::get(getContext(), 0)); 6145f757f3fSDimitry Andric Op<0>().set(CPN); 6155f757f3fSDimitry Andric Op<1>().set(CPN); 6165f757f3fSDimitry Andric Op<2>().set(CPN); 6175f757f3fSDimitry Andric } 6180b57cec5SDimitry Andric setValueSubclassData(getSubclassDataFromValue() & ~0xe); 6190b57cec5SDimitry Andric } 6200b57cec5SDimitry Andric 6210b57cec5SDimitry Andric // Metadata is stored in a side-table. 6220b57cec5SDimitry Andric clearMetadata(); 6230b57cec5SDimitry Andric } 6240b57cec5SDimitry Andric 625349cc55cSDimitry Andric void Function::addAttributeAtIndex(unsigned i, Attribute Attr) { 626349cc55cSDimitry Andric AttributeSets = AttributeSets.addAttributeAtIndex(getContext(), i, Attr); 6270b57cec5SDimitry Andric } 6280b57cec5SDimitry Andric 629349cc55cSDimitry Andric void Function::addFnAttr(Attribute::AttrKind Kind) { 630349cc55cSDimitry Andric AttributeSets = AttributeSets.addFnAttribute(getContext(), Kind); 6310b57cec5SDimitry Andric } 6320b57cec5SDimitry Andric 633349cc55cSDimitry Andric void Function::addFnAttr(StringRef Kind, StringRef Val) { 634349cc55cSDimitry Andric AttributeSets = AttributeSets.addFnAttribute(getContext(), Kind, Val); 635349cc55cSDimitry Andric } 636349cc55cSDimitry Andric 637349cc55cSDimitry Andric void Function::addFnAttr(Attribute Attr) { 638349cc55cSDimitry Andric AttributeSets = AttributeSets.addFnAttribute(getContext(), Attr); 639349cc55cSDimitry Andric } 640349cc55cSDimitry Andric 641349cc55cSDimitry Andric void Function::addFnAttrs(const AttrBuilder &Attrs) { 642349cc55cSDimitry Andric AttributeSets = AttributeSets.addFnAttributes(getContext(), Attrs); 643349cc55cSDimitry Andric } 644349cc55cSDimitry Andric 645349cc55cSDimitry Andric void Function::addRetAttr(Attribute::AttrKind Kind) { 646349cc55cSDimitry Andric AttributeSets = AttributeSets.addRetAttribute(getContext(), Kind); 647349cc55cSDimitry Andric } 648349cc55cSDimitry Andric 649349cc55cSDimitry Andric void Function::addRetAttr(Attribute Attr) { 650349cc55cSDimitry Andric AttributeSets = AttributeSets.addRetAttribute(getContext(), Attr); 651349cc55cSDimitry Andric } 652349cc55cSDimitry Andric 653349cc55cSDimitry Andric void Function::addRetAttrs(const AttrBuilder &Attrs) { 654349cc55cSDimitry Andric AttributeSets = AttributeSets.addRetAttributes(getContext(), Attrs); 6550b57cec5SDimitry Andric } 6560b57cec5SDimitry Andric 6570b57cec5SDimitry Andric void Function::addParamAttr(unsigned ArgNo, Attribute::AttrKind Kind) { 658349cc55cSDimitry Andric AttributeSets = AttributeSets.addParamAttribute(getContext(), ArgNo, Kind); 6590b57cec5SDimitry Andric } 6600b57cec5SDimitry Andric 6610b57cec5SDimitry Andric void Function::addParamAttr(unsigned ArgNo, Attribute Attr) { 662349cc55cSDimitry Andric AttributeSets = AttributeSets.addParamAttribute(getContext(), ArgNo, Attr); 6630b57cec5SDimitry Andric } 6640b57cec5SDimitry Andric 6650b57cec5SDimitry Andric void Function::addParamAttrs(unsigned ArgNo, const AttrBuilder &Attrs) { 666349cc55cSDimitry Andric AttributeSets = AttributeSets.addParamAttributes(getContext(), ArgNo, Attrs); 6670b57cec5SDimitry Andric } 6680b57cec5SDimitry Andric 669349cc55cSDimitry Andric void Function::removeAttributeAtIndex(unsigned i, Attribute::AttrKind Kind) { 670349cc55cSDimitry Andric AttributeSets = AttributeSets.removeAttributeAtIndex(getContext(), i, Kind); 6710b57cec5SDimitry Andric } 6720b57cec5SDimitry Andric 673349cc55cSDimitry Andric void Function::removeAttributeAtIndex(unsigned i, StringRef Kind) { 674349cc55cSDimitry Andric AttributeSets = AttributeSets.removeAttributeAtIndex(getContext(), i, Kind); 6750b57cec5SDimitry Andric } 6760b57cec5SDimitry Andric 677349cc55cSDimitry Andric void Function::removeFnAttr(Attribute::AttrKind Kind) { 678349cc55cSDimitry Andric AttributeSets = AttributeSets.removeFnAttribute(getContext(), Kind); 679349cc55cSDimitry Andric } 680349cc55cSDimitry Andric 681349cc55cSDimitry Andric void Function::removeFnAttr(StringRef Kind) { 682349cc55cSDimitry Andric AttributeSets = AttributeSets.removeFnAttribute(getContext(), Kind); 683349cc55cSDimitry Andric } 684349cc55cSDimitry Andric 68504eeddc0SDimitry Andric void Function::removeFnAttrs(const AttributeMask &AM) { 68604eeddc0SDimitry Andric AttributeSets = AttributeSets.removeFnAttributes(getContext(), AM); 687349cc55cSDimitry Andric } 688349cc55cSDimitry Andric 689349cc55cSDimitry Andric void Function::removeRetAttr(Attribute::AttrKind Kind) { 690349cc55cSDimitry Andric AttributeSets = AttributeSets.removeRetAttribute(getContext(), Kind); 691349cc55cSDimitry Andric } 692349cc55cSDimitry Andric 693349cc55cSDimitry Andric void Function::removeRetAttr(StringRef Kind) { 694349cc55cSDimitry Andric AttributeSets = AttributeSets.removeRetAttribute(getContext(), Kind); 695349cc55cSDimitry Andric } 696349cc55cSDimitry Andric 69704eeddc0SDimitry Andric void Function::removeRetAttrs(const AttributeMask &Attrs) { 698349cc55cSDimitry Andric AttributeSets = AttributeSets.removeRetAttributes(getContext(), Attrs); 6990b57cec5SDimitry Andric } 7000b57cec5SDimitry Andric 7010b57cec5SDimitry Andric void Function::removeParamAttr(unsigned ArgNo, Attribute::AttrKind Kind) { 702349cc55cSDimitry Andric AttributeSets = AttributeSets.removeParamAttribute(getContext(), ArgNo, Kind); 7030b57cec5SDimitry Andric } 7040b57cec5SDimitry Andric 7050b57cec5SDimitry Andric void Function::removeParamAttr(unsigned ArgNo, StringRef Kind) { 706349cc55cSDimitry Andric AttributeSets = AttributeSets.removeParamAttribute(getContext(), ArgNo, Kind); 7070b57cec5SDimitry Andric } 7080b57cec5SDimitry Andric 70904eeddc0SDimitry Andric void Function::removeParamAttrs(unsigned ArgNo, const AttributeMask &Attrs) { 710349cc55cSDimitry Andric AttributeSets = 711349cc55cSDimitry Andric AttributeSets.removeParamAttributes(getContext(), ArgNo, Attrs); 7120b57cec5SDimitry Andric } 7130b57cec5SDimitry Andric 7140b57cec5SDimitry Andric void Function::addDereferenceableParamAttr(unsigned ArgNo, uint64_t Bytes) { 715349cc55cSDimitry Andric AttributeSets = 716349cc55cSDimitry Andric AttributeSets.addDereferenceableParamAttr(getContext(), ArgNo, Bytes); 7170b57cec5SDimitry Andric } 7180b57cec5SDimitry Andric 719349cc55cSDimitry Andric bool Function::hasFnAttribute(Attribute::AttrKind Kind) const { 720349cc55cSDimitry Andric return AttributeSets.hasFnAttr(Kind); 721349cc55cSDimitry Andric } 722349cc55cSDimitry Andric 723349cc55cSDimitry Andric bool Function::hasFnAttribute(StringRef Kind) const { 724349cc55cSDimitry Andric return AttributeSets.hasFnAttr(Kind); 725349cc55cSDimitry Andric } 726349cc55cSDimitry Andric 727349cc55cSDimitry Andric bool Function::hasRetAttribute(Attribute::AttrKind Kind) const { 728349cc55cSDimitry Andric return AttributeSets.hasRetAttr(Kind); 729349cc55cSDimitry Andric } 730349cc55cSDimitry Andric 731349cc55cSDimitry Andric bool Function::hasParamAttribute(unsigned ArgNo, 732349cc55cSDimitry Andric Attribute::AttrKind Kind) const { 733349cc55cSDimitry Andric return AttributeSets.hasParamAttr(ArgNo, Kind); 734349cc55cSDimitry Andric } 735349cc55cSDimitry Andric 736349cc55cSDimitry Andric Attribute Function::getAttributeAtIndex(unsigned i, 737349cc55cSDimitry Andric Attribute::AttrKind Kind) const { 738349cc55cSDimitry Andric return AttributeSets.getAttributeAtIndex(i, Kind); 739349cc55cSDimitry Andric } 740349cc55cSDimitry Andric 741349cc55cSDimitry Andric Attribute Function::getAttributeAtIndex(unsigned i, StringRef Kind) const { 742349cc55cSDimitry Andric return AttributeSets.getAttributeAtIndex(i, Kind); 743349cc55cSDimitry Andric } 744349cc55cSDimitry Andric 745349cc55cSDimitry Andric Attribute Function::getFnAttribute(Attribute::AttrKind Kind) const { 746349cc55cSDimitry Andric return AttributeSets.getFnAttr(Kind); 747349cc55cSDimitry Andric } 748349cc55cSDimitry Andric 749349cc55cSDimitry Andric Attribute Function::getFnAttribute(StringRef Kind) const { 750349cc55cSDimitry Andric return AttributeSets.getFnAttr(Kind); 751349cc55cSDimitry Andric } 752349cc55cSDimitry Andric 753*0fca6ea1SDimitry Andric Attribute Function::getRetAttribute(Attribute::AttrKind Kind) const { 754*0fca6ea1SDimitry Andric return AttributeSets.getRetAttr(Kind); 755*0fca6ea1SDimitry Andric } 756*0fca6ea1SDimitry Andric 757bdd1243dSDimitry Andric uint64_t Function::getFnAttributeAsParsedInteger(StringRef Name, 758bdd1243dSDimitry Andric uint64_t Default) const { 759bdd1243dSDimitry Andric Attribute A = getFnAttribute(Name); 760bdd1243dSDimitry Andric uint64_t Result = Default; 761bdd1243dSDimitry Andric if (A.isStringAttribute()) { 762bdd1243dSDimitry Andric StringRef Str = A.getValueAsString(); 763bdd1243dSDimitry Andric if (Str.getAsInteger(0, Result)) 764bdd1243dSDimitry Andric getContext().emitError("cannot parse integer attribute " + Name); 765bdd1243dSDimitry Andric } 766bdd1243dSDimitry Andric 767bdd1243dSDimitry Andric return Result; 768bdd1243dSDimitry Andric } 769bdd1243dSDimitry Andric 770349cc55cSDimitry Andric /// gets the specified attribute from the list of attributes. 771349cc55cSDimitry Andric Attribute Function::getParamAttribute(unsigned ArgNo, 772349cc55cSDimitry Andric Attribute::AttrKind Kind) const { 773349cc55cSDimitry Andric return AttributeSets.getParamAttr(ArgNo, Kind); 7740b57cec5SDimitry Andric } 7750b57cec5SDimitry Andric 7760b57cec5SDimitry Andric void Function::addDereferenceableOrNullParamAttr(unsigned ArgNo, 7770b57cec5SDimitry Andric uint64_t Bytes) { 778349cc55cSDimitry Andric AttributeSets = AttributeSets.addDereferenceableOrNullParamAttr(getContext(), 779349cc55cSDimitry Andric ArgNo, Bytes); 7800b57cec5SDimitry Andric } 7810b57cec5SDimitry Andric 782*0fca6ea1SDimitry Andric void Function::addRangeRetAttr(const ConstantRange &CR) { 783*0fca6ea1SDimitry Andric AttributeSets = AttributeSets.addRangeRetAttr(getContext(), CR); 784*0fca6ea1SDimitry Andric } 785*0fca6ea1SDimitry Andric 786e8d8bef9SDimitry Andric DenormalMode Function::getDenormalMode(const fltSemantics &FPType) const { 787e8d8bef9SDimitry Andric if (&FPType == &APFloat::IEEEsingle()) { 78806c3fb27SDimitry Andric DenormalMode Mode = getDenormalModeF32Raw(); 789e8d8bef9SDimitry Andric // If the f32 variant of the attribute isn't specified, try to use the 790e8d8bef9SDimitry Andric // generic one. 79106c3fb27SDimitry Andric if (Mode.isValid()) 79206c3fb27SDimitry Andric return Mode; 793e8d8bef9SDimitry Andric } 794e8d8bef9SDimitry Andric 79506c3fb27SDimitry Andric return getDenormalModeRaw(); 79606c3fb27SDimitry Andric } 79706c3fb27SDimitry Andric 79806c3fb27SDimitry Andric DenormalMode Function::getDenormalModeRaw() const { 799e8d8bef9SDimitry Andric Attribute Attr = getFnAttribute("denormal-fp-math"); 80006c3fb27SDimitry Andric StringRef Val = Attr.getValueAsString(); 80106c3fb27SDimitry Andric return parseDenormalFPAttribute(Val); 80206c3fb27SDimitry Andric } 80306c3fb27SDimitry Andric 80406c3fb27SDimitry Andric DenormalMode Function::getDenormalModeF32Raw() const { 80506c3fb27SDimitry Andric Attribute Attr = getFnAttribute("denormal-fp-math-f32"); 80606c3fb27SDimitry Andric if (Attr.isValid()) { 80706c3fb27SDimitry Andric StringRef Val = Attr.getValueAsString(); 80806c3fb27SDimitry Andric return parseDenormalFPAttribute(Val); 80906c3fb27SDimitry Andric } 81006c3fb27SDimitry Andric 81106c3fb27SDimitry Andric return DenormalMode::getInvalid(); 812e8d8bef9SDimitry Andric } 813e8d8bef9SDimitry Andric 8140b57cec5SDimitry Andric const std::string &Function::getGC() const { 8150b57cec5SDimitry Andric assert(hasGC() && "Function has no collector"); 8160b57cec5SDimitry Andric return getContext().getGC(*this); 8170b57cec5SDimitry Andric } 8180b57cec5SDimitry Andric 8190b57cec5SDimitry Andric void Function::setGC(std::string Str) { 8200b57cec5SDimitry Andric setValueSubclassDataBit(14, !Str.empty()); 8210b57cec5SDimitry Andric getContext().setGC(*this, std::move(Str)); 8220b57cec5SDimitry Andric } 8230b57cec5SDimitry Andric 8240b57cec5SDimitry Andric void Function::clearGC() { 8250b57cec5SDimitry Andric if (!hasGC()) 8260b57cec5SDimitry Andric return; 8270b57cec5SDimitry Andric getContext().deleteGC(*this); 8280b57cec5SDimitry Andric setValueSubclassDataBit(14, false); 8290b57cec5SDimitry Andric } 8300b57cec5SDimitry Andric 831e8d8bef9SDimitry Andric bool Function::hasStackProtectorFnAttr() const { 832e8d8bef9SDimitry Andric return hasFnAttribute(Attribute::StackProtect) || 833e8d8bef9SDimitry Andric hasFnAttribute(Attribute::StackProtectStrong) || 834e8d8bef9SDimitry Andric hasFnAttribute(Attribute::StackProtectReq); 835e8d8bef9SDimitry Andric } 836e8d8bef9SDimitry Andric 8370b57cec5SDimitry Andric /// Copy all additional attributes (those not needed to create a Function) from 8380b57cec5SDimitry Andric /// the Function Src to this one. 8390b57cec5SDimitry Andric void Function::copyAttributesFrom(const Function *Src) { 8400b57cec5SDimitry Andric GlobalObject::copyAttributesFrom(Src); 8410b57cec5SDimitry Andric setCallingConv(Src->getCallingConv()); 8420b57cec5SDimitry Andric setAttributes(Src->getAttributes()); 8430b57cec5SDimitry Andric if (Src->hasGC()) 8440b57cec5SDimitry Andric setGC(Src->getGC()); 8450b57cec5SDimitry Andric else 8460b57cec5SDimitry Andric clearGC(); 8470b57cec5SDimitry Andric if (Src->hasPersonalityFn()) 8480b57cec5SDimitry Andric setPersonalityFn(Src->getPersonalityFn()); 8490b57cec5SDimitry Andric if (Src->hasPrefixData()) 8500b57cec5SDimitry Andric setPrefixData(Src->getPrefixData()); 8510b57cec5SDimitry Andric if (Src->hasPrologueData()) 8520b57cec5SDimitry Andric setPrologueData(Src->getPrologueData()); 8530b57cec5SDimitry Andric } 8540b57cec5SDimitry Andric 855bdd1243dSDimitry Andric MemoryEffects Function::getMemoryEffects() const { 856bdd1243dSDimitry Andric return getAttributes().getMemoryEffects(); 857bdd1243dSDimitry Andric } 858bdd1243dSDimitry Andric void Function::setMemoryEffects(MemoryEffects ME) { 859bdd1243dSDimitry Andric addFnAttr(Attribute::getWithMemoryEffects(getContext(), ME)); 860bdd1243dSDimitry Andric } 861bdd1243dSDimitry Andric 862bdd1243dSDimitry Andric /// Determine if the function does not access memory. 863bdd1243dSDimitry Andric bool Function::doesNotAccessMemory() const { 864bdd1243dSDimitry Andric return getMemoryEffects().doesNotAccessMemory(); 865bdd1243dSDimitry Andric } 866bdd1243dSDimitry Andric void Function::setDoesNotAccessMemory() { 867bdd1243dSDimitry Andric setMemoryEffects(MemoryEffects::none()); 868bdd1243dSDimitry Andric } 869bdd1243dSDimitry Andric 870bdd1243dSDimitry Andric /// Determine if the function does not access or only reads memory. 871bdd1243dSDimitry Andric bool Function::onlyReadsMemory() const { 872bdd1243dSDimitry Andric return getMemoryEffects().onlyReadsMemory(); 873bdd1243dSDimitry Andric } 874bdd1243dSDimitry Andric void Function::setOnlyReadsMemory() { 875bdd1243dSDimitry Andric setMemoryEffects(getMemoryEffects() & MemoryEffects::readOnly()); 876bdd1243dSDimitry Andric } 877bdd1243dSDimitry Andric 878bdd1243dSDimitry Andric /// Determine if the function does not access or only writes memory. 879bdd1243dSDimitry Andric bool Function::onlyWritesMemory() const { 880bdd1243dSDimitry Andric return getMemoryEffects().onlyWritesMemory(); 881bdd1243dSDimitry Andric } 882bdd1243dSDimitry Andric void Function::setOnlyWritesMemory() { 883bdd1243dSDimitry Andric setMemoryEffects(getMemoryEffects() & MemoryEffects::writeOnly()); 884bdd1243dSDimitry Andric } 885bdd1243dSDimitry Andric 886bdd1243dSDimitry Andric /// Determine if the call can access memmory only using pointers based 887bdd1243dSDimitry Andric /// on its arguments. 888bdd1243dSDimitry Andric bool Function::onlyAccessesArgMemory() const { 889bdd1243dSDimitry Andric return getMemoryEffects().onlyAccessesArgPointees(); 890bdd1243dSDimitry Andric } 891bdd1243dSDimitry Andric void Function::setOnlyAccessesArgMemory() { 892bdd1243dSDimitry Andric setMemoryEffects(getMemoryEffects() & MemoryEffects::argMemOnly()); 893bdd1243dSDimitry Andric } 894bdd1243dSDimitry Andric 895bdd1243dSDimitry Andric /// Determine if the function may only access memory that is 896bdd1243dSDimitry Andric /// inaccessible from the IR. 897bdd1243dSDimitry Andric bool Function::onlyAccessesInaccessibleMemory() const { 898bdd1243dSDimitry Andric return getMemoryEffects().onlyAccessesInaccessibleMem(); 899bdd1243dSDimitry Andric } 900bdd1243dSDimitry Andric void Function::setOnlyAccessesInaccessibleMemory() { 901bdd1243dSDimitry Andric setMemoryEffects(getMemoryEffects() & MemoryEffects::inaccessibleMemOnly()); 902bdd1243dSDimitry Andric } 903bdd1243dSDimitry Andric 904bdd1243dSDimitry Andric /// Determine if the function may only access memory that is 905bdd1243dSDimitry Andric /// either inaccessible from the IR or pointed to by its arguments. 906bdd1243dSDimitry Andric bool Function::onlyAccessesInaccessibleMemOrArgMem() const { 907bdd1243dSDimitry Andric return getMemoryEffects().onlyAccessesInaccessibleOrArgMem(); 908bdd1243dSDimitry Andric } 909bdd1243dSDimitry Andric void Function::setOnlyAccessesInaccessibleMemOrArgMem() { 910bdd1243dSDimitry Andric setMemoryEffects(getMemoryEffects() & 911bdd1243dSDimitry Andric MemoryEffects::inaccessibleOrArgMemOnly()); 912bdd1243dSDimitry Andric } 913bdd1243dSDimitry Andric 9140b57cec5SDimitry Andric /// Table of string intrinsic names indexed by enum value. 9150b57cec5SDimitry Andric static const char * const IntrinsicNameTable[] = { 9160b57cec5SDimitry Andric "not_intrinsic", 9170b57cec5SDimitry Andric #define GET_INTRINSIC_NAME_TABLE 9180b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 9190b57cec5SDimitry Andric #undef GET_INTRINSIC_NAME_TABLE 9200b57cec5SDimitry Andric }; 9210b57cec5SDimitry Andric 9220b57cec5SDimitry Andric /// Table of per-target intrinsic name tables. 9230b57cec5SDimitry Andric #define GET_INTRINSIC_TARGET_DATA 9240b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 9250b57cec5SDimitry Andric #undef GET_INTRINSIC_TARGET_DATA 9260b57cec5SDimitry Andric 927e8d8bef9SDimitry Andric bool Function::isTargetIntrinsic(Intrinsic::ID IID) { 928e8d8bef9SDimitry Andric return IID > TargetInfos[0].Count; 929e8d8bef9SDimitry Andric } 930e8d8bef9SDimitry Andric 931e8d8bef9SDimitry Andric bool Function::isTargetIntrinsic() const { 932e8d8bef9SDimitry Andric return isTargetIntrinsic(IntID); 933e8d8bef9SDimitry Andric } 934e8d8bef9SDimitry Andric 9350b57cec5SDimitry Andric /// Find the segment of \c IntrinsicNameTable for intrinsics with the same 9360b57cec5SDimitry Andric /// target as \c Name, or the generic table if \c Name is not target specific. 9370b57cec5SDimitry Andric /// 9380b57cec5SDimitry Andric /// Returns the relevant slice of \c IntrinsicNameTable 9390b57cec5SDimitry Andric static ArrayRef<const char *> findTargetSubtable(StringRef Name) { 9405f757f3fSDimitry Andric assert(Name.starts_with("llvm.")); 9410b57cec5SDimitry Andric 9420b57cec5SDimitry Andric ArrayRef<IntrinsicTargetInfo> Targets(TargetInfos); 9430b57cec5SDimitry Andric // Drop "llvm." and take the first dotted component. That will be the target 9440b57cec5SDimitry Andric // if this is target specific. 9450b57cec5SDimitry Andric StringRef Target = Name.drop_front(5).split('.').first; 9460b57cec5SDimitry Andric auto It = partition_point( 9470b57cec5SDimitry Andric Targets, [=](const IntrinsicTargetInfo &TI) { return TI.Name < Target; }); 9480b57cec5SDimitry Andric // We've either found the target or just fall back to the generic set, which 9490b57cec5SDimitry Andric // is always first. 9500b57cec5SDimitry Andric const auto &TI = It != Targets.end() && It->Name == Target ? *It : Targets[0]; 951bdd1243dSDimitry Andric return ArrayRef(&IntrinsicNameTable[1] + TI.Offset, TI.Count); 9520b57cec5SDimitry Andric } 9530b57cec5SDimitry Andric 9540b57cec5SDimitry Andric /// This does the actual lookup of an intrinsic ID which 9550b57cec5SDimitry Andric /// matches the given function name. 9560b57cec5SDimitry Andric Intrinsic::ID Function::lookupIntrinsicID(StringRef Name) { 9570b57cec5SDimitry Andric ArrayRef<const char *> NameTable = findTargetSubtable(Name); 9580b57cec5SDimitry Andric int Idx = Intrinsic::lookupLLVMIntrinsicByName(NameTable, Name); 9590b57cec5SDimitry Andric if (Idx == -1) 9600b57cec5SDimitry Andric return Intrinsic::not_intrinsic; 9610b57cec5SDimitry Andric 9620b57cec5SDimitry Andric // Intrinsic IDs correspond to the location in IntrinsicNameTable, but we have 9630b57cec5SDimitry Andric // an index into a sub-table. 9640b57cec5SDimitry Andric int Adjust = NameTable.data() - IntrinsicNameTable; 9650b57cec5SDimitry Andric Intrinsic::ID ID = static_cast<Intrinsic::ID>(Idx + Adjust); 9660b57cec5SDimitry Andric 9670b57cec5SDimitry Andric // If the intrinsic is not overloaded, require an exact match. If it is 9680b57cec5SDimitry Andric // overloaded, require either exact or prefix match. 9690b57cec5SDimitry Andric const auto MatchSize = strlen(NameTable[Idx]); 9700b57cec5SDimitry Andric assert(Name.size() >= MatchSize && "Expected either exact or prefix match"); 9710b57cec5SDimitry Andric bool IsExactMatch = Name.size() == MatchSize; 972480093f4SDimitry Andric return IsExactMatch || Intrinsic::isOverloaded(ID) ? ID 973480093f4SDimitry Andric : Intrinsic::not_intrinsic; 9740b57cec5SDimitry Andric } 9750b57cec5SDimitry Andric 9765f757f3fSDimitry Andric void Function::updateAfterNameChange() { 9775f757f3fSDimitry Andric LibFuncCache = UnknownLibFunc; 9780b57cec5SDimitry Andric StringRef Name = getName(); 9795f757f3fSDimitry Andric if (!Name.starts_with("llvm.")) { 9800b57cec5SDimitry Andric HasLLVMReservedName = false; 9810b57cec5SDimitry Andric IntID = Intrinsic::not_intrinsic; 9820b57cec5SDimitry Andric return; 9830b57cec5SDimitry Andric } 9840b57cec5SDimitry Andric HasLLVMReservedName = true; 9850b57cec5SDimitry Andric IntID = lookupIntrinsicID(Name); 9860b57cec5SDimitry Andric } 9870b57cec5SDimitry Andric 9880b57cec5SDimitry Andric /// Returns a stable mangling for the type specified for use in the name 9890b57cec5SDimitry Andric /// mangling scheme used by 'any' types in intrinsic signatures. The mangling 9900b57cec5SDimitry Andric /// of named types is simply their name. Manglings for unnamed types consist 9910b57cec5SDimitry Andric /// of a prefix ('p' for pointers, 'a' for arrays, 'f_' for functions) 9920b57cec5SDimitry Andric /// combined with the mangling of their component types. A vararg function 9930b57cec5SDimitry Andric /// type will have a suffix of 'vararg'. Since function types can contain 9940b57cec5SDimitry Andric /// other function types, we close a function type mangling with suffix 'f' 9950b57cec5SDimitry Andric /// which can't be confused with it's prefix. This ensures we don't have 9960b57cec5SDimitry Andric /// collisions between two unrelated function types. Otherwise, you might 9970b57cec5SDimitry Andric /// parse ffXX as f(fXX) or f(fX)X. (X is a placeholder for any other type.) 998fe6060f1SDimitry Andric /// The HasUnnamedType boolean is set if an unnamed type was encountered, 999fe6060f1SDimitry Andric /// indicating that extra care must be taken to ensure a unique name. 1000fe6060f1SDimitry Andric static std::string getMangledTypeStr(Type *Ty, bool &HasUnnamedType) { 10010b57cec5SDimitry Andric std::string Result; 10020b57cec5SDimitry Andric if (PointerType *PTyp = dyn_cast<PointerType>(Ty)) { 1003fe6060f1SDimitry Andric Result += "p" + utostr(PTyp->getAddressSpace()); 10040b57cec5SDimitry Andric } else if (ArrayType *ATyp = dyn_cast<ArrayType>(Ty)) { 10050b57cec5SDimitry Andric Result += "a" + utostr(ATyp->getNumElements()) + 1006fe6060f1SDimitry Andric getMangledTypeStr(ATyp->getElementType(), HasUnnamedType); 10070b57cec5SDimitry Andric } else if (StructType *STyp = dyn_cast<StructType>(Ty)) { 10080b57cec5SDimitry Andric if (!STyp->isLiteral()) { 10090b57cec5SDimitry Andric Result += "s_"; 1010fe6060f1SDimitry Andric if (STyp->hasName()) 10110b57cec5SDimitry Andric Result += STyp->getName(); 1012fe6060f1SDimitry Andric else 1013fe6060f1SDimitry Andric HasUnnamedType = true; 10140b57cec5SDimitry Andric } else { 10150b57cec5SDimitry Andric Result += "sl_"; 1016bdd1243dSDimitry Andric for (auto *Elem : STyp->elements()) 1017fe6060f1SDimitry Andric Result += getMangledTypeStr(Elem, HasUnnamedType); 10180b57cec5SDimitry Andric } 10190b57cec5SDimitry Andric // Ensure nested structs are distinguishable. 10200b57cec5SDimitry Andric Result += "s"; 10210b57cec5SDimitry Andric } else if (FunctionType *FT = dyn_cast<FunctionType>(Ty)) { 1022fe6060f1SDimitry Andric Result += "f_" + getMangledTypeStr(FT->getReturnType(), HasUnnamedType); 10230b57cec5SDimitry Andric for (size_t i = 0; i < FT->getNumParams(); i++) 1024fe6060f1SDimitry Andric Result += getMangledTypeStr(FT->getParamType(i), HasUnnamedType); 10250b57cec5SDimitry Andric if (FT->isVarArg()) 10260b57cec5SDimitry Andric Result += "vararg"; 10270b57cec5SDimitry Andric // Ensure nested function types are distinguishable. 10280b57cec5SDimitry Andric Result += "f"; 10298bcb0991SDimitry Andric } else if (VectorType *VTy = dyn_cast<VectorType>(Ty)) { 10305ffd83dbSDimitry Andric ElementCount EC = VTy->getElementCount(); 1031e8d8bef9SDimitry Andric if (EC.isScalable()) 10328bcb0991SDimitry Andric Result += "nx"; 1033e8d8bef9SDimitry Andric Result += "v" + utostr(EC.getKnownMinValue()) + 1034fe6060f1SDimitry Andric getMangledTypeStr(VTy->getElementType(), HasUnnamedType); 1035bdd1243dSDimitry Andric } else if (TargetExtType *TETy = dyn_cast<TargetExtType>(Ty)) { 1036bdd1243dSDimitry Andric Result += "t"; 1037bdd1243dSDimitry Andric Result += TETy->getName(); 1038bdd1243dSDimitry Andric for (Type *ParamTy : TETy->type_params()) 1039bdd1243dSDimitry Andric Result += "_" + getMangledTypeStr(ParamTy, HasUnnamedType); 1040bdd1243dSDimitry Andric for (unsigned IntParam : TETy->int_params()) 1041bdd1243dSDimitry Andric Result += "_" + utostr(IntParam); 1042bdd1243dSDimitry Andric // Ensure nested target extension types are distinguishable. 1043bdd1243dSDimitry Andric Result += "t"; 10440b57cec5SDimitry Andric } else if (Ty) { 10450b57cec5SDimitry Andric switch (Ty->getTypeID()) { 10460b57cec5SDimitry Andric default: llvm_unreachable("Unhandled type"); 10470b57cec5SDimitry Andric case Type::VoidTyID: Result += "isVoid"; break; 10480b57cec5SDimitry Andric case Type::MetadataTyID: Result += "Metadata"; break; 10490b57cec5SDimitry Andric case Type::HalfTyID: Result += "f16"; break; 10505ffd83dbSDimitry Andric case Type::BFloatTyID: Result += "bf16"; break; 10510b57cec5SDimitry Andric case Type::FloatTyID: Result += "f32"; break; 10520b57cec5SDimitry Andric case Type::DoubleTyID: Result += "f64"; break; 10530b57cec5SDimitry Andric case Type::X86_FP80TyID: Result += "f80"; break; 10540b57cec5SDimitry Andric case Type::FP128TyID: Result += "f128"; break; 10550b57cec5SDimitry Andric case Type::PPC_FP128TyID: Result += "ppcf128"; break; 10560b57cec5SDimitry Andric case Type::X86_MMXTyID: Result += "x86mmx"; break; 1057e8d8bef9SDimitry Andric case Type::X86_AMXTyID: Result += "x86amx"; break; 10580b57cec5SDimitry Andric case Type::IntegerTyID: 10590b57cec5SDimitry Andric Result += "i" + utostr(cast<IntegerType>(Ty)->getBitWidth()); 10600b57cec5SDimitry Andric break; 10610b57cec5SDimitry Andric } 10620b57cec5SDimitry Andric } 10630b57cec5SDimitry Andric return Result; 10640b57cec5SDimitry Andric } 10650b57cec5SDimitry Andric 1066fe6060f1SDimitry Andric StringRef Intrinsic::getBaseName(ID id) { 1067fe6060f1SDimitry Andric assert(id < num_intrinsics && "Invalid intrinsic ID!"); 1068fe6060f1SDimitry Andric return IntrinsicNameTable[id]; 1069fe6060f1SDimitry Andric } 1070fe6060f1SDimitry Andric 10710b57cec5SDimitry Andric StringRef Intrinsic::getName(ID id) { 10720b57cec5SDimitry Andric assert(id < num_intrinsics && "Invalid intrinsic ID!"); 1073480093f4SDimitry Andric assert(!Intrinsic::isOverloaded(id) && 10740b57cec5SDimitry Andric "This version of getName does not support overloading"); 1075fe6060f1SDimitry Andric return getBaseName(id); 10760b57cec5SDimitry Andric } 10770b57cec5SDimitry Andric 1078fe6060f1SDimitry Andric static std::string getIntrinsicNameImpl(Intrinsic::ID Id, ArrayRef<Type *> Tys, 1079fe6060f1SDimitry Andric Module *M, FunctionType *FT, 1080fe6060f1SDimitry Andric bool EarlyModuleCheck) { 1081fe6060f1SDimitry Andric 1082fe6060f1SDimitry Andric assert(Id < Intrinsic::num_intrinsics && "Invalid intrinsic ID!"); 1083fe6060f1SDimitry Andric assert((Tys.empty() || Intrinsic::isOverloaded(Id)) && 1084e8d8bef9SDimitry Andric "This version of getName is for overloaded intrinsics only"); 1085fe6060f1SDimitry Andric (void)EarlyModuleCheck; 1086fe6060f1SDimitry Andric assert((!EarlyModuleCheck || M || 1087fe6060f1SDimitry Andric !any_of(Tys, [](Type *T) { return isa<PointerType>(T); })) && 1088fe6060f1SDimitry Andric "Intrinsic overloading on pointer types need to provide a Module"); 1089fe6060f1SDimitry Andric bool HasUnnamedType = false; 1090fe6060f1SDimitry Andric std::string Result(Intrinsic::getBaseName(Id)); 1091fe6060f1SDimitry Andric for (Type *Ty : Tys) 1092fe6060f1SDimitry Andric Result += "." + getMangledTypeStr(Ty, HasUnnamedType); 1093fe6060f1SDimitry Andric if (HasUnnamedType) { 1094fe6060f1SDimitry Andric assert(M && "unnamed types need a module"); 1095fe6060f1SDimitry Andric if (!FT) 1096fe6060f1SDimitry Andric FT = Intrinsic::getType(M->getContext(), Id, Tys); 1097fe6060f1SDimitry Andric else 1098fe6060f1SDimitry Andric assert((FT == Intrinsic::getType(M->getContext(), Id, Tys)) && 1099fe6060f1SDimitry Andric "Provided FunctionType must match arguments"); 1100fe6060f1SDimitry Andric return M->getUniqueIntrinsicName(Result, Id, FT); 11010b57cec5SDimitry Andric } 11020b57cec5SDimitry Andric return Result; 11030b57cec5SDimitry Andric } 11040b57cec5SDimitry Andric 1105fe6060f1SDimitry Andric std::string Intrinsic::getName(ID Id, ArrayRef<Type *> Tys, Module *M, 1106fe6060f1SDimitry Andric FunctionType *FT) { 1107fe6060f1SDimitry Andric assert(M && "We need to have a Module"); 1108fe6060f1SDimitry Andric return getIntrinsicNameImpl(Id, Tys, M, FT, true); 1109fe6060f1SDimitry Andric } 1110fe6060f1SDimitry Andric 1111fe6060f1SDimitry Andric std::string Intrinsic::getNameNoUnnamedTypes(ID Id, ArrayRef<Type *> Tys) { 1112fe6060f1SDimitry Andric return getIntrinsicNameImpl(Id, Tys, nullptr, nullptr, false); 1113fe6060f1SDimitry Andric } 1114fe6060f1SDimitry Andric 11150b57cec5SDimitry Andric /// IIT_Info - These are enumerators that describe the entries returned by the 11160b57cec5SDimitry Andric /// getIntrinsicInfoTableEntries function. 11170b57cec5SDimitry Andric /// 111806c3fb27SDimitry Andric /// Defined in Intrinsics.td. 11190b57cec5SDimitry Andric enum IIT_Info { 112006c3fb27SDimitry Andric #define GET_INTRINSIC_IITINFO 112106c3fb27SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 112206c3fb27SDimitry Andric #undef GET_INTRINSIC_IITINFO 11230b57cec5SDimitry Andric }; 11240b57cec5SDimitry Andric 11250b57cec5SDimitry Andric static void DecodeIITType(unsigned &NextElt, ArrayRef<unsigned char> Infos, 11265ffd83dbSDimitry Andric IIT_Info LastInfo, 11270b57cec5SDimitry Andric SmallVectorImpl<Intrinsic::IITDescriptor> &OutputTable) { 11280b57cec5SDimitry Andric using namespace Intrinsic; 11290b57cec5SDimitry Andric 11305ffd83dbSDimitry Andric bool IsScalableVector = (LastInfo == IIT_SCALABLE_VEC); 11315ffd83dbSDimitry Andric 11320b57cec5SDimitry Andric IIT_Info Info = IIT_Info(Infos[NextElt++]); 11330b57cec5SDimitry Andric unsigned StructElts = 2; 11340b57cec5SDimitry Andric 11350b57cec5SDimitry Andric switch (Info) { 11360b57cec5SDimitry Andric case IIT_Done: 11370b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Void, 0)); 11380b57cec5SDimitry Andric return; 11390b57cec5SDimitry Andric case IIT_VARARG: 11400b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::VarArg, 0)); 11410b57cec5SDimitry Andric return; 11420b57cec5SDimitry Andric case IIT_MMX: 11430b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::MMX, 0)); 11440b57cec5SDimitry Andric return; 1145e8d8bef9SDimitry Andric case IIT_AMX: 1146e8d8bef9SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::AMX, 0)); 1147e8d8bef9SDimitry Andric return; 11480b57cec5SDimitry Andric case IIT_TOKEN: 11490b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Token, 0)); 11500b57cec5SDimitry Andric return; 11510b57cec5SDimitry Andric case IIT_METADATA: 11520b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Metadata, 0)); 11530b57cec5SDimitry Andric return; 11540b57cec5SDimitry Andric case IIT_F16: 11550b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Half, 0)); 11560b57cec5SDimitry Andric return; 11575ffd83dbSDimitry Andric case IIT_BF16: 11585ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::BFloat, 0)); 11595ffd83dbSDimitry Andric return; 11600b57cec5SDimitry Andric case IIT_F32: 11610b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Float, 0)); 11620b57cec5SDimitry Andric return; 11630b57cec5SDimitry Andric case IIT_F64: 11640b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Double, 0)); 11650b57cec5SDimitry Andric return; 11660b57cec5SDimitry Andric case IIT_F128: 11670b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Quad, 0)); 11680b57cec5SDimitry Andric return; 1169349cc55cSDimitry Andric case IIT_PPCF128: 1170349cc55cSDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::PPCQuad, 0)); 1171349cc55cSDimitry Andric return; 11720b57cec5SDimitry Andric case IIT_I1: 11730b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 1)); 11740b57cec5SDimitry Andric return; 117581ad6265SDimitry Andric case IIT_I2: 117681ad6265SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 2)); 117781ad6265SDimitry Andric return; 117881ad6265SDimitry Andric case IIT_I4: 117981ad6265SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 4)); 118081ad6265SDimitry Andric return; 118106c3fb27SDimitry Andric case IIT_AARCH64_SVCOUNT: 118206c3fb27SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::AArch64Svcount, 0)); 118306c3fb27SDimitry Andric return; 11840b57cec5SDimitry Andric case IIT_I8: 11850b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 8)); 11860b57cec5SDimitry Andric return; 11870b57cec5SDimitry Andric case IIT_I16: 11880b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer,16)); 11890b57cec5SDimitry Andric return; 11900b57cec5SDimitry Andric case IIT_I32: 11910b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 32)); 11920b57cec5SDimitry Andric return; 11930b57cec5SDimitry Andric case IIT_I64: 11940b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 64)); 11950b57cec5SDimitry Andric return; 11960b57cec5SDimitry Andric case IIT_I128: 11970b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 128)); 11980b57cec5SDimitry Andric return; 11990b57cec5SDimitry Andric case IIT_V1: 12005ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(1, IsScalableVector)); 12015ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12020b57cec5SDimitry Andric return; 12030b57cec5SDimitry Andric case IIT_V2: 12045ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(2, IsScalableVector)); 12055ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12060b57cec5SDimitry Andric return; 12070eae32dcSDimitry Andric case IIT_V3: 12080eae32dcSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(3, IsScalableVector)); 12090eae32dcSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12100eae32dcSDimitry Andric return; 12110b57cec5SDimitry Andric case IIT_V4: 12125ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(4, IsScalableVector)); 12135ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12140b57cec5SDimitry Andric return; 1215*0fca6ea1SDimitry Andric case IIT_V6: 1216*0fca6ea1SDimitry Andric OutputTable.push_back(IITDescriptor::getVector(6, IsScalableVector)); 1217*0fca6ea1SDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 1218*0fca6ea1SDimitry Andric return; 12190b57cec5SDimitry Andric case IIT_V8: 12205ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(8, IsScalableVector)); 12215ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12220b57cec5SDimitry Andric return; 1223*0fca6ea1SDimitry Andric case IIT_V10: 1224*0fca6ea1SDimitry Andric OutputTable.push_back(IITDescriptor::getVector(10, IsScalableVector)); 1225*0fca6ea1SDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 1226*0fca6ea1SDimitry Andric return; 12270b57cec5SDimitry Andric case IIT_V16: 12285ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(16, IsScalableVector)); 12295ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12300b57cec5SDimitry Andric return; 12310b57cec5SDimitry Andric case IIT_V32: 12325ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(32, IsScalableVector)); 12335ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12340b57cec5SDimitry Andric return; 12350b57cec5SDimitry Andric case IIT_V64: 12365ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(64, IsScalableVector)); 12375ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12385ffd83dbSDimitry Andric return; 12395ffd83dbSDimitry Andric case IIT_V128: 12405ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(128, IsScalableVector)); 12415ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12420b57cec5SDimitry Andric return; 1243e8d8bef9SDimitry Andric case IIT_V256: 1244e8d8bef9SDimitry Andric OutputTable.push_back(IITDescriptor::getVector(256, IsScalableVector)); 1245e8d8bef9SDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 1246e8d8bef9SDimitry Andric return; 12470b57cec5SDimitry Andric case IIT_V512: 12485ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(512, IsScalableVector)); 12495ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12500b57cec5SDimitry Andric return; 12510b57cec5SDimitry Andric case IIT_V1024: 12525ffd83dbSDimitry Andric OutputTable.push_back(IITDescriptor::getVector(1024, IsScalableVector)); 12535ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 12540b57cec5SDimitry Andric return; 12550eae32dcSDimitry Andric case IIT_EXTERNREF: 12560eae32dcSDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 10)); 12570eae32dcSDimitry Andric return; 12580eae32dcSDimitry Andric case IIT_FUNCREF: 12590eae32dcSDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 20)); 12600eae32dcSDimitry Andric return; 12610b57cec5SDimitry Andric case IIT_PTR: 12620b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 0)); 12630b57cec5SDimitry Andric return; 126406c3fb27SDimitry Andric case IIT_ANYPTR: // [ANYPTR addrspace] 12650b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 12660b57cec5SDimitry Andric Infos[NextElt++])); 12670b57cec5SDimitry Andric return; 12680b57cec5SDimitry Andric case IIT_ARG: { 12690b57cec5SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 12700b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Argument, ArgInfo)); 12710b57cec5SDimitry Andric return; 12720b57cec5SDimitry Andric } 12730b57cec5SDimitry Andric case IIT_EXTEND_ARG: { 12740b57cec5SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 12750b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::ExtendArgument, 12760b57cec5SDimitry Andric ArgInfo)); 12770b57cec5SDimitry Andric return; 12780b57cec5SDimitry Andric } 12790b57cec5SDimitry Andric case IIT_TRUNC_ARG: { 12800b57cec5SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 12810b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::TruncArgument, 12820b57cec5SDimitry Andric ArgInfo)); 12830b57cec5SDimitry Andric return; 12840b57cec5SDimitry Andric } 12850b57cec5SDimitry Andric case IIT_HALF_VEC_ARG: { 12860b57cec5SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 12870b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::HalfVecArgument, 12880b57cec5SDimitry Andric ArgInfo)); 12890b57cec5SDimitry Andric return; 12900b57cec5SDimitry Andric } 12910b57cec5SDimitry Andric case IIT_SAME_VEC_WIDTH_ARG: { 12920b57cec5SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 12930b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::SameVecWidthArgument, 12940b57cec5SDimitry Andric ArgInfo)); 12950b57cec5SDimitry Andric return; 12960b57cec5SDimitry Andric } 12970b57cec5SDimitry Andric case IIT_VEC_OF_ANYPTRS_TO_ELT: { 12980b57cec5SDimitry Andric unsigned short ArgNo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 12990b57cec5SDimitry Andric unsigned short RefNo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 13000b57cec5SDimitry Andric OutputTable.push_back( 13010b57cec5SDimitry Andric IITDescriptor::get(IITDescriptor::VecOfAnyPtrsToElt, ArgNo, RefNo)); 13020b57cec5SDimitry Andric return; 13030b57cec5SDimitry Andric } 13040b57cec5SDimitry Andric case IIT_EMPTYSTRUCT: 13050b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Struct, 0)); 13060b57cec5SDimitry Andric return; 1307bdd1243dSDimitry Andric case IIT_STRUCT9: ++StructElts; [[fallthrough]]; 1308bdd1243dSDimitry Andric case IIT_STRUCT8: ++StructElts; [[fallthrough]]; 1309bdd1243dSDimitry Andric case IIT_STRUCT7: ++StructElts; [[fallthrough]]; 1310bdd1243dSDimitry Andric case IIT_STRUCT6: ++StructElts; [[fallthrough]]; 1311bdd1243dSDimitry Andric case IIT_STRUCT5: ++StructElts; [[fallthrough]]; 1312bdd1243dSDimitry Andric case IIT_STRUCT4: ++StructElts; [[fallthrough]]; 1313bdd1243dSDimitry Andric case IIT_STRUCT3: ++StructElts; [[fallthrough]]; 13140b57cec5SDimitry Andric case IIT_STRUCT2: { 13150b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Struct,StructElts)); 13160b57cec5SDimitry Andric 13170b57cec5SDimitry Andric for (unsigned i = 0; i != StructElts; ++i) 13185ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 13190b57cec5SDimitry Andric return; 13200b57cec5SDimitry Andric } 13218bcb0991SDimitry Andric case IIT_SUBDIVIDE2_ARG: { 13228bcb0991SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 13238bcb0991SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Subdivide2Argument, 13248bcb0991SDimitry Andric ArgInfo)); 13258bcb0991SDimitry Andric return; 13268bcb0991SDimitry Andric } 13278bcb0991SDimitry Andric case IIT_SUBDIVIDE4_ARG: { 13288bcb0991SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 13298bcb0991SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::Subdivide4Argument, 13308bcb0991SDimitry Andric ArgInfo)); 13318bcb0991SDimitry Andric return; 13328bcb0991SDimitry Andric } 13330b57cec5SDimitry Andric case IIT_VEC_ELEMENT: { 13340b57cec5SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 13350b57cec5SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::VecElementArgument, 13360b57cec5SDimitry Andric ArgInfo)); 13370b57cec5SDimitry Andric return; 13380b57cec5SDimitry Andric } 13398bcb0991SDimitry Andric case IIT_SCALABLE_VEC: { 13405ffd83dbSDimitry Andric DecodeIITType(NextElt, Infos, Info, OutputTable); 13418bcb0991SDimitry Andric return; 13428bcb0991SDimitry Andric } 13438bcb0991SDimitry Andric case IIT_VEC_OF_BITCASTS_TO_INT: { 13448bcb0991SDimitry Andric unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]); 13458bcb0991SDimitry Andric OutputTable.push_back(IITDescriptor::get(IITDescriptor::VecOfBitcastsToInt, 13468bcb0991SDimitry Andric ArgInfo)); 13478bcb0991SDimitry Andric return; 13488bcb0991SDimitry Andric } 13490b57cec5SDimitry Andric } 13500b57cec5SDimitry Andric llvm_unreachable("unhandled"); 13510b57cec5SDimitry Andric } 13520b57cec5SDimitry Andric 13530b57cec5SDimitry Andric #define GET_INTRINSIC_GENERATOR_GLOBAL 13540b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 13550b57cec5SDimitry Andric #undef GET_INTRINSIC_GENERATOR_GLOBAL 13560b57cec5SDimitry Andric 13570b57cec5SDimitry Andric void Intrinsic::getIntrinsicInfoTableEntries(ID id, 13580b57cec5SDimitry Andric SmallVectorImpl<IITDescriptor> &T){ 13590b57cec5SDimitry Andric // Check to see if the intrinsic's type was expressible by the table. 13600b57cec5SDimitry Andric unsigned TableVal = IIT_Table[id-1]; 13610b57cec5SDimitry Andric 13620b57cec5SDimitry Andric // Decode the TableVal into an array of IITValues. 13630b57cec5SDimitry Andric SmallVector<unsigned char, 8> IITValues; 13640b57cec5SDimitry Andric ArrayRef<unsigned char> IITEntries; 13650b57cec5SDimitry Andric unsigned NextElt = 0; 13660b57cec5SDimitry Andric if ((TableVal >> 31) != 0) { 13670b57cec5SDimitry Andric // This is an offset into the IIT_LongEncodingTable. 13680b57cec5SDimitry Andric IITEntries = IIT_LongEncodingTable; 13690b57cec5SDimitry Andric 13700b57cec5SDimitry Andric // Strip sentinel bit. 13710b57cec5SDimitry Andric NextElt = (TableVal << 1) >> 1; 13720b57cec5SDimitry Andric } else { 13730b57cec5SDimitry Andric // Decode the TableVal into an array of IITValues. If the entry was encoded 13740b57cec5SDimitry Andric // into a single word in the table itself, decode it now. 13750b57cec5SDimitry Andric do { 13760b57cec5SDimitry Andric IITValues.push_back(TableVal & 0xF); 13770b57cec5SDimitry Andric TableVal >>= 4; 13780b57cec5SDimitry Andric } while (TableVal); 13790b57cec5SDimitry Andric 13800b57cec5SDimitry Andric IITEntries = IITValues; 13810b57cec5SDimitry Andric NextElt = 0; 13820b57cec5SDimitry Andric } 13830b57cec5SDimitry Andric 13840b57cec5SDimitry Andric // Okay, decode the table into the output vector of IITDescriptors. 13855ffd83dbSDimitry Andric DecodeIITType(NextElt, IITEntries, IIT_Done, T); 13860b57cec5SDimitry Andric while (NextElt != IITEntries.size() && IITEntries[NextElt] != 0) 13875ffd83dbSDimitry Andric DecodeIITType(NextElt, IITEntries, IIT_Done, T); 13880b57cec5SDimitry Andric } 13890b57cec5SDimitry Andric 13900b57cec5SDimitry Andric static Type *DecodeFixedType(ArrayRef<Intrinsic::IITDescriptor> &Infos, 13910b57cec5SDimitry Andric ArrayRef<Type*> Tys, LLVMContext &Context) { 13920b57cec5SDimitry Andric using namespace Intrinsic; 13930b57cec5SDimitry Andric 13940b57cec5SDimitry Andric IITDescriptor D = Infos.front(); 13950b57cec5SDimitry Andric Infos = Infos.slice(1); 13960b57cec5SDimitry Andric 13970b57cec5SDimitry Andric switch (D.Kind) { 13980b57cec5SDimitry Andric case IITDescriptor::Void: return Type::getVoidTy(Context); 13990b57cec5SDimitry Andric case IITDescriptor::VarArg: return Type::getVoidTy(Context); 14000b57cec5SDimitry Andric case IITDescriptor::MMX: return Type::getX86_MMXTy(Context); 1401e8d8bef9SDimitry Andric case IITDescriptor::AMX: return Type::getX86_AMXTy(Context); 14020b57cec5SDimitry Andric case IITDescriptor::Token: return Type::getTokenTy(Context); 14030b57cec5SDimitry Andric case IITDescriptor::Metadata: return Type::getMetadataTy(Context); 14040b57cec5SDimitry Andric case IITDescriptor::Half: return Type::getHalfTy(Context); 14055ffd83dbSDimitry Andric case IITDescriptor::BFloat: return Type::getBFloatTy(Context); 14060b57cec5SDimitry Andric case IITDescriptor::Float: return Type::getFloatTy(Context); 14070b57cec5SDimitry Andric case IITDescriptor::Double: return Type::getDoubleTy(Context); 14080b57cec5SDimitry Andric case IITDescriptor::Quad: return Type::getFP128Ty(Context); 1409349cc55cSDimitry Andric case IITDescriptor::PPCQuad: return Type::getPPC_FP128Ty(Context); 141006c3fb27SDimitry Andric case IITDescriptor::AArch64Svcount: 141106c3fb27SDimitry Andric return TargetExtType::get(Context, "aarch64.svcount"); 14120b57cec5SDimitry Andric 14130b57cec5SDimitry Andric case IITDescriptor::Integer: 14140b57cec5SDimitry Andric return IntegerType::get(Context, D.Integer_Width); 14150b57cec5SDimitry Andric case IITDescriptor::Vector: 14165ffd83dbSDimitry Andric return VectorType::get(DecodeFixedType(Infos, Tys, Context), 14175ffd83dbSDimitry Andric D.Vector_Width); 14180b57cec5SDimitry Andric case IITDescriptor::Pointer: 141906c3fb27SDimitry Andric return PointerType::get(Context, D.Pointer_AddressSpace); 14200b57cec5SDimitry Andric case IITDescriptor::Struct: { 14210b57cec5SDimitry Andric SmallVector<Type *, 8> Elts; 14220b57cec5SDimitry Andric for (unsigned i = 0, e = D.Struct_NumElements; i != e; ++i) 14230b57cec5SDimitry Andric Elts.push_back(DecodeFixedType(Infos, Tys, Context)); 14240b57cec5SDimitry Andric return StructType::get(Context, Elts); 14250b57cec5SDimitry Andric } 14260b57cec5SDimitry Andric case IITDescriptor::Argument: 14270b57cec5SDimitry Andric return Tys[D.getArgumentNumber()]; 14280b57cec5SDimitry Andric case IITDescriptor::ExtendArgument: { 14290b57cec5SDimitry Andric Type *Ty = Tys[D.getArgumentNumber()]; 14300b57cec5SDimitry Andric if (VectorType *VTy = dyn_cast<VectorType>(Ty)) 14310b57cec5SDimitry Andric return VectorType::getExtendedElementVectorType(VTy); 14320b57cec5SDimitry Andric 14330b57cec5SDimitry Andric return IntegerType::get(Context, 2 * cast<IntegerType>(Ty)->getBitWidth()); 14340b57cec5SDimitry Andric } 14350b57cec5SDimitry Andric case IITDescriptor::TruncArgument: { 14360b57cec5SDimitry Andric Type *Ty = Tys[D.getArgumentNumber()]; 14370b57cec5SDimitry Andric if (VectorType *VTy = dyn_cast<VectorType>(Ty)) 14380b57cec5SDimitry Andric return VectorType::getTruncatedElementVectorType(VTy); 14390b57cec5SDimitry Andric 14400b57cec5SDimitry Andric IntegerType *ITy = cast<IntegerType>(Ty); 14410b57cec5SDimitry Andric assert(ITy->getBitWidth() % 2 == 0); 14420b57cec5SDimitry Andric return IntegerType::get(Context, ITy->getBitWidth() / 2); 14430b57cec5SDimitry Andric } 14448bcb0991SDimitry Andric case IITDescriptor::Subdivide2Argument: 14458bcb0991SDimitry Andric case IITDescriptor::Subdivide4Argument: { 14468bcb0991SDimitry Andric Type *Ty = Tys[D.getArgumentNumber()]; 14478bcb0991SDimitry Andric VectorType *VTy = dyn_cast<VectorType>(Ty); 14488bcb0991SDimitry Andric assert(VTy && "Expected an argument of Vector Type"); 14498bcb0991SDimitry Andric int SubDivs = D.Kind == IITDescriptor::Subdivide2Argument ? 1 : 2; 14508bcb0991SDimitry Andric return VectorType::getSubdividedVectorType(VTy, SubDivs); 14518bcb0991SDimitry Andric } 14520b57cec5SDimitry Andric case IITDescriptor::HalfVecArgument: 14530b57cec5SDimitry Andric return VectorType::getHalfElementsVectorType(cast<VectorType>( 14540b57cec5SDimitry Andric Tys[D.getArgumentNumber()])); 14550b57cec5SDimitry Andric case IITDescriptor::SameVecWidthArgument: { 14560b57cec5SDimitry Andric Type *EltTy = DecodeFixedType(Infos, Tys, Context); 14570b57cec5SDimitry Andric Type *Ty = Tys[D.getArgumentNumber()]; 14580b57cec5SDimitry Andric if (auto *VTy = dyn_cast<VectorType>(Ty)) 14598bcb0991SDimitry Andric return VectorType::get(EltTy, VTy->getElementCount()); 14600b57cec5SDimitry Andric return EltTy; 14610b57cec5SDimitry Andric } 14620b57cec5SDimitry Andric case IITDescriptor::VecElementArgument: { 14630b57cec5SDimitry Andric Type *Ty = Tys[D.getArgumentNumber()]; 14640b57cec5SDimitry Andric if (VectorType *VTy = dyn_cast<VectorType>(Ty)) 14650b57cec5SDimitry Andric return VTy->getElementType(); 14660b57cec5SDimitry Andric llvm_unreachable("Expected an argument of Vector Type"); 14670b57cec5SDimitry Andric } 14688bcb0991SDimitry Andric case IITDescriptor::VecOfBitcastsToInt: { 14698bcb0991SDimitry Andric Type *Ty = Tys[D.getArgumentNumber()]; 14708bcb0991SDimitry Andric VectorType *VTy = dyn_cast<VectorType>(Ty); 14718bcb0991SDimitry Andric assert(VTy && "Expected an argument of Vector Type"); 14728bcb0991SDimitry Andric return VectorType::getInteger(VTy); 14738bcb0991SDimitry Andric } 14740b57cec5SDimitry Andric case IITDescriptor::VecOfAnyPtrsToElt: 14750b57cec5SDimitry Andric // Return the overloaded type (which determines the pointers address space) 14760b57cec5SDimitry Andric return Tys[D.getOverloadArgNumber()]; 14770b57cec5SDimitry Andric } 14780b57cec5SDimitry Andric llvm_unreachable("unhandled"); 14790b57cec5SDimitry Andric } 14800b57cec5SDimitry Andric 14810b57cec5SDimitry Andric FunctionType *Intrinsic::getType(LLVMContext &Context, 14820b57cec5SDimitry Andric ID id, ArrayRef<Type*> Tys) { 14830b57cec5SDimitry Andric SmallVector<IITDescriptor, 8> Table; 14840b57cec5SDimitry Andric getIntrinsicInfoTableEntries(id, Table); 14850b57cec5SDimitry Andric 14860b57cec5SDimitry Andric ArrayRef<IITDescriptor> TableRef = Table; 14870b57cec5SDimitry Andric Type *ResultTy = DecodeFixedType(TableRef, Tys, Context); 14880b57cec5SDimitry Andric 14890b57cec5SDimitry Andric SmallVector<Type*, 8> ArgTys; 14900b57cec5SDimitry Andric while (!TableRef.empty()) 14910b57cec5SDimitry Andric ArgTys.push_back(DecodeFixedType(TableRef, Tys, Context)); 14920b57cec5SDimitry Andric 14930b57cec5SDimitry Andric // DecodeFixedType returns Void for IITDescriptor::Void and IITDescriptor::VarArg 14940b57cec5SDimitry Andric // If we see void type as the type of the last argument, it is vararg intrinsic 14950b57cec5SDimitry Andric if (!ArgTys.empty() && ArgTys.back()->isVoidTy()) { 14960b57cec5SDimitry Andric ArgTys.pop_back(); 14970b57cec5SDimitry Andric return FunctionType::get(ResultTy, ArgTys, true); 14980b57cec5SDimitry Andric } 14990b57cec5SDimitry Andric return FunctionType::get(ResultTy, ArgTys, false); 15000b57cec5SDimitry Andric } 15010b57cec5SDimitry Andric 15020b57cec5SDimitry Andric bool Intrinsic::isOverloaded(ID id) { 15030b57cec5SDimitry Andric #define GET_INTRINSIC_OVERLOAD_TABLE 15040b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 15050b57cec5SDimitry Andric #undef GET_INTRINSIC_OVERLOAD_TABLE 15060b57cec5SDimitry Andric } 15070b57cec5SDimitry Andric 15080b57cec5SDimitry Andric /// This defines the "Intrinsic::getAttributes(ID id)" method. 15090b57cec5SDimitry Andric #define GET_INTRINSIC_ATTRIBUTES 15100b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 15110b57cec5SDimitry Andric #undef GET_INTRINSIC_ATTRIBUTES 15120b57cec5SDimitry Andric 15130b57cec5SDimitry Andric Function *Intrinsic::getDeclaration(Module *M, ID id, ArrayRef<Type*> Tys) { 15140b57cec5SDimitry Andric // There can never be multiple globals with the same name of different types, 15150b57cec5SDimitry Andric // because intrinsics must be a specific type. 1516fe6060f1SDimitry Andric auto *FT = getType(M->getContext(), id, Tys); 15170b57cec5SDimitry Andric return cast<Function>( 1518bdd1243dSDimitry Andric M->getOrInsertFunction( 1519bdd1243dSDimitry Andric Tys.empty() ? getName(id) : getName(id, Tys, M, FT), FT) 15200b57cec5SDimitry Andric .getCallee()); 15210b57cec5SDimitry Andric } 15220b57cec5SDimitry Andric 152381ad6265SDimitry Andric // This defines the "Intrinsic::getIntrinsicForClangBuiltin()" method. 152481ad6265SDimitry Andric #define GET_LLVM_INTRINSIC_FOR_CLANG_BUILTIN 15250b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 152681ad6265SDimitry Andric #undef GET_LLVM_INTRINSIC_FOR_CLANG_BUILTIN 15270b57cec5SDimitry Andric 15280b57cec5SDimitry Andric // This defines the "Intrinsic::getIntrinsicForMSBuiltin()" method. 15290b57cec5SDimitry Andric #define GET_LLVM_INTRINSIC_FOR_MS_BUILTIN 15300b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc" 15310b57cec5SDimitry Andric #undef GET_LLVM_INTRINSIC_FOR_MS_BUILTIN 15320b57cec5SDimitry Andric 1533*0fca6ea1SDimitry Andric bool Intrinsic::isConstrainedFPIntrinsic(ID QID) { 1534*0fca6ea1SDimitry Andric switch (QID) { 1535*0fca6ea1SDimitry Andric #define INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC) \ 1536*0fca6ea1SDimitry Andric case Intrinsic::INTRINSIC: 1537*0fca6ea1SDimitry Andric #include "llvm/IR/ConstrainedOps.def" 1538*0fca6ea1SDimitry Andric #undef INSTRUCTION 1539*0fca6ea1SDimitry Andric return true; 1540*0fca6ea1SDimitry Andric default: 1541*0fca6ea1SDimitry Andric return false; 1542*0fca6ea1SDimitry Andric } 1543*0fca6ea1SDimitry Andric } 1544*0fca6ea1SDimitry Andric 1545*0fca6ea1SDimitry Andric bool Intrinsic::hasConstrainedFPRoundingModeOperand(Intrinsic::ID QID) { 1546*0fca6ea1SDimitry Andric switch (QID) { 1547*0fca6ea1SDimitry Andric #define INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC) \ 1548*0fca6ea1SDimitry Andric case Intrinsic::INTRINSIC: \ 1549*0fca6ea1SDimitry Andric return ROUND_MODE == 1; 1550*0fca6ea1SDimitry Andric #include "llvm/IR/ConstrainedOps.def" 1551*0fca6ea1SDimitry Andric #undef INSTRUCTION 1552*0fca6ea1SDimitry Andric default: 1553*0fca6ea1SDimitry Andric return false; 1554*0fca6ea1SDimitry Andric } 1555*0fca6ea1SDimitry Andric } 1556*0fca6ea1SDimitry Andric 15570b57cec5SDimitry Andric using DeferredIntrinsicMatchPair = 15580b57cec5SDimitry Andric std::pair<Type *, ArrayRef<Intrinsic::IITDescriptor>>; 15590b57cec5SDimitry Andric 15600b57cec5SDimitry Andric static bool matchIntrinsicType( 15610b57cec5SDimitry Andric Type *Ty, ArrayRef<Intrinsic::IITDescriptor> &Infos, 15620b57cec5SDimitry Andric SmallVectorImpl<Type *> &ArgTys, 15630b57cec5SDimitry Andric SmallVectorImpl<DeferredIntrinsicMatchPair> &DeferredChecks, 15640b57cec5SDimitry Andric bool IsDeferredCheck) { 15650b57cec5SDimitry Andric using namespace Intrinsic; 15660b57cec5SDimitry Andric 15670b57cec5SDimitry Andric // If we ran out of descriptors, there are too many arguments. 15680b57cec5SDimitry Andric if (Infos.empty()) return true; 15690b57cec5SDimitry Andric 15700b57cec5SDimitry Andric // Do this before slicing off the 'front' part 15710b57cec5SDimitry Andric auto InfosRef = Infos; 15720b57cec5SDimitry Andric auto DeferCheck = [&DeferredChecks, &InfosRef](Type *T) { 15730b57cec5SDimitry Andric DeferredChecks.emplace_back(T, InfosRef); 15740b57cec5SDimitry Andric return false; 15750b57cec5SDimitry Andric }; 15760b57cec5SDimitry Andric 15770b57cec5SDimitry Andric IITDescriptor D = Infos.front(); 15780b57cec5SDimitry Andric Infos = Infos.slice(1); 15790b57cec5SDimitry Andric 15800b57cec5SDimitry Andric switch (D.Kind) { 15810b57cec5SDimitry Andric case IITDescriptor::Void: return !Ty->isVoidTy(); 15820b57cec5SDimitry Andric case IITDescriptor::VarArg: return true; 15830b57cec5SDimitry Andric case IITDescriptor::MMX: return !Ty->isX86_MMXTy(); 1584e8d8bef9SDimitry Andric case IITDescriptor::AMX: return !Ty->isX86_AMXTy(); 15850b57cec5SDimitry Andric case IITDescriptor::Token: return !Ty->isTokenTy(); 15860b57cec5SDimitry Andric case IITDescriptor::Metadata: return !Ty->isMetadataTy(); 15870b57cec5SDimitry Andric case IITDescriptor::Half: return !Ty->isHalfTy(); 15885ffd83dbSDimitry Andric case IITDescriptor::BFloat: return !Ty->isBFloatTy(); 15890b57cec5SDimitry Andric case IITDescriptor::Float: return !Ty->isFloatTy(); 15900b57cec5SDimitry Andric case IITDescriptor::Double: return !Ty->isDoubleTy(); 15910b57cec5SDimitry Andric case IITDescriptor::Quad: return !Ty->isFP128Ty(); 1592349cc55cSDimitry Andric case IITDescriptor::PPCQuad: return !Ty->isPPC_FP128Ty(); 15930b57cec5SDimitry Andric case IITDescriptor::Integer: return !Ty->isIntegerTy(D.Integer_Width); 159406c3fb27SDimitry Andric case IITDescriptor::AArch64Svcount: 159506c3fb27SDimitry Andric return !isa<TargetExtType>(Ty) || 159606c3fb27SDimitry Andric cast<TargetExtType>(Ty)->getName() != "aarch64.svcount"; 15970b57cec5SDimitry Andric case IITDescriptor::Vector: { 15980b57cec5SDimitry Andric VectorType *VT = dyn_cast<VectorType>(Ty); 15995ffd83dbSDimitry Andric return !VT || VT->getElementCount() != D.Vector_Width || 16000b57cec5SDimitry Andric matchIntrinsicType(VT->getElementType(), Infos, ArgTys, 16010b57cec5SDimitry Andric DeferredChecks, IsDeferredCheck); 16020b57cec5SDimitry Andric } 16030b57cec5SDimitry Andric case IITDescriptor::Pointer: { 16040b57cec5SDimitry Andric PointerType *PT = dyn_cast<PointerType>(Ty); 160506c3fb27SDimitry Andric return !PT || PT->getAddressSpace() != D.Pointer_AddressSpace; 16060b57cec5SDimitry Andric } 16070b57cec5SDimitry Andric 16080b57cec5SDimitry Andric case IITDescriptor::Struct: { 16090b57cec5SDimitry Andric StructType *ST = dyn_cast<StructType>(Ty); 161081ad6265SDimitry Andric if (!ST || !ST->isLiteral() || ST->isPacked() || 161181ad6265SDimitry Andric ST->getNumElements() != D.Struct_NumElements) 16120b57cec5SDimitry Andric return true; 16130b57cec5SDimitry Andric 16140b57cec5SDimitry Andric for (unsigned i = 0, e = D.Struct_NumElements; i != e; ++i) 16150b57cec5SDimitry Andric if (matchIntrinsicType(ST->getElementType(i), Infos, ArgTys, 16160b57cec5SDimitry Andric DeferredChecks, IsDeferredCheck)) 16170b57cec5SDimitry Andric return true; 16180b57cec5SDimitry Andric return false; 16190b57cec5SDimitry Andric } 16200b57cec5SDimitry Andric 16210b57cec5SDimitry Andric case IITDescriptor::Argument: 16220b57cec5SDimitry Andric // If this is the second occurrence of an argument, 16230b57cec5SDimitry Andric // verify that the later instance matches the previous instance. 16240b57cec5SDimitry Andric if (D.getArgumentNumber() < ArgTys.size()) 16250b57cec5SDimitry Andric return Ty != ArgTys[D.getArgumentNumber()]; 16260b57cec5SDimitry Andric 16270b57cec5SDimitry Andric if (D.getArgumentNumber() > ArgTys.size() || 16280b57cec5SDimitry Andric D.getArgumentKind() == IITDescriptor::AK_MatchType) 16290b57cec5SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 16300b57cec5SDimitry Andric 16310b57cec5SDimitry Andric assert(D.getArgumentNumber() == ArgTys.size() && !IsDeferredCheck && 16320b57cec5SDimitry Andric "Table consistency error"); 16330b57cec5SDimitry Andric ArgTys.push_back(Ty); 16340b57cec5SDimitry Andric 16350b57cec5SDimitry Andric switch (D.getArgumentKind()) { 16360b57cec5SDimitry Andric case IITDescriptor::AK_Any: return false; // Success 16370b57cec5SDimitry Andric case IITDescriptor::AK_AnyInteger: return !Ty->isIntOrIntVectorTy(); 16380b57cec5SDimitry Andric case IITDescriptor::AK_AnyFloat: return !Ty->isFPOrFPVectorTy(); 16390b57cec5SDimitry Andric case IITDescriptor::AK_AnyVector: return !isa<VectorType>(Ty); 16400b57cec5SDimitry Andric case IITDescriptor::AK_AnyPointer: return !isa<PointerType>(Ty); 16410b57cec5SDimitry Andric default: break; 16420b57cec5SDimitry Andric } 16430b57cec5SDimitry Andric llvm_unreachable("all argument kinds not covered"); 16440b57cec5SDimitry Andric 16450b57cec5SDimitry Andric case IITDescriptor::ExtendArgument: { 16460b57cec5SDimitry Andric // If this is a forward reference, defer the check for later. 16470b57cec5SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) 16480b57cec5SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 16490b57cec5SDimitry Andric 16500b57cec5SDimitry Andric Type *NewTy = ArgTys[D.getArgumentNumber()]; 16510b57cec5SDimitry Andric if (VectorType *VTy = dyn_cast<VectorType>(NewTy)) 16520b57cec5SDimitry Andric NewTy = VectorType::getExtendedElementVectorType(VTy); 16530b57cec5SDimitry Andric else if (IntegerType *ITy = dyn_cast<IntegerType>(NewTy)) 16540b57cec5SDimitry Andric NewTy = IntegerType::get(ITy->getContext(), 2 * ITy->getBitWidth()); 16550b57cec5SDimitry Andric else 16560b57cec5SDimitry Andric return true; 16570b57cec5SDimitry Andric 16580b57cec5SDimitry Andric return Ty != NewTy; 16590b57cec5SDimitry Andric } 16600b57cec5SDimitry Andric case IITDescriptor::TruncArgument: { 16610b57cec5SDimitry Andric // If this is a forward reference, defer the check for later. 16620b57cec5SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) 16630b57cec5SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 16640b57cec5SDimitry Andric 16650b57cec5SDimitry Andric Type *NewTy = ArgTys[D.getArgumentNumber()]; 16660b57cec5SDimitry Andric if (VectorType *VTy = dyn_cast<VectorType>(NewTy)) 16670b57cec5SDimitry Andric NewTy = VectorType::getTruncatedElementVectorType(VTy); 16680b57cec5SDimitry Andric else if (IntegerType *ITy = dyn_cast<IntegerType>(NewTy)) 16690b57cec5SDimitry Andric NewTy = IntegerType::get(ITy->getContext(), ITy->getBitWidth() / 2); 16700b57cec5SDimitry Andric else 16710b57cec5SDimitry Andric return true; 16720b57cec5SDimitry Andric 16730b57cec5SDimitry Andric return Ty != NewTy; 16740b57cec5SDimitry Andric } 16750b57cec5SDimitry Andric case IITDescriptor::HalfVecArgument: 16760b57cec5SDimitry Andric // If this is a forward reference, defer the check for later. 16778bcb0991SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) 16788bcb0991SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 16798bcb0991SDimitry Andric return !isa<VectorType>(ArgTys[D.getArgumentNumber()]) || 16800b57cec5SDimitry Andric VectorType::getHalfElementsVectorType( 16810b57cec5SDimitry Andric cast<VectorType>(ArgTys[D.getArgumentNumber()])) != Ty; 16820b57cec5SDimitry Andric case IITDescriptor::SameVecWidthArgument: { 16830b57cec5SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) { 16840b57cec5SDimitry Andric // Defer check and subsequent check for the vector element type. 16850b57cec5SDimitry Andric Infos = Infos.slice(1); 16860b57cec5SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 16870b57cec5SDimitry Andric } 16880b57cec5SDimitry Andric auto *ReferenceType = dyn_cast<VectorType>(ArgTys[D.getArgumentNumber()]); 16890b57cec5SDimitry Andric auto *ThisArgType = dyn_cast<VectorType>(Ty); 16900b57cec5SDimitry Andric // Both must be vectors of the same number of elements or neither. 16910b57cec5SDimitry Andric if ((ReferenceType != nullptr) != (ThisArgType != nullptr)) 16920b57cec5SDimitry Andric return true; 16930b57cec5SDimitry Andric Type *EltTy = Ty; 16940b57cec5SDimitry Andric if (ThisArgType) { 16958bcb0991SDimitry Andric if (ReferenceType->getElementCount() != 16968bcb0991SDimitry Andric ThisArgType->getElementCount()) 16970b57cec5SDimitry Andric return true; 16985ffd83dbSDimitry Andric EltTy = ThisArgType->getElementType(); 16990b57cec5SDimitry Andric } 17000b57cec5SDimitry Andric return matchIntrinsicType(EltTy, Infos, ArgTys, DeferredChecks, 17010b57cec5SDimitry Andric IsDeferredCheck); 17020b57cec5SDimitry Andric } 17030b57cec5SDimitry Andric case IITDescriptor::VecOfAnyPtrsToElt: { 17040b57cec5SDimitry Andric unsigned RefArgNumber = D.getRefArgNumber(); 17050b57cec5SDimitry Andric if (RefArgNumber >= ArgTys.size()) { 17060b57cec5SDimitry Andric if (IsDeferredCheck) 17070b57cec5SDimitry Andric return true; 17080b57cec5SDimitry Andric // If forward referencing, already add the pointer-vector type and 17090b57cec5SDimitry Andric // defer the checks for later. 17100b57cec5SDimitry Andric ArgTys.push_back(Ty); 17110b57cec5SDimitry Andric return DeferCheck(Ty); 17120b57cec5SDimitry Andric } 17130b57cec5SDimitry Andric 17140b57cec5SDimitry Andric if (!IsDeferredCheck){ 17150b57cec5SDimitry Andric assert(D.getOverloadArgNumber() == ArgTys.size() && 17160b57cec5SDimitry Andric "Table consistency error"); 17170b57cec5SDimitry Andric ArgTys.push_back(Ty); 17180b57cec5SDimitry Andric } 17190b57cec5SDimitry Andric 17200b57cec5SDimitry Andric // Verify the overloaded type "matches" the Ref type. 17210b57cec5SDimitry Andric // i.e. Ty is a vector with the same width as Ref. 17220b57cec5SDimitry Andric // Composed of pointers to the same element type as Ref. 1723e8d8bef9SDimitry Andric auto *ReferenceType = dyn_cast<VectorType>(ArgTys[RefArgNumber]); 1724e8d8bef9SDimitry Andric auto *ThisArgVecTy = dyn_cast<VectorType>(Ty); 17250b57cec5SDimitry Andric if (!ThisArgVecTy || !ReferenceType || 1726e8d8bef9SDimitry Andric (ReferenceType->getElementCount() != ThisArgVecTy->getElementCount())) 17270b57cec5SDimitry Andric return true; 172806c3fb27SDimitry Andric return !ThisArgVecTy->getElementType()->isPointerTy(); 17290b57cec5SDimitry Andric } 17300b57cec5SDimitry Andric case IITDescriptor::VecElementArgument: { 17310b57cec5SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) 17320b57cec5SDimitry Andric return IsDeferredCheck ? true : DeferCheck(Ty); 17330b57cec5SDimitry Andric auto *ReferenceType = dyn_cast<VectorType>(ArgTys[D.getArgumentNumber()]); 17340b57cec5SDimitry Andric return !ReferenceType || Ty != ReferenceType->getElementType(); 17350b57cec5SDimitry Andric } 17368bcb0991SDimitry Andric case IITDescriptor::Subdivide2Argument: 17378bcb0991SDimitry Andric case IITDescriptor::Subdivide4Argument: { 17388bcb0991SDimitry Andric // If this is a forward reference, defer the check for later. 17398bcb0991SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) 17408bcb0991SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 17418bcb0991SDimitry Andric 17428bcb0991SDimitry Andric Type *NewTy = ArgTys[D.getArgumentNumber()]; 17438bcb0991SDimitry Andric if (auto *VTy = dyn_cast<VectorType>(NewTy)) { 17448bcb0991SDimitry Andric int SubDivs = D.Kind == IITDescriptor::Subdivide2Argument ? 1 : 2; 17458bcb0991SDimitry Andric NewTy = VectorType::getSubdividedVectorType(VTy, SubDivs); 17468bcb0991SDimitry Andric return Ty != NewTy; 17478bcb0991SDimitry Andric } 17488bcb0991SDimitry Andric return true; 17498bcb0991SDimitry Andric } 17508bcb0991SDimitry Andric case IITDescriptor::VecOfBitcastsToInt: { 17518bcb0991SDimitry Andric if (D.getArgumentNumber() >= ArgTys.size()) 17528bcb0991SDimitry Andric return IsDeferredCheck || DeferCheck(Ty); 17538bcb0991SDimitry Andric auto *ReferenceType = dyn_cast<VectorType>(ArgTys[D.getArgumentNumber()]); 17548bcb0991SDimitry Andric auto *ThisArgVecTy = dyn_cast<VectorType>(Ty); 17558bcb0991SDimitry Andric if (!ThisArgVecTy || !ReferenceType) 17568bcb0991SDimitry Andric return true; 17578bcb0991SDimitry Andric return ThisArgVecTy != VectorType::getInteger(ReferenceType); 17588bcb0991SDimitry Andric } 17590b57cec5SDimitry Andric } 17600b57cec5SDimitry Andric llvm_unreachable("unhandled"); 17610b57cec5SDimitry Andric } 17620b57cec5SDimitry Andric 17630b57cec5SDimitry Andric Intrinsic::MatchIntrinsicTypesResult 17640b57cec5SDimitry Andric Intrinsic::matchIntrinsicSignature(FunctionType *FTy, 17650b57cec5SDimitry Andric ArrayRef<Intrinsic::IITDescriptor> &Infos, 17660b57cec5SDimitry Andric SmallVectorImpl<Type *> &ArgTys) { 17670b57cec5SDimitry Andric SmallVector<DeferredIntrinsicMatchPair, 2> DeferredChecks; 17680b57cec5SDimitry Andric if (matchIntrinsicType(FTy->getReturnType(), Infos, ArgTys, DeferredChecks, 17690b57cec5SDimitry Andric false)) 17700b57cec5SDimitry Andric return MatchIntrinsicTypes_NoMatchRet; 17710b57cec5SDimitry Andric 17720b57cec5SDimitry Andric unsigned NumDeferredReturnChecks = DeferredChecks.size(); 17730b57cec5SDimitry Andric 1774bdd1243dSDimitry Andric for (auto *Ty : FTy->params()) 17750b57cec5SDimitry Andric if (matchIntrinsicType(Ty, Infos, ArgTys, DeferredChecks, false)) 17760b57cec5SDimitry Andric return MatchIntrinsicTypes_NoMatchArg; 17770b57cec5SDimitry Andric 17780b57cec5SDimitry Andric for (unsigned I = 0, E = DeferredChecks.size(); I != E; ++I) { 17790b57cec5SDimitry Andric DeferredIntrinsicMatchPair &Check = DeferredChecks[I]; 17800b57cec5SDimitry Andric if (matchIntrinsicType(Check.first, Check.second, ArgTys, DeferredChecks, 17810b57cec5SDimitry Andric true)) 17820b57cec5SDimitry Andric return I < NumDeferredReturnChecks ? MatchIntrinsicTypes_NoMatchRet 17830b57cec5SDimitry Andric : MatchIntrinsicTypes_NoMatchArg; 17840b57cec5SDimitry Andric } 17850b57cec5SDimitry Andric 17860b57cec5SDimitry Andric return MatchIntrinsicTypes_Match; 17870b57cec5SDimitry Andric } 17880b57cec5SDimitry Andric 17890b57cec5SDimitry Andric bool 17900b57cec5SDimitry Andric Intrinsic::matchIntrinsicVarArg(bool isVarArg, 17910b57cec5SDimitry Andric ArrayRef<Intrinsic::IITDescriptor> &Infos) { 17920b57cec5SDimitry Andric // If there are no descriptors left, then it can't be a vararg. 17930b57cec5SDimitry Andric if (Infos.empty()) 17940b57cec5SDimitry Andric return isVarArg; 17950b57cec5SDimitry Andric 17960b57cec5SDimitry Andric // There should be only one descriptor remaining at this point. 17970b57cec5SDimitry Andric if (Infos.size() != 1) 17980b57cec5SDimitry Andric return true; 17990b57cec5SDimitry Andric 18000b57cec5SDimitry Andric // Check and verify the descriptor. 18010b57cec5SDimitry Andric IITDescriptor D = Infos.front(); 18020b57cec5SDimitry Andric Infos = Infos.slice(1); 18030b57cec5SDimitry Andric if (D.Kind == IITDescriptor::VarArg) 18040b57cec5SDimitry Andric return !isVarArg; 18050b57cec5SDimitry Andric 18060b57cec5SDimitry Andric return true; 18070b57cec5SDimitry Andric } 18080b57cec5SDimitry Andric 1809*0fca6ea1SDimitry Andric bool Intrinsic::getIntrinsicSignature(Intrinsic::ID ID, FunctionType *FT, 18105ffd83dbSDimitry Andric SmallVectorImpl<Type *> &ArgTys) { 18110b57cec5SDimitry Andric if (!ID) 18125ffd83dbSDimitry Andric return false; 18130b57cec5SDimitry Andric 18140b57cec5SDimitry Andric SmallVector<Intrinsic::IITDescriptor, 8> Table; 18150b57cec5SDimitry Andric getIntrinsicInfoTableEntries(ID, Table); 18160b57cec5SDimitry Andric ArrayRef<Intrinsic::IITDescriptor> TableRef = Table; 18170b57cec5SDimitry Andric 1818*0fca6ea1SDimitry Andric if (Intrinsic::matchIntrinsicSignature(FT, TableRef, ArgTys) != 18195ffd83dbSDimitry Andric Intrinsic::MatchIntrinsicTypesResult::MatchIntrinsicTypes_Match) { 18205ffd83dbSDimitry Andric return false; 18215ffd83dbSDimitry Andric } 1822*0fca6ea1SDimitry Andric if (Intrinsic::matchIntrinsicVarArg(FT->isVarArg(), TableRef)) 18235ffd83dbSDimitry Andric return false; 18245ffd83dbSDimitry Andric return true; 18250b57cec5SDimitry Andric } 18260b57cec5SDimitry Andric 1827*0fca6ea1SDimitry Andric bool Intrinsic::getIntrinsicSignature(Function *F, 1828*0fca6ea1SDimitry Andric SmallVectorImpl<Type *> &ArgTys) { 1829*0fca6ea1SDimitry Andric return getIntrinsicSignature(F->getIntrinsicID(), F->getFunctionType(), 1830*0fca6ea1SDimitry Andric ArgTys); 1831*0fca6ea1SDimitry Andric } 1832*0fca6ea1SDimitry Andric 1833bdd1243dSDimitry Andric std::optional<Function *> Intrinsic::remangleIntrinsicFunction(Function *F) { 18345ffd83dbSDimitry Andric SmallVector<Type *, 4> ArgTys; 18355ffd83dbSDimitry Andric if (!getIntrinsicSignature(F, ArgTys)) 1836bdd1243dSDimitry Andric return std::nullopt; 18375ffd83dbSDimitry Andric 18385ffd83dbSDimitry Andric Intrinsic::ID ID = F->getIntrinsicID(); 18390b57cec5SDimitry Andric StringRef Name = F->getName(); 1840fe6060f1SDimitry Andric std::string WantedName = 1841fe6060f1SDimitry Andric Intrinsic::getName(ID, ArgTys, F->getParent(), F->getFunctionType()); 1842fe6060f1SDimitry Andric if (Name == WantedName) 1843bdd1243dSDimitry Andric return std::nullopt; 18440b57cec5SDimitry Andric 1845fe6060f1SDimitry Andric Function *NewDecl = [&] { 1846fe6060f1SDimitry Andric if (auto *ExistingGV = F->getParent()->getNamedValue(WantedName)) { 1847fe6060f1SDimitry Andric if (auto *ExistingF = dyn_cast<Function>(ExistingGV)) 1848fe6060f1SDimitry Andric if (ExistingF->getFunctionType() == F->getFunctionType()) 1849fe6060f1SDimitry Andric return ExistingF; 1850fe6060f1SDimitry Andric 1851fe6060f1SDimitry Andric // The name already exists, but is not a function or has the wrong 1852fe6060f1SDimitry Andric // prototype. Make place for the new one by renaming the old version. 1853fe6060f1SDimitry Andric // Either this old version will be removed later on or the module is 1854fe6060f1SDimitry Andric // invalid and we'll get an error. 1855fe6060f1SDimitry Andric ExistingGV->setName(WantedName + ".renamed"); 1856fe6060f1SDimitry Andric } 1857fe6060f1SDimitry Andric return Intrinsic::getDeclaration(F->getParent(), ID, ArgTys); 1858fe6060f1SDimitry Andric }(); 1859fe6060f1SDimitry Andric 18600b57cec5SDimitry Andric NewDecl->setCallingConv(F->getCallingConv()); 18615ffd83dbSDimitry Andric assert(NewDecl->getFunctionType() == F->getFunctionType() && 18625ffd83dbSDimitry Andric "Shouldn't change the signature"); 18630b57cec5SDimitry Andric return NewDecl; 18640b57cec5SDimitry Andric } 18650b57cec5SDimitry Andric 18660b57cec5SDimitry Andric /// hasAddressTaken - returns true if there are any uses of this function 18675ffd83dbSDimitry Andric /// other than direct calls or invokes to it. Optionally ignores callback 1868fe6060f1SDimitry Andric /// uses, assume like pointer annotation calls, and references in llvm.used 1869fe6060f1SDimitry Andric /// and llvm.compiler.used variables. 18705ffd83dbSDimitry Andric bool Function::hasAddressTaken(const User **PutOffender, 1871fe6060f1SDimitry Andric bool IgnoreCallbackUses, 1872349cc55cSDimitry Andric bool IgnoreAssumeLikeCalls, bool IgnoreLLVMUsed, 18735f757f3fSDimitry Andric bool IgnoreARCAttachedCall, 18745f757f3fSDimitry Andric bool IgnoreCastedDirectCall) const { 18750b57cec5SDimitry Andric for (const Use &U : uses()) { 18760b57cec5SDimitry Andric const User *FU = U.getUser(); 18770b57cec5SDimitry Andric if (isa<BlockAddress>(FU)) 18780b57cec5SDimitry Andric continue; 18795ffd83dbSDimitry Andric 18805ffd83dbSDimitry Andric if (IgnoreCallbackUses) { 18815ffd83dbSDimitry Andric AbstractCallSite ACS(&U); 18825ffd83dbSDimitry Andric if (ACS && ACS.isCallbackCall()) 18835ffd83dbSDimitry Andric continue; 18845ffd83dbSDimitry Andric } 18855ffd83dbSDimitry Andric 18860b57cec5SDimitry Andric const auto *Call = dyn_cast<CallBase>(FU); 18870b57cec5SDimitry Andric if (!Call) { 1888bdd1243dSDimitry Andric if (IgnoreAssumeLikeCalls && 1889bdd1243dSDimitry Andric isa<BitCastOperator, AddrSpaceCastOperator>(FU) && 1890bdd1243dSDimitry Andric all_of(FU->users(), [](const User *U) { 1891fe6060f1SDimitry Andric if (const auto *I = dyn_cast<IntrinsicInst>(U)) 1892fe6060f1SDimitry Andric return I->isAssumeLikeIntrinsic(); 1893fe6060f1SDimitry Andric return false; 1894bdd1243dSDimitry Andric })) { 1895fe6060f1SDimitry Andric continue; 1896fe6060f1SDimitry Andric } 1897bdd1243dSDimitry Andric 1898fe6060f1SDimitry Andric if (IgnoreLLVMUsed && !FU->user_empty()) { 1899fe6060f1SDimitry Andric const User *FUU = FU; 1900bdd1243dSDimitry Andric if (isa<BitCastOperator, AddrSpaceCastOperator>(FU) && 1901bdd1243dSDimitry Andric FU->hasOneUse() && !FU->user_begin()->user_empty()) 1902fe6060f1SDimitry Andric FUU = *FU->user_begin(); 1903fe6060f1SDimitry Andric if (llvm::all_of(FUU->users(), [](const User *U) { 1904fe6060f1SDimitry Andric if (const auto *GV = dyn_cast<GlobalVariable>(U)) 1905fe6060f1SDimitry Andric return GV->hasName() && 1906*0fca6ea1SDimitry Andric (GV->getName() == "llvm.compiler.used" || 1907*0fca6ea1SDimitry Andric GV->getName() == "llvm.used"); 1908fe6060f1SDimitry Andric return false; 1909fe6060f1SDimitry Andric })) 1910fe6060f1SDimitry Andric continue; 1911fe6060f1SDimitry Andric } 19120b57cec5SDimitry Andric if (PutOffender) 19130b57cec5SDimitry Andric *PutOffender = FU; 19140b57cec5SDimitry Andric return true; 19150b57cec5SDimitry Andric } 1916bdd1243dSDimitry Andric 1917bdd1243dSDimitry Andric if (IgnoreAssumeLikeCalls) { 1918bdd1243dSDimitry Andric if (const auto *I = dyn_cast<IntrinsicInst>(Call)) 1919bdd1243dSDimitry Andric if (I->isAssumeLikeIntrinsic()) 1920bdd1243dSDimitry Andric continue; 1921bdd1243dSDimitry Andric } 1922bdd1243dSDimitry Andric 19235f757f3fSDimitry Andric if (!Call->isCallee(&U) || (!IgnoreCastedDirectCall && 19245f757f3fSDimitry Andric Call->getFunctionType() != getFunctionType())) { 1925349cc55cSDimitry Andric if (IgnoreARCAttachedCall && 1926349cc55cSDimitry Andric Call->isOperandBundleOfType(LLVMContext::OB_clang_arc_attachedcall, 1927349cc55cSDimitry Andric U.getOperandNo())) 1928349cc55cSDimitry Andric continue; 1929349cc55cSDimitry Andric 19300b57cec5SDimitry Andric if (PutOffender) 19310b57cec5SDimitry Andric *PutOffender = FU; 19320b57cec5SDimitry Andric return true; 19330b57cec5SDimitry Andric } 19340b57cec5SDimitry Andric } 19350b57cec5SDimitry Andric return false; 19360b57cec5SDimitry Andric } 19370b57cec5SDimitry Andric 19380b57cec5SDimitry Andric bool Function::isDefTriviallyDead() const { 19390b57cec5SDimitry Andric // Check the linkage 19400b57cec5SDimitry Andric if (!hasLinkOnceLinkage() && !hasLocalLinkage() && 19410b57cec5SDimitry Andric !hasAvailableExternallyLinkage()) 19420b57cec5SDimitry Andric return false; 19430b57cec5SDimitry Andric 19440b57cec5SDimitry Andric // Check if the function is used by anything other than a blockaddress. 19450b57cec5SDimitry Andric for (const User *U : users()) 19460b57cec5SDimitry Andric if (!isa<BlockAddress>(U)) 19470b57cec5SDimitry Andric return false; 19480b57cec5SDimitry Andric 19490b57cec5SDimitry Andric return true; 19500b57cec5SDimitry Andric } 19510b57cec5SDimitry Andric 19520b57cec5SDimitry Andric /// callsFunctionThatReturnsTwice - Return true if the function has a call to 19530b57cec5SDimitry Andric /// setjmp or other function that gcc recognizes as "returning twice". 19540b57cec5SDimitry Andric bool Function::callsFunctionThatReturnsTwice() const { 19550b57cec5SDimitry Andric for (const Instruction &I : instructions(this)) 19560b57cec5SDimitry Andric if (const auto *Call = dyn_cast<CallBase>(&I)) 19570b57cec5SDimitry Andric if (Call->hasFnAttr(Attribute::ReturnsTwice)) 19580b57cec5SDimitry Andric return true; 19590b57cec5SDimitry Andric 19600b57cec5SDimitry Andric return false; 19610b57cec5SDimitry Andric } 19620b57cec5SDimitry Andric 19630b57cec5SDimitry Andric Constant *Function::getPersonalityFn() const { 19640b57cec5SDimitry Andric assert(hasPersonalityFn() && getNumOperands()); 19650b57cec5SDimitry Andric return cast<Constant>(Op<0>()); 19660b57cec5SDimitry Andric } 19670b57cec5SDimitry Andric 19680b57cec5SDimitry Andric void Function::setPersonalityFn(Constant *Fn) { 19690b57cec5SDimitry Andric setHungoffOperand<0>(Fn); 19700b57cec5SDimitry Andric setValueSubclassDataBit(3, Fn != nullptr); 19710b57cec5SDimitry Andric } 19720b57cec5SDimitry Andric 19730b57cec5SDimitry Andric Constant *Function::getPrefixData() const { 19740b57cec5SDimitry Andric assert(hasPrefixData() && getNumOperands()); 19750b57cec5SDimitry Andric return cast<Constant>(Op<1>()); 19760b57cec5SDimitry Andric } 19770b57cec5SDimitry Andric 19780b57cec5SDimitry Andric void Function::setPrefixData(Constant *PrefixData) { 19790b57cec5SDimitry Andric setHungoffOperand<1>(PrefixData); 19800b57cec5SDimitry Andric setValueSubclassDataBit(1, PrefixData != nullptr); 19810b57cec5SDimitry Andric } 19820b57cec5SDimitry Andric 19830b57cec5SDimitry Andric Constant *Function::getPrologueData() const { 19840b57cec5SDimitry Andric assert(hasPrologueData() && getNumOperands()); 19850b57cec5SDimitry Andric return cast<Constant>(Op<2>()); 19860b57cec5SDimitry Andric } 19870b57cec5SDimitry Andric 19880b57cec5SDimitry Andric void Function::setPrologueData(Constant *PrologueData) { 19890b57cec5SDimitry Andric setHungoffOperand<2>(PrologueData); 19900b57cec5SDimitry Andric setValueSubclassDataBit(2, PrologueData != nullptr); 19910b57cec5SDimitry Andric } 19920b57cec5SDimitry Andric 19930b57cec5SDimitry Andric void Function::allocHungoffUselist() { 19940b57cec5SDimitry Andric // If we've already allocated a uselist, stop here. 19950b57cec5SDimitry Andric if (getNumOperands()) 19960b57cec5SDimitry Andric return; 19970b57cec5SDimitry Andric 19980b57cec5SDimitry Andric allocHungoffUses(3, /*IsPhi=*/ false); 19990b57cec5SDimitry Andric setNumHungOffUseOperands(3); 20000b57cec5SDimitry Andric 20010b57cec5SDimitry Andric // Initialize the uselist with placeholder operands to allow traversal. 20025f757f3fSDimitry Andric auto *CPN = ConstantPointerNull::get(PointerType::get(getContext(), 0)); 20030b57cec5SDimitry Andric Op<0>().set(CPN); 20040b57cec5SDimitry Andric Op<1>().set(CPN); 20050b57cec5SDimitry Andric Op<2>().set(CPN); 20060b57cec5SDimitry Andric } 20070b57cec5SDimitry Andric 20080b57cec5SDimitry Andric template <int Idx> 20090b57cec5SDimitry Andric void Function::setHungoffOperand(Constant *C) { 20100b57cec5SDimitry Andric if (C) { 20110b57cec5SDimitry Andric allocHungoffUselist(); 20120b57cec5SDimitry Andric Op<Idx>().set(C); 20130b57cec5SDimitry Andric } else if (getNumOperands()) { 20145f757f3fSDimitry Andric Op<Idx>().set(ConstantPointerNull::get(PointerType::get(getContext(), 0))); 20150b57cec5SDimitry Andric } 20160b57cec5SDimitry Andric } 20170b57cec5SDimitry Andric 20180b57cec5SDimitry Andric void Function::setValueSubclassDataBit(unsigned Bit, bool On) { 20190b57cec5SDimitry Andric assert(Bit < 16 && "SubclassData contains only 16 bits"); 20200b57cec5SDimitry Andric if (On) 20210b57cec5SDimitry Andric setValueSubclassData(getSubclassDataFromValue() | (1 << Bit)); 20220b57cec5SDimitry Andric else 20230b57cec5SDimitry Andric setValueSubclassData(getSubclassDataFromValue() & ~(1 << Bit)); 20240b57cec5SDimitry Andric } 20250b57cec5SDimitry Andric 20260b57cec5SDimitry Andric void Function::setEntryCount(ProfileCount Count, 20270b57cec5SDimitry Andric const DenseSet<GlobalValue::GUID> *S) { 20280b57cec5SDimitry Andric #if !defined(NDEBUG) 20290b57cec5SDimitry Andric auto PrevCount = getEntryCount(); 203081ad6265SDimitry Andric assert(!PrevCount || PrevCount->getType() == Count.getType()); 20310b57cec5SDimitry Andric #endif 2032480093f4SDimitry Andric 2033480093f4SDimitry Andric auto ImportGUIDs = getImportGUIDs(); 2034480093f4SDimitry Andric if (S == nullptr && ImportGUIDs.size()) 2035480093f4SDimitry Andric S = &ImportGUIDs; 2036480093f4SDimitry Andric 20370b57cec5SDimitry Andric MDBuilder MDB(getContext()); 20380b57cec5SDimitry Andric setMetadata( 20390b57cec5SDimitry Andric LLVMContext::MD_prof, 20400b57cec5SDimitry Andric MDB.createFunctionEntryCount(Count.getCount(), Count.isSynthetic(), S)); 20410b57cec5SDimitry Andric } 20420b57cec5SDimitry Andric 20430b57cec5SDimitry Andric void Function::setEntryCount(uint64_t Count, Function::ProfileCountType Type, 20440b57cec5SDimitry Andric const DenseSet<GlobalValue::GUID> *Imports) { 20450b57cec5SDimitry Andric setEntryCount(ProfileCount(Count, Type), Imports); 20460b57cec5SDimitry Andric } 20470b57cec5SDimitry Andric 2048bdd1243dSDimitry Andric std::optional<ProfileCount> Function::getEntryCount(bool AllowSynthetic) const { 20490b57cec5SDimitry Andric MDNode *MD = getMetadata(LLVMContext::MD_prof); 20500b57cec5SDimitry Andric if (MD && MD->getOperand(0)) 20510b57cec5SDimitry Andric if (MDString *MDS = dyn_cast<MDString>(MD->getOperand(0))) { 2052*0fca6ea1SDimitry Andric if (MDS->getString() == "function_entry_count") { 20530b57cec5SDimitry Andric ConstantInt *CI = mdconst::extract<ConstantInt>(MD->getOperand(1)); 20540b57cec5SDimitry Andric uint64_t Count = CI->getValue().getZExtValue(); 20550b57cec5SDimitry Andric // A value of -1 is used for SamplePGO when there were no samples. 20560b57cec5SDimitry Andric // Treat this the same as unknown. 20570b57cec5SDimitry Andric if (Count == (uint64_t)-1) 2058bdd1243dSDimitry Andric return std::nullopt; 20590b57cec5SDimitry Andric return ProfileCount(Count, PCT_Real); 20600b57cec5SDimitry Andric } else if (AllowSynthetic && 2061*0fca6ea1SDimitry Andric MDS->getString() == "synthetic_function_entry_count") { 20620b57cec5SDimitry Andric ConstantInt *CI = mdconst::extract<ConstantInt>(MD->getOperand(1)); 20630b57cec5SDimitry Andric uint64_t Count = CI->getValue().getZExtValue(); 20640b57cec5SDimitry Andric return ProfileCount(Count, PCT_Synthetic); 20650b57cec5SDimitry Andric } 20660b57cec5SDimitry Andric } 2067bdd1243dSDimitry Andric return std::nullopt; 20680b57cec5SDimitry Andric } 20690b57cec5SDimitry Andric 20700b57cec5SDimitry Andric DenseSet<GlobalValue::GUID> Function::getImportGUIDs() const { 20710b57cec5SDimitry Andric DenseSet<GlobalValue::GUID> R; 20720b57cec5SDimitry Andric if (MDNode *MD = getMetadata(LLVMContext::MD_prof)) 20730b57cec5SDimitry Andric if (MDString *MDS = dyn_cast<MDString>(MD->getOperand(0))) 2074*0fca6ea1SDimitry Andric if (MDS->getString() == "function_entry_count") 20750b57cec5SDimitry Andric for (unsigned i = 2; i < MD->getNumOperands(); i++) 20760b57cec5SDimitry Andric R.insert(mdconst::extract<ConstantInt>(MD->getOperand(i)) 20770b57cec5SDimitry Andric ->getValue() 20780b57cec5SDimitry Andric .getZExtValue()); 20790b57cec5SDimitry Andric return R; 20800b57cec5SDimitry Andric } 20810b57cec5SDimitry Andric 20820b57cec5SDimitry Andric void Function::setSectionPrefix(StringRef Prefix) { 20830b57cec5SDimitry Andric MDBuilder MDB(getContext()); 20840b57cec5SDimitry Andric setMetadata(LLVMContext::MD_section_prefix, 20850b57cec5SDimitry Andric MDB.createFunctionSectionPrefix(Prefix)); 20860b57cec5SDimitry Andric } 20870b57cec5SDimitry Andric 2088bdd1243dSDimitry Andric std::optional<StringRef> Function::getSectionPrefix() const { 20890b57cec5SDimitry Andric if (MDNode *MD = getMetadata(LLVMContext::MD_section_prefix)) { 2090*0fca6ea1SDimitry Andric assert(cast<MDString>(MD->getOperand(0))->getString() == 2091*0fca6ea1SDimitry Andric "function_section_prefix" && 20920b57cec5SDimitry Andric "Metadata not match"); 20930b57cec5SDimitry Andric return cast<MDString>(MD->getOperand(1))->getString(); 20940b57cec5SDimitry Andric } 2095bdd1243dSDimitry Andric return std::nullopt; 20960b57cec5SDimitry Andric } 20970b57cec5SDimitry Andric 20980b57cec5SDimitry Andric bool Function::nullPointerIsDefined() const { 20995ffd83dbSDimitry Andric return hasFnAttribute(Attribute::NullPointerIsValid); 21000b57cec5SDimitry Andric } 21010b57cec5SDimitry Andric 21020b57cec5SDimitry Andric bool llvm::NullPointerIsDefined(const Function *F, unsigned AS) { 21030b57cec5SDimitry Andric if (F && F->nullPointerIsDefined()) 21040b57cec5SDimitry Andric return true; 21050b57cec5SDimitry Andric 21060b57cec5SDimitry Andric if (AS != 0) 21070b57cec5SDimitry Andric return true; 21080b57cec5SDimitry Andric 21090b57cec5SDimitry Andric return false; 21100b57cec5SDimitry Andric } 2111