• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===-- NVPTXISelLowering.h - NVPTX DAG Lowering Interface ------*- 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 // This file defines the interfaces that NVPTX uses to lower LLVM code into a
11 // selection DAG.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef NVPTXISELLOWERING_H
16 #define NVPTXISELLOWERING_H
17 
18 #include "NVPTX.h"
19 #include "NVPTXSubtarget.h"
20 #include "llvm/CodeGen/SelectionDAG.h"
21 #include "llvm/Target/TargetLowering.h"
22 
23 namespace llvm {
24 namespace NVPTXISD {
25 enum NodeType {
26   // Start the numbering from where ISD NodeType finishes.
27   FIRST_NUMBER = ISD::BUILTIN_OP_END,
28   Wrapper,
29   CALL,
30   RET_FLAG,
31   LOAD_PARAM,
32   NVBuiltin,
33   DeclareParam,
34   DeclareScalarParam,
35   DeclareRetParam,
36   DeclareRet,
37   DeclareScalarRet,
38   LoadParam,
39   StoreParam,
40   StoreParamS32, // to sext and store a <32bit value, not used currently
41   StoreParamU32, // to zext and store a <32bit value, not used currently
42   MoveToParam,
43   PrintCall,
44   PrintCallUni,
45   CallArgBegin,
46   CallArg,
47   LastCallArg,
48   CallArgEnd,
49   CallVoid,
50   CallVal,
51   CallSymbol,
52   Prototype,
53   MoveParam,
54   MoveRetval,
55   MoveToRetval,
56   StoreRetval,
57   PseudoUseParam,
58   RETURN,
59   CallSeqBegin,
60   CallSeqEnd,
61   Dummy,
62 
63   LoadV2 = ISD::FIRST_TARGET_MEMORY_OPCODE,
64   LoadV4,
65   LDGV2, // LDG.v2
66   LDGV4, // LDG.v4
67   LDUV2, // LDU.v2
68   LDUV4, // LDU.v4
69   StoreV2,
70   StoreV4
71 };
72 }
73 
74 //===--------------------------------------------------------------------===//
75 // TargetLowering Implementation
76 //===--------------------------------------------------------------------===//
77 class NVPTXTargetLowering : public TargetLowering {
78 public:
79   explicit NVPTXTargetLowering(NVPTXTargetMachine &TM);
80   virtual SDValue LowerOperation(SDValue Op, SelectionDAG &DAG) const;
81 
82   SDValue LowerGlobalAddress(SDValue Op, SelectionDAG &DAG) const;
83   SDValue LowerGlobalAddress(const GlobalValue *GV, int64_t Offset,
84                              SelectionDAG &DAG) const;
85 
86   virtual const char *getTargetNodeName(unsigned Opcode) const;
87 
88   bool isTypeSupportedInIntrinsic(MVT VT) const;
89 
90   bool getTgtMemIntrinsic(IntrinsicInfo& Info, const CallInst &I,
91                           unsigned Intrinsic) const;
92 
93   /// isLegalAddressingMode - Return true if the addressing mode represented
94   /// by AM is legal for this target, for a load/store of the specified type
95   /// Used to guide target specific optimizations, like loop strength
96   /// reduction (LoopStrengthReduce.cpp) and memory optimization for
97   /// address mode (CodeGenPrepare.cpp)
98   virtual bool isLegalAddressingMode(const AddrMode &AM, Type *Ty) const;
99 
100   /// getFunctionAlignment - Return the Log2 alignment of this function.
101   virtual unsigned getFunctionAlignment(const Function *F) const;
102 
getSetCCResultType(EVT VT)103   virtual EVT getSetCCResultType(EVT VT) const {
104     if (VT.isVector())
105       return MVT::getVectorVT(MVT::i1, VT.getVectorNumElements());
106     return MVT::i1;
107   }
108 
109   ConstraintType getConstraintType(const std::string &Constraint) const;
110   std::pair<unsigned, const TargetRegisterClass*>
111   getRegForInlineAsmConstraint(const std::string &Constraint, EVT VT) const;
112 
113   virtual SDValue
114   LowerFormalArguments(SDValue Chain, CallingConv::ID CallConv, bool isVarArg,
115                        const SmallVectorImpl<ISD::InputArg> &Ins, DebugLoc dl,
116                        SelectionDAG &DAG,
117                        SmallVectorImpl<SDValue> &InVals) const;
118 
119   virtual SDValue
120   LowerCall(CallLoweringInfo &CLI, SmallVectorImpl<SDValue> &InVals) const;
121 
122   std::string getPrototype(Type *, const ArgListTy &,
123                            const SmallVectorImpl<ISD::OutputArg> &,
124                            unsigned retAlignment) const;
125 
126   virtual SDValue
127   LowerReturn(SDValue Chain, CallingConv::ID CallConv, bool isVarArg,
128               const SmallVectorImpl<ISD::OutputArg> &Outs,
129               const SmallVectorImpl<SDValue> &OutVals, DebugLoc dl,
130               SelectionDAG &DAG) const;
131 
132   virtual void LowerAsmOperandForConstraint(SDValue Op, std::string &Constraint,
133                                             std::vector<SDValue> &Ops,
134                                             SelectionDAG &DAG) const;
135 
136   NVPTXTargetMachine *nvTM;
137 
138   // PTX always uses 32-bit shift amounts
getScalarShiftAmountTy(EVT LHSTy)139   virtual MVT getScalarShiftAmountTy(EVT LHSTy) const {
140     return MVT::i32;
141   }
142 
143   virtual bool shouldSplitVectorElementType(EVT VT) const;
144 
145 private:
146   const NVPTXSubtarget &nvptxSubtarget;  // cache the subtarget here
147 
148   SDValue getExtSymb(SelectionDAG &DAG, const char *name, int idx, EVT =
149                          MVT::i32) const;
150   SDValue getParamSymbol(SelectionDAG &DAG, int idx, EVT = MVT::i32) const;
151   SDValue getParamHelpSymbol(SelectionDAG &DAG, int idx);
152 
153   SDValue LowerCONCAT_VECTORS(SDValue Op, SelectionDAG &DAG) const;
154 
155   SDValue LowerLOAD(SDValue Op, SelectionDAG &DAG) const;
156   SDValue LowerLOADi1(SDValue Op, SelectionDAG &DAG) const;
157 
158   SDValue LowerSTORE(SDValue Op, SelectionDAG &DAG) const;
159   SDValue LowerSTOREi1(SDValue Op, SelectionDAG &DAG) const;
160   SDValue LowerSTOREVector(SDValue Op, SelectionDAG &DAG) const;
161 
162   virtual void ReplaceNodeResults(SDNode *N,
163                                   SmallVectorImpl<SDValue> &Results,
164                                   SelectionDAG &DAG) const;
165 };
166 } // namespace llvm
167 
168 #endif // NVPTXISELLOWERING_H
169