Push LLVMContexts through the IntegerType APIs.
[oota-llvm.git] / lib / Target / Alpha / AlphaISelDAGToDAG.cpp
1 //===-- AlphaISelDAGToDAG.cpp - Alpha pattern matching inst selector ------===//
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 a pattern matching instruction selector for Alpha,
11 // converting from a legalized dag to a Alpha dag.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #include "Alpha.h"
16 #include "AlphaTargetMachine.h"
17 #include "AlphaISelLowering.h"
18 #include "llvm/CodeGen/MachineInstrBuilder.h"
19 #include "llvm/CodeGen/MachineFrameInfo.h"
20 #include "llvm/CodeGen/MachineFunction.h"
21 #include "llvm/CodeGen/MachineRegisterInfo.h"
22 #include "llvm/CodeGen/SelectionDAG.h"
23 #include "llvm/CodeGen/SelectionDAGISel.h"
24 #include "llvm/Target/TargetOptions.h"
25 #include "llvm/Constants.h"
26 #include "llvm/DerivedTypes.h"
27 #include "llvm/GlobalValue.h"
28 #include "llvm/Intrinsics.h"
29 #include "llvm/LLVMContext.h"
30 #include "llvm/Support/Compiler.h"
31 #include "llvm/Support/Debug.h"
32 #include "llvm/Support/ErrorHandling.h"
33 #include "llvm/Support/MathExtras.h"
34 #include "llvm/Support/raw_ostream.h"
35 #include <algorithm>
36 using namespace llvm;
37
38 namespace {
39
40   //===--------------------------------------------------------------------===//
41   /// AlphaDAGToDAGISel - Alpha specific code to select Alpha machine
42   /// instructions for SelectionDAG operations.
43   class AlphaDAGToDAGISel : public SelectionDAGISel {
44     static const int64_t IMM_LOW  = -32768;
45     static const int64_t IMM_HIGH = 32767;
46     static const int64_t IMM_MULT = 65536;
47     static const int64_t IMM_FULLHIGH = IMM_HIGH + IMM_HIGH * IMM_MULT;
48     static const int64_t IMM_FULLLOW = IMM_LOW + IMM_LOW  * IMM_MULT;
49
50     static int64_t get_ldah16(int64_t x) {
51       int64_t y = x / IMM_MULT;
52       if (x % IMM_MULT > IMM_HIGH)
53         ++y;
54       return y;
55     }
56
57     static int64_t get_lda16(int64_t x) {
58       return x - get_ldah16(x) * IMM_MULT;
59     }
60
61     /// get_zapImm - Return a zap mask if X is a valid immediate for a zapnot
62     /// instruction (if not, return 0).  Note that this code accepts partial
63     /// zap masks.  For example (and LHS, 1) is a valid zap, as long we know
64     /// that the bits 1-7 of LHS are already zero.  If LHS is non-null, we are
65     /// in checking mode.  If LHS is null, we assume that the mask has already
66     /// been validated before.
67     uint64_t get_zapImm(SDValue LHS, uint64_t Constant) {
68       uint64_t BitsToCheck = 0;
69       unsigned Result = 0;
70       for (unsigned i = 0; i != 8; ++i) {
71         if (((Constant >> 8*i) & 0xFF) == 0) {
72           // nothing to do.
73         } else {
74           Result |= 1 << i;
75           if (((Constant >> 8*i) & 0xFF) == 0xFF) {
76             // If the entire byte is set, zapnot the byte.
77           } else if (LHS.getNode() == 0) {
78             // Otherwise, if the mask was previously validated, we know its okay
79             // to zapnot this entire byte even though all the bits aren't set.
80           } else {
81             // Otherwise we don't know that the it's okay to zapnot this entire
82             // byte.  Only do this iff we can prove that the missing bits are
83             // already null, so the bytezap doesn't need to really null them.
84             BitsToCheck |= ~Constant & (0xFF << 8*i);
85           }
86         }
87       }
88       
89       // If there are missing bits in a byte (for example, X & 0xEF00), check to
90       // see if the missing bits (0x1000) are already known zero if not, the zap
91       // isn't okay to do, as it won't clear all the required bits.
92       if (BitsToCheck &&
93           !CurDAG->MaskedValueIsZero(LHS,
94                                      APInt(LHS.getValueSizeInBits(),
95                                            BitsToCheck)))
96         return 0;
97       
98       return Result;
99     }
100     
101     static uint64_t get_zapImm(uint64_t x) {
102       unsigned build = 0;
103       for(int i = 0; i != 8; ++i) {
104         if ((x & 0x00FF) == 0x00FF)
105           build |= 1 << i;
106         else if ((x & 0x00FF) != 0)
107           return 0;
108         x >>= 8;
109       }
110       return build;
111     }
112       
113     
114     static uint64_t getNearPower2(uint64_t x) {
115       if (!x) return 0;
116       unsigned at = CountLeadingZeros_64(x);
117       uint64_t complow = 1 << (63 - at);
118       uint64_t comphigh = 1 << (64 - at);
119       //cerr << x << ":" << complow << ":" << comphigh << "\n";
120       if (abs64(complow - x) <= abs64(comphigh - x))
121         return complow;
122       else
123         return comphigh;
124     }
125
126     static bool chkRemNearPower2(uint64_t x, uint64_t r, bool swap) {
127       uint64_t y = getNearPower2(x);
128       if (swap)
129         return (y - x) == r;
130       else
131         return (x - y) == r;
132     }
133
134     static bool isFPZ(SDValue N) {
135       ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(N);
136       return (CN && (CN->getValueAPF().isZero()));
137     }
138     static bool isFPZn(SDValue N) {
139       ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(N);
140       return (CN && CN->getValueAPF().isNegZero());
141     }
142     static bool isFPZp(SDValue N) {
143       ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(N);
144       return (CN && CN->getValueAPF().isPosZero());
145     }
146
147   public:
148     explicit AlphaDAGToDAGISel(AlphaTargetMachine &TM)
149       : SelectionDAGISel(TM)
150     {}
151
152     /// getI64Imm - Return a target constant with the specified value, of type
153     /// i64.
154     inline SDValue getI64Imm(int64_t Imm) {
155       return CurDAG->getTargetConstant(Imm, MVT::i64);
156     }
157
158     // Select - Convert the specified operand from a target-independent to a
159     // target-specific node if it hasn't already been changed.
160     SDNode *Select(SDValue Op);
161     
162     /// InstructionSelect - This callback is invoked by
163     /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
164     virtual void InstructionSelect();
165     
166     virtual const char *getPassName() const {
167       return "Alpha DAG->DAG Pattern Instruction Selection";
168     } 
169
170     /// SelectInlineAsmMemoryOperand - Implement addressing mode selection for
171     /// inline asm expressions.
172     virtual bool SelectInlineAsmMemoryOperand(const SDValue &Op,
173                                               char ConstraintCode,
174                                               std::vector<SDValue> &OutOps) {
175       SDValue Op0;
176       switch (ConstraintCode) {
177       default: return true;
178       case 'm':   // memory
179         Op0 = Op;
180         break;
181       }
182       
183       OutOps.push_back(Op0);
184       return false;
185     }
186     
187 // Include the pieces autogenerated from the target description.
188 #include "AlphaGenDAGISel.inc"
189     
190 private:
191     /// getTargetMachine - Return a reference to the TargetMachine, casted
192     /// to the target-specific type.
193     const AlphaTargetMachine &getTargetMachine() {
194       return static_cast<const AlphaTargetMachine &>(TM);
195     }
196
197     /// getInstrInfo - Return a reference to the TargetInstrInfo, casted
198     /// to the target-specific type.
199     const AlphaInstrInfo *getInstrInfo() {
200       return getTargetMachine().getInstrInfo();
201     }
202
203     SDNode *getGlobalBaseReg();
204     SDNode *getGlobalRetAddr();
205     void SelectCALL(SDValue Op);
206
207   };
208 }
209
210 /// getGlobalBaseReg - Output the instructions required to put the
211 /// GOT address into a register.
212 ///
213 SDNode *AlphaDAGToDAGISel::getGlobalBaseReg() {
214   MachineFunction *MF = BB->getParent();
215   unsigned GlobalBaseReg = getInstrInfo()->getGlobalBaseReg(MF);
216   return CurDAG->getRegister(GlobalBaseReg, TLI.getPointerTy()).getNode();
217 }
218
219 /// getGlobalRetAddr - Grab the return address.
220 ///
221 SDNode *AlphaDAGToDAGISel::getGlobalRetAddr() {
222   MachineFunction *MF = BB->getParent();
223   unsigned GlobalRetAddr = getInstrInfo()->getGlobalRetAddr(MF);
224   return CurDAG->getRegister(GlobalRetAddr, TLI.getPointerTy()).getNode();
225 }
226
227 /// InstructionSelect - This callback is invoked by
228 /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
229 void AlphaDAGToDAGISel::InstructionSelect() {
230   DEBUG(BB->dump());
231   
232   // Select target instructions for the DAG.
233   SelectRoot(*CurDAG);
234   CurDAG->RemoveDeadNodes();
235 }
236
237 // Select - Convert the specified operand from a target-independent to a
238 // target-specific node if it hasn't already been changed.
239 SDNode *AlphaDAGToDAGISel::Select(SDValue Op) {
240   SDNode *N = Op.getNode();
241   if (N->isMachineOpcode()) {
242     return NULL;   // Already selected.
243   }
244   DebugLoc dl = N->getDebugLoc();
245
246   switch (N->getOpcode()) {
247   default: break;
248   case AlphaISD::CALL:
249     SelectCALL(Op);
250     return NULL;
251
252   case ISD::FrameIndex: {
253     int FI = cast<FrameIndexSDNode>(N)->getIndex();
254     return CurDAG->SelectNodeTo(N, Alpha::LDA, MVT::i64,
255                                 CurDAG->getTargetFrameIndex(FI, MVT::i32),
256                                 getI64Imm(0));
257   }
258   case ISD::GLOBAL_OFFSET_TABLE:
259     return getGlobalBaseReg();
260   case AlphaISD::GlobalRetAddr:
261     return getGlobalRetAddr();
262   
263   case AlphaISD::DivCall: {
264     SDValue Chain = CurDAG->getEntryNode();
265     SDValue N0 = Op.getOperand(0);
266     SDValue N1 = Op.getOperand(1);
267     SDValue N2 = Op.getOperand(2);
268     Chain = CurDAG->getCopyToReg(Chain, dl, Alpha::R24, N1, 
269                                  SDValue(0,0));
270     Chain = CurDAG->getCopyToReg(Chain, dl, Alpha::R25, N2, 
271                                  Chain.getValue(1));
272     Chain = CurDAG->getCopyToReg(Chain, dl, Alpha::R27, N0, 
273                                  Chain.getValue(1));
274     SDNode *CNode =
275       CurDAG->getTargetNode(Alpha::JSRs, dl, MVT::Other, MVT::Flag, 
276                             Chain, Chain.getValue(1));
277     Chain = CurDAG->getCopyFromReg(Chain, dl, Alpha::R27, MVT::i64, 
278                                    SDValue(CNode, 1));
279     return CurDAG->SelectNodeTo(N, Alpha::BISr, MVT::i64, Chain, Chain);
280   }
281
282   case ISD::READCYCLECOUNTER: {
283     SDValue Chain = N->getOperand(0);
284     return CurDAG->getTargetNode(Alpha::RPCC, dl, MVT::i64, MVT::Other,
285                                  Chain);
286   }
287
288   case ISD::Constant: {
289     uint64_t uval = cast<ConstantSDNode>(N)->getZExtValue();
290     
291     if (uval == 0) {
292       SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
293                                                 Alpha::R31, MVT::i64);
294       ReplaceUses(Op, Result);
295       return NULL;
296     }
297
298     int64_t val = (int64_t)uval;
299     int32_t val32 = (int32_t)val;
300     if (val <= IMM_HIGH + IMM_HIGH * IMM_MULT &&
301         val >= IMM_LOW  + IMM_LOW  * IMM_MULT)
302       break; //(LDAH (LDA))
303     if ((uval >> 32) == 0 && //empty upper bits
304         val32 <= IMM_HIGH + IMM_HIGH * IMM_MULT)
305       // val32 >= IMM_LOW  + IMM_LOW  * IMM_MULT) //always true
306       break; //(zext (LDAH (LDA)))
307     //Else use the constant pool
308     ConstantInt *C = ConstantInt::get(
309                                 Type::getInt64Ty(*CurDAG->getContext()), uval);
310     SDValue CPI = CurDAG->getTargetConstantPool(C, MVT::i64);
311     SDNode *Tmp = CurDAG->getTargetNode(Alpha::LDAHr, dl, MVT::i64, CPI,
312                                         SDValue(getGlobalBaseReg(), 0));
313     return CurDAG->SelectNodeTo(N, Alpha::LDQr, MVT::i64, MVT::Other, 
314                                 CPI, SDValue(Tmp, 0), CurDAG->getEntryNode());
315   }
316   case ISD::TargetConstantFP:
317   case ISD::ConstantFP: {
318     ConstantFPSDNode *CN = cast<ConstantFPSDNode>(N);
319     bool isDouble = N->getValueType(0) == MVT::f64;
320     EVT T = isDouble ? MVT::f64 : MVT::f32;
321     if (CN->getValueAPF().isPosZero()) {
322       return CurDAG->SelectNodeTo(N, isDouble ? Alpha::CPYST : Alpha::CPYSS,
323                                   T, CurDAG->getRegister(Alpha::F31, T),
324                                   CurDAG->getRegister(Alpha::F31, T));
325     } else if (CN->getValueAPF().isNegZero()) {
326       return CurDAG->SelectNodeTo(N, isDouble ? Alpha::CPYSNT : Alpha::CPYSNS,
327                                   T, CurDAG->getRegister(Alpha::F31, T),
328                                   CurDAG->getRegister(Alpha::F31, T));
329     } else {
330       llvm_report_error("Unhandled FP constant type");
331     }
332     break;
333   }
334
335   case ISD::SETCC:
336     if (N->getOperand(0).getNode()->getValueType(0).isFloatingPoint()) {
337       ISD::CondCode CC = cast<CondCodeSDNode>(N->getOperand(2))->get();
338
339       unsigned Opc = Alpha::WTF;
340       bool rev = false;
341       bool inv = false;
342       switch(CC) {
343       default: DEBUG(N->dump(CurDAG)); llvm_unreachable("Unknown FP comparison!");
344       case ISD::SETEQ: case ISD::SETOEQ: case ISD::SETUEQ:
345         Opc = Alpha::CMPTEQ; break;
346       case ISD::SETLT: case ISD::SETOLT: case ISD::SETULT: 
347         Opc = Alpha::CMPTLT; break;
348       case ISD::SETLE: case ISD::SETOLE: case ISD::SETULE: 
349         Opc = Alpha::CMPTLE; break;
350       case ISD::SETGT: case ISD::SETOGT: case ISD::SETUGT: 
351         Opc = Alpha::CMPTLT; rev = true; break;
352       case ISD::SETGE: case ISD::SETOGE: case ISD::SETUGE: 
353         Opc = Alpha::CMPTLE; rev = true; break;
354       case ISD::SETNE: case ISD::SETONE: case ISD::SETUNE:
355         Opc = Alpha::CMPTEQ; inv = true; break;
356       case ISD::SETO:
357         Opc = Alpha::CMPTUN; inv = true; break;
358       case ISD::SETUO:
359         Opc = Alpha::CMPTUN; break;
360       };
361       SDValue tmp1 = N->getOperand(rev?1:0);
362       SDValue tmp2 = N->getOperand(rev?0:1);
363       SDNode *cmp = CurDAG->getTargetNode(Opc, dl, MVT::f64, tmp1, tmp2);
364       if (inv) 
365         cmp = CurDAG->getTargetNode(Alpha::CMPTEQ, dl, 
366                                     MVT::f64, SDValue(cmp, 0), 
367                                     CurDAG->getRegister(Alpha::F31, MVT::f64));
368       switch(CC) {
369       case ISD::SETUEQ: case ISD::SETULT: case ISD::SETULE:
370       case ISD::SETUNE: case ISD::SETUGT: case ISD::SETUGE:
371        {
372          SDNode* cmp2 = CurDAG->getTargetNode(Alpha::CMPTUN, dl, MVT::f64,
373                                               tmp1, tmp2);
374          cmp = CurDAG->getTargetNode(Alpha::ADDT, dl, MVT::f64, 
375                                      SDValue(cmp2, 0), SDValue(cmp, 0));
376          break;
377        }
378       default: break;
379       }
380
381       SDNode* LD = CurDAG->getTargetNode(Alpha::FTOIT, dl,
382                                          MVT::i64, SDValue(cmp, 0));
383       return CurDAG->getTargetNode(Alpha::CMPULT, dl, MVT::i64, 
384                                    CurDAG->getRegister(Alpha::R31, MVT::i64),
385                                    SDValue(LD,0));
386     }
387     break;
388
389   case ISD::AND: {
390     ConstantSDNode* SC = NULL;
391     ConstantSDNode* MC = NULL;
392     if (N->getOperand(0).getOpcode() == ISD::SRL &&
393         (MC = dyn_cast<ConstantSDNode>(N->getOperand(1))) &&
394         (SC = dyn_cast<ConstantSDNode>(N->getOperand(0).getOperand(1)))) {
395       uint64_t sval = SC->getZExtValue();
396       uint64_t mval = MC->getZExtValue();
397       // If the result is a zap, let the autogened stuff handle it.
398       if (get_zapImm(N->getOperand(0), mval))
399         break;
400       // given mask X, and shift S, we want to see if there is any zap in the
401       // mask if we play around with the botton S bits
402       uint64_t dontcare = (~0ULL) >> (64 - sval);
403       uint64_t mask = mval << sval;
404       
405       if (get_zapImm(mask | dontcare))
406         mask = mask | dontcare;
407       
408       if (get_zapImm(mask)) {
409         SDValue Z = 
410           SDValue(CurDAG->getTargetNode(Alpha::ZAPNOTi, dl, MVT::i64,
411                                           N->getOperand(0).getOperand(0),
412                                           getI64Imm(get_zapImm(mask))), 0);
413         return CurDAG->getTargetNode(Alpha::SRLr, dl, MVT::i64, Z, 
414                                      getI64Imm(sval));
415       }
416     }
417     break;
418   }
419
420   }
421
422   return SelectCode(Op);
423 }
424
425 void AlphaDAGToDAGISel::SelectCALL(SDValue Op) {
426   //TODO: add flag stuff to prevent nondeturministic breakage!
427
428   SDNode *N = Op.getNode();
429   SDValue Chain = N->getOperand(0);
430   SDValue Addr = N->getOperand(1);
431   SDValue InFlag = N->getOperand(N->getNumOperands() - 1);
432   DebugLoc dl = N->getDebugLoc();
433
434    if (Addr.getOpcode() == AlphaISD::GPRelLo) {
435      SDValue GOT = SDValue(getGlobalBaseReg(), 0);
436      Chain = CurDAG->getCopyToReg(Chain, dl, Alpha::R29, GOT, InFlag);
437      InFlag = Chain.getValue(1);
438      Chain = SDValue(CurDAG->getTargetNode(Alpha::BSR, dl, MVT::Other, 
439                                            MVT::Flag, Addr.getOperand(0), 
440                                            Chain, InFlag), 0);
441    } else {
442      Chain = CurDAG->getCopyToReg(Chain, dl, Alpha::R27, Addr, InFlag);
443      InFlag = Chain.getValue(1);
444      Chain = SDValue(CurDAG->getTargetNode(Alpha::JSR, dl, MVT::Other,
445                                              MVT::Flag, Chain, InFlag), 0);
446    }
447    InFlag = Chain.getValue(1);
448
449   ReplaceUses(Op.getValue(0), Chain);
450   ReplaceUses(Op.getValue(1), InFlag);
451 }
452
453
454 /// createAlphaISelDag - This pass converts a legalized DAG into a 
455 /// Alpha-specific DAG, ready for instruction scheduling.
456 ///
457 FunctionPass *llvm::createAlphaISelDag(AlphaTargetMachine &TM) {
458   return new AlphaDAGToDAGISel(TM);
459 }