xref: /freebsd/contrib/llvm-project/llvm/lib/IR/Function.cpp (revision bdd1243df58e60e85101c09001d9812a789b6bc4)
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"
270b57cec5SDimitry Andric #include "llvm/IR/Constants.h"
280b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h"
290b57cec5SDimitry Andric #include "llvm/IR/GlobalValue.h"
300b57cec5SDimitry Andric #include "llvm/IR/InstIterator.h"
310b57cec5SDimitry Andric #include "llvm/IR/Instruction.h"
32fe6060f1SDimitry Andric #include "llvm/IR/IntrinsicInst.h"
330b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h"
34480093f4SDimitry Andric #include "llvm/IR/IntrinsicsAArch64.h"
35480093f4SDimitry Andric #include "llvm/IR/IntrinsicsAMDGPU.h"
36480093f4SDimitry Andric #include "llvm/IR/IntrinsicsARM.h"
37480093f4SDimitry Andric #include "llvm/IR/IntrinsicsBPF.h"
3881ad6265SDimitry Andric #include "llvm/IR/IntrinsicsDirectX.h"
39480093f4SDimitry Andric #include "llvm/IR/IntrinsicsHexagon.h"
40480093f4SDimitry Andric #include "llvm/IR/IntrinsicsMips.h"
41480093f4SDimitry Andric #include "llvm/IR/IntrinsicsNVPTX.h"
42480093f4SDimitry Andric #include "llvm/IR/IntrinsicsPowerPC.h"
43480093f4SDimitry Andric #include "llvm/IR/IntrinsicsR600.h"
44480093f4SDimitry Andric #include "llvm/IR/IntrinsicsRISCV.h"
45480093f4SDimitry Andric #include "llvm/IR/IntrinsicsS390.h"
46e8d8bef9SDimitry Andric #include "llvm/IR/IntrinsicsVE.h"
47480093f4SDimitry Andric #include "llvm/IR/IntrinsicsWebAssembly.h"
48480093f4SDimitry Andric #include "llvm/IR/IntrinsicsX86.h"
49480093f4SDimitry Andric #include "llvm/IR/IntrinsicsXCore.h"
500b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
510b57cec5SDimitry Andric #include "llvm/IR/MDBuilder.h"
520b57cec5SDimitry Andric #include "llvm/IR/Metadata.h"
530b57cec5SDimitry Andric #include "llvm/IR/Module.h"
54fe6060f1SDimitry Andric #include "llvm/IR/Operator.h"
550b57cec5SDimitry Andric #include "llvm/IR/SymbolTableListTraits.h"
560b57cec5SDimitry Andric #include "llvm/IR/Type.h"
570b57cec5SDimitry Andric #include "llvm/IR/Use.h"
580b57cec5SDimitry Andric #include "llvm/IR/User.h"
590b57cec5SDimitry Andric #include "llvm/IR/Value.h"
600b57cec5SDimitry Andric #include "llvm/IR/ValueSymbolTable.h"
610b57cec5SDimitry Andric #include "llvm/Support/Casting.h"
62fe6060f1SDimitry Andric #include "llvm/Support/CommandLine.h"
630b57cec5SDimitry Andric #include "llvm/Support/Compiler.h"
640b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
65*bdd1243dSDimitry Andric #include "llvm/Support/ModRef.h"
660b57cec5SDimitry Andric #include <cassert>
670b57cec5SDimitry Andric #include <cstddef>
680b57cec5SDimitry Andric #include <cstdint>
690b57cec5SDimitry Andric #include <cstring>
700b57cec5SDimitry Andric #include <string>
710b57cec5SDimitry Andric 
720b57cec5SDimitry Andric using namespace llvm;
730b57cec5SDimitry Andric using ProfileCount = Function::ProfileCount;
740b57cec5SDimitry Andric 
750b57cec5SDimitry Andric // Explicit instantiations of SymbolTableListTraits since some of the methods
760b57cec5SDimitry Andric // are not in the public header file...
770b57cec5SDimitry Andric template class llvm::SymbolTableListTraits<BasicBlock>;
780b57cec5SDimitry Andric 
79fe6060f1SDimitry Andric static cl::opt<unsigned> NonGlobalValueMaxNameSize(
80fe6060f1SDimitry Andric     "non-global-value-max-name-size", cl::Hidden, cl::init(1024),
81fe6060f1SDimitry Andric     cl::desc("Maximum size for the name of non-global values."));
82fe6060f1SDimitry Andric 
830b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
840b57cec5SDimitry Andric // Argument Implementation
850b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
860b57cec5SDimitry Andric 
870b57cec5SDimitry Andric Argument::Argument(Type *Ty, const Twine &Name, Function *Par, unsigned ArgNo)
880b57cec5SDimitry Andric     : Value(Ty, Value::ArgumentVal), Parent(Par), ArgNo(ArgNo) {
890b57cec5SDimitry Andric   setName(Name);
900b57cec5SDimitry Andric }
910b57cec5SDimitry Andric 
920b57cec5SDimitry Andric void Argument::setParent(Function *parent) {
930b57cec5SDimitry Andric   Parent = parent;
940b57cec5SDimitry Andric }
950b57cec5SDimitry Andric 
96e8d8bef9SDimitry Andric bool Argument::hasNonNullAttr(bool AllowUndefOrPoison) const {
970b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
98e8d8bef9SDimitry Andric   if (getParent()->hasParamAttribute(getArgNo(), Attribute::NonNull) &&
99e8d8bef9SDimitry Andric       (AllowUndefOrPoison ||
100e8d8bef9SDimitry Andric        getParent()->hasParamAttribute(getArgNo(), Attribute::NoUndef)))
1010b57cec5SDimitry Andric     return true;
1020b57cec5SDimitry Andric   else if (getDereferenceableBytes() > 0 &&
1030b57cec5SDimitry Andric            !NullPointerIsDefined(getParent(),
1040b57cec5SDimitry Andric                                  getType()->getPointerAddressSpace()))
1050b57cec5SDimitry Andric     return true;
1060b57cec5SDimitry Andric   return false;
1070b57cec5SDimitry Andric }
1080b57cec5SDimitry Andric 
1090b57cec5SDimitry Andric bool Argument::hasByValAttr() const {
1100b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
1110b57cec5SDimitry Andric   return hasAttribute(Attribute::ByVal);
1120b57cec5SDimitry Andric }
1130b57cec5SDimitry Andric 
114e8d8bef9SDimitry Andric bool Argument::hasByRefAttr() const {
115e8d8bef9SDimitry Andric   if (!getType()->isPointerTy())
116e8d8bef9SDimitry Andric     return false;
117e8d8bef9SDimitry Andric   return hasAttribute(Attribute::ByRef);
118e8d8bef9SDimitry Andric }
119e8d8bef9SDimitry Andric 
1200b57cec5SDimitry Andric bool Argument::hasSwiftSelfAttr() const {
1210b57cec5SDimitry Andric   return getParent()->hasParamAttribute(getArgNo(), Attribute::SwiftSelf);
1220b57cec5SDimitry Andric }
1230b57cec5SDimitry Andric 
1240b57cec5SDimitry Andric bool Argument::hasSwiftErrorAttr() const {
1250b57cec5SDimitry Andric   return getParent()->hasParamAttribute(getArgNo(), Attribute::SwiftError);
1260b57cec5SDimitry Andric }
1270b57cec5SDimitry Andric 
1280b57cec5SDimitry Andric bool Argument::hasInAllocaAttr() const {
1290b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
1300b57cec5SDimitry Andric   return hasAttribute(Attribute::InAlloca);
1310b57cec5SDimitry Andric }
1320b57cec5SDimitry Andric 
1335ffd83dbSDimitry Andric bool Argument::hasPreallocatedAttr() const {
1345ffd83dbSDimitry Andric   if (!getType()->isPointerTy())
1355ffd83dbSDimitry Andric     return false;
1365ffd83dbSDimitry Andric   return hasAttribute(Attribute::Preallocated);
1375ffd83dbSDimitry Andric }
1385ffd83dbSDimitry Andric 
139e8d8bef9SDimitry Andric bool Argument::hasPassPointeeByValueCopyAttr() const {
1400b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
1410b57cec5SDimitry Andric   AttributeList Attrs = getParent()->getAttributes();
142349cc55cSDimitry Andric   return Attrs.hasParamAttr(getArgNo(), Attribute::ByVal) ||
143349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::InAlloca) ||
144349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::Preallocated);
1455ffd83dbSDimitry Andric }
1465ffd83dbSDimitry Andric 
147e8d8bef9SDimitry Andric bool Argument::hasPointeeInMemoryValueAttr() const {
148e8d8bef9SDimitry Andric   if (!getType()->isPointerTy())
149e8d8bef9SDimitry Andric     return false;
150e8d8bef9SDimitry Andric   AttributeList Attrs = getParent()->getAttributes();
151349cc55cSDimitry Andric   return Attrs.hasParamAttr(getArgNo(), Attribute::ByVal) ||
152349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::StructRet) ||
153349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::InAlloca) ||
154349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::Preallocated) ||
155349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::ByRef);
156e8d8bef9SDimitry Andric }
1575ffd83dbSDimitry Andric 
158e8d8bef9SDimitry Andric /// For a byval, sret, inalloca, or preallocated parameter, get the in-memory
159e8d8bef9SDimitry Andric /// parameter type.
160349cc55cSDimitry Andric static Type *getMemoryParamAllocType(AttributeSet ParamAttrs) {
1615ffd83dbSDimitry Andric   // FIXME: All the type carrying attributes are mutually exclusive, so there
1625ffd83dbSDimitry Andric   // should be a single query to get the stored type that handles any of them.
1635ffd83dbSDimitry Andric   if (Type *ByValTy = ParamAttrs.getByValType())
164e8d8bef9SDimitry Andric     return ByValTy;
165e8d8bef9SDimitry Andric   if (Type *ByRefTy = ParamAttrs.getByRefType())
166e8d8bef9SDimitry Andric     return ByRefTy;
1675ffd83dbSDimitry Andric   if (Type *PreAllocTy = ParamAttrs.getPreallocatedType())
168e8d8bef9SDimitry Andric     return PreAllocTy;
169fe6060f1SDimitry Andric   if (Type *InAllocaTy = ParamAttrs.getInAllocaType())
170fe6060f1SDimitry Andric     return InAllocaTy;
171fe6060f1SDimitry Andric   if (Type *SRetTy = ParamAttrs.getStructRetType())
172fe6060f1SDimitry Andric     return SRetTy;
1735ffd83dbSDimitry Andric 
174e8d8bef9SDimitry Andric   return nullptr;
175e8d8bef9SDimitry Andric }
176e8d8bef9SDimitry Andric 
177e8d8bef9SDimitry Andric uint64_t Argument::getPassPointeeByValueCopySize(const DataLayout &DL) const {
178e8d8bef9SDimitry Andric   AttributeSet ParamAttrs =
179349cc55cSDimitry Andric       getParent()->getAttributes().getParamAttrs(getArgNo());
180349cc55cSDimitry Andric   if (Type *MemTy = getMemoryParamAllocType(ParamAttrs))
181e8d8bef9SDimitry Andric     return DL.getTypeAllocSize(MemTy);
1825ffd83dbSDimitry Andric   return 0;
1830b57cec5SDimitry Andric }
1840b57cec5SDimitry Andric 
185e8d8bef9SDimitry Andric Type *Argument::getPointeeInMemoryValueType() const {
186e8d8bef9SDimitry Andric   AttributeSet ParamAttrs =
187349cc55cSDimitry Andric       getParent()->getAttributes().getParamAttrs(getArgNo());
188349cc55cSDimitry Andric   return getMemoryParamAllocType(ParamAttrs);
189e8d8bef9SDimitry Andric }
190e8d8bef9SDimitry Andric 
191480093f4SDimitry Andric MaybeAlign Argument::getParamAlign() const {
192480093f4SDimitry Andric   assert(getType()->isPointerTy() && "Only pointers have alignments");
193480093f4SDimitry Andric   return getParent()->getParamAlign(getArgNo());
194480093f4SDimitry Andric }
195480093f4SDimitry Andric 
196fe6060f1SDimitry Andric MaybeAlign Argument::getParamStackAlign() const {
197fe6060f1SDimitry Andric   return getParent()->getParamStackAlign(getArgNo());
198fe6060f1SDimitry Andric }
199fe6060f1SDimitry Andric 
2000b57cec5SDimitry Andric Type *Argument::getParamByValType() const {
2010b57cec5SDimitry Andric   assert(getType()->isPointerTy() && "Only pointers have byval types");
2020b57cec5SDimitry Andric   return getParent()->getParamByValType(getArgNo());
2030b57cec5SDimitry Andric }
2040b57cec5SDimitry Andric 
205e8d8bef9SDimitry Andric Type *Argument::getParamStructRetType() const {
206e8d8bef9SDimitry Andric   assert(getType()->isPointerTy() && "Only pointers have sret types");
207e8d8bef9SDimitry Andric   return getParent()->getParamStructRetType(getArgNo());
208e8d8bef9SDimitry Andric }
209e8d8bef9SDimitry Andric 
210e8d8bef9SDimitry Andric Type *Argument::getParamByRefType() const {
211fe6060f1SDimitry Andric   assert(getType()->isPointerTy() && "Only pointers have byref types");
212e8d8bef9SDimitry Andric   return getParent()->getParamByRefType(getArgNo());
213e8d8bef9SDimitry Andric }
214e8d8bef9SDimitry Andric 
215fe6060f1SDimitry Andric Type *Argument::getParamInAllocaType() const {
216fe6060f1SDimitry Andric   assert(getType()->isPointerTy() && "Only pointers have inalloca types");
217fe6060f1SDimitry Andric   return getParent()->getParamInAllocaType(getArgNo());
218fe6060f1SDimitry Andric }
219fe6060f1SDimitry Andric 
2200b57cec5SDimitry Andric uint64_t Argument::getDereferenceableBytes() const {
2210b57cec5SDimitry Andric   assert(getType()->isPointerTy() &&
2220b57cec5SDimitry Andric          "Only pointers have dereferenceable bytes");
2230b57cec5SDimitry Andric   return getParent()->getParamDereferenceableBytes(getArgNo());
2240b57cec5SDimitry Andric }
2250b57cec5SDimitry Andric 
2260b57cec5SDimitry Andric uint64_t Argument::getDereferenceableOrNullBytes() const {
2270b57cec5SDimitry Andric   assert(getType()->isPointerTy() &&
2280b57cec5SDimitry Andric          "Only pointers have dereferenceable bytes");
2290b57cec5SDimitry Andric   return getParent()->getParamDereferenceableOrNullBytes(getArgNo());
2300b57cec5SDimitry Andric }
2310b57cec5SDimitry Andric 
2320b57cec5SDimitry Andric bool Argument::hasNestAttr() const {
2330b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
2340b57cec5SDimitry Andric   return hasAttribute(Attribute::Nest);
2350b57cec5SDimitry Andric }
2360b57cec5SDimitry Andric 
2370b57cec5SDimitry Andric bool Argument::hasNoAliasAttr() const {
2380b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
2390b57cec5SDimitry Andric   return hasAttribute(Attribute::NoAlias);
2400b57cec5SDimitry Andric }
2410b57cec5SDimitry Andric 
2420b57cec5SDimitry Andric bool Argument::hasNoCaptureAttr() const {
2430b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
2440b57cec5SDimitry Andric   return hasAttribute(Attribute::NoCapture);
2450b57cec5SDimitry Andric }
2460b57cec5SDimitry Andric 
247fe6060f1SDimitry Andric bool Argument::hasNoFreeAttr() const {
248fe6060f1SDimitry Andric   if (!getType()->isPointerTy()) return false;
249fe6060f1SDimitry Andric   return hasAttribute(Attribute::NoFree);
250fe6060f1SDimitry Andric }
251fe6060f1SDimitry Andric 
2520b57cec5SDimitry Andric bool Argument::hasStructRetAttr() const {
2530b57cec5SDimitry Andric   if (!getType()->isPointerTy()) return false;
2540b57cec5SDimitry Andric   return hasAttribute(Attribute::StructRet);
2550b57cec5SDimitry Andric }
2560b57cec5SDimitry Andric 
2570b57cec5SDimitry Andric bool Argument::hasInRegAttr() const {
2580b57cec5SDimitry Andric   return hasAttribute(Attribute::InReg);
2590b57cec5SDimitry Andric }
2600b57cec5SDimitry Andric 
2610b57cec5SDimitry Andric bool Argument::hasReturnedAttr() const {
2620b57cec5SDimitry Andric   return hasAttribute(Attribute::Returned);
2630b57cec5SDimitry Andric }
2640b57cec5SDimitry Andric 
2650b57cec5SDimitry Andric bool Argument::hasZExtAttr() const {
2660b57cec5SDimitry Andric   return hasAttribute(Attribute::ZExt);
2670b57cec5SDimitry Andric }
2680b57cec5SDimitry Andric 
2690b57cec5SDimitry Andric bool Argument::hasSExtAttr() const {
2700b57cec5SDimitry Andric   return hasAttribute(Attribute::SExt);
2710b57cec5SDimitry Andric }
2720b57cec5SDimitry Andric 
2730b57cec5SDimitry Andric bool Argument::onlyReadsMemory() const {
2740b57cec5SDimitry Andric   AttributeList Attrs = getParent()->getAttributes();
275349cc55cSDimitry Andric   return Attrs.hasParamAttr(getArgNo(), Attribute::ReadOnly) ||
276349cc55cSDimitry Andric          Attrs.hasParamAttr(getArgNo(), Attribute::ReadNone);
2770b57cec5SDimitry Andric }
2780b57cec5SDimitry Andric 
2790b57cec5SDimitry Andric void Argument::addAttrs(AttrBuilder &B) {
2800b57cec5SDimitry Andric   AttributeList AL = getParent()->getAttributes();
2810b57cec5SDimitry Andric   AL = AL.addParamAttributes(Parent->getContext(), getArgNo(), B);
2820b57cec5SDimitry Andric   getParent()->setAttributes(AL);
2830b57cec5SDimitry Andric }
2840b57cec5SDimitry Andric 
2850b57cec5SDimitry Andric void Argument::addAttr(Attribute::AttrKind Kind) {
2860b57cec5SDimitry Andric   getParent()->addParamAttr(getArgNo(), Kind);
2870b57cec5SDimitry Andric }
2880b57cec5SDimitry Andric 
2890b57cec5SDimitry Andric void Argument::addAttr(Attribute Attr) {
2900b57cec5SDimitry Andric   getParent()->addParamAttr(getArgNo(), Attr);
2910b57cec5SDimitry Andric }
2920b57cec5SDimitry Andric 
2930b57cec5SDimitry Andric void Argument::removeAttr(Attribute::AttrKind Kind) {
2940b57cec5SDimitry Andric   getParent()->removeParamAttr(getArgNo(), Kind);
2950b57cec5SDimitry Andric }
2960b57cec5SDimitry Andric 
29704eeddc0SDimitry Andric void Argument::removeAttrs(const AttributeMask &AM) {
298fe6060f1SDimitry Andric   AttributeList AL = getParent()->getAttributes();
29904eeddc0SDimitry Andric   AL = AL.removeParamAttributes(Parent->getContext(), getArgNo(), AM);
300fe6060f1SDimitry Andric   getParent()->setAttributes(AL);
301fe6060f1SDimitry Andric }
302fe6060f1SDimitry Andric 
3030b57cec5SDimitry Andric bool Argument::hasAttribute(Attribute::AttrKind Kind) const {
3040b57cec5SDimitry Andric   return getParent()->hasParamAttribute(getArgNo(), Kind);
3050b57cec5SDimitry Andric }
3060b57cec5SDimitry Andric 
3070b57cec5SDimitry Andric Attribute Argument::getAttribute(Attribute::AttrKind Kind) const {
3080b57cec5SDimitry Andric   return getParent()->getParamAttribute(getArgNo(), Kind);
3090b57cec5SDimitry Andric }
3100b57cec5SDimitry Andric 
3110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3120b57cec5SDimitry Andric // Helper Methods in Function
3130b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3140b57cec5SDimitry Andric 
3150b57cec5SDimitry Andric LLVMContext &Function::getContext() const {
3160b57cec5SDimitry Andric   return getType()->getContext();
3170b57cec5SDimitry Andric }
3180b57cec5SDimitry Andric 
3190b57cec5SDimitry Andric unsigned Function::getInstructionCount() const {
3200b57cec5SDimitry Andric   unsigned NumInstrs = 0;
3210b57cec5SDimitry Andric   for (const BasicBlock &BB : BasicBlocks)
3220b57cec5SDimitry Andric     NumInstrs += std::distance(BB.instructionsWithoutDebug().begin(),
3230b57cec5SDimitry Andric                                BB.instructionsWithoutDebug().end());
3240b57cec5SDimitry Andric   return NumInstrs;
3250b57cec5SDimitry Andric }
3260b57cec5SDimitry Andric 
3270b57cec5SDimitry Andric Function *Function::Create(FunctionType *Ty, LinkageTypes Linkage,
3280b57cec5SDimitry Andric                            const Twine &N, Module &M) {
3290b57cec5SDimitry Andric   return Create(Ty, Linkage, M.getDataLayout().getProgramAddressSpace(), N, &M);
3300b57cec5SDimitry Andric }
3310b57cec5SDimitry Andric 
332fe6060f1SDimitry Andric Function *Function::createWithDefaultAttr(FunctionType *Ty,
333fe6060f1SDimitry Andric                                           LinkageTypes Linkage,
334fe6060f1SDimitry Andric                                           unsigned AddrSpace, const Twine &N,
335fe6060f1SDimitry Andric                                           Module *M) {
336fe6060f1SDimitry Andric   auto *F = new Function(Ty, Linkage, AddrSpace, N, M);
33704eeddc0SDimitry Andric   AttrBuilder B(F->getContext());
33881ad6265SDimitry Andric   UWTableKind UWTable = M->getUwtable();
33981ad6265SDimitry Andric   if (UWTable != UWTableKind::None)
34081ad6265SDimitry Andric     B.addUWTableAttr(UWTable);
341fe6060f1SDimitry Andric   switch (M->getFramePointer()) {
342fe6060f1SDimitry Andric   case FramePointerKind::None:
343fe6060f1SDimitry Andric     // 0 ("none") is the default.
344fe6060f1SDimitry Andric     break;
345fe6060f1SDimitry Andric   case FramePointerKind::NonLeaf:
346fe6060f1SDimitry Andric     B.addAttribute("frame-pointer", "non-leaf");
347fe6060f1SDimitry Andric     break;
348fe6060f1SDimitry Andric   case FramePointerKind::All:
349fe6060f1SDimitry Andric     B.addAttribute("frame-pointer", "all");
350fe6060f1SDimitry Andric     break;
351fe6060f1SDimitry Andric   }
352fcaf7f86SDimitry Andric   if (M->getModuleFlag("function_return_thunk_extern"))
353fcaf7f86SDimitry Andric     B.addAttribute(Attribute::FnRetThunkExtern);
354349cc55cSDimitry Andric   F->addFnAttrs(B);
355fe6060f1SDimitry Andric   return F;
356fe6060f1SDimitry Andric }
357fe6060f1SDimitry Andric 
3580b57cec5SDimitry Andric void Function::removeFromParent() {
3590b57cec5SDimitry Andric   getParent()->getFunctionList().remove(getIterator());
3600b57cec5SDimitry Andric }
3610b57cec5SDimitry Andric 
3620b57cec5SDimitry Andric void Function::eraseFromParent() {
3630b57cec5SDimitry Andric   getParent()->getFunctionList().erase(getIterator());
3640b57cec5SDimitry Andric }
3650b57cec5SDimitry Andric 
366*bdd1243dSDimitry Andric void Function::splice(Function::iterator ToIt, Function *FromF,
367*bdd1243dSDimitry Andric                       Function::iterator FromBeginIt,
368*bdd1243dSDimitry Andric                       Function::iterator FromEndIt) {
369*bdd1243dSDimitry Andric #ifdef EXPENSIVE_CHECKS
370*bdd1243dSDimitry Andric   // Check that FromBeginIt is before FromEndIt.
371*bdd1243dSDimitry Andric   auto FromFEnd = FromF->end();
372*bdd1243dSDimitry Andric   for (auto It = FromBeginIt; It != FromEndIt; ++It)
373*bdd1243dSDimitry Andric     assert(It != FromFEnd && "FromBeginIt not before FromEndIt!");
374*bdd1243dSDimitry Andric #endif // EXPENSIVE_CHECKS
375*bdd1243dSDimitry Andric   BasicBlocks.splice(ToIt, FromF->BasicBlocks, FromBeginIt, FromEndIt);
376*bdd1243dSDimitry Andric }
377*bdd1243dSDimitry Andric 
378*bdd1243dSDimitry Andric Function::iterator Function::erase(Function::iterator FromIt,
379*bdd1243dSDimitry Andric                                    Function::iterator ToIt) {
380*bdd1243dSDimitry Andric   return BasicBlocks.erase(FromIt, ToIt);
381*bdd1243dSDimitry Andric }
382*bdd1243dSDimitry Andric 
3830b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3840b57cec5SDimitry Andric // Function Implementation
3850b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
3860b57cec5SDimitry Andric 
3870b57cec5SDimitry Andric static unsigned computeAddrSpace(unsigned AddrSpace, Module *M) {
3880b57cec5SDimitry Andric   // If AS == -1 and we are passed a valid module pointer we place the function
3890b57cec5SDimitry Andric   // in the program address space. Otherwise we default to AS0.
3900b57cec5SDimitry Andric   if (AddrSpace == static_cast<unsigned>(-1))
3910b57cec5SDimitry Andric     return M ? M->getDataLayout().getProgramAddressSpace() : 0;
3920b57cec5SDimitry Andric   return AddrSpace;
3930b57cec5SDimitry Andric }
3940b57cec5SDimitry Andric 
3950b57cec5SDimitry Andric Function::Function(FunctionType *Ty, LinkageTypes Linkage, unsigned AddrSpace,
3960b57cec5SDimitry Andric                    const Twine &name, Module *ParentModule)
3970b57cec5SDimitry Andric     : GlobalObject(Ty, Value::FunctionVal,
3980b57cec5SDimitry Andric                    OperandTraits<Function>::op_begin(this), 0, Linkage, name,
3990b57cec5SDimitry Andric                    computeAddrSpace(AddrSpace, ParentModule)),
4000b57cec5SDimitry Andric       NumArgs(Ty->getNumParams()) {
4010b57cec5SDimitry Andric   assert(FunctionType::isValidReturnType(getReturnType()) &&
4020b57cec5SDimitry Andric          "invalid return type");
4030b57cec5SDimitry Andric   setGlobalObjectSubClassData(0);
4040b57cec5SDimitry Andric 
4050b57cec5SDimitry Andric   // We only need a symbol table for a function if the context keeps value names
4060b57cec5SDimitry Andric   if (!getContext().shouldDiscardValueNames())
407fe6060f1SDimitry Andric     SymTab = std::make_unique<ValueSymbolTable>(NonGlobalValueMaxNameSize);
4080b57cec5SDimitry Andric 
4090b57cec5SDimitry Andric   // If the function has arguments, mark them as lazily built.
4100b57cec5SDimitry Andric   if (Ty->getNumParams())
4110b57cec5SDimitry Andric     setValueSubclassData(1);   // Set the "has lazy arguments" bit.
4120b57cec5SDimitry Andric 
4130b57cec5SDimitry Andric   if (ParentModule)
4140b57cec5SDimitry Andric     ParentModule->getFunctionList().push_back(this);
4150b57cec5SDimitry Andric 
4160b57cec5SDimitry Andric   HasLLVMReservedName = getName().startswith("llvm.");
4170b57cec5SDimitry Andric   // Ensure intrinsics have the right parameter attributes.
4180b57cec5SDimitry Andric   // Note, the IntID field will have been set in Value::setName if this function
4190b57cec5SDimitry Andric   // name is a valid intrinsic ID.
4200b57cec5SDimitry Andric   if (IntID)
4210b57cec5SDimitry Andric     setAttributes(Intrinsic::getAttributes(getContext(), IntID));
4220b57cec5SDimitry Andric }
4230b57cec5SDimitry Andric 
4240b57cec5SDimitry Andric Function::~Function() {
4250b57cec5SDimitry Andric   dropAllReferences();    // After this it is safe to delete instructions.
4260b57cec5SDimitry Andric 
4270b57cec5SDimitry Andric   // Delete all of the method arguments and unlink from symbol table...
4280b57cec5SDimitry Andric   if (Arguments)
4290b57cec5SDimitry Andric     clearArguments();
4300b57cec5SDimitry Andric 
4310b57cec5SDimitry Andric   // Remove the function from the on-the-side GC table.
4320b57cec5SDimitry Andric   clearGC();
4330b57cec5SDimitry Andric }
4340b57cec5SDimitry Andric 
4350b57cec5SDimitry Andric void Function::BuildLazyArguments() const {
4360b57cec5SDimitry Andric   // Create the arguments vector, all arguments start out unnamed.
4370b57cec5SDimitry Andric   auto *FT = getFunctionType();
4380b57cec5SDimitry Andric   if (NumArgs > 0) {
4390b57cec5SDimitry Andric     Arguments = std::allocator<Argument>().allocate(NumArgs);
4400b57cec5SDimitry Andric     for (unsigned i = 0, e = NumArgs; i != e; ++i) {
4410b57cec5SDimitry Andric       Type *ArgTy = FT->getParamType(i);
4420b57cec5SDimitry Andric       assert(!ArgTy->isVoidTy() && "Cannot have void typed arguments!");
4430b57cec5SDimitry Andric       new (Arguments + i) Argument(ArgTy, "", const_cast<Function *>(this), i);
4440b57cec5SDimitry Andric     }
4450b57cec5SDimitry Andric   }
4460b57cec5SDimitry Andric 
4470b57cec5SDimitry Andric   // Clear the lazy arguments bit.
4480b57cec5SDimitry Andric   unsigned SDC = getSubclassDataFromValue();
4498bcb0991SDimitry Andric   SDC &= ~(1 << 0);
4508bcb0991SDimitry Andric   const_cast<Function*>(this)->setValueSubclassData(SDC);
4510b57cec5SDimitry Andric   assert(!hasLazyArguments());
4520b57cec5SDimitry Andric }
4530b57cec5SDimitry Andric 
4540b57cec5SDimitry Andric static MutableArrayRef<Argument> makeArgArray(Argument *Args, size_t Count) {
4550b57cec5SDimitry Andric   return MutableArrayRef<Argument>(Args, Count);
4560b57cec5SDimitry Andric }
4570b57cec5SDimitry Andric 
4585ffd83dbSDimitry Andric bool Function::isConstrainedFPIntrinsic() const {
4595ffd83dbSDimitry Andric   switch (getIntrinsicID()) {
4605ffd83dbSDimitry Andric #define INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC)                         \
4615ffd83dbSDimitry Andric   case Intrinsic::INTRINSIC:
4625ffd83dbSDimitry Andric #include "llvm/IR/ConstrainedOps.def"
4635ffd83dbSDimitry Andric     return true;
4645ffd83dbSDimitry Andric #undef INSTRUCTION
4655ffd83dbSDimitry Andric   default:
4665ffd83dbSDimitry Andric     return false;
4675ffd83dbSDimitry Andric   }
4685ffd83dbSDimitry Andric }
4695ffd83dbSDimitry Andric 
4700b57cec5SDimitry Andric void Function::clearArguments() {
4710b57cec5SDimitry Andric   for (Argument &A : makeArgArray(Arguments, NumArgs)) {
4720b57cec5SDimitry Andric     A.setName("");
4730b57cec5SDimitry Andric     A.~Argument();
4740b57cec5SDimitry Andric   }
4750b57cec5SDimitry Andric   std::allocator<Argument>().deallocate(Arguments, NumArgs);
4760b57cec5SDimitry Andric   Arguments = nullptr;
4770b57cec5SDimitry Andric }
4780b57cec5SDimitry Andric 
4790b57cec5SDimitry Andric void Function::stealArgumentListFrom(Function &Src) {
4800b57cec5SDimitry Andric   assert(isDeclaration() && "Expected no references to current arguments");
4810b57cec5SDimitry Andric 
4820b57cec5SDimitry Andric   // Drop the current arguments, if any, and set the lazy argument bit.
4830b57cec5SDimitry Andric   if (!hasLazyArguments()) {
4840b57cec5SDimitry Andric     assert(llvm::all_of(makeArgArray(Arguments, NumArgs),
4850b57cec5SDimitry Andric                         [](const Argument &A) { return A.use_empty(); }) &&
4860b57cec5SDimitry Andric            "Expected arguments to be unused in declaration");
4870b57cec5SDimitry Andric     clearArguments();
4880b57cec5SDimitry Andric     setValueSubclassData(getSubclassDataFromValue() | (1 << 0));
4890b57cec5SDimitry Andric   }
4900b57cec5SDimitry Andric 
4910b57cec5SDimitry Andric   // Nothing to steal if Src has lazy arguments.
4920b57cec5SDimitry Andric   if (Src.hasLazyArguments())
4930b57cec5SDimitry Andric     return;
4940b57cec5SDimitry Andric 
4950b57cec5SDimitry Andric   // Steal arguments from Src, and fix the lazy argument bits.
4960b57cec5SDimitry Andric   assert(arg_size() == Src.arg_size());
4970b57cec5SDimitry Andric   Arguments = Src.Arguments;
4980b57cec5SDimitry Andric   Src.Arguments = nullptr;
4990b57cec5SDimitry Andric   for (Argument &A : makeArgArray(Arguments, NumArgs)) {
5000b57cec5SDimitry Andric     // FIXME: This does the work of transferNodesFromList inefficiently.
5010b57cec5SDimitry Andric     SmallString<128> Name;
5020b57cec5SDimitry Andric     if (A.hasName())
5030b57cec5SDimitry Andric       Name = A.getName();
5040b57cec5SDimitry Andric     if (!Name.empty())
5050b57cec5SDimitry Andric       A.setName("");
5060b57cec5SDimitry Andric     A.setParent(this);
5070b57cec5SDimitry Andric     if (!Name.empty())
5080b57cec5SDimitry Andric       A.setName(Name);
5090b57cec5SDimitry Andric   }
5100b57cec5SDimitry Andric 
5110b57cec5SDimitry Andric   setValueSubclassData(getSubclassDataFromValue() & ~(1 << 0));
5120b57cec5SDimitry Andric   assert(!hasLazyArguments());
5130b57cec5SDimitry Andric   Src.setValueSubclassData(Src.getSubclassDataFromValue() | (1 << 0));
5140b57cec5SDimitry Andric }
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric // dropAllReferences() - This function causes all the subinstructions to "let
5170b57cec5SDimitry Andric // go" of all references that they are maintaining.  This allows one to
5180b57cec5SDimitry Andric // 'delete' a whole class at a time, even though there may be circular
5190b57cec5SDimitry Andric // references... first all references are dropped, and all use counts go to
5200b57cec5SDimitry Andric // zero.  Then everything is deleted for real.  Note that no operations are
5210b57cec5SDimitry Andric // valid on an object that has "dropped all references", except operator
5220b57cec5SDimitry Andric // delete.
5230b57cec5SDimitry Andric //
5240b57cec5SDimitry Andric void Function::dropAllReferences() {
5250b57cec5SDimitry Andric   setIsMaterializable(false);
5260b57cec5SDimitry Andric 
5270b57cec5SDimitry Andric   for (BasicBlock &BB : *this)
5280b57cec5SDimitry Andric     BB.dropAllReferences();
5290b57cec5SDimitry Andric 
5300b57cec5SDimitry Andric   // Delete all basic blocks. They are now unused, except possibly by
5310b57cec5SDimitry Andric   // blockaddresses, but BasicBlock's destructor takes care of those.
5320b57cec5SDimitry Andric   while (!BasicBlocks.empty())
5330b57cec5SDimitry Andric     BasicBlocks.begin()->eraseFromParent();
5340b57cec5SDimitry Andric 
5350b57cec5SDimitry Andric   // Drop uses of any optional data (real or placeholder).
5360b57cec5SDimitry Andric   if (getNumOperands()) {
5370b57cec5SDimitry Andric     User::dropAllReferences();
5380b57cec5SDimitry Andric     setNumHungOffUseOperands(0);
5390b57cec5SDimitry Andric     setValueSubclassData(getSubclassDataFromValue() & ~0xe);
5400b57cec5SDimitry Andric   }
5410b57cec5SDimitry Andric 
5420b57cec5SDimitry Andric   // Metadata is stored in a side-table.
5430b57cec5SDimitry Andric   clearMetadata();
5440b57cec5SDimitry Andric }
5450b57cec5SDimitry Andric 
546349cc55cSDimitry Andric void Function::addAttributeAtIndex(unsigned i, Attribute Attr) {
547349cc55cSDimitry Andric   AttributeSets = AttributeSets.addAttributeAtIndex(getContext(), i, Attr);
5480b57cec5SDimitry Andric }
5490b57cec5SDimitry Andric 
550349cc55cSDimitry Andric void Function::addFnAttr(Attribute::AttrKind Kind) {
551349cc55cSDimitry Andric   AttributeSets = AttributeSets.addFnAttribute(getContext(), Kind);
5520b57cec5SDimitry Andric }
5530b57cec5SDimitry Andric 
554349cc55cSDimitry Andric void Function::addFnAttr(StringRef Kind, StringRef Val) {
555349cc55cSDimitry Andric   AttributeSets = AttributeSets.addFnAttribute(getContext(), Kind, Val);
556349cc55cSDimitry Andric }
557349cc55cSDimitry Andric 
558349cc55cSDimitry Andric void Function::addFnAttr(Attribute Attr) {
559349cc55cSDimitry Andric   AttributeSets = AttributeSets.addFnAttribute(getContext(), Attr);
560349cc55cSDimitry Andric }
561349cc55cSDimitry Andric 
562349cc55cSDimitry Andric void Function::addFnAttrs(const AttrBuilder &Attrs) {
563349cc55cSDimitry Andric   AttributeSets = AttributeSets.addFnAttributes(getContext(), Attrs);
564349cc55cSDimitry Andric }
565349cc55cSDimitry Andric 
566349cc55cSDimitry Andric void Function::addRetAttr(Attribute::AttrKind Kind) {
567349cc55cSDimitry Andric   AttributeSets = AttributeSets.addRetAttribute(getContext(), Kind);
568349cc55cSDimitry Andric }
569349cc55cSDimitry Andric 
570349cc55cSDimitry Andric void Function::addRetAttr(Attribute Attr) {
571349cc55cSDimitry Andric   AttributeSets = AttributeSets.addRetAttribute(getContext(), Attr);
572349cc55cSDimitry Andric }
573349cc55cSDimitry Andric 
574349cc55cSDimitry Andric void Function::addRetAttrs(const AttrBuilder &Attrs) {
575349cc55cSDimitry Andric   AttributeSets = AttributeSets.addRetAttributes(getContext(), Attrs);
5760b57cec5SDimitry Andric }
5770b57cec5SDimitry Andric 
5780b57cec5SDimitry Andric void Function::addParamAttr(unsigned ArgNo, Attribute::AttrKind Kind) {
579349cc55cSDimitry Andric   AttributeSets = AttributeSets.addParamAttribute(getContext(), ArgNo, Kind);
5800b57cec5SDimitry Andric }
5810b57cec5SDimitry Andric 
5820b57cec5SDimitry Andric void Function::addParamAttr(unsigned ArgNo, Attribute Attr) {
583349cc55cSDimitry Andric   AttributeSets = AttributeSets.addParamAttribute(getContext(), ArgNo, Attr);
5840b57cec5SDimitry Andric }
5850b57cec5SDimitry Andric 
5860b57cec5SDimitry Andric void Function::addParamAttrs(unsigned ArgNo, const AttrBuilder &Attrs) {
587349cc55cSDimitry Andric   AttributeSets = AttributeSets.addParamAttributes(getContext(), ArgNo, Attrs);
5880b57cec5SDimitry Andric }
5890b57cec5SDimitry Andric 
590349cc55cSDimitry Andric void Function::removeAttributeAtIndex(unsigned i, Attribute::AttrKind Kind) {
591349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeAttributeAtIndex(getContext(), i, Kind);
5920b57cec5SDimitry Andric }
5930b57cec5SDimitry Andric 
594349cc55cSDimitry Andric void Function::removeAttributeAtIndex(unsigned i, StringRef Kind) {
595349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeAttributeAtIndex(getContext(), i, Kind);
5960b57cec5SDimitry Andric }
5970b57cec5SDimitry Andric 
598349cc55cSDimitry Andric void Function::removeFnAttr(Attribute::AttrKind Kind) {
599349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeFnAttribute(getContext(), Kind);
600349cc55cSDimitry Andric }
601349cc55cSDimitry Andric 
602349cc55cSDimitry Andric void Function::removeFnAttr(StringRef Kind) {
603349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeFnAttribute(getContext(), Kind);
604349cc55cSDimitry Andric }
605349cc55cSDimitry Andric 
60604eeddc0SDimitry Andric void Function::removeFnAttrs(const AttributeMask &AM) {
60704eeddc0SDimitry Andric   AttributeSets = AttributeSets.removeFnAttributes(getContext(), AM);
608349cc55cSDimitry Andric }
609349cc55cSDimitry Andric 
610349cc55cSDimitry Andric void Function::removeRetAttr(Attribute::AttrKind Kind) {
611349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeRetAttribute(getContext(), Kind);
612349cc55cSDimitry Andric }
613349cc55cSDimitry Andric 
614349cc55cSDimitry Andric void Function::removeRetAttr(StringRef Kind) {
615349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeRetAttribute(getContext(), Kind);
616349cc55cSDimitry Andric }
617349cc55cSDimitry Andric 
61804eeddc0SDimitry Andric void Function::removeRetAttrs(const AttributeMask &Attrs) {
619349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeRetAttributes(getContext(), Attrs);
6200b57cec5SDimitry Andric }
6210b57cec5SDimitry Andric 
6220b57cec5SDimitry Andric void Function::removeParamAttr(unsigned ArgNo, Attribute::AttrKind Kind) {
623349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeParamAttribute(getContext(), ArgNo, Kind);
6240b57cec5SDimitry Andric }
6250b57cec5SDimitry Andric 
6260b57cec5SDimitry Andric void Function::removeParamAttr(unsigned ArgNo, StringRef Kind) {
627349cc55cSDimitry Andric   AttributeSets = AttributeSets.removeParamAttribute(getContext(), ArgNo, Kind);
6280b57cec5SDimitry Andric }
6290b57cec5SDimitry Andric 
63004eeddc0SDimitry Andric void Function::removeParamAttrs(unsigned ArgNo, const AttributeMask &Attrs) {
631349cc55cSDimitry Andric   AttributeSets =
632349cc55cSDimitry Andric       AttributeSets.removeParamAttributes(getContext(), ArgNo, Attrs);
6330b57cec5SDimitry Andric }
6340b57cec5SDimitry Andric 
6350b57cec5SDimitry Andric void Function::addDereferenceableParamAttr(unsigned ArgNo, uint64_t Bytes) {
636349cc55cSDimitry Andric   AttributeSets =
637349cc55cSDimitry Andric       AttributeSets.addDereferenceableParamAttr(getContext(), ArgNo, Bytes);
6380b57cec5SDimitry Andric }
6390b57cec5SDimitry Andric 
640349cc55cSDimitry Andric bool Function::hasFnAttribute(Attribute::AttrKind Kind) const {
641349cc55cSDimitry Andric   return AttributeSets.hasFnAttr(Kind);
642349cc55cSDimitry Andric }
643349cc55cSDimitry Andric 
644349cc55cSDimitry Andric bool Function::hasFnAttribute(StringRef Kind) const {
645349cc55cSDimitry Andric   return AttributeSets.hasFnAttr(Kind);
646349cc55cSDimitry Andric }
647349cc55cSDimitry Andric 
648349cc55cSDimitry Andric bool Function::hasRetAttribute(Attribute::AttrKind Kind) const {
649349cc55cSDimitry Andric   return AttributeSets.hasRetAttr(Kind);
650349cc55cSDimitry Andric }
651349cc55cSDimitry Andric 
652349cc55cSDimitry Andric bool Function::hasParamAttribute(unsigned ArgNo,
653349cc55cSDimitry Andric                                  Attribute::AttrKind Kind) const {
654349cc55cSDimitry Andric   return AttributeSets.hasParamAttr(ArgNo, Kind);
655349cc55cSDimitry Andric }
656349cc55cSDimitry Andric 
657349cc55cSDimitry Andric Attribute Function::getAttributeAtIndex(unsigned i,
658349cc55cSDimitry Andric                                         Attribute::AttrKind Kind) const {
659349cc55cSDimitry Andric   return AttributeSets.getAttributeAtIndex(i, Kind);
660349cc55cSDimitry Andric }
661349cc55cSDimitry Andric 
662349cc55cSDimitry Andric Attribute Function::getAttributeAtIndex(unsigned i, StringRef Kind) const {
663349cc55cSDimitry Andric   return AttributeSets.getAttributeAtIndex(i, Kind);
664349cc55cSDimitry Andric }
665349cc55cSDimitry Andric 
666349cc55cSDimitry Andric Attribute Function::getFnAttribute(Attribute::AttrKind Kind) const {
667349cc55cSDimitry Andric   return AttributeSets.getFnAttr(Kind);
668349cc55cSDimitry Andric }
669349cc55cSDimitry Andric 
670349cc55cSDimitry Andric Attribute Function::getFnAttribute(StringRef Kind) const {
671349cc55cSDimitry Andric   return AttributeSets.getFnAttr(Kind);
672349cc55cSDimitry Andric }
673349cc55cSDimitry Andric 
674*bdd1243dSDimitry Andric uint64_t Function::getFnAttributeAsParsedInteger(StringRef Name,
675*bdd1243dSDimitry Andric                                                  uint64_t Default) const {
676*bdd1243dSDimitry Andric   Attribute A = getFnAttribute(Name);
677*bdd1243dSDimitry Andric   uint64_t Result = Default;
678*bdd1243dSDimitry Andric   if (A.isStringAttribute()) {
679*bdd1243dSDimitry Andric     StringRef Str = A.getValueAsString();
680*bdd1243dSDimitry Andric     if (Str.getAsInteger(0, Result))
681*bdd1243dSDimitry Andric       getContext().emitError("cannot parse integer attribute " + Name);
682*bdd1243dSDimitry Andric   }
683*bdd1243dSDimitry Andric 
684*bdd1243dSDimitry Andric   return Result;
685*bdd1243dSDimitry Andric }
686*bdd1243dSDimitry Andric 
687349cc55cSDimitry Andric /// gets the specified attribute from the list of attributes.
688349cc55cSDimitry Andric Attribute Function::getParamAttribute(unsigned ArgNo,
689349cc55cSDimitry Andric                                       Attribute::AttrKind Kind) const {
690349cc55cSDimitry Andric   return AttributeSets.getParamAttr(ArgNo, Kind);
6910b57cec5SDimitry Andric }
6920b57cec5SDimitry Andric 
6930b57cec5SDimitry Andric void Function::addDereferenceableOrNullParamAttr(unsigned ArgNo,
6940b57cec5SDimitry Andric                                                  uint64_t Bytes) {
695349cc55cSDimitry Andric   AttributeSets = AttributeSets.addDereferenceableOrNullParamAttr(getContext(),
696349cc55cSDimitry Andric                                                                   ArgNo, Bytes);
6970b57cec5SDimitry Andric }
6980b57cec5SDimitry Andric 
699e8d8bef9SDimitry Andric DenormalMode Function::getDenormalMode(const fltSemantics &FPType) const {
700e8d8bef9SDimitry Andric   if (&FPType == &APFloat::IEEEsingle()) {
701e8d8bef9SDimitry Andric     Attribute Attr = getFnAttribute("denormal-fp-math-f32");
702e8d8bef9SDimitry Andric     StringRef Val = Attr.getValueAsString();
703e8d8bef9SDimitry Andric     if (!Val.empty())
704e8d8bef9SDimitry Andric       return parseDenormalFPAttribute(Val);
705e8d8bef9SDimitry Andric 
706e8d8bef9SDimitry Andric     // If the f32 variant of the attribute isn't specified, try to use the
707e8d8bef9SDimitry Andric     // generic one.
708e8d8bef9SDimitry Andric   }
709e8d8bef9SDimitry Andric 
710e8d8bef9SDimitry Andric   Attribute Attr = getFnAttribute("denormal-fp-math");
711e8d8bef9SDimitry Andric   return parseDenormalFPAttribute(Attr.getValueAsString());
712e8d8bef9SDimitry Andric }
713e8d8bef9SDimitry Andric 
7140b57cec5SDimitry Andric const std::string &Function::getGC() const {
7150b57cec5SDimitry Andric   assert(hasGC() && "Function has no collector");
7160b57cec5SDimitry Andric   return getContext().getGC(*this);
7170b57cec5SDimitry Andric }
7180b57cec5SDimitry Andric 
7190b57cec5SDimitry Andric void Function::setGC(std::string Str) {
7200b57cec5SDimitry Andric   setValueSubclassDataBit(14, !Str.empty());
7210b57cec5SDimitry Andric   getContext().setGC(*this, std::move(Str));
7220b57cec5SDimitry Andric }
7230b57cec5SDimitry Andric 
7240b57cec5SDimitry Andric void Function::clearGC() {
7250b57cec5SDimitry Andric   if (!hasGC())
7260b57cec5SDimitry Andric     return;
7270b57cec5SDimitry Andric   getContext().deleteGC(*this);
7280b57cec5SDimitry Andric   setValueSubclassDataBit(14, false);
7290b57cec5SDimitry Andric }
7300b57cec5SDimitry Andric 
731e8d8bef9SDimitry Andric bool Function::hasStackProtectorFnAttr() const {
732e8d8bef9SDimitry Andric   return hasFnAttribute(Attribute::StackProtect) ||
733e8d8bef9SDimitry Andric          hasFnAttribute(Attribute::StackProtectStrong) ||
734e8d8bef9SDimitry Andric          hasFnAttribute(Attribute::StackProtectReq);
735e8d8bef9SDimitry Andric }
736e8d8bef9SDimitry Andric 
7370b57cec5SDimitry Andric /// Copy all additional attributes (those not needed to create a Function) from
7380b57cec5SDimitry Andric /// the Function Src to this one.
7390b57cec5SDimitry Andric void Function::copyAttributesFrom(const Function *Src) {
7400b57cec5SDimitry Andric   GlobalObject::copyAttributesFrom(Src);
7410b57cec5SDimitry Andric   setCallingConv(Src->getCallingConv());
7420b57cec5SDimitry Andric   setAttributes(Src->getAttributes());
7430b57cec5SDimitry Andric   if (Src->hasGC())
7440b57cec5SDimitry Andric     setGC(Src->getGC());
7450b57cec5SDimitry Andric   else
7460b57cec5SDimitry Andric     clearGC();
7470b57cec5SDimitry Andric   if (Src->hasPersonalityFn())
7480b57cec5SDimitry Andric     setPersonalityFn(Src->getPersonalityFn());
7490b57cec5SDimitry Andric   if (Src->hasPrefixData())
7500b57cec5SDimitry Andric     setPrefixData(Src->getPrefixData());
7510b57cec5SDimitry Andric   if (Src->hasPrologueData())
7520b57cec5SDimitry Andric     setPrologueData(Src->getPrologueData());
7530b57cec5SDimitry Andric }
7540b57cec5SDimitry Andric 
755*bdd1243dSDimitry Andric MemoryEffects Function::getMemoryEffects() const {
756*bdd1243dSDimitry Andric   return getAttributes().getMemoryEffects();
757*bdd1243dSDimitry Andric }
758*bdd1243dSDimitry Andric void Function::setMemoryEffects(MemoryEffects ME) {
759*bdd1243dSDimitry Andric   addFnAttr(Attribute::getWithMemoryEffects(getContext(), ME));
760*bdd1243dSDimitry Andric }
761*bdd1243dSDimitry Andric 
762*bdd1243dSDimitry Andric /// Determine if the function does not access memory.
763*bdd1243dSDimitry Andric bool Function::doesNotAccessMemory() const {
764*bdd1243dSDimitry Andric   return getMemoryEffects().doesNotAccessMemory();
765*bdd1243dSDimitry Andric }
766*bdd1243dSDimitry Andric void Function::setDoesNotAccessMemory() {
767*bdd1243dSDimitry Andric   setMemoryEffects(MemoryEffects::none());
768*bdd1243dSDimitry Andric }
769*bdd1243dSDimitry Andric 
770*bdd1243dSDimitry Andric /// Determine if the function does not access or only reads memory.
771*bdd1243dSDimitry Andric bool Function::onlyReadsMemory() const {
772*bdd1243dSDimitry Andric   return getMemoryEffects().onlyReadsMemory();
773*bdd1243dSDimitry Andric }
774*bdd1243dSDimitry Andric void Function::setOnlyReadsMemory() {
775*bdd1243dSDimitry Andric   setMemoryEffects(getMemoryEffects() & MemoryEffects::readOnly());
776*bdd1243dSDimitry Andric }
777*bdd1243dSDimitry Andric 
778*bdd1243dSDimitry Andric /// Determine if the function does not access or only writes memory.
779*bdd1243dSDimitry Andric bool Function::onlyWritesMemory() const {
780*bdd1243dSDimitry Andric   return getMemoryEffects().onlyWritesMemory();
781*bdd1243dSDimitry Andric }
782*bdd1243dSDimitry Andric void Function::setOnlyWritesMemory() {
783*bdd1243dSDimitry Andric   setMemoryEffects(getMemoryEffects() & MemoryEffects::writeOnly());
784*bdd1243dSDimitry Andric }
785*bdd1243dSDimitry Andric 
786*bdd1243dSDimitry Andric /// Determine if the call can access memmory only using pointers based
787*bdd1243dSDimitry Andric /// on its arguments.
788*bdd1243dSDimitry Andric bool Function::onlyAccessesArgMemory() const {
789*bdd1243dSDimitry Andric   return getMemoryEffects().onlyAccessesArgPointees();
790*bdd1243dSDimitry Andric }
791*bdd1243dSDimitry Andric void Function::setOnlyAccessesArgMemory() {
792*bdd1243dSDimitry Andric   setMemoryEffects(getMemoryEffects() & MemoryEffects::argMemOnly());
793*bdd1243dSDimitry Andric }
794*bdd1243dSDimitry Andric 
795*bdd1243dSDimitry Andric /// Determine if the function may only access memory that is
796*bdd1243dSDimitry Andric ///  inaccessible from the IR.
797*bdd1243dSDimitry Andric bool Function::onlyAccessesInaccessibleMemory() const {
798*bdd1243dSDimitry Andric   return getMemoryEffects().onlyAccessesInaccessibleMem();
799*bdd1243dSDimitry Andric }
800*bdd1243dSDimitry Andric void Function::setOnlyAccessesInaccessibleMemory() {
801*bdd1243dSDimitry Andric   setMemoryEffects(getMemoryEffects() & MemoryEffects::inaccessibleMemOnly());
802*bdd1243dSDimitry Andric }
803*bdd1243dSDimitry Andric 
804*bdd1243dSDimitry Andric /// Determine if the function may only access memory that is
805*bdd1243dSDimitry Andric ///  either inaccessible from the IR or pointed to by its arguments.
806*bdd1243dSDimitry Andric bool Function::onlyAccessesInaccessibleMemOrArgMem() const {
807*bdd1243dSDimitry Andric   return getMemoryEffects().onlyAccessesInaccessibleOrArgMem();
808*bdd1243dSDimitry Andric }
809*bdd1243dSDimitry Andric void Function::setOnlyAccessesInaccessibleMemOrArgMem() {
810*bdd1243dSDimitry Andric   setMemoryEffects(getMemoryEffects() &
811*bdd1243dSDimitry Andric                    MemoryEffects::inaccessibleOrArgMemOnly());
812*bdd1243dSDimitry Andric }
813*bdd1243dSDimitry Andric 
8140b57cec5SDimitry Andric /// Table of string intrinsic names indexed by enum value.
8150b57cec5SDimitry Andric static const char * const IntrinsicNameTable[] = {
8160b57cec5SDimitry Andric   "not_intrinsic",
8170b57cec5SDimitry Andric #define GET_INTRINSIC_NAME_TABLE
8180b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
8190b57cec5SDimitry Andric #undef GET_INTRINSIC_NAME_TABLE
8200b57cec5SDimitry Andric };
8210b57cec5SDimitry Andric 
8220b57cec5SDimitry Andric /// Table of per-target intrinsic name tables.
8230b57cec5SDimitry Andric #define GET_INTRINSIC_TARGET_DATA
8240b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
8250b57cec5SDimitry Andric #undef GET_INTRINSIC_TARGET_DATA
8260b57cec5SDimitry Andric 
827e8d8bef9SDimitry Andric bool Function::isTargetIntrinsic(Intrinsic::ID IID) {
828e8d8bef9SDimitry Andric   return IID > TargetInfos[0].Count;
829e8d8bef9SDimitry Andric }
830e8d8bef9SDimitry Andric 
831e8d8bef9SDimitry Andric bool Function::isTargetIntrinsic() const {
832e8d8bef9SDimitry Andric   return isTargetIntrinsic(IntID);
833e8d8bef9SDimitry Andric }
834e8d8bef9SDimitry Andric 
8350b57cec5SDimitry Andric /// Find the segment of \c IntrinsicNameTable for intrinsics with the same
8360b57cec5SDimitry Andric /// target as \c Name, or the generic table if \c Name is not target specific.
8370b57cec5SDimitry Andric ///
8380b57cec5SDimitry Andric /// Returns the relevant slice of \c IntrinsicNameTable
8390b57cec5SDimitry Andric static ArrayRef<const char *> findTargetSubtable(StringRef Name) {
8400b57cec5SDimitry Andric   assert(Name.startswith("llvm."));
8410b57cec5SDimitry Andric 
8420b57cec5SDimitry Andric   ArrayRef<IntrinsicTargetInfo> Targets(TargetInfos);
8430b57cec5SDimitry Andric   // Drop "llvm." and take the first dotted component. That will be the target
8440b57cec5SDimitry Andric   // if this is target specific.
8450b57cec5SDimitry Andric   StringRef Target = Name.drop_front(5).split('.').first;
8460b57cec5SDimitry Andric   auto It = partition_point(
8470b57cec5SDimitry Andric       Targets, [=](const IntrinsicTargetInfo &TI) { return TI.Name < Target; });
8480b57cec5SDimitry Andric   // We've either found the target or just fall back to the generic set, which
8490b57cec5SDimitry Andric   // is always first.
8500b57cec5SDimitry Andric   const auto &TI = It != Targets.end() && It->Name == Target ? *It : Targets[0];
851*bdd1243dSDimitry Andric   return ArrayRef(&IntrinsicNameTable[1] + TI.Offset, TI.Count);
8520b57cec5SDimitry Andric }
8530b57cec5SDimitry Andric 
8540b57cec5SDimitry Andric /// This does the actual lookup of an intrinsic ID which
8550b57cec5SDimitry Andric /// matches the given function name.
8560b57cec5SDimitry Andric Intrinsic::ID Function::lookupIntrinsicID(StringRef Name) {
8570b57cec5SDimitry Andric   ArrayRef<const char *> NameTable = findTargetSubtable(Name);
8580b57cec5SDimitry Andric   int Idx = Intrinsic::lookupLLVMIntrinsicByName(NameTable, Name);
8590b57cec5SDimitry Andric   if (Idx == -1)
8600b57cec5SDimitry Andric     return Intrinsic::not_intrinsic;
8610b57cec5SDimitry Andric 
8620b57cec5SDimitry Andric   // Intrinsic IDs correspond to the location in IntrinsicNameTable, but we have
8630b57cec5SDimitry Andric   // an index into a sub-table.
8640b57cec5SDimitry Andric   int Adjust = NameTable.data() - IntrinsicNameTable;
8650b57cec5SDimitry Andric   Intrinsic::ID ID = static_cast<Intrinsic::ID>(Idx + Adjust);
8660b57cec5SDimitry Andric 
8670b57cec5SDimitry Andric   // If the intrinsic is not overloaded, require an exact match. If it is
8680b57cec5SDimitry Andric   // overloaded, require either exact or prefix match.
8690b57cec5SDimitry Andric   const auto MatchSize = strlen(NameTable[Idx]);
8700b57cec5SDimitry Andric   assert(Name.size() >= MatchSize && "Expected either exact or prefix match");
8710b57cec5SDimitry Andric   bool IsExactMatch = Name.size() == MatchSize;
872480093f4SDimitry Andric   return IsExactMatch || Intrinsic::isOverloaded(ID) ? ID
873480093f4SDimitry Andric                                                      : Intrinsic::not_intrinsic;
8740b57cec5SDimitry Andric }
8750b57cec5SDimitry Andric 
8760b57cec5SDimitry Andric void Function::recalculateIntrinsicID() {
8770b57cec5SDimitry Andric   StringRef Name = getName();
8780b57cec5SDimitry Andric   if (!Name.startswith("llvm.")) {
8790b57cec5SDimitry Andric     HasLLVMReservedName = false;
8800b57cec5SDimitry Andric     IntID = Intrinsic::not_intrinsic;
8810b57cec5SDimitry Andric     return;
8820b57cec5SDimitry Andric   }
8830b57cec5SDimitry Andric   HasLLVMReservedName = true;
8840b57cec5SDimitry Andric   IntID = lookupIntrinsicID(Name);
8850b57cec5SDimitry Andric }
8860b57cec5SDimitry Andric 
8870b57cec5SDimitry Andric /// Returns a stable mangling for the type specified for use in the name
8880b57cec5SDimitry Andric /// mangling scheme used by 'any' types in intrinsic signatures.  The mangling
8890b57cec5SDimitry Andric /// of named types is simply their name.  Manglings for unnamed types consist
8900b57cec5SDimitry Andric /// of a prefix ('p' for pointers, 'a' for arrays, 'f_' for functions)
8910b57cec5SDimitry Andric /// combined with the mangling of their component types.  A vararg function
8920b57cec5SDimitry Andric /// type will have a suffix of 'vararg'.  Since function types can contain
8930b57cec5SDimitry Andric /// other function types, we close a function type mangling with suffix 'f'
8940b57cec5SDimitry Andric /// which can't be confused with it's prefix.  This ensures we don't have
8950b57cec5SDimitry Andric /// collisions between two unrelated function types. Otherwise, you might
8960b57cec5SDimitry Andric /// parse ffXX as f(fXX) or f(fX)X.  (X is a placeholder for any other type.)
897fe6060f1SDimitry Andric /// The HasUnnamedType boolean is set if an unnamed type was encountered,
898fe6060f1SDimitry Andric /// indicating that extra care must be taken to ensure a unique name.
899fe6060f1SDimitry Andric static std::string getMangledTypeStr(Type *Ty, bool &HasUnnamedType) {
9000b57cec5SDimitry Andric   std::string Result;
9010b57cec5SDimitry Andric   if (PointerType *PTyp = dyn_cast<PointerType>(Ty)) {
902fe6060f1SDimitry Andric     Result += "p" + utostr(PTyp->getAddressSpace());
903fe6060f1SDimitry Andric     // Opaque pointer doesn't have pointee type information, so we just mangle
904fe6060f1SDimitry Andric     // address space for opaque pointer.
905fe6060f1SDimitry Andric     if (!PTyp->isOpaque())
90604eeddc0SDimitry Andric       Result += getMangledTypeStr(PTyp->getNonOpaquePointerElementType(),
90704eeddc0SDimitry Andric                                   HasUnnamedType);
9080b57cec5SDimitry Andric   } else if (ArrayType *ATyp = dyn_cast<ArrayType>(Ty)) {
9090b57cec5SDimitry Andric     Result += "a" + utostr(ATyp->getNumElements()) +
910fe6060f1SDimitry Andric               getMangledTypeStr(ATyp->getElementType(), HasUnnamedType);
9110b57cec5SDimitry Andric   } else if (StructType *STyp = dyn_cast<StructType>(Ty)) {
9120b57cec5SDimitry Andric     if (!STyp->isLiteral()) {
9130b57cec5SDimitry Andric       Result += "s_";
914fe6060f1SDimitry Andric       if (STyp->hasName())
9150b57cec5SDimitry Andric         Result += STyp->getName();
916fe6060f1SDimitry Andric       else
917fe6060f1SDimitry Andric         HasUnnamedType = true;
9180b57cec5SDimitry Andric     } else {
9190b57cec5SDimitry Andric       Result += "sl_";
920*bdd1243dSDimitry Andric       for (auto *Elem : STyp->elements())
921fe6060f1SDimitry Andric         Result += getMangledTypeStr(Elem, HasUnnamedType);
9220b57cec5SDimitry Andric     }
9230b57cec5SDimitry Andric     // Ensure nested structs are distinguishable.
9240b57cec5SDimitry Andric     Result += "s";
9250b57cec5SDimitry Andric   } else if (FunctionType *FT = dyn_cast<FunctionType>(Ty)) {
926fe6060f1SDimitry Andric     Result += "f_" + getMangledTypeStr(FT->getReturnType(), HasUnnamedType);
9270b57cec5SDimitry Andric     for (size_t i = 0; i < FT->getNumParams(); i++)
928fe6060f1SDimitry Andric       Result += getMangledTypeStr(FT->getParamType(i), HasUnnamedType);
9290b57cec5SDimitry Andric     if (FT->isVarArg())
9300b57cec5SDimitry Andric       Result += "vararg";
9310b57cec5SDimitry Andric     // Ensure nested function types are distinguishable.
9320b57cec5SDimitry Andric     Result += "f";
9338bcb0991SDimitry Andric   } else if (VectorType *VTy = dyn_cast<VectorType>(Ty)) {
9345ffd83dbSDimitry Andric     ElementCount EC = VTy->getElementCount();
935e8d8bef9SDimitry Andric     if (EC.isScalable())
9368bcb0991SDimitry Andric       Result += "nx";
937e8d8bef9SDimitry Andric     Result += "v" + utostr(EC.getKnownMinValue()) +
938fe6060f1SDimitry Andric               getMangledTypeStr(VTy->getElementType(), HasUnnamedType);
939*bdd1243dSDimitry Andric   } else if (TargetExtType *TETy = dyn_cast<TargetExtType>(Ty)) {
940*bdd1243dSDimitry Andric     Result += "t";
941*bdd1243dSDimitry Andric     Result += TETy->getName();
942*bdd1243dSDimitry Andric     for (Type *ParamTy : TETy->type_params())
943*bdd1243dSDimitry Andric       Result += "_" + getMangledTypeStr(ParamTy, HasUnnamedType);
944*bdd1243dSDimitry Andric     for (unsigned IntParam : TETy->int_params())
945*bdd1243dSDimitry Andric       Result += "_" + utostr(IntParam);
946*bdd1243dSDimitry Andric     // Ensure nested target extension types are distinguishable.
947*bdd1243dSDimitry Andric     Result += "t";
9480b57cec5SDimitry Andric   } else if (Ty) {
9490b57cec5SDimitry Andric     switch (Ty->getTypeID()) {
9500b57cec5SDimitry Andric     default: llvm_unreachable("Unhandled type");
9510b57cec5SDimitry Andric     case Type::VoidTyID:      Result += "isVoid";   break;
9520b57cec5SDimitry Andric     case Type::MetadataTyID:  Result += "Metadata"; break;
9530b57cec5SDimitry Andric     case Type::HalfTyID:      Result += "f16";      break;
9545ffd83dbSDimitry Andric     case Type::BFloatTyID:    Result += "bf16";     break;
9550b57cec5SDimitry Andric     case Type::FloatTyID:     Result += "f32";      break;
9560b57cec5SDimitry Andric     case Type::DoubleTyID:    Result += "f64";      break;
9570b57cec5SDimitry Andric     case Type::X86_FP80TyID:  Result += "f80";      break;
9580b57cec5SDimitry Andric     case Type::FP128TyID:     Result += "f128";     break;
9590b57cec5SDimitry Andric     case Type::PPC_FP128TyID: Result += "ppcf128";  break;
9600b57cec5SDimitry Andric     case Type::X86_MMXTyID:   Result += "x86mmx";   break;
961e8d8bef9SDimitry Andric     case Type::X86_AMXTyID:   Result += "x86amx";   break;
9620b57cec5SDimitry Andric     case Type::IntegerTyID:
9630b57cec5SDimitry Andric       Result += "i" + utostr(cast<IntegerType>(Ty)->getBitWidth());
9640b57cec5SDimitry Andric       break;
9650b57cec5SDimitry Andric     }
9660b57cec5SDimitry Andric   }
9670b57cec5SDimitry Andric   return Result;
9680b57cec5SDimitry Andric }
9690b57cec5SDimitry Andric 
970fe6060f1SDimitry Andric StringRef Intrinsic::getBaseName(ID id) {
971fe6060f1SDimitry Andric   assert(id < num_intrinsics && "Invalid intrinsic ID!");
972fe6060f1SDimitry Andric   return IntrinsicNameTable[id];
973fe6060f1SDimitry Andric }
974fe6060f1SDimitry Andric 
9750b57cec5SDimitry Andric StringRef Intrinsic::getName(ID id) {
9760b57cec5SDimitry Andric   assert(id < num_intrinsics && "Invalid intrinsic ID!");
977480093f4SDimitry Andric   assert(!Intrinsic::isOverloaded(id) &&
9780b57cec5SDimitry Andric          "This version of getName does not support overloading");
979fe6060f1SDimitry Andric   return getBaseName(id);
9800b57cec5SDimitry Andric }
9810b57cec5SDimitry Andric 
982fe6060f1SDimitry Andric static std::string getIntrinsicNameImpl(Intrinsic::ID Id, ArrayRef<Type *> Tys,
983fe6060f1SDimitry Andric                                         Module *M, FunctionType *FT,
984fe6060f1SDimitry Andric                                         bool EarlyModuleCheck) {
985fe6060f1SDimitry Andric 
986fe6060f1SDimitry Andric   assert(Id < Intrinsic::num_intrinsics && "Invalid intrinsic ID!");
987fe6060f1SDimitry Andric   assert((Tys.empty() || Intrinsic::isOverloaded(Id)) &&
988e8d8bef9SDimitry Andric          "This version of getName is for overloaded intrinsics only");
989fe6060f1SDimitry Andric   (void)EarlyModuleCheck;
990fe6060f1SDimitry Andric   assert((!EarlyModuleCheck || M ||
991fe6060f1SDimitry Andric           !any_of(Tys, [](Type *T) { return isa<PointerType>(T); })) &&
992fe6060f1SDimitry Andric          "Intrinsic overloading on pointer types need to provide a Module");
993fe6060f1SDimitry Andric   bool HasUnnamedType = false;
994fe6060f1SDimitry Andric   std::string Result(Intrinsic::getBaseName(Id));
995fe6060f1SDimitry Andric   for (Type *Ty : Tys)
996fe6060f1SDimitry Andric     Result += "." + getMangledTypeStr(Ty, HasUnnamedType);
997fe6060f1SDimitry Andric   if (HasUnnamedType) {
998fe6060f1SDimitry Andric     assert(M && "unnamed types need a module");
999fe6060f1SDimitry Andric     if (!FT)
1000fe6060f1SDimitry Andric       FT = Intrinsic::getType(M->getContext(), Id, Tys);
1001fe6060f1SDimitry Andric     else
1002fe6060f1SDimitry Andric       assert((FT == Intrinsic::getType(M->getContext(), Id, Tys)) &&
1003fe6060f1SDimitry Andric              "Provided FunctionType must match arguments");
1004fe6060f1SDimitry Andric     return M->getUniqueIntrinsicName(Result, Id, FT);
10050b57cec5SDimitry Andric   }
10060b57cec5SDimitry Andric   return Result;
10070b57cec5SDimitry Andric }
10080b57cec5SDimitry Andric 
1009fe6060f1SDimitry Andric std::string Intrinsic::getName(ID Id, ArrayRef<Type *> Tys, Module *M,
1010fe6060f1SDimitry Andric                                FunctionType *FT) {
1011fe6060f1SDimitry Andric   assert(M && "We need to have a Module");
1012fe6060f1SDimitry Andric   return getIntrinsicNameImpl(Id, Tys, M, FT, true);
1013fe6060f1SDimitry Andric }
1014fe6060f1SDimitry Andric 
1015fe6060f1SDimitry Andric std::string Intrinsic::getNameNoUnnamedTypes(ID Id, ArrayRef<Type *> Tys) {
1016fe6060f1SDimitry Andric   return getIntrinsicNameImpl(Id, Tys, nullptr, nullptr, false);
1017fe6060f1SDimitry Andric }
1018fe6060f1SDimitry Andric 
10190b57cec5SDimitry Andric /// IIT_Info - These are enumerators that describe the entries returned by the
10200b57cec5SDimitry Andric /// getIntrinsicInfoTableEntries function.
10210b57cec5SDimitry Andric ///
10220b57cec5SDimitry Andric /// NOTE: This must be kept in synch with the copy in TblGen/IntrinsicEmitter!
10230b57cec5SDimitry Andric enum IIT_Info {
10240b57cec5SDimitry Andric   // Common values should be encoded with 0-15.
10250b57cec5SDimitry Andric   IIT_Done = 0,
10260b57cec5SDimitry Andric   IIT_I1 = 1,
10270b57cec5SDimitry Andric   IIT_I8 = 2,
10280b57cec5SDimitry Andric   IIT_I16 = 3,
10290b57cec5SDimitry Andric   IIT_I32 = 4,
10300b57cec5SDimitry Andric   IIT_I64 = 5,
10310b57cec5SDimitry Andric   IIT_F16 = 6,
10320b57cec5SDimitry Andric   IIT_F32 = 7,
10330b57cec5SDimitry Andric   IIT_F64 = 8,
10340b57cec5SDimitry Andric   IIT_V2 = 9,
10350b57cec5SDimitry Andric   IIT_V4 = 10,
10360b57cec5SDimitry Andric   IIT_V8 = 11,
10370b57cec5SDimitry Andric   IIT_V16 = 12,
10380b57cec5SDimitry Andric   IIT_V32 = 13,
10390b57cec5SDimitry Andric   IIT_PTR = 14,
10400b57cec5SDimitry Andric   IIT_ARG = 15,
10410b57cec5SDimitry Andric 
10420b57cec5SDimitry Andric   // Values from 16+ are only encodable with the inefficient encoding.
10430b57cec5SDimitry Andric   IIT_V64 = 16,
10440b57cec5SDimitry Andric   IIT_MMX = 17,
10450b57cec5SDimitry Andric   IIT_TOKEN = 18,
10460b57cec5SDimitry Andric   IIT_METADATA = 19,
10470b57cec5SDimitry Andric   IIT_EMPTYSTRUCT = 20,
10480b57cec5SDimitry Andric   IIT_STRUCT2 = 21,
10490b57cec5SDimitry Andric   IIT_STRUCT3 = 22,
10500b57cec5SDimitry Andric   IIT_STRUCT4 = 23,
10510b57cec5SDimitry Andric   IIT_STRUCT5 = 24,
10520b57cec5SDimitry Andric   IIT_EXTEND_ARG = 25,
10530b57cec5SDimitry Andric   IIT_TRUNC_ARG = 26,
10540b57cec5SDimitry Andric   IIT_ANYPTR = 27,
10550b57cec5SDimitry Andric   IIT_V1 = 28,
10560b57cec5SDimitry Andric   IIT_VARARG = 29,
10570b57cec5SDimitry Andric   IIT_HALF_VEC_ARG = 30,
10580b57cec5SDimitry Andric   IIT_SAME_VEC_WIDTH_ARG = 31,
10590b57cec5SDimitry Andric   IIT_PTR_TO_ARG = 32,
10600b57cec5SDimitry Andric   IIT_PTR_TO_ELT = 33,
10610b57cec5SDimitry Andric   IIT_VEC_OF_ANYPTRS_TO_ELT = 34,
10620b57cec5SDimitry Andric   IIT_I128 = 35,
10630b57cec5SDimitry Andric   IIT_V512 = 36,
10640b57cec5SDimitry Andric   IIT_V1024 = 37,
10650b57cec5SDimitry Andric   IIT_STRUCT6 = 38,
10660b57cec5SDimitry Andric   IIT_STRUCT7 = 39,
10670b57cec5SDimitry Andric   IIT_STRUCT8 = 40,
10680b57cec5SDimitry Andric   IIT_F128 = 41,
10698bcb0991SDimitry Andric   IIT_VEC_ELEMENT = 42,
10708bcb0991SDimitry Andric   IIT_SCALABLE_VEC = 43,
10718bcb0991SDimitry Andric   IIT_SUBDIVIDE2_ARG = 44,
10728bcb0991SDimitry Andric   IIT_SUBDIVIDE4_ARG = 45,
10735ffd83dbSDimitry Andric   IIT_VEC_OF_BITCASTS_TO_INT = 46,
10745ffd83dbSDimitry Andric   IIT_V128 = 47,
1075e8d8bef9SDimitry Andric   IIT_BF16 = 48,
1076e8d8bef9SDimitry Andric   IIT_STRUCT9 = 49,
1077e8d8bef9SDimitry Andric   IIT_V256 = 50,
1078349cc55cSDimitry Andric   IIT_AMX = 51,
10790eae32dcSDimitry Andric   IIT_PPCF128 = 52,
10800eae32dcSDimitry Andric   IIT_V3 = 53,
10810eae32dcSDimitry Andric   IIT_EXTERNREF = 54,
108281ad6265SDimitry Andric   IIT_FUNCREF = 55,
108381ad6265SDimitry Andric   IIT_ANYPTR_TO_ELT = 56,
108481ad6265SDimitry Andric   IIT_I2 = 57,
108581ad6265SDimitry Andric   IIT_I4 = 58,
10860b57cec5SDimitry Andric };
10870b57cec5SDimitry Andric 
10880b57cec5SDimitry Andric static void DecodeIITType(unsigned &NextElt, ArrayRef<unsigned char> Infos,
10895ffd83dbSDimitry Andric                       IIT_Info LastInfo,
10900b57cec5SDimitry Andric                       SmallVectorImpl<Intrinsic::IITDescriptor> &OutputTable) {
10910b57cec5SDimitry Andric   using namespace Intrinsic;
10920b57cec5SDimitry Andric 
10935ffd83dbSDimitry Andric   bool IsScalableVector = (LastInfo == IIT_SCALABLE_VEC);
10945ffd83dbSDimitry Andric 
10950b57cec5SDimitry Andric   IIT_Info Info = IIT_Info(Infos[NextElt++]);
10960b57cec5SDimitry Andric   unsigned StructElts = 2;
10970b57cec5SDimitry Andric 
10980b57cec5SDimitry Andric   switch (Info) {
10990b57cec5SDimitry Andric   case IIT_Done:
11000b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Void, 0));
11010b57cec5SDimitry Andric     return;
11020b57cec5SDimitry Andric   case IIT_VARARG:
11030b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::VarArg, 0));
11040b57cec5SDimitry Andric     return;
11050b57cec5SDimitry Andric   case IIT_MMX:
11060b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::MMX, 0));
11070b57cec5SDimitry Andric     return;
1108e8d8bef9SDimitry Andric   case IIT_AMX:
1109e8d8bef9SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::AMX, 0));
1110e8d8bef9SDimitry Andric     return;
11110b57cec5SDimitry Andric   case IIT_TOKEN:
11120b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Token, 0));
11130b57cec5SDimitry Andric     return;
11140b57cec5SDimitry Andric   case IIT_METADATA:
11150b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Metadata, 0));
11160b57cec5SDimitry Andric     return;
11170b57cec5SDimitry Andric   case IIT_F16:
11180b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Half, 0));
11190b57cec5SDimitry Andric     return;
11205ffd83dbSDimitry Andric   case IIT_BF16:
11215ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::BFloat, 0));
11225ffd83dbSDimitry Andric     return;
11230b57cec5SDimitry Andric   case IIT_F32:
11240b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Float, 0));
11250b57cec5SDimitry Andric     return;
11260b57cec5SDimitry Andric   case IIT_F64:
11270b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Double, 0));
11280b57cec5SDimitry Andric     return;
11290b57cec5SDimitry Andric   case IIT_F128:
11300b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Quad, 0));
11310b57cec5SDimitry Andric     return;
1132349cc55cSDimitry Andric   case IIT_PPCF128:
1133349cc55cSDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::PPCQuad, 0));
1134349cc55cSDimitry Andric     return;
11350b57cec5SDimitry Andric   case IIT_I1:
11360b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 1));
11370b57cec5SDimitry Andric     return;
113881ad6265SDimitry Andric   case IIT_I2:
113981ad6265SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 2));
114081ad6265SDimitry Andric     return;
114181ad6265SDimitry Andric   case IIT_I4:
114281ad6265SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 4));
114381ad6265SDimitry Andric     return;
11440b57cec5SDimitry Andric   case IIT_I8:
11450b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 8));
11460b57cec5SDimitry Andric     return;
11470b57cec5SDimitry Andric   case IIT_I16:
11480b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer,16));
11490b57cec5SDimitry Andric     return;
11500b57cec5SDimitry Andric   case IIT_I32:
11510b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 32));
11520b57cec5SDimitry Andric     return;
11530b57cec5SDimitry Andric   case IIT_I64:
11540b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 64));
11550b57cec5SDimitry Andric     return;
11560b57cec5SDimitry Andric   case IIT_I128:
11570b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 128));
11580b57cec5SDimitry Andric     return;
11590b57cec5SDimitry Andric   case IIT_V1:
11605ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(1, IsScalableVector));
11615ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11620b57cec5SDimitry Andric     return;
11630b57cec5SDimitry Andric   case IIT_V2:
11645ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(2, IsScalableVector));
11655ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11660b57cec5SDimitry Andric     return;
11670eae32dcSDimitry Andric   case IIT_V3:
11680eae32dcSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(3, IsScalableVector));
11690eae32dcSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11700eae32dcSDimitry Andric     return;
11710b57cec5SDimitry Andric   case IIT_V4:
11725ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(4, IsScalableVector));
11735ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11740b57cec5SDimitry Andric     return;
11750b57cec5SDimitry Andric   case IIT_V8:
11765ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(8, IsScalableVector));
11775ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11780b57cec5SDimitry Andric     return;
11790b57cec5SDimitry Andric   case IIT_V16:
11805ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(16, IsScalableVector));
11815ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11820b57cec5SDimitry Andric     return;
11830b57cec5SDimitry Andric   case IIT_V32:
11845ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(32, IsScalableVector));
11855ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11860b57cec5SDimitry Andric     return;
11870b57cec5SDimitry Andric   case IIT_V64:
11885ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(64, IsScalableVector));
11895ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11905ffd83dbSDimitry Andric     return;
11915ffd83dbSDimitry Andric   case IIT_V128:
11925ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(128, IsScalableVector));
11935ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
11940b57cec5SDimitry Andric     return;
1195e8d8bef9SDimitry Andric   case IIT_V256:
1196e8d8bef9SDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(256, IsScalableVector));
1197e8d8bef9SDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
1198e8d8bef9SDimitry Andric     return;
11990b57cec5SDimitry Andric   case IIT_V512:
12005ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(512, IsScalableVector));
12015ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
12020b57cec5SDimitry Andric     return;
12030b57cec5SDimitry Andric   case IIT_V1024:
12045ffd83dbSDimitry Andric     OutputTable.push_back(IITDescriptor::getVector(1024, IsScalableVector));
12055ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
12060b57cec5SDimitry Andric     return;
12070eae32dcSDimitry Andric   case IIT_EXTERNREF:
12080eae32dcSDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 10));
12090eae32dcSDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Struct, 0));
12100eae32dcSDimitry Andric     return;
12110eae32dcSDimitry Andric   case IIT_FUNCREF:
12120eae32dcSDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 20));
12130eae32dcSDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Integer, 8));
12140eae32dcSDimitry Andric     return;
12150b57cec5SDimitry Andric   case IIT_PTR:
12160b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer, 0));
12175ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
12180b57cec5SDimitry Andric     return;
12190b57cec5SDimitry Andric   case IIT_ANYPTR: {  // [ANYPTR addrspace, subtype]
12200b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Pointer,
12210b57cec5SDimitry Andric                                              Infos[NextElt++]));
12225ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
12230b57cec5SDimitry Andric     return;
12240b57cec5SDimitry Andric   }
12250b57cec5SDimitry Andric   case IIT_ARG: {
12260b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12270b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Argument, ArgInfo));
12280b57cec5SDimitry Andric     return;
12290b57cec5SDimitry Andric   }
12300b57cec5SDimitry Andric   case IIT_EXTEND_ARG: {
12310b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12320b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::ExtendArgument,
12330b57cec5SDimitry Andric                                              ArgInfo));
12340b57cec5SDimitry Andric     return;
12350b57cec5SDimitry Andric   }
12360b57cec5SDimitry Andric   case IIT_TRUNC_ARG: {
12370b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12380b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::TruncArgument,
12390b57cec5SDimitry Andric                                              ArgInfo));
12400b57cec5SDimitry Andric     return;
12410b57cec5SDimitry Andric   }
12420b57cec5SDimitry Andric   case IIT_HALF_VEC_ARG: {
12430b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12440b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::HalfVecArgument,
12450b57cec5SDimitry Andric                                              ArgInfo));
12460b57cec5SDimitry Andric     return;
12470b57cec5SDimitry Andric   }
12480b57cec5SDimitry Andric   case IIT_SAME_VEC_WIDTH_ARG: {
12490b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12500b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::SameVecWidthArgument,
12510b57cec5SDimitry Andric                                              ArgInfo));
12520b57cec5SDimitry Andric     return;
12530b57cec5SDimitry Andric   }
12540b57cec5SDimitry Andric   case IIT_PTR_TO_ARG: {
12550b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12560b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::PtrToArgument,
12570b57cec5SDimitry Andric                                              ArgInfo));
12580b57cec5SDimitry Andric     return;
12590b57cec5SDimitry Andric   }
12600b57cec5SDimitry Andric   case IIT_PTR_TO_ELT: {
12610b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12620b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::PtrToElt, ArgInfo));
12630b57cec5SDimitry Andric     return;
12640b57cec5SDimitry Andric   }
126581ad6265SDimitry Andric   case IIT_ANYPTR_TO_ELT: {
126681ad6265SDimitry Andric     unsigned short ArgNo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
126781ad6265SDimitry Andric     unsigned short RefNo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
126881ad6265SDimitry Andric     OutputTable.push_back(
126981ad6265SDimitry Andric         IITDescriptor::get(IITDescriptor::AnyPtrToElt, ArgNo, RefNo));
127081ad6265SDimitry Andric     return;
127181ad6265SDimitry Andric   }
12720b57cec5SDimitry Andric   case IIT_VEC_OF_ANYPTRS_TO_ELT: {
12730b57cec5SDimitry Andric     unsigned short ArgNo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12740b57cec5SDimitry Andric     unsigned short RefNo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12750b57cec5SDimitry Andric     OutputTable.push_back(
12760b57cec5SDimitry Andric         IITDescriptor::get(IITDescriptor::VecOfAnyPtrsToElt, ArgNo, RefNo));
12770b57cec5SDimitry Andric     return;
12780b57cec5SDimitry Andric   }
12790b57cec5SDimitry Andric   case IIT_EMPTYSTRUCT:
12800b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Struct, 0));
12810b57cec5SDimitry Andric     return;
1282*bdd1243dSDimitry Andric   case IIT_STRUCT9: ++StructElts; [[fallthrough]];
1283*bdd1243dSDimitry Andric   case IIT_STRUCT8: ++StructElts; [[fallthrough]];
1284*bdd1243dSDimitry Andric   case IIT_STRUCT7: ++StructElts; [[fallthrough]];
1285*bdd1243dSDimitry Andric   case IIT_STRUCT6: ++StructElts; [[fallthrough]];
1286*bdd1243dSDimitry Andric   case IIT_STRUCT5: ++StructElts; [[fallthrough]];
1287*bdd1243dSDimitry Andric   case IIT_STRUCT4: ++StructElts; [[fallthrough]];
1288*bdd1243dSDimitry Andric   case IIT_STRUCT3: ++StructElts; [[fallthrough]];
12890b57cec5SDimitry Andric   case IIT_STRUCT2: {
12900b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Struct,StructElts));
12910b57cec5SDimitry Andric 
12920b57cec5SDimitry Andric     for (unsigned i = 0; i != StructElts; ++i)
12935ffd83dbSDimitry Andric       DecodeIITType(NextElt, Infos, Info, OutputTable);
12940b57cec5SDimitry Andric     return;
12950b57cec5SDimitry Andric   }
12968bcb0991SDimitry Andric   case IIT_SUBDIVIDE2_ARG: {
12978bcb0991SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
12988bcb0991SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Subdivide2Argument,
12998bcb0991SDimitry Andric                                              ArgInfo));
13008bcb0991SDimitry Andric     return;
13018bcb0991SDimitry Andric   }
13028bcb0991SDimitry Andric   case IIT_SUBDIVIDE4_ARG: {
13038bcb0991SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
13048bcb0991SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::Subdivide4Argument,
13058bcb0991SDimitry Andric                                              ArgInfo));
13068bcb0991SDimitry Andric     return;
13078bcb0991SDimitry Andric   }
13080b57cec5SDimitry Andric   case IIT_VEC_ELEMENT: {
13090b57cec5SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
13100b57cec5SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::VecElementArgument,
13110b57cec5SDimitry Andric                                              ArgInfo));
13120b57cec5SDimitry Andric     return;
13130b57cec5SDimitry Andric   }
13148bcb0991SDimitry Andric   case IIT_SCALABLE_VEC: {
13155ffd83dbSDimitry Andric     DecodeIITType(NextElt, Infos, Info, OutputTable);
13168bcb0991SDimitry Andric     return;
13178bcb0991SDimitry Andric   }
13188bcb0991SDimitry Andric   case IIT_VEC_OF_BITCASTS_TO_INT: {
13198bcb0991SDimitry Andric     unsigned ArgInfo = (NextElt == Infos.size() ? 0 : Infos[NextElt++]);
13208bcb0991SDimitry Andric     OutputTable.push_back(IITDescriptor::get(IITDescriptor::VecOfBitcastsToInt,
13218bcb0991SDimitry Andric                                              ArgInfo));
13228bcb0991SDimitry Andric     return;
13238bcb0991SDimitry Andric   }
13240b57cec5SDimitry Andric   }
13250b57cec5SDimitry Andric   llvm_unreachable("unhandled");
13260b57cec5SDimitry Andric }
13270b57cec5SDimitry Andric 
13280b57cec5SDimitry Andric #define GET_INTRINSIC_GENERATOR_GLOBAL
13290b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
13300b57cec5SDimitry Andric #undef GET_INTRINSIC_GENERATOR_GLOBAL
13310b57cec5SDimitry Andric 
13320b57cec5SDimitry Andric void Intrinsic::getIntrinsicInfoTableEntries(ID id,
13330b57cec5SDimitry Andric                                              SmallVectorImpl<IITDescriptor> &T){
13340b57cec5SDimitry Andric   // Check to see if the intrinsic's type was expressible by the table.
13350b57cec5SDimitry Andric   unsigned TableVal = IIT_Table[id-1];
13360b57cec5SDimitry Andric 
13370b57cec5SDimitry Andric   // Decode the TableVal into an array of IITValues.
13380b57cec5SDimitry Andric   SmallVector<unsigned char, 8> IITValues;
13390b57cec5SDimitry Andric   ArrayRef<unsigned char> IITEntries;
13400b57cec5SDimitry Andric   unsigned NextElt = 0;
13410b57cec5SDimitry Andric   if ((TableVal >> 31) != 0) {
13420b57cec5SDimitry Andric     // This is an offset into the IIT_LongEncodingTable.
13430b57cec5SDimitry Andric     IITEntries = IIT_LongEncodingTable;
13440b57cec5SDimitry Andric 
13450b57cec5SDimitry Andric     // Strip sentinel bit.
13460b57cec5SDimitry Andric     NextElt = (TableVal << 1) >> 1;
13470b57cec5SDimitry Andric   } else {
13480b57cec5SDimitry Andric     // Decode the TableVal into an array of IITValues.  If the entry was encoded
13490b57cec5SDimitry Andric     // into a single word in the table itself, decode it now.
13500b57cec5SDimitry Andric     do {
13510b57cec5SDimitry Andric       IITValues.push_back(TableVal & 0xF);
13520b57cec5SDimitry Andric       TableVal >>= 4;
13530b57cec5SDimitry Andric     } while (TableVal);
13540b57cec5SDimitry Andric 
13550b57cec5SDimitry Andric     IITEntries = IITValues;
13560b57cec5SDimitry Andric     NextElt = 0;
13570b57cec5SDimitry Andric   }
13580b57cec5SDimitry Andric 
13590b57cec5SDimitry Andric   // Okay, decode the table into the output vector of IITDescriptors.
13605ffd83dbSDimitry Andric   DecodeIITType(NextElt, IITEntries, IIT_Done, T);
13610b57cec5SDimitry Andric   while (NextElt != IITEntries.size() && IITEntries[NextElt] != 0)
13625ffd83dbSDimitry Andric     DecodeIITType(NextElt, IITEntries, IIT_Done, T);
13630b57cec5SDimitry Andric }
13640b57cec5SDimitry Andric 
13650b57cec5SDimitry Andric static Type *DecodeFixedType(ArrayRef<Intrinsic::IITDescriptor> &Infos,
13660b57cec5SDimitry Andric                              ArrayRef<Type*> Tys, LLVMContext &Context) {
13670b57cec5SDimitry Andric   using namespace Intrinsic;
13680b57cec5SDimitry Andric 
13690b57cec5SDimitry Andric   IITDescriptor D = Infos.front();
13700b57cec5SDimitry Andric   Infos = Infos.slice(1);
13710b57cec5SDimitry Andric 
13720b57cec5SDimitry Andric   switch (D.Kind) {
13730b57cec5SDimitry Andric   case IITDescriptor::Void: return Type::getVoidTy(Context);
13740b57cec5SDimitry Andric   case IITDescriptor::VarArg: return Type::getVoidTy(Context);
13750b57cec5SDimitry Andric   case IITDescriptor::MMX: return Type::getX86_MMXTy(Context);
1376e8d8bef9SDimitry Andric   case IITDescriptor::AMX: return Type::getX86_AMXTy(Context);
13770b57cec5SDimitry Andric   case IITDescriptor::Token: return Type::getTokenTy(Context);
13780b57cec5SDimitry Andric   case IITDescriptor::Metadata: return Type::getMetadataTy(Context);
13790b57cec5SDimitry Andric   case IITDescriptor::Half: return Type::getHalfTy(Context);
13805ffd83dbSDimitry Andric   case IITDescriptor::BFloat: return Type::getBFloatTy(Context);
13810b57cec5SDimitry Andric   case IITDescriptor::Float: return Type::getFloatTy(Context);
13820b57cec5SDimitry Andric   case IITDescriptor::Double: return Type::getDoubleTy(Context);
13830b57cec5SDimitry Andric   case IITDescriptor::Quad: return Type::getFP128Ty(Context);
1384349cc55cSDimitry Andric   case IITDescriptor::PPCQuad: return Type::getPPC_FP128Ty(Context);
13850b57cec5SDimitry Andric 
13860b57cec5SDimitry Andric   case IITDescriptor::Integer:
13870b57cec5SDimitry Andric     return IntegerType::get(Context, D.Integer_Width);
13880b57cec5SDimitry Andric   case IITDescriptor::Vector:
13895ffd83dbSDimitry Andric     return VectorType::get(DecodeFixedType(Infos, Tys, Context),
13905ffd83dbSDimitry Andric                            D.Vector_Width);
13910b57cec5SDimitry Andric   case IITDescriptor::Pointer:
13920b57cec5SDimitry Andric     return PointerType::get(DecodeFixedType(Infos, Tys, Context),
13930b57cec5SDimitry Andric                             D.Pointer_AddressSpace);
13940b57cec5SDimitry Andric   case IITDescriptor::Struct: {
13950b57cec5SDimitry Andric     SmallVector<Type *, 8> Elts;
13960b57cec5SDimitry Andric     for (unsigned i = 0, e = D.Struct_NumElements; i != e; ++i)
13970b57cec5SDimitry Andric       Elts.push_back(DecodeFixedType(Infos, Tys, Context));
13980b57cec5SDimitry Andric     return StructType::get(Context, Elts);
13990b57cec5SDimitry Andric   }
14000b57cec5SDimitry Andric   case IITDescriptor::Argument:
14010b57cec5SDimitry Andric     return Tys[D.getArgumentNumber()];
14020b57cec5SDimitry Andric   case IITDescriptor::ExtendArgument: {
14030b57cec5SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14040b57cec5SDimitry Andric     if (VectorType *VTy = dyn_cast<VectorType>(Ty))
14050b57cec5SDimitry Andric       return VectorType::getExtendedElementVectorType(VTy);
14060b57cec5SDimitry Andric 
14070b57cec5SDimitry Andric     return IntegerType::get(Context, 2 * cast<IntegerType>(Ty)->getBitWidth());
14080b57cec5SDimitry Andric   }
14090b57cec5SDimitry Andric   case IITDescriptor::TruncArgument: {
14100b57cec5SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14110b57cec5SDimitry Andric     if (VectorType *VTy = dyn_cast<VectorType>(Ty))
14120b57cec5SDimitry Andric       return VectorType::getTruncatedElementVectorType(VTy);
14130b57cec5SDimitry Andric 
14140b57cec5SDimitry Andric     IntegerType *ITy = cast<IntegerType>(Ty);
14150b57cec5SDimitry Andric     assert(ITy->getBitWidth() % 2 == 0);
14160b57cec5SDimitry Andric     return IntegerType::get(Context, ITy->getBitWidth() / 2);
14170b57cec5SDimitry Andric   }
14188bcb0991SDimitry Andric   case IITDescriptor::Subdivide2Argument:
14198bcb0991SDimitry Andric   case IITDescriptor::Subdivide4Argument: {
14208bcb0991SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14218bcb0991SDimitry Andric     VectorType *VTy = dyn_cast<VectorType>(Ty);
14228bcb0991SDimitry Andric     assert(VTy && "Expected an argument of Vector Type");
14238bcb0991SDimitry Andric     int SubDivs = D.Kind == IITDescriptor::Subdivide2Argument ? 1 : 2;
14248bcb0991SDimitry Andric     return VectorType::getSubdividedVectorType(VTy, SubDivs);
14258bcb0991SDimitry Andric   }
14260b57cec5SDimitry Andric   case IITDescriptor::HalfVecArgument:
14270b57cec5SDimitry Andric     return VectorType::getHalfElementsVectorType(cast<VectorType>(
14280b57cec5SDimitry Andric                                                   Tys[D.getArgumentNumber()]));
14290b57cec5SDimitry Andric   case IITDescriptor::SameVecWidthArgument: {
14300b57cec5SDimitry Andric     Type *EltTy = DecodeFixedType(Infos, Tys, Context);
14310b57cec5SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14320b57cec5SDimitry Andric     if (auto *VTy = dyn_cast<VectorType>(Ty))
14338bcb0991SDimitry Andric       return VectorType::get(EltTy, VTy->getElementCount());
14340b57cec5SDimitry Andric     return EltTy;
14350b57cec5SDimitry Andric   }
14360b57cec5SDimitry Andric   case IITDescriptor::PtrToArgument: {
14370b57cec5SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14380b57cec5SDimitry Andric     return PointerType::getUnqual(Ty);
14390b57cec5SDimitry Andric   }
14400b57cec5SDimitry Andric   case IITDescriptor::PtrToElt: {
14410b57cec5SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14420b57cec5SDimitry Andric     VectorType *VTy = dyn_cast<VectorType>(Ty);
14430b57cec5SDimitry Andric     if (!VTy)
14440b57cec5SDimitry Andric       llvm_unreachable("Expected an argument of Vector Type");
14455ffd83dbSDimitry Andric     Type *EltTy = VTy->getElementType();
14460b57cec5SDimitry Andric     return PointerType::getUnqual(EltTy);
14470b57cec5SDimitry Andric   }
14480b57cec5SDimitry Andric   case IITDescriptor::VecElementArgument: {
14490b57cec5SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14500b57cec5SDimitry Andric     if (VectorType *VTy = dyn_cast<VectorType>(Ty))
14510b57cec5SDimitry Andric       return VTy->getElementType();
14520b57cec5SDimitry Andric     llvm_unreachable("Expected an argument of Vector Type");
14530b57cec5SDimitry Andric   }
14548bcb0991SDimitry Andric   case IITDescriptor::VecOfBitcastsToInt: {
14558bcb0991SDimitry Andric     Type *Ty = Tys[D.getArgumentNumber()];
14568bcb0991SDimitry Andric     VectorType *VTy = dyn_cast<VectorType>(Ty);
14578bcb0991SDimitry Andric     assert(VTy && "Expected an argument of Vector Type");
14588bcb0991SDimitry Andric     return VectorType::getInteger(VTy);
14598bcb0991SDimitry Andric   }
14600b57cec5SDimitry Andric   case IITDescriptor::VecOfAnyPtrsToElt:
14610b57cec5SDimitry Andric     // Return the overloaded type (which determines the pointers address space)
14620b57cec5SDimitry Andric     return Tys[D.getOverloadArgNumber()];
146381ad6265SDimitry Andric   case IITDescriptor::AnyPtrToElt:
146481ad6265SDimitry Andric     // Return the overloaded type (which determines the pointers address space)
146581ad6265SDimitry Andric     return Tys[D.getOverloadArgNumber()];
14660b57cec5SDimitry Andric   }
14670b57cec5SDimitry Andric   llvm_unreachable("unhandled");
14680b57cec5SDimitry Andric }
14690b57cec5SDimitry Andric 
14700b57cec5SDimitry Andric FunctionType *Intrinsic::getType(LLVMContext &Context,
14710b57cec5SDimitry Andric                                  ID id, ArrayRef<Type*> Tys) {
14720b57cec5SDimitry Andric   SmallVector<IITDescriptor, 8> Table;
14730b57cec5SDimitry Andric   getIntrinsicInfoTableEntries(id, Table);
14740b57cec5SDimitry Andric 
14750b57cec5SDimitry Andric   ArrayRef<IITDescriptor> TableRef = Table;
14760b57cec5SDimitry Andric   Type *ResultTy = DecodeFixedType(TableRef, Tys, Context);
14770b57cec5SDimitry Andric 
14780b57cec5SDimitry Andric   SmallVector<Type*, 8> ArgTys;
14790b57cec5SDimitry Andric   while (!TableRef.empty())
14800b57cec5SDimitry Andric     ArgTys.push_back(DecodeFixedType(TableRef, Tys, Context));
14810b57cec5SDimitry Andric 
14820b57cec5SDimitry Andric   // DecodeFixedType returns Void for IITDescriptor::Void and IITDescriptor::VarArg
14830b57cec5SDimitry Andric   // If we see void type as the type of the last argument, it is vararg intrinsic
14840b57cec5SDimitry Andric   if (!ArgTys.empty() && ArgTys.back()->isVoidTy()) {
14850b57cec5SDimitry Andric     ArgTys.pop_back();
14860b57cec5SDimitry Andric     return FunctionType::get(ResultTy, ArgTys, true);
14870b57cec5SDimitry Andric   }
14880b57cec5SDimitry Andric   return FunctionType::get(ResultTy, ArgTys, false);
14890b57cec5SDimitry Andric }
14900b57cec5SDimitry Andric 
14910b57cec5SDimitry Andric bool Intrinsic::isOverloaded(ID id) {
14920b57cec5SDimitry Andric #define GET_INTRINSIC_OVERLOAD_TABLE
14930b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
14940b57cec5SDimitry Andric #undef GET_INTRINSIC_OVERLOAD_TABLE
14950b57cec5SDimitry Andric }
14960b57cec5SDimitry Andric 
14970b57cec5SDimitry Andric /// This defines the "Intrinsic::getAttributes(ID id)" method.
14980b57cec5SDimitry Andric #define GET_INTRINSIC_ATTRIBUTES
14990b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
15000b57cec5SDimitry Andric #undef GET_INTRINSIC_ATTRIBUTES
15010b57cec5SDimitry Andric 
15020b57cec5SDimitry Andric Function *Intrinsic::getDeclaration(Module *M, ID id, ArrayRef<Type*> Tys) {
15030b57cec5SDimitry Andric   // There can never be multiple globals with the same name of different types,
15040b57cec5SDimitry Andric   // because intrinsics must be a specific type.
1505fe6060f1SDimitry Andric   auto *FT = getType(M->getContext(), id, Tys);
15060b57cec5SDimitry Andric   return cast<Function>(
1507*bdd1243dSDimitry Andric       M->getOrInsertFunction(
1508*bdd1243dSDimitry Andric            Tys.empty() ? getName(id) : getName(id, Tys, M, FT), FT)
15090b57cec5SDimitry Andric           .getCallee());
15100b57cec5SDimitry Andric }
15110b57cec5SDimitry Andric 
151281ad6265SDimitry Andric // This defines the "Intrinsic::getIntrinsicForClangBuiltin()" method.
151381ad6265SDimitry Andric #define GET_LLVM_INTRINSIC_FOR_CLANG_BUILTIN
15140b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
151581ad6265SDimitry Andric #undef GET_LLVM_INTRINSIC_FOR_CLANG_BUILTIN
15160b57cec5SDimitry Andric 
15170b57cec5SDimitry Andric // This defines the "Intrinsic::getIntrinsicForMSBuiltin()" method.
15180b57cec5SDimitry Andric #define GET_LLVM_INTRINSIC_FOR_MS_BUILTIN
15190b57cec5SDimitry Andric #include "llvm/IR/IntrinsicImpl.inc"
15200b57cec5SDimitry Andric #undef GET_LLVM_INTRINSIC_FOR_MS_BUILTIN
15210b57cec5SDimitry Andric 
15220b57cec5SDimitry Andric using DeferredIntrinsicMatchPair =
15230b57cec5SDimitry Andric     std::pair<Type *, ArrayRef<Intrinsic::IITDescriptor>>;
15240b57cec5SDimitry Andric 
15250b57cec5SDimitry Andric static bool matchIntrinsicType(
15260b57cec5SDimitry Andric     Type *Ty, ArrayRef<Intrinsic::IITDescriptor> &Infos,
15270b57cec5SDimitry Andric     SmallVectorImpl<Type *> &ArgTys,
15280b57cec5SDimitry Andric     SmallVectorImpl<DeferredIntrinsicMatchPair> &DeferredChecks,
15290b57cec5SDimitry Andric     bool IsDeferredCheck) {
15300b57cec5SDimitry Andric   using namespace Intrinsic;
15310b57cec5SDimitry Andric 
15320b57cec5SDimitry Andric   // If we ran out of descriptors, there are too many arguments.
15330b57cec5SDimitry Andric   if (Infos.empty()) return true;
15340b57cec5SDimitry Andric 
15350b57cec5SDimitry Andric   // Do this before slicing off the 'front' part
15360b57cec5SDimitry Andric   auto InfosRef = Infos;
15370b57cec5SDimitry Andric   auto DeferCheck = [&DeferredChecks, &InfosRef](Type *T) {
15380b57cec5SDimitry Andric     DeferredChecks.emplace_back(T, InfosRef);
15390b57cec5SDimitry Andric     return false;
15400b57cec5SDimitry Andric   };
15410b57cec5SDimitry Andric 
15420b57cec5SDimitry Andric   IITDescriptor D = Infos.front();
15430b57cec5SDimitry Andric   Infos = Infos.slice(1);
15440b57cec5SDimitry Andric 
15450b57cec5SDimitry Andric   switch (D.Kind) {
15460b57cec5SDimitry Andric     case IITDescriptor::Void: return !Ty->isVoidTy();
15470b57cec5SDimitry Andric     case IITDescriptor::VarArg: return true;
15480b57cec5SDimitry Andric     case IITDescriptor::MMX:  return !Ty->isX86_MMXTy();
1549e8d8bef9SDimitry Andric     case IITDescriptor::AMX:  return !Ty->isX86_AMXTy();
15500b57cec5SDimitry Andric     case IITDescriptor::Token: return !Ty->isTokenTy();
15510b57cec5SDimitry Andric     case IITDescriptor::Metadata: return !Ty->isMetadataTy();
15520b57cec5SDimitry Andric     case IITDescriptor::Half: return !Ty->isHalfTy();
15535ffd83dbSDimitry Andric     case IITDescriptor::BFloat: return !Ty->isBFloatTy();
15540b57cec5SDimitry Andric     case IITDescriptor::Float: return !Ty->isFloatTy();
15550b57cec5SDimitry Andric     case IITDescriptor::Double: return !Ty->isDoubleTy();
15560b57cec5SDimitry Andric     case IITDescriptor::Quad: return !Ty->isFP128Ty();
1557349cc55cSDimitry Andric     case IITDescriptor::PPCQuad: return !Ty->isPPC_FP128Ty();
15580b57cec5SDimitry Andric     case IITDescriptor::Integer: return !Ty->isIntegerTy(D.Integer_Width);
15590b57cec5SDimitry Andric     case IITDescriptor::Vector: {
15600b57cec5SDimitry Andric       VectorType *VT = dyn_cast<VectorType>(Ty);
15615ffd83dbSDimitry Andric       return !VT || VT->getElementCount() != D.Vector_Width ||
15620b57cec5SDimitry Andric              matchIntrinsicType(VT->getElementType(), Infos, ArgTys,
15630b57cec5SDimitry Andric                                 DeferredChecks, IsDeferredCheck);
15640b57cec5SDimitry Andric     }
15650b57cec5SDimitry Andric     case IITDescriptor::Pointer: {
15660b57cec5SDimitry Andric       PointerType *PT = dyn_cast<PointerType>(Ty);
1567fe6060f1SDimitry Andric       if (!PT || PT->getAddressSpace() != D.Pointer_AddressSpace)
1568fe6060f1SDimitry Andric         return true;
156981ad6265SDimitry Andric       if (!PT->isOpaque()) {
157081ad6265SDimitry Andric         /* Manually consume a pointer to empty struct descriptor, which is
157181ad6265SDimitry Andric          * used for externref. We don't want to enforce that the struct is
157281ad6265SDimitry Andric          * anonymous in this case. (This renders externref intrinsics
157381ad6265SDimitry Andric          * non-unique, but this will go away with opaque pointers anyway.) */
157481ad6265SDimitry Andric         if (Infos.front().Kind == IITDescriptor::Struct &&
157581ad6265SDimitry Andric             Infos.front().Struct_NumElements == 0) {
157681ad6265SDimitry Andric           Infos = Infos.slice(1);
157781ad6265SDimitry Andric           return false;
157881ad6265SDimitry Andric         }
157904eeddc0SDimitry Andric         return matchIntrinsicType(PT->getNonOpaquePointerElementType(), Infos,
158004eeddc0SDimitry Andric                                   ArgTys, DeferredChecks, IsDeferredCheck);
158181ad6265SDimitry Andric       }
1582fe6060f1SDimitry Andric       // Consume IIT descriptors relating to the pointer element type.
158381ad6265SDimitry Andric       // FIXME: Intrinsic type matching of nested single value types or even
158481ad6265SDimitry Andric       // aggregates doesn't work properly with opaque pointers but hopefully
158581ad6265SDimitry Andric       // doesn't happen in practice.
158681ad6265SDimitry Andric       while (Infos.front().Kind == IITDescriptor::Pointer ||
158781ad6265SDimitry Andric              Infos.front().Kind == IITDescriptor::Vector)
1588fe6060f1SDimitry Andric         Infos = Infos.slice(1);
158981ad6265SDimitry Andric       assert((Infos.front().Kind != IITDescriptor::Argument ||
159081ad6265SDimitry Andric               Infos.front().getArgumentKind() == IITDescriptor::AK_MatchType) &&
159181ad6265SDimitry Andric              "Unsupported polymorphic pointer type with opaque pointer");
1592fe6060f1SDimitry Andric       Infos = Infos.slice(1);
1593fe6060f1SDimitry Andric       return false;
15940b57cec5SDimitry Andric     }
15950b57cec5SDimitry Andric 
15960b57cec5SDimitry Andric     case IITDescriptor::Struct: {
15970b57cec5SDimitry Andric       StructType *ST = dyn_cast<StructType>(Ty);
159881ad6265SDimitry Andric       if (!ST || !ST->isLiteral() || ST->isPacked() ||
159981ad6265SDimitry Andric           ST->getNumElements() != D.Struct_NumElements)
16000b57cec5SDimitry Andric         return true;
16010b57cec5SDimitry Andric 
16020b57cec5SDimitry Andric       for (unsigned i = 0, e = D.Struct_NumElements; i != e; ++i)
16030b57cec5SDimitry Andric         if (matchIntrinsicType(ST->getElementType(i), Infos, ArgTys,
16040b57cec5SDimitry Andric                                DeferredChecks, IsDeferredCheck))
16050b57cec5SDimitry Andric           return true;
16060b57cec5SDimitry Andric       return false;
16070b57cec5SDimitry Andric     }
16080b57cec5SDimitry Andric 
16090b57cec5SDimitry Andric     case IITDescriptor::Argument:
16100b57cec5SDimitry Andric       // If this is the second occurrence of an argument,
16110b57cec5SDimitry Andric       // verify that the later instance matches the previous instance.
16120b57cec5SDimitry Andric       if (D.getArgumentNumber() < ArgTys.size())
16130b57cec5SDimitry Andric         return Ty != ArgTys[D.getArgumentNumber()];
16140b57cec5SDimitry Andric 
16150b57cec5SDimitry Andric       if (D.getArgumentNumber() > ArgTys.size() ||
16160b57cec5SDimitry Andric           D.getArgumentKind() == IITDescriptor::AK_MatchType)
16170b57cec5SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
16180b57cec5SDimitry Andric 
16190b57cec5SDimitry Andric       assert(D.getArgumentNumber() == ArgTys.size() && !IsDeferredCheck &&
16200b57cec5SDimitry Andric              "Table consistency error");
16210b57cec5SDimitry Andric       ArgTys.push_back(Ty);
16220b57cec5SDimitry Andric 
16230b57cec5SDimitry Andric       switch (D.getArgumentKind()) {
16240b57cec5SDimitry Andric         case IITDescriptor::AK_Any:        return false; // Success
16250b57cec5SDimitry Andric         case IITDescriptor::AK_AnyInteger: return !Ty->isIntOrIntVectorTy();
16260b57cec5SDimitry Andric         case IITDescriptor::AK_AnyFloat:   return !Ty->isFPOrFPVectorTy();
16270b57cec5SDimitry Andric         case IITDescriptor::AK_AnyVector:  return !isa<VectorType>(Ty);
16280b57cec5SDimitry Andric         case IITDescriptor::AK_AnyPointer: return !isa<PointerType>(Ty);
16290b57cec5SDimitry Andric         default:                           break;
16300b57cec5SDimitry Andric       }
16310b57cec5SDimitry Andric       llvm_unreachable("all argument kinds not covered");
16320b57cec5SDimitry Andric 
16330b57cec5SDimitry Andric     case IITDescriptor::ExtendArgument: {
16340b57cec5SDimitry Andric       // If this is a forward reference, defer the check for later.
16350b57cec5SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
16360b57cec5SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
16370b57cec5SDimitry Andric 
16380b57cec5SDimitry Andric       Type *NewTy = ArgTys[D.getArgumentNumber()];
16390b57cec5SDimitry Andric       if (VectorType *VTy = dyn_cast<VectorType>(NewTy))
16400b57cec5SDimitry Andric         NewTy = VectorType::getExtendedElementVectorType(VTy);
16410b57cec5SDimitry Andric       else if (IntegerType *ITy = dyn_cast<IntegerType>(NewTy))
16420b57cec5SDimitry Andric         NewTy = IntegerType::get(ITy->getContext(), 2 * ITy->getBitWidth());
16430b57cec5SDimitry Andric       else
16440b57cec5SDimitry Andric         return true;
16450b57cec5SDimitry Andric 
16460b57cec5SDimitry Andric       return Ty != NewTy;
16470b57cec5SDimitry Andric     }
16480b57cec5SDimitry Andric     case IITDescriptor::TruncArgument: {
16490b57cec5SDimitry Andric       // If this is a forward reference, defer the check for later.
16500b57cec5SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
16510b57cec5SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
16520b57cec5SDimitry Andric 
16530b57cec5SDimitry Andric       Type *NewTy = ArgTys[D.getArgumentNumber()];
16540b57cec5SDimitry Andric       if (VectorType *VTy = dyn_cast<VectorType>(NewTy))
16550b57cec5SDimitry Andric         NewTy = VectorType::getTruncatedElementVectorType(VTy);
16560b57cec5SDimitry Andric       else if (IntegerType *ITy = dyn_cast<IntegerType>(NewTy))
16570b57cec5SDimitry Andric         NewTy = IntegerType::get(ITy->getContext(), ITy->getBitWidth() / 2);
16580b57cec5SDimitry Andric       else
16590b57cec5SDimitry Andric         return true;
16600b57cec5SDimitry Andric 
16610b57cec5SDimitry Andric       return Ty != NewTy;
16620b57cec5SDimitry Andric     }
16630b57cec5SDimitry Andric     case IITDescriptor::HalfVecArgument:
16640b57cec5SDimitry Andric       // If this is a forward reference, defer the check for later.
16658bcb0991SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
16668bcb0991SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
16678bcb0991SDimitry Andric       return !isa<VectorType>(ArgTys[D.getArgumentNumber()]) ||
16680b57cec5SDimitry Andric              VectorType::getHalfElementsVectorType(
16690b57cec5SDimitry Andric                      cast<VectorType>(ArgTys[D.getArgumentNumber()])) != Ty;
16700b57cec5SDimitry Andric     case IITDescriptor::SameVecWidthArgument: {
16710b57cec5SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size()) {
16720b57cec5SDimitry Andric         // Defer check and subsequent check for the vector element type.
16730b57cec5SDimitry Andric         Infos = Infos.slice(1);
16740b57cec5SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
16750b57cec5SDimitry Andric       }
16760b57cec5SDimitry Andric       auto *ReferenceType = dyn_cast<VectorType>(ArgTys[D.getArgumentNumber()]);
16770b57cec5SDimitry Andric       auto *ThisArgType = dyn_cast<VectorType>(Ty);
16780b57cec5SDimitry Andric       // Both must be vectors of the same number of elements or neither.
16790b57cec5SDimitry Andric       if ((ReferenceType != nullptr) != (ThisArgType != nullptr))
16800b57cec5SDimitry Andric         return true;
16810b57cec5SDimitry Andric       Type *EltTy = Ty;
16820b57cec5SDimitry Andric       if (ThisArgType) {
16838bcb0991SDimitry Andric         if (ReferenceType->getElementCount() !=
16848bcb0991SDimitry Andric             ThisArgType->getElementCount())
16850b57cec5SDimitry Andric           return true;
16865ffd83dbSDimitry Andric         EltTy = ThisArgType->getElementType();
16870b57cec5SDimitry Andric       }
16880b57cec5SDimitry Andric       return matchIntrinsicType(EltTy, Infos, ArgTys, DeferredChecks,
16890b57cec5SDimitry Andric                                 IsDeferredCheck);
16900b57cec5SDimitry Andric     }
16910b57cec5SDimitry Andric     case IITDescriptor::PtrToArgument: {
16920b57cec5SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
16930b57cec5SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
16940b57cec5SDimitry Andric       Type * ReferenceType = ArgTys[D.getArgumentNumber()];
16950b57cec5SDimitry Andric       PointerType *ThisArgType = dyn_cast<PointerType>(Ty);
169604eeddc0SDimitry Andric       return (!ThisArgType ||
169704eeddc0SDimitry Andric               !ThisArgType->isOpaqueOrPointeeTypeMatches(ReferenceType));
16980b57cec5SDimitry Andric     }
16990b57cec5SDimitry Andric     case IITDescriptor::PtrToElt: {
17000b57cec5SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
17010b57cec5SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
17020b57cec5SDimitry Andric       VectorType * ReferenceType =
17030b57cec5SDimitry Andric         dyn_cast<VectorType> (ArgTys[D.getArgumentNumber()]);
17040b57cec5SDimitry Andric       PointerType *ThisArgType = dyn_cast<PointerType>(Ty);
17050b57cec5SDimitry Andric 
1706fe6060f1SDimitry Andric       if (!ThisArgType || !ReferenceType)
1707fe6060f1SDimitry Andric         return true;
1708349cc55cSDimitry Andric       return !ThisArgType->isOpaqueOrPointeeTypeMatches(
1709349cc55cSDimitry Andric           ReferenceType->getElementType());
17100b57cec5SDimitry Andric     }
171181ad6265SDimitry Andric     case IITDescriptor::AnyPtrToElt: {
171281ad6265SDimitry Andric       unsigned RefArgNumber = D.getRefArgNumber();
171381ad6265SDimitry Andric       if (RefArgNumber >= ArgTys.size()) {
171481ad6265SDimitry Andric         if (IsDeferredCheck)
171581ad6265SDimitry Andric           return true;
171681ad6265SDimitry Andric         // If forward referencing, already add the pointer type and
171781ad6265SDimitry Andric         // defer the checks for later.
171881ad6265SDimitry Andric         ArgTys.push_back(Ty);
171981ad6265SDimitry Andric         return DeferCheck(Ty);
172081ad6265SDimitry Andric       }
172181ad6265SDimitry Andric 
172281ad6265SDimitry Andric       if (!IsDeferredCheck) {
172381ad6265SDimitry Andric         assert(D.getOverloadArgNumber() == ArgTys.size() &&
172481ad6265SDimitry Andric                "Table consistency error");
172581ad6265SDimitry Andric         ArgTys.push_back(Ty);
172681ad6265SDimitry Andric       }
172781ad6265SDimitry Andric 
172881ad6265SDimitry Andric       auto *ReferenceType = dyn_cast<VectorType>(ArgTys[RefArgNumber]);
172981ad6265SDimitry Andric       auto *ThisArgType = dyn_cast<PointerType>(Ty);
173081ad6265SDimitry Andric       if (!ThisArgType || !ReferenceType)
173181ad6265SDimitry Andric         return true;
173281ad6265SDimitry Andric       return !ThisArgType->isOpaqueOrPointeeTypeMatches(
173381ad6265SDimitry Andric           ReferenceType->getElementType());
173481ad6265SDimitry Andric     }
17350b57cec5SDimitry Andric     case IITDescriptor::VecOfAnyPtrsToElt: {
17360b57cec5SDimitry Andric       unsigned RefArgNumber = D.getRefArgNumber();
17370b57cec5SDimitry Andric       if (RefArgNumber >= ArgTys.size()) {
17380b57cec5SDimitry Andric         if (IsDeferredCheck)
17390b57cec5SDimitry Andric           return true;
17400b57cec5SDimitry Andric         // If forward referencing, already add the pointer-vector type and
17410b57cec5SDimitry Andric         // defer the checks for later.
17420b57cec5SDimitry Andric         ArgTys.push_back(Ty);
17430b57cec5SDimitry Andric         return DeferCheck(Ty);
17440b57cec5SDimitry Andric       }
17450b57cec5SDimitry Andric 
17460b57cec5SDimitry Andric       if (!IsDeferredCheck){
17470b57cec5SDimitry Andric         assert(D.getOverloadArgNumber() == ArgTys.size() &&
17480b57cec5SDimitry Andric                "Table consistency error");
17490b57cec5SDimitry Andric         ArgTys.push_back(Ty);
17500b57cec5SDimitry Andric       }
17510b57cec5SDimitry Andric 
17520b57cec5SDimitry Andric       // Verify the overloaded type "matches" the Ref type.
17530b57cec5SDimitry Andric       // i.e. Ty is a vector with the same width as Ref.
17540b57cec5SDimitry Andric       // Composed of pointers to the same element type as Ref.
1755e8d8bef9SDimitry Andric       auto *ReferenceType = dyn_cast<VectorType>(ArgTys[RefArgNumber]);
1756e8d8bef9SDimitry Andric       auto *ThisArgVecTy = dyn_cast<VectorType>(Ty);
17570b57cec5SDimitry Andric       if (!ThisArgVecTy || !ReferenceType ||
1758e8d8bef9SDimitry Andric           (ReferenceType->getElementCount() != ThisArgVecTy->getElementCount()))
17590b57cec5SDimitry Andric         return true;
17600b57cec5SDimitry Andric       PointerType *ThisArgEltTy =
17615ffd83dbSDimitry Andric           dyn_cast<PointerType>(ThisArgVecTy->getElementType());
17620b57cec5SDimitry Andric       if (!ThisArgEltTy)
17630b57cec5SDimitry Andric         return true;
1764fe6060f1SDimitry Andric       return !ThisArgEltTy->isOpaqueOrPointeeTypeMatches(
1765fe6060f1SDimitry Andric           ReferenceType->getElementType());
17660b57cec5SDimitry Andric     }
17670b57cec5SDimitry Andric     case IITDescriptor::VecElementArgument: {
17680b57cec5SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
17690b57cec5SDimitry Andric         return IsDeferredCheck ? true : DeferCheck(Ty);
17700b57cec5SDimitry Andric       auto *ReferenceType = dyn_cast<VectorType>(ArgTys[D.getArgumentNumber()]);
17710b57cec5SDimitry Andric       return !ReferenceType || Ty != ReferenceType->getElementType();
17720b57cec5SDimitry Andric     }
17738bcb0991SDimitry Andric     case IITDescriptor::Subdivide2Argument:
17748bcb0991SDimitry Andric     case IITDescriptor::Subdivide4Argument: {
17758bcb0991SDimitry Andric       // If this is a forward reference, defer the check for later.
17768bcb0991SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
17778bcb0991SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
17788bcb0991SDimitry Andric 
17798bcb0991SDimitry Andric       Type *NewTy = ArgTys[D.getArgumentNumber()];
17808bcb0991SDimitry Andric       if (auto *VTy = dyn_cast<VectorType>(NewTy)) {
17818bcb0991SDimitry Andric         int SubDivs = D.Kind == IITDescriptor::Subdivide2Argument ? 1 : 2;
17828bcb0991SDimitry Andric         NewTy = VectorType::getSubdividedVectorType(VTy, SubDivs);
17838bcb0991SDimitry Andric         return Ty != NewTy;
17848bcb0991SDimitry Andric       }
17858bcb0991SDimitry Andric       return true;
17868bcb0991SDimitry Andric     }
17878bcb0991SDimitry Andric     case IITDescriptor::VecOfBitcastsToInt: {
17888bcb0991SDimitry Andric       if (D.getArgumentNumber() >= ArgTys.size())
17898bcb0991SDimitry Andric         return IsDeferredCheck || DeferCheck(Ty);
17908bcb0991SDimitry Andric       auto *ReferenceType = dyn_cast<VectorType>(ArgTys[D.getArgumentNumber()]);
17918bcb0991SDimitry Andric       auto *ThisArgVecTy = dyn_cast<VectorType>(Ty);
17928bcb0991SDimitry Andric       if (!ThisArgVecTy || !ReferenceType)
17938bcb0991SDimitry Andric         return true;
17948bcb0991SDimitry Andric       return ThisArgVecTy != VectorType::getInteger(ReferenceType);
17958bcb0991SDimitry Andric     }
17960b57cec5SDimitry Andric   }
17970b57cec5SDimitry Andric   llvm_unreachable("unhandled");
17980b57cec5SDimitry Andric }
17990b57cec5SDimitry Andric 
18000b57cec5SDimitry Andric Intrinsic::MatchIntrinsicTypesResult
18010b57cec5SDimitry Andric Intrinsic::matchIntrinsicSignature(FunctionType *FTy,
18020b57cec5SDimitry Andric                                    ArrayRef<Intrinsic::IITDescriptor> &Infos,
18030b57cec5SDimitry Andric                                    SmallVectorImpl<Type *> &ArgTys) {
18040b57cec5SDimitry Andric   SmallVector<DeferredIntrinsicMatchPair, 2> DeferredChecks;
18050b57cec5SDimitry Andric   if (matchIntrinsicType(FTy->getReturnType(), Infos, ArgTys, DeferredChecks,
18060b57cec5SDimitry Andric                          false))
18070b57cec5SDimitry Andric     return MatchIntrinsicTypes_NoMatchRet;
18080b57cec5SDimitry Andric 
18090b57cec5SDimitry Andric   unsigned NumDeferredReturnChecks = DeferredChecks.size();
18100b57cec5SDimitry Andric 
1811*bdd1243dSDimitry Andric   for (auto *Ty : FTy->params())
18120b57cec5SDimitry Andric     if (matchIntrinsicType(Ty, Infos, ArgTys, DeferredChecks, false))
18130b57cec5SDimitry Andric       return MatchIntrinsicTypes_NoMatchArg;
18140b57cec5SDimitry Andric 
18150b57cec5SDimitry Andric   for (unsigned I = 0, E = DeferredChecks.size(); I != E; ++I) {
18160b57cec5SDimitry Andric     DeferredIntrinsicMatchPair &Check = DeferredChecks[I];
18170b57cec5SDimitry Andric     if (matchIntrinsicType(Check.first, Check.second, ArgTys, DeferredChecks,
18180b57cec5SDimitry Andric                            true))
18190b57cec5SDimitry Andric       return I < NumDeferredReturnChecks ? MatchIntrinsicTypes_NoMatchRet
18200b57cec5SDimitry Andric                                          : MatchIntrinsicTypes_NoMatchArg;
18210b57cec5SDimitry Andric   }
18220b57cec5SDimitry Andric 
18230b57cec5SDimitry Andric   return MatchIntrinsicTypes_Match;
18240b57cec5SDimitry Andric }
18250b57cec5SDimitry Andric 
18260b57cec5SDimitry Andric bool
18270b57cec5SDimitry Andric Intrinsic::matchIntrinsicVarArg(bool isVarArg,
18280b57cec5SDimitry Andric                                 ArrayRef<Intrinsic::IITDescriptor> &Infos) {
18290b57cec5SDimitry Andric   // If there are no descriptors left, then it can't be a vararg.
18300b57cec5SDimitry Andric   if (Infos.empty())
18310b57cec5SDimitry Andric     return isVarArg;
18320b57cec5SDimitry Andric 
18330b57cec5SDimitry Andric   // There should be only one descriptor remaining at this point.
18340b57cec5SDimitry Andric   if (Infos.size() != 1)
18350b57cec5SDimitry Andric     return true;
18360b57cec5SDimitry Andric 
18370b57cec5SDimitry Andric   // Check and verify the descriptor.
18380b57cec5SDimitry Andric   IITDescriptor D = Infos.front();
18390b57cec5SDimitry Andric   Infos = Infos.slice(1);
18400b57cec5SDimitry Andric   if (D.Kind == IITDescriptor::VarArg)
18410b57cec5SDimitry Andric     return !isVarArg;
18420b57cec5SDimitry Andric 
18430b57cec5SDimitry Andric   return true;
18440b57cec5SDimitry Andric }
18450b57cec5SDimitry Andric 
18465ffd83dbSDimitry Andric bool Intrinsic::getIntrinsicSignature(Function *F,
18475ffd83dbSDimitry Andric                                       SmallVectorImpl<Type *> &ArgTys) {
18480b57cec5SDimitry Andric   Intrinsic::ID ID = F->getIntrinsicID();
18490b57cec5SDimitry Andric   if (!ID)
18505ffd83dbSDimitry Andric     return false;
18510b57cec5SDimitry Andric 
18520b57cec5SDimitry Andric   SmallVector<Intrinsic::IITDescriptor, 8> Table;
18530b57cec5SDimitry Andric   getIntrinsicInfoTableEntries(ID, Table);
18540b57cec5SDimitry Andric   ArrayRef<Intrinsic::IITDescriptor> TableRef = Table;
18550b57cec5SDimitry Andric 
18565ffd83dbSDimitry Andric   if (Intrinsic::matchIntrinsicSignature(F->getFunctionType(), TableRef,
18575ffd83dbSDimitry Andric                                          ArgTys) !=
18585ffd83dbSDimitry Andric       Intrinsic::MatchIntrinsicTypesResult::MatchIntrinsicTypes_Match) {
18595ffd83dbSDimitry Andric     return false;
18605ffd83dbSDimitry Andric   }
18615ffd83dbSDimitry Andric   if (Intrinsic::matchIntrinsicVarArg(F->getFunctionType()->isVarArg(),
18625ffd83dbSDimitry Andric                                       TableRef))
18635ffd83dbSDimitry Andric     return false;
18645ffd83dbSDimitry Andric   return true;
18650b57cec5SDimitry Andric }
18660b57cec5SDimitry Andric 
1867*bdd1243dSDimitry Andric std::optional<Function *> Intrinsic::remangleIntrinsicFunction(Function *F) {
18685ffd83dbSDimitry Andric   SmallVector<Type *, 4> ArgTys;
18695ffd83dbSDimitry Andric   if (!getIntrinsicSignature(F, ArgTys))
1870*bdd1243dSDimitry Andric     return std::nullopt;
18715ffd83dbSDimitry Andric 
18725ffd83dbSDimitry Andric   Intrinsic::ID ID = F->getIntrinsicID();
18730b57cec5SDimitry Andric   StringRef Name = F->getName();
1874fe6060f1SDimitry Andric   std::string WantedName =
1875fe6060f1SDimitry Andric       Intrinsic::getName(ID, ArgTys, F->getParent(), F->getFunctionType());
1876fe6060f1SDimitry Andric   if (Name == WantedName)
1877*bdd1243dSDimitry Andric     return std::nullopt;
18780b57cec5SDimitry Andric 
1879fe6060f1SDimitry Andric   Function *NewDecl = [&] {
1880fe6060f1SDimitry Andric     if (auto *ExistingGV = F->getParent()->getNamedValue(WantedName)) {
1881fe6060f1SDimitry Andric       if (auto *ExistingF = dyn_cast<Function>(ExistingGV))
1882fe6060f1SDimitry Andric         if (ExistingF->getFunctionType() == F->getFunctionType())
1883fe6060f1SDimitry Andric           return ExistingF;
1884fe6060f1SDimitry Andric 
1885fe6060f1SDimitry Andric       // The name already exists, but is not a function or has the wrong
1886fe6060f1SDimitry Andric       // prototype. Make place for the new one by renaming the old version.
1887fe6060f1SDimitry Andric       // Either this old version will be removed later on or the module is
1888fe6060f1SDimitry Andric       // invalid and we'll get an error.
1889fe6060f1SDimitry Andric       ExistingGV->setName(WantedName + ".renamed");
1890fe6060f1SDimitry Andric     }
1891fe6060f1SDimitry Andric     return Intrinsic::getDeclaration(F->getParent(), ID, ArgTys);
1892fe6060f1SDimitry Andric   }();
1893fe6060f1SDimitry Andric 
18940b57cec5SDimitry Andric   NewDecl->setCallingConv(F->getCallingConv());
18955ffd83dbSDimitry Andric   assert(NewDecl->getFunctionType() == F->getFunctionType() &&
18965ffd83dbSDimitry Andric          "Shouldn't change the signature");
18970b57cec5SDimitry Andric   return NewDecl;
18980b57cec5SDimitry Andric }
18990b57cec5SDimitry Andric 
19000b57cec5SDimitry Andric /// hasAddressTaken - returns true if there are any uses of this function
19015ffd83dbSDimitry Andric /// other than direct calls or invokes to it. Optionally ignores callback
1902fe6060f1SDimitry Andric /// uses, assume like pointer annotation calls, and references in llvm.used
1903fe6060f1SDimitry Andric /// and llvm.compiler.used variables.
19045ffd83dbSDimitry Andric bool Function::hasAddressTaken(const User **PutOffender,
1905fe6060f1SDimitry Andric                                bool IgnoreCallbackUses,
1906349cc55cSDimitry Andric                                bool IgnoreAssumeLikeCalls, bool IgnoreLLVMUsed,
1907349cc55cSDimitry Andric                                bool IgnoreARCAttachedCall) const {
19080b57cec5SDimitry Andric   for (const Use &U : uses()) {
19090b57cec5SDimitry Andric     const User *FU = U.getUser();
19100b57cec5SDimitry Andric     if (isa<BlockAddress>(FU))
19110b57cec5SDimitry Andric       continue;
19125ffd83dbSDimitry Andric 
19135ffd83dbSDimitry Andric     if (IgnoreCallbackUses) {
19145ffd83dbSDimitry Andric       AbstractCallSite ACS(&U);
19155ffd83dbSDimitry Andric       if (ACS && ACS.isCallbackCall())
19165ffd83dbSDimitry Andric         continue;
19175ffd83dbSDimitry Andric     }
19185ffd83dbSDimitry Andric 
19190b57cec5SDimitry Andric     const auto *Call = dyn_cast<CallBase>(FU);
19200b57cec5SDimitry Andric     if (!Call) {
1921*bdd1243dSDimitry Andric       if (IgnoreAssumeLikeCalls &&
1922*bdd1243dSDimitry Andric           isa<BitCastOperator, AddrSpaceCastOperator>(FU) &&
1923*bdd1243dSDimitry Andric           all_of(FU->users(), [](const User *U) {
1924fe6060f1SDimitry Andric             if (const auto *I = dyn_cast<IntrinsicInst>(U))
1925fe6060f1SDimitry Andric               return I->isAssumeLikeIntrinsic();
1926fe6060f1SDimitry Andric             return false;
1927*bdd1243dSDimitry Andric           })) {
1928fe6060f1SDimitry Andric         continue;
1929fe6060f1SDimitry Andric       }
1930*bdd1243dSDimitry Andric 
1931fe6060f1SDimitry Andric       if (IgnoreLLVMUsed && !FU->user_empty()) {
1932fe6060f1SDimitry Andric         const User *FUU = FU;
1933*bdd1243dSDimitry Andric         if (isa<BitCastOperator, AddrSpaceCastOperator>(FU) &&
1934*bdd1243dSDimitry Andric             FU->hasOneUse() && !FU->user_begin()->user_empty())
1935fe6060f1SDimitry Andric           FUU = *FU->user_begin();
1936fe6060f1SDimitry Andric         if (llvm::all_of(FUU->users(), [](const User *U) {
1937fe6060f1SDimitry Andric               if (const auto *GV = dyn_cast<GlobalVariable>(U))
1938fe6060f1SDimitry Andric                 return GV->hasName() &&
1939fe6060f1SDimitry Andric                        (GV->getName().equals("llvm.compiler.used") ||
1940fe6060f1SDimitry Andric                         GV->getName().equals("llvm.used"));
1941fe6060f1SDimitry Andric               return false;
1942fe6060f1SDimitry Andric             }))
1943fe6060f1SDimitry Andric           continue;
1944fe6060f1SDimitry Andric       }
19450b57cec5SDimitry Andric       if (PutOffender)
19460b57cec5SDimitry Andric         *PutOffender = FU;
19470b57cec5SDimitry Andric       return true;
19480b57cec5SDimitry Andric     }
1949*bdd1243dSDimitry Andric 
1950*bdd1243dSDimitry Andric     if (IgnoreAssumeLikeCalls) {
1951*bdd1243dSDimitry Andric       if (const auto *I = dyn_cast<IntrinsicInst>(Call))
1952*bdd1243dSDimitry Andric         if (I->isAssumeLikeIntrinsic())
1953*bdd1243dSDimitry Andric           continue;
1954*bdd1243dSDimitry Andric     }
1955*bdd1243dSDimitry Andric 
195681ad6265SDimitry Andric     if (!Call->isCallee(&U) || Call->getFunctionType() != getFunctionType()) {
1957349cc55cSDimitry Andric       if (IgnoreARCAttachedCall &&
1958349cc55cSDimitry Andric           Call->isOperandBundleOfType(LLVMContext::OB_clang_arc_attachedcall,
1959349cc55cSDimitry Andric                                       U.getOperandNo()))
1960349cc55cSDimitry Andric         continue;
1961349cc55cSDimitry Andric 
19620b57cec5SDimitry Andric       if (PutOffender)
19630b57cec5SDimitry Andric         *PutOffender = FU;
19640b57cec5SDimitry Andric       return true;
19650b57cec5SDimitry Andric     }
19660b57cec5SDimitry Andric   }
19670b57cec5SDimitry Andric   return false;
19680b57cec5SDimitry Andric }
19690b57cec5SDimitry Andric 
19700b57cec5SDimitry Andric bool Function::isDefTriviallyDead() const {
19710b57cec5SDimitry Andric   // Check the linkage
19720b57cec5SDimitry Andric   if (!hasLinkOnceLinkage() && !hasLocalLinkage() &&
19730b57cec5SDimitry Andric       !hasAvailableExternallyLinkage())
19740b57cec5SDimitry Andric     return false;
19750b57cec5SDimitry Andric 
19760b57cec5SDimitry Andric   // Check if the function is used by anything other than a blockaddress.
19770b57cec5SDimitry Andric   for (const User *U : users())
19780b57cec5SDimitry Andric     if (!isa<BlockAddress>(U))
19790b57cec5SDimitry Andric       return false;
19800b57cec5SDimitry Andric 
19810b57cec5SDimitry Andric   return true;
19820b57cec5SDimitry Andric }
19830b57cec5SDimitry Andric 
19840b57cec5SDimitry Andric /// callsFunctionThatReturnsTwice - Return true if the function has a call to
19850b57cec5SDimitry Andric /// setjmp or other function that gcc recognizes as "returning twice".
19860b57cec5SDimitry Andric bool Function::callsFunctionThatReturnsTwice() const {
19870b57cec5SDimitry Andric   for (const Instruction &I : instructions(this))
19880b57cec5SDimitry Andric     if (const auto *Call = dyn_cast<CallBase>(&I))
19890b57cec5SDimitry Andric       if (Call->hasFnAttr(Attribute::ReturnsTwice))
19900b57cec5SDimitry Andric         return true;
19910b57cec5SDimitry Andric 
19920b57cec5SDimitry Andric   return false;
19930b57cec5SDimitry Andric }
19940b57cec5SDimitry Andric 
19950b57cec5SDimitry Andric Constant *Function::getPersonalityFn() const {
19960b57cec5SDimitry Andric   assert(hasPersonalityFn() && getNumOperands());
19970b57cec5SDimitry Andric   return cast<Constant>(Op<0>());
19980b57cec5SDimitry Andric }
19990b57cec5SDimitry Andric 
20000b57cec5SDimitry Andric void Function::setPersonalityFn(Constant *Fn) {
20010b57cec5SDimitry Andric   setHungoffOperand<0>(Fn);
20020b57cec5SDimitry Andric   setValueSubclassDataBit(3, Fn != nullptr);
20030b57cec5SDimitry Andric }
20040b57cec5SDimitry Andric 
20050b57cec5SDimitry Andric Constant *Function::getPrefixData() const {
20060b57cec5SDimitry Andric   assert(hasPrefixData() && getNumOperands());
20070b57cec5SDimitry Andric   return cast<Constant>(Op<1>());
20080b57cec5SDimitry Andric }
20090b57cec5SDimitry Andric 
20100b57cec5SDimitry Andric void Function::setPrefixData(Constant *PrefixData) {
20110b57cec5SDimitry Andric   setHungoffOperand<1>(PrefixData);
20120b57cec5SDimitry Andric   setValueSubclassDataBit(1, PrefixData != nullptr);
20130b57cec5SDimitry Andric }
20140b57cec5SDimitry Andric 
20150b57cec5SDimitry Andric Constant *Function::getPrologueData() const {
20160b57cec5SDimitry Andric   assert(hasPrologueData() && getNumOperands());
20170b57cec5SDimitry Andric   return cast<Constant>(Op<2>());
20180b57cec5SDimitry Andric }
20190b57cec5SDimitry Andric 
20200b57cec5SDimitry Andric void Function::setPrologueData(Constant *PrologueData) {
20210b57cec5SDimitry Andric   setHungoffOperand<2>(PrologueData);
20220b57cec5SDimitry Andric   setValueSubclassDataBit(2, PrologueData != nullptr);
20230b57cec5SDimitry Andric }
20240b57cec5SDimitry Andric 
20250b57cec5SDimitry Andric void Function::allocHungoffUselist() {
20260b57cec5SDimitry Andric   // If we've already allocated a uselist, stop here.
20270b57cec5SDimitry Andric   if (getNumOperands())
20280b57cec5SDimitry Andric     return;
20290b57cec5SDimitry Andric 
20300b57cec5SDimitry Andric   allocHungoffUses(3, /*IsPhi=*/ false);
20310b57cec5SDimitry Andric   setNumHungOffUseOperands(3);
20320b57cec5SDimitry Andric 
20330b57cec5SDimitry Andric   // Initialize the uselist with placeholder operands to allow traversal.
20340b57cec5SDimitry Andric   auto *CPN = ConstantPointerNull::get(Type::getInt1PtrTy(getContext(), 0));
20350b57cec5SDimitry Andric   Op<0>().set(CPN);
20360b57cec5SDimitry Andric   Op<1>().set(CPN);
20370b57cec5SDimitry Andric   Op<2>().set(CPN);
20380b57cec5SDimitry Andric }
20390b57cec5SDimitry Andric 
20400b57cec5SDimitry Andric template <int Idx>
20410b57cec5SDimitry Andric void Function::setHungoffOperand(Constant *C) {
20420b57cec5SDimitry Andric   if (C) {
20430b57cec5SDimitry Andric     allocHungoffUselist();
20440b57cec5SDimitry Andric     Op<Idx>().set(C);
20450b57cec5SDimitry Andric   } else if (getNumOperands()) {
20460b57cec5SDimitry Andric     Op<Idx>().set(
20470b57cec5SDimitry Andric         ConstantPointerNull::get(Type::getInt1PtrTy(getContext(), 0)));
20480b57cec5SDimitry Andric   }
20490b57cec5SDimitry Andric }
20500b57cec5SDimitry Andric 
20510b57cec5SDimitry Andric void Function::setValueSubclassDataBit(unsigned Bit, bool On) {
20520b57cec5SDimitry Andric   assert(Bit < 16 && "SubclassData contains only 16 bits");
20530b57cec5SDimitry Andric   if (On)
20540b57cec5SDimitry Andric     setValueSubclassData(getSubclassDataFromValue() | (1 << Bit));
20550b57cec5SDimitry Andric   else
20560b57cec5SDimitry Andric     setValueSubclassData(getSubclassDataFromValue() & ~(1 << Bit));
20570b57cec5SDimitry Andric }
20580b57cec5SDimitry Andric 
20590b57cec5SDimitry Andric void Function::setEntryCount(ProfileCount Count,
20600b57cec5SDimitry Andric                              const DenseSet<GlobalValue::GUID> *S) {
20610b57cec5SDimitry Andric #if !defined(NDEBUG)
20620b57cec5SDimitry Andric   auto PrevCount = getEntryCount();
206381ad6265SDimitry Andric   assert(!PrevCount || PrevCount->getType() == Count.getType());
20640b57cec5SDimitry Andric #endif
2065480093f4SDimitry Andric 
2066480093f4SDimitry Andric   auto ImportGUIDs = getImportGUIDs();
2067480093f4SDimitry Andric   if (S == nullptr && ImportGUIDs.size())
2068480093f4SDimitry Andric     S = &ImportGUIDs;
2069480093f4SDimitry Andric 
20700b57cec5SDimitry Andric   MDBuilder MDB(getContext());
20710b57cec5SDimitry Andric   setMetadata(
20720b57cec5SDimitry Andric       LLVMContext::MD_prof,
20730b57cec5SDimitry Andric       MDB.createFunctionEntryCount(Count.getCount(), Count.isSynthetic(), S));
20740b57cec5SDimitry Andric }
20750b57cec5SDimitry Andric 
20760b57cec5SDimitry Andric void Function::setEntryCount(uint64_t Count, Function::ProfileCountType Type,
20770b57cec5SDimitry Andric                              const DenseSet<GlobalValue::GUID> *Imports) {
20780b57cec5SDimitry Andric   setEntryCount(ProfileCount(Count, Type), Imports);
20790b57cec5SDimitry Andric }
20800b57cec5SDimitry Andric 
2081*bdd1243dSDimitry Andric std::optional<ProfileCount> Function::getEntryCount(bool AllowSynthetic) const {
20820b57cec5SDimitry Andric   MDNode *MD = getMetadata(LLVMContext::MD_prof);
20830b57cec5SDimitry Andric   if (MD && MD->getOperand(0))
20840b57cec5SDimitry Andric     if (MDString *MDS = dyn_cast<MDString>(MD->getOperand(0))) {
20850b57cec5SDimitry Andric       if (MDS->getString().equals("function_entry_count")) {
20860b57cec5SDimitry Andric         ConstantInt *CI = mdconst::extract<ConstantInt>(MD->getOperand(1));
20870b57cec5SDimitry Andric         uint64_t Count = CI->getValue().getZExtValue();
20880b57cec5SDimitry Andric         // A value of -1 is used for SamplePGO when there were no samples.
20890b57cec5SDimitry Andric         // Treat this the same as unknown.
20900b57cec5SDimitry Andric         if (Count == (uint64_t)-1)
2091*bdd1243dSDimitry Andric           return std::nullopt;
20920b57cec5SDimitry Andric         return ProfileCount(Count, PCT_Real);
20930b57cec5SDimitry Andric       } else if (AllowSynthetic &&
20940b57cec5SDimitry Andric                  MDS->getString().equals("synthetic_function_entry_count")) {
20950b57cec5SDimitry Andric         ConstantInt *CI = mdconst::extract<ConstantInt>(MD->getOperand(1));
20960b57cec5SDimitry Andric         uint64_t Count = CI->getValue().getZExtValue();
20970b57cec5SDimitry Andric         return ProfileCount(Count, PCT_Synthetic);
20980b57cec5SDimitry Andric       }
20990b57cec5SDimitry Andric     }
2100*bdd1243dSDimitry Andric   return std::nullopt;
21010b57cec5SDimitry Andric }
21020b57cec5SDimitry Andric 
21030b57cec5SDimitry Andric DenseSet<GlobalValue::GUID> Function::getImportGUIDs() const {
21040b57cec5SDimitry Andric   DenseSet<GlobalValue::GUID> R;
21050b57cec5SDimitry Andric   if (MDNode *MD = getMetadata(LLVMContext::MD_prof))
21060b57cec5SDimitry Andric     if (MDString *MDS = dyn_cast<MDString>(MD->getOperand(0)))
21070b57cec5SDimitry Andric       if (MDS->getString().equals("function_entry_count"))
21080b57cec5SDimitry Andric         for (unsigned i = 2; i < MD->getNumOperands(); i++)
21090b57cec5SDimitry Andric           R.insert(mdconst::extract<ConstantInt>(MD->getOperand(i))
21100b57cec5SDimitry Andric                        ->getValue()
21110b57cec5SDimitry Andric                        .getZExtValue());
21120b57cec5SDimitry Andric   return R;
21130b57cec5SDimitry Andric }
21140b57cec5SDimitry Andric 
21150b57cec5SDimitry Andric void Function::setSectionPrefix(StringRef Prefix) {
21160b57cec5SDimitry Andric   MDBuilder MDB(getContext());
21170b57cec5SDimitry Andric   setMetadata(LLVMContext::MD_section_prefix,
21180b57cec5SDimitry Andric               MDB.createFunctionSectionPrefix(Prefix));
21190b57cec5SDimitry Andric }
21200b57cec5SDimitry Andric 
2121*bdd1243dSDimitry Andric std::optional<StringRef> Function::getSectionPrefix() const {
21220b57cec5SDimitry Andric   if (MDNode *MD = getMetadata(LLVMContext::MD_section_prefix)) {
21230b57cec5SDimitry Andric     assert(cast<MDString>(MD->getOperand(0))
21240b57cec5SDimitry Andric                ->getString()
21250b57cec5SDimitry Andric                .equals("function_section_prefix") &&
21260b57cec5SDimitry Andric            "Metadata not match");
21270b57cec5SDimitry Andric     return cast<MDString>(MD->getOperand(1))->getString();
21280b57cec5SDimitry Andric   }
2129*bdd1243dSDimitry Andric   return std::nullopt;
21300b57cec5SDimitry Andric }
21310b57cec5SDimitry Andric 
21320b57cec5SDimitry Andric bool Function::nullPointerIsDefined() const {
21335ffd83dbSDimitry Andric   return hasFnAttribute(Attribute::NullPointerIsValid);
21340b57cec5SDimitry Andric }
21350b57cec5SDimitry Andric 
21360b57cec5SDimitry Andric bool llvm::NullPointerIsDefined(const Function *F, unsigned AS) {
21370b57cec5SDimitry Andric   if (F && F->nullPointerIsDefined())
21380b57cec5SDimitry Andric     return true;
21390b57cec5SDimitry Andric 
21400b57cec5SDimitry Andric   if (AS != 0)
21410b57cec5SDimitry Andric     return true;
21420b57cec5SDimitry Andric 
21430b57cec5SDimitry Andric   return false;
21440b57cec5SDimitry Andric }
2145