Make TargetTransformInfo keeping a reference to the Module DataLayout
[oota-llvm.git] / lib / Target / Sparc / SparcISelDAGToDAG.cpp
1 //===-- SparcISelDAGToDAG.cpp - A dag to dag inst selector for Sparc ------===//
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 an instruction selector for the SPARC target.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "SparcTargetMachine.h"
15 #include "llvm/CodeGen/SelectionDAGISel.h"
16 #include "llvm/IR/Intrinsics.h"
17 #include "llvm/Support/Compiler.h"
18 #include "llvm/Support/Debug.h"
19 #include "llvm/Support/ErrorHandling.h"
20 #include "llvm/Support/raw_ostream.h"
21 using namespace llvm;
22
23 //===----------------------------------------------------------------------===//
24 // Instruction Selector Implementation
25 //===----------------------------------------------------------------------===//
26
27 //===--------------------------------------------------------------------===//
28 /// SparcDAGToDAGISel - SPARC specific code to select SPARC machine
29 /// instructions for SelectionDAG operations.
30 ///
31 namespace {
32 class SparcDAGToDAGISel : public SelectionDAGISel {
33   /// Subtarget - Keep a pointer to the Sparc Subtarget around so that we can
34   /// make the right decision when generating code for different targets.
35   const SparcSubtarget *Subtarget;
36 public:
37   explicit SparcDAGToDAGISel(SparcTargetMachine &tm) : SelectionDAGISel(tm) {}
38
39   bool runOnMachineFunction(MachineFunction &MF) override {
40     Subtarget = &MF.getSubtarget<SparcSubtarget>();
41     return SelectionDAGISel::runOnMachineFunction(MF);
42   }
43
44   SDNode *Select(SDNode *N) override;
45
46   // Complex Pattern Selectors.
47   bool SelectADDRrr(SDValue N, SDValue &R1, SDValue &R2);
48   bool SelectADDRri(SDValue N, SDValue &Base, SDValue &Offset);
49
50   /// SelectInlineAsmMemoryOperand - Implement addressing mode selection for
51   /// inline asm expressions.
52   bool SelectInlineAsmMemoryOperand(const SDValue &Op,
53                                     unsigned ConstraintID,
54                                     std::vector<SDValue> &OutOps) override;
55
56   const char *getPassName() const override {
57     return "SPARC DAG->DAG Pattern Instruction Selection";
58   }
59
60   // Include the pieces autogenerated from the target description.
61 #include "SparcGenDAGISel.inc"
62
63 private:
64   SDNode* getGlobalBaseReg();
65 };
66 }  // end anonymous namespace
67
68 SDNode* SparcDAGToDAGISel::getGlobalBaseReg() {
69   unsigned GlobalBaseReg = Subtarget->getInstrInfo()->getGlobalBaseReg(MF);
70   return CurDAG->getRegister(GlobalBaseReg, TLI->getPointerTy()).getNode();
71 }
72
73 bool SparcDAGToDAGISel::SelectADDRri(SDValue Addr,
74                                      SDValue &Base, SDValue &Offset) {
75   if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
76     Base = CurDAG->getTargetFrameIndex(FIN->getIndex(), TLI->getPointerTy());
77     Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
78     return true;
79   }
80   if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
81       Addr.getOpcode() == ISD::TargetGlobalAddress ||
82       Addr.getOpcode() == ISD::TargetGlobalTLSAddress)
83     return false;  // direct calls.
84
85   if (Addr.getOpcode() == ISD::ADD) {
86     if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Addr.getOperand(1))) {
87       if (isInt<13>(CN->getSExtValue())) {
88         if (FrameIndexSDNode *FIN =
89                 dyn_cast<FrameIndexSDNode>(Addr.getOperand(0))) {
90           // Constant offset from frame ref.
91           Base =
92               CurDAG->getTargetFrameIndex(FIN->getIndex(), TLI->getPointerTy());
93         } else {
94           Base = Addr.getOperand(0);
95         }
96         Offset = CurDAG->getTargetConstant(CN->getZExtValue(), SDLoc(Addr),
97                                            MVT::i32);
98         return true;
99       }
100     }
101     if (Addr.getOperand(0).getOpcode() == SPISD::Lo) {
102       Base = Addr.getOperand(1);
103       Offset = Addr.getOperand(0).getOperand(0);
104       return true;
105     }
106     if (Addr.getOperand(1).getOpcode() == SPISD::Lo) {
107       Base = Addr.getOperand(0);
108       Offset = Addr.getOperand(1).getOperand(0);
109       return true;
110     }
111   }
112   Base = Addr;
113   Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
114   return true;
115 }
116
117 bool SparcDAGToDAGISel::SelectADDRrr(SDValue Addr, SDValue &R1, SDValue &R2) {
118   if (Addr.getOpcode() == ISD::FrameIndex) return false;
119   if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
120       Addr.getOpcode() == ISD::TargetGlobalAddress ||
121       Addr.getOpcode() == ISD::TargetGlobalTLSAddress)
122     return false;  // direct calls.
123
124   if (Addr.getOpcode() == ISD::ADD) {
125     if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))
126       if (isInt<13>(CN->getSExtValue()))
127         return false;  // Let the reg+imm pattern catch this!
128     if (Addr.getOperand(0).getOpcode() == SPISD::Lo ||
129         Addr.getOperand(1).getOpcode() == SPISD::Lo)
130       return false;  // Let the reg+imm pattern catch this!
131     R1 = Addr.getOperand(0);
132     R2 = Addr.getOperand(1);
133     return true;
134   }
135
136   R1 = Addr;
137   R2 = CurDAG->getRegister(SP::G0, TLI->getPointerTy());
138   return true;
139 }
140
141 SDNode *SparcDAGToDAGISel::Select(SDNode *N) {
142   SDLoc dl(N);
143   if (N->isMachineOpcode()) {
144     N->setNodeId(-1);
145     return nullptr;   // Already selected.
146   }
147
148   switch (N->getOpcode()) {
149   default: break;
150   case SPISD::GLOBAL_BASE_REG:
151     return getGlobalBaseReg();
152
153   case ISD::SDIV:
154   case ISD::UDIV: {
155     // sdivx / udivx handle 64-bit divides.
156     if (N->getValueType(0) == MVT::i64)
157       break;
158     // FIXME: should use a custom expander to expose the SRA to the dag.
159     SDValue DivLHS = N->getOperand(0);
160     SDValue DivRHS = N->getOperand(1);
161
162     // Set the Y register to the high-part.
163     SDValue TopPart;
164     if (N->getOpcode() == ISD::SDIV) {
165       TopPart = SDValue(CurDAG->getMachineNode(SP::SRAri, dl, MVT::i32, DivLHS,
166                                    CurDAG->getTargetConstant(31, dl, MVT::i32)),
167                         0);
168     } else {
169       TopPart = CurDAG->getRegister(SP::G0, MVT::i32);
170     }
171     TopPart = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, SP::Y, TopPart,
172                                    SDValue())
173                   .getValue(1);
174
175     // FIXME: Handle div by immediate.
176     unsigned Opcode = N->getOpcode() == ISD::SDIV ? SP::SDIVrr : SP::UDIVrr;
177     return CurDAG->SelectNodeTo(N, Opcode, MVT::i32, DivLHS, DivRHS,
178                                 TopPart);
179   }
180   case ISD::MULHU:
181   case ISD::MULHS: {
182     // FIXME: Handle mul by immediate.
183     SDValue MulLHS = N->getOperand(0);
184     SDValue MulRHS = N->getOperand(1);
185     unsigned Opcode = N->getOpcode() == ISD::MULHU ? SP::UMULrr : SP::SMULrr;
186     SDNode *Mul =
187         CurDAG->getMachineNode(Opcode, dl, MVT::i32, MVT::i32, MulLHS, MulRHS);
188     SDValue ResultHigh = SDValue(Mul, 1);
189     ReplaceUses(SDValue(N, 0), ResultHigh);
190     return nullptr;
191   }
192   }
193
194   return SelectCode(N);
195 }
196
197
198 /// SelectInlineAsmMemoryOperand - Implement addressing mode selection for
199 /// inline asm expressions.
200 bool
201 SparcDAGToDAGISel::SelectInlineAsmMemoryOperand(const SDValue &Op,
202                                                 unsigned ConstraintID,
203                                                 std::vector<SDValue> &OutOps) {
204   SDValue Op0, Op1;
205   switch (ConstraintID) {
206   default: return true;
207   case InlineAsm::Constraint_i:
208   case InlineAsm::Constraint_m: // memory
209    if (!SelectADDRrr(Op, Op0, Op1))
210      SelectADDRri(Op, Op0, Op1);
211    break;
212   }
213
214   OutOps.push_back(Op0);
215   OutOps.push_back(Op1);
216   return false;
217 }
218
219 /// createSparcISelDag - This pass converts a legalized DAG into a
220 /// SPARC-specific DAG, ready for instruction scheduling.
221 ///
222 FunctionPass *llvm::createSparcISelDag(SparcTargetMachine &TM) {
223   return new SparcDAGToDAGISel(TM);
224 }