1 //===- AMDGPUIntrinsicInfo.cpp - AMDGPU Intrinsic Information ---*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //==-----------------------------------------------------------------------===//
9 //
10 /// \file
11 /// AMDGPU Implementation of the IntrinsicInfo class.
12 //
13 //===-----------------------------------------------------------------------===//
14
15 #include "AMDGPUIntrinsicInfo.h"
16 #include "AMDGPUSubtarget.h"
17 #include "llvm/IR/DerivedTypes.h"
18 #include "llvm/IR/Intrinsics.h"
19 #include "llvm/IR/Module.h"
20
21 using namespace llvm;
22
AMDGPUIntrinsicInfo()23 AMDGPUIntrinsicInfo::AMDGPUIntrinsicInfo()
24 : TargetIntrinsicInfo() {}
25
26 static const char *const IntrinsicNameTable[] = {
27 #define GET_INTRINSIC_NAME_TABLE
28 #include "AMDGPUGenIntrinsicImpl.inc"
29 #undef GET_INTRINSIC_NAME_TABLE
30 };
31
32 namespace {
33 #define GET_INTRINSIC_ATTRIBUTES
34 #include "AMDGPUGenIntrinsicImpl.inc"
35 #undef GET_INTRINSIC_ATTRIBUTES
36 }
37
getName(unsigned IntrID,ArrayRef<Type * > Tys) const38 StringRef AMDGPUIntrinsicInfo::getName(unsigned IntrID,
39 ArrayRef<Type *> Tys) const {
40 if (IntrID < Intrinsic::num_intrinsics)
41 return StringRef();
42
43 assert(IntrID < AMDGPUIntrinsic::num_AMDGPU_intrinsics &&
44 "Invalid intrinsic ID");
45
46 return IntrinsicNameTable[IntrID - Intrinsic::num_intrinsics];
47 }
48
getName(unsigned IntrID,Type ** Tys,unsigned NumTys) const49 std::string AMDGPUIntrinsicInfo::getName(unsigned IntrID, Type **Tys,
50 unsigned NumTys) const {
51 return getName(IntrID, makeArrayRef(Tys, NumTys)).str();
52 }
53
getType(LLVMContext & Context,unsigned ID,ArrayRef<Type * > Tys) const54 FunctionType *AMDGPUIntrinsicInfo::getType(LLVMContext &Context, unsigned ID,
55 ArrayRef<Type*> Tys) const {
56 // FIXME: Re-use Intrinsic::getType machinery
57 llvm_unreachable("unhandled intrinsic");
58 }
59
lookupName(const char * NameData,unsigned Len) const60 unsigned AMDGPUIntrinsicInfo::lookupName(const char *NameData,
61 unsigned Len) const {
62 StringRef Name(NameData, Len);
63 if (!Name.startswith("llvm."))
64 return 0; // All intrinsics start with 'llvm.'
65
66 // Look for a name match in our table. If the intrinsic is not overloaded,
67 // require an exact match. If it is overloaded, require a prefix match. The
68 // AMDGPU enum enum starts at Intrinsic::num_intrinsics.
69 int Idx = Intrinsic::lookupLLVMIntrinsicByName(IntrinsicNameTable, Name);
70 if (Idx >= 0) {
71 bool IsPrefixMatch = Name.size() > strlen(IntrinsicNameTable[Idx]);
72 return IsPrefixMatch == isOverloaded(Idx + 1)
73 ? Intrinsic::num_intrinsics + Idx
74 : 0;
75 }
76
77 return 0;
78 }
79
isOverloaded(unsigned id) const80 bool AMDGPUIntrinsicInfo::isOverloaded(unsigned id) const {
81 // Overload Table
82 #define GET_INTRINSIC_OVERLOAD_TABLE
83 #include "AMDGPUGenIntrinsicImpl.inc"
84 #undef GET_INTRINSIC_OVERLOAD_TABLE
85 }
86
getDeclaration(Module * M,unsigned IntrID,ArrayRef<Type * > Tys) const87 Function *AMDGPUIntrinsicInfo::getDeclaration(Module *M, unsigned IntrID,
88 ArrayRef<Type *> Tys) const {
89 FunctionType *FTy = getType(M->getContext(), IntrID, Tys);
90 Function *F
91 = cast<Function>(M->getOrInsertFunction(getName(IntrID, Tys), FTy));
92
93 AttributeList AS =
94 getAttributes(M->getContext(), static_cast<AMDGPUIntrinsic::ID>(IntrID));
95 F->setAttributes(AS);
96 return F;
97 }
98
getDeclaration(Module * M,unsigned IntrID,Type ** Tys,unsigned NumTys) const99 Function *AMDGPUIntrinsicInfo::getDeclaration(Module *M, unsigned IntrID,
100 Type **Tys,
101 unsigned NumTys) const {
102 return getDeclaration(M, IntrID, makeArrayRef(Tys, NumTys));
103 }
104