blob: 22d676e5df0998b834a90a8b9f2b2c99114e58fb [file] [log] [blame]
Chris Lattner7a125372005-11-16 22:59:19 +00001//===- X86ISelDAGToDAG.cpp - A DAG pattern matching inst selector for X86 -===//
Chris Lattnerc961eea2005-11-16 01:54:32 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattnerc961eea2005-11-16 01:54:32 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file defines a DAG pattern matching instruction selector for X86,
11// converting from a legalized dag to a X86 dag.
12//
13//===----------------------------------------------------------------------===//
14
Evan Cheng2ef88a02006-08-07 22:28:20 +000015#define DEBUG_TYPE "x86-isel"
Chris Lattnerc961eea2005-11-16 01:54:32 +000016#include "X86.h"
Evan Cheng8700e142006-01-11 06:09:51 +000017#include "X86InstrBuilder.h"
Evan Chengc4c62572006-03-13 23:20:37 +000018#include "X86ISelLowering.h"
Evan Cheng0475ab52008-01-05 00:41:47 +000019#include "X86MachineFunctionInfo.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000020#include "X86RegisterInfo.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000021#include "X86Subtarget.h"
Evan Chengc4c62572006-03-13 23:20:37 +000022#include "X86TargetMachine.h"
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000023#include "llvm/GlobalValue.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000024#include "llvm/Instructions.h"
Chris Lattner420736d2006-03-25 06:47:10 +000025#include "llvm/Intrinsics.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000026#include "llvm/Support/CFG.h"
Reid Spencer7aa8a452007-01-12 23:22:14 +000027#include "llvm/Type.h"
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000028#include "llvm/CodeGen/MachineConstantPool.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000029#include "llvm/CodeGen/MachineFunction.h"
Evan Chengaaca22c2006-01-10 20:26:56 +000030#include "llvm/CodeGen/MachineFrameInfo.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000031#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000032#include "llvm/CodeGen/MachineRegisterInfo.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000033#include "llvm/CodeGen/SelectionDAGISel.h"
34#include "llvm/Target/TargetMachine.h"
Evan Chengb7a75a52008-09-26 23:41:32 +000035#include "llvm/Target/TargetOptions.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000036#include "llvm/Support/Compiler.h"
Evan Cheng25ab6902006-09-08 06:48:29 +000037#include "llvm/Support/Debug.h"
Torok Edwindac237e2009-07-08 20:53:28 +000038#include "llvm/Support/ErrorHandling.h"
Evan Cheng25ab6902006-09-08 06:48:29 +000039#include "llvm/Support/MathExtras.h"
Torok Edwindac237e2009-07-08 20:53:28 +000040#include "llvm/Support/raw_ostream.h"
Evan Chengcdda25d2008-04-25 08:22:20 +000041#include "llvm/ADT/SmallPtrSet.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000042#include "llvm/ADT/Statistic.h"
43using namespace llvm;
44
Evan Cheng4d952322009-03-31 01:13:53 +000045#include "llvm/Support/CommandLine.h"
46static cl::opt<bool> AvoidDupAddrCompute("x86-avoid-dup-address", cl::Hidden);
47
Chris Lattner95b2c7d2006-12-19 22:59:26 +000048STATISTIC(NumLoadMoved, "Number of loads moved below TokenFactor");
49
Chris Lattnerc961eea2005-11-16 01:54:32 +000050//===----------------------------------------------------------------------===//
51// Pattern Matcher Implementation
52//===----------------------------------------------------------------------===//
53
54namespace {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000055 /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses
Dan Gohman475871a2008-07-27 21:46:04 +000056 /// SDValue's instead of register numbers for the leaves of the matched
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000057 /// tree.
58 struct X86ISelAddressMode {
59 enum {
60 RegBase,
Chris Lattnerd74ea2b2006-05-24 17:04:05 +000061 FrameIndexBase
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000062 } BaseType;
63
64 struct { // This is really a union, discriminated by BaseType!
Dan Gohman475871a2008-07-27 21:46:04 +000065 SDValue Reg;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000066 int FrameIndex;
67 } Base;
68
69 unsigned Scale;
Dan Gohman475871a2008-07-27 21:46:04 +000070 SDValue IndexReg;
Dan Gohman27cae7b2008-11-11 15:52:29 +000071 int32_t Disp;
Rafael Espindola094fad32009-04-08 21:14:34 +000072 SDValue Segment;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000073 GlobalValue *GV;
Evan Cheng51a9ed92006-02-25 10:09:08 +000074 Constant *CP;
Evan Cheng25ab6902006-09-08 06:48:29 +000075 const char *ES;
76 int JT;
Evan Cheng51a9ed92006-02-25 10:09:08 +000077 unsigned Align; // CP alignment.
Chris Lattnerb8afeb92009-06-26 05:51:45 +000078 unsigned char SymbolFlags; // X86II::MO_*
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000079
80 X86ISelAddressMode()
Chris Lattner18c59872009-06-27 04:16:01 +000081 : BaseType(RegBase), Scale(1), IndexReg(), Disp(0),
Dan Gohman79b765d2009-08-25 17:47:44 +000082 Segment(), GV(0), CP(0), ES(0), JT(-1), Align(0),
83 SymbolFlags(X86II::MO_NO_FLAG) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000084 }
Dan Gohman2d0a1cc2009-02-07 00:43:41 +000085
86 bool hasSymbolicDisplacement() const {
87 return GV != 0 || CP != 0 || ES != 0 || JT != -1;
88 }
Chris Lattner18c59872009-06-27 04:16:01 +000089
90 bool hasBaseOrIndexReg() const {
91 return IndexReg.getNode() != 0 || Base.Reg.getNode() != 0;
92 }
93
94 /// isRIPRelative - Return true if this addressing mode is already RIP
95 /// relative.
96 bool isRIPRelative() const {
97 if (BaseType != RegBase) return false;
98 if (RegisterSDNode *RegNode =
99 dyn_cast_or_null<RegisterSDNode>(Base.Reg.getNode()))
100 return RegNode->getReg() == X86::RIP;
101 return false;
102 }
103
104 void setBaseReg(SDValue Reg) {
105 BaseType = RegBase;
106 Base.Reg = Reg;
107 }
Dan Gohman2d0a1cc2009-02-07 00:43:41 +0000108
Dale Johannesen50dd1d02008-08-11 23:46:25 +0000109 void dump() {
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000110 errs() << "X86ISelAddressMode " << this << '\n';
111 errs() << "Base.Reg ";
Bill Wendling12321672009-08-07 21:33:25 +0000112 if (Base.Reg.getNode() != 0)
113 Base.Reg.getNode()->dump();
114 else
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000115 errs() << "nul";
116 errs() << " Base.FrameIndex " << Base.FrameIndex << '\n'
117 << " Scale" << Scale << '\n'
118 << "IndexReg ";
Bill Wendling12321672009-08-07 21:33:25 +0000119 if (IndexReg.getNode() != 0)
120 IndexReg.getNode()->dump();
121 else
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000122 errs() << "nul";
123 errs() << " Disp " << Disp << '\n'
124 << "GV ";
Bill Wendling12321672009-08-07 21:33:25 +0000125 if (GV)
126 GV->dump();
127 else
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000128 errs() << "nul";
129 errs() << " CP ";
Bill Wendling12321672009-08-07 21:33:25 +0000130 if (CP)
131 CP->dump();
132 else
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000133 errs() << "nul";
134 errs() << '\n'
135 << "ES ";
Bill Wendling12321672009-08-07 21:33:25 +0000136 if (ES)
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000137 errs() << ES;
Bill Wendling12321672009-08-07 21:33:25 +0000138 else
Benjamin Kramer072a56e2009-08-23 11:52:17 +0000139 errs() << "nul";
140 errs() << " JT" << JT << " Align" << Align << '\n';
Dale Johannesen50dd1d02008-08-11 23:46:25 +0000141 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000142 };
143}
144
145namespace {
Chris Lattnerc961eea2005-11-16 01:54:32 +0000146 //===--------------------------------------------------------------------===//
147 /// ISel - X86 specific code to select X86 machine instructions for
148 /// SelectionDAG operations.
149 ///
Chris Lattner2c79de82006-06-28 23:27:49 +0000150 class VISIBILITY_HIDDEN X86DAGToDAGISel : public SelectionDAGISel {
Chris Lattnerc961eea2005-11-16 01:54:32 +0000151 /// X86Lowering - This object fully describes how to lower LLVM code to an
152 /// X86-specific SelectionDAG.
Dan Gohmanda8ac5f2008-10-03 16:55:19 +0000153 X86TargetLowering &X86Lowering;
Chris Lattnerc961eea2005-11-16 01:54:32 +0000154
155 /// Subtarget - Keep a pointer to the X86Subtarget around so that we can
156 /// make the right decision when generating code for different targets.
157 const X86Subtarget *Subtarget;
Evan Cheng7ccced62006-02-18 00:15:05 +0000158
Evan Chengb7a75a52008-09-26 23:41:32 +0000159 /// OptForSize - If true, selector should try to optimize for code size
160 /// instead of performance.
161 bool OptForSize;
162
Chris Lattnerc961eea2005-11-16 01:54:32 +0000163 public:
Bill Wendling98a366d2009-04-29 23:29:43 +0000164 explicit X86DAGToDAGISel(X86TargetMachine &tm, CodeGenOpt::Level OptLevel)
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000165 : SelectionDAGISel(tm, OptLevel),
Dan Gohmanc5534622009-06-03 20:20:00 +0000166 X86Lowering(*tm.getTargetLowering()),
167 Subtarget(&tm.getSubtarget<X86Subtarget>()),
Devang Patel4ae641f2008-10-01 23:18:38 +0000168 OptForSize(false) {}
Chris Lattnerc961eea2005-11-16 01:54:32 +0000169
170 virtual const char *getPassName() const {
171 return "X86 DAG->DAG Instruction Selection";
172 }
173
Evan Chengdb8d56b2008-06-30 20:45:06 +0000174 /// InstructionSelect - This callback is invoked by
Chris Lattnerc961eea2005-11-16 01:54:32 +0000175 /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
Dan Gohmanf350b272008-08-23 02:25:05 +0000176 virtual void InstructionSelect();
Evan Chengdb8d56b2008-06-30 20:45:06 +0000177
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000178 virtual void EmitFunctionEntryCode(Function &Fn, MachineFunction &MF);
179
Evan Cheng884c70c2008-11-27 00:49:46 +0000180 virtual
181 bool IsLegalAndProfitableToFold(SDNode *N, SDNode *U, SDNode *Root) const;
Evan Chenga8df1b42006-07-27 16:44:36 +0000182
Chris Lattnerc961eea2005-11-16 01:54:32 +0000183// Include the pieces autogenerated from the target description.
184#include "X86GenDAGISel.inc"
185
186 private:
Dan Gohman475871a2008-07-27 21:46:04 +0000187 SDNode *Select(SDValue N);
Dale Johannesen48c1bc22008-10-02 18:53:47 +0000188 SDNode *SelectAtomic64(SDNode *Node, unsigned Opc);
Owen Andersone50ed302009-08-10 22:56:29 +0000189 SDNode *SelectAtomicLoadAdd(SDNode *Node, EVT NVT);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000190
Rafael Espindola094fad32009-04-08 21:14:34 +0000191 bool MatchSegmentBaseAddress(SDValue N, X86ISelAddressMode &AM);
192 bool MatchLoad(SDValue N, X86ISelAddressMode &AM);
Rafael Espindola49a168d2009-04-12 21:55:03 +0000193 bool MatchWrapper(SDValue N, X86ISelAddressMode &AM);
Dan Gohman41d0b9d2009-07-22 23:26:55 +0000194 bool MatchAddress(SDValue N, X86ISelAddressMode &AM);
195 bool MatchAddressRecursively(SDValue N, X86ISelAddressMode &AM,
196 unsigned Depth);
Rafael Espindola523249f2009-03-31 16:16:57 +0000197 bool MatchAddressBase(SDValue N, X86ISelAddressMode &AM);
Dan Gohman475871a2008-07-27 21:46:04 +0000198 bool SelectAddr(SDValue Op, SDValue N, SDValue &Base,
Rafael Espindola094fad32009-04-08 21:14:34 +0000199 SDValue &Scale, SDValue &Index, SDValue &Disp,
200 SDValue &Segment);
Dan Gohman475871a2008-07-27 21:46:04 +0000201 bool SelectLEAAddr(SDValue Op, SDValue N, SDValue &Base,
202 SDValue &Scale, SDValue &Index, SDValue &Disp);
Chris Lattner5c0b16d2009-06-20 20:38:48 +0000203 bool SelectTLSADDRAddr(SDValue Op, SDValue N, SDValue &Base,
204 SDValue &Scale, SDValue &Index, SDValue &Disp);
Dan Gohman475871a2008-07-27 21:46:04 +0000205 bool SelectScalarSSELoad(SDValue Op, SDValue Pred,
206 SDValue N, SDValue &Base, SDValue &Scale,
207 SDValue &Index, SDValue &Disp,
Rafael Espindola094fad32009-04-08 21:14:34 +0000208 SDValue &Segment,
Dan Gohman475871a2008-07-27 21:46:04 +0000209 SDValue &InChain, SDValue &OutChain);
210 bool TryFoldLoad(SDValue P, SDValue N,
211 SDValue &Base, SDValue &Scale,
Rafael Espindola094fad32009-04-08 21:14:34 +0000212 SDValue &Index, SDValue &Disp,
213 SDValue &Segment);
Dan Gohmanf350b272008-08-23 02:25:05 +0000214 void PreprocessForRMW();
215 void PreprocessForFPConvert();
Evan Cheng2ef88a02006-08-07 22:28:20 +0000216
Chris Lattnerc0bad572006-06-08 18:03:49 +0000217 /// SelectInlineAsmMemoryOperand - Implement addressing mode selection for
218 /// inline asm expressions.
Dan Gohman475871a2008-07-27 21:46:04 +0000219 virtual bool SelectInlineAsmMemoryOperand(const SDValue &Op,
Chris Lattnerc0bad572006-06-08 18:03:49 +0000220 char ConstraintCode,
Dan Gohmanf350b272008-08-23 02:25:05 +0000221 std::vector<SDValue> &OutOps);
Chris Lattnerc0bad572006-06-08 18:03:49 +0000222
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000223 void EmitSpecialCodeForMain(MachineBasicBlock *BB, MachineFrameInfo *MFI);
224
Dan Gohman475871a2008-07-27 21:46:04 +0000225 inline void getAddressOperands(X86ISelAddressMode &AM, SDValue &Base,
226 SDValue &Scale, SDValue &Index,
Rafael Espindola094fad32009-04-08 21:14:34 +0000227 SDValue &Disp, SDValue &Segment) {
Evan Chenge5280532005-12-12 21:49:40 +0000228 Base = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) ?
Evan Cheng25ab6902006-09-08 06:48:29 +0000229 CurDAG->getTargetFrameIndex(AM.Base.FrameIndex, TLI.getPointerTy()) :
230 AM.Base.Reg;
Evan Chengbdce7b42005-12-17 09:13:43 +0000231 Scale = getI8Imm(AM.Scale);
Evan Chenge5280532005-12-12 21:49:40 +0000232 Index = AM.IndexReg;
Evan Cheng25ab6902006-09-08 06:48:29 +0000233 // These are 32-bit even in 64-bit mode since RIP relative offset
234 // is 32-bit.
235 if (AM.GV)
Owen Anderson825b72b2009-08-11 20:47:22 +0000236 Disp = CurDAG->getTargetGlobalAddress(AM.GV, MVT::i32, AM.Disp,
Chris Lattnerb8afeb92009-06-26 05:51:45 +0000237 AM.SymbolFlags);
Evan Cheng25ab6902006-09-08 06:48:29 +0000238 else if (AM.CP)
Owen Anderson825b72b2009-08-11 20:47:22 +0000239 Disp = CurDAG->getTargetConstantPool(AM.CP, MVT::i32,
Chris Lattnerb8afeb92009-06-26 05:51:45 +0000240 AM.Align, AM.Disp, AM.SymbolFlags);
Evan Cheng25ab6902006-09-08 06:48:29 +0000241 else if (AM.ES)
Owen Anderson825b72b2009-08-11 20:47:22 +0000242 Disp = CurDAG->getTargetExternalSymbol(AM.ES, MVT::i32, AM.SymbolFlags);
Evan Cheng25ab6902006-09-08 06:48:29 +0000243 else if (AM.JT != -1)
Owen Anderson825b72b2009-08-11 20:47:22 +0000244 Disp = CurDAG->getTargetJumpTable(AM.JT, MVT::i32, AM.SymbolFlags);
Evan Cheng25ab6902006-09-08 06:48:29 +0000245 else
Owen Anderson825b72b2009-08-11 20:47:22 +0000246 Disp = CurDAG->getTargetConstant(AM.Disp, MVT::i32);
Rafael Espindola094fad32009-04-08 21:14:34 +0000247
248 if (AM.Segment.getNode())
249 Segment = AM.Segment;
250 else
Owen Anderson825b72b2009-08-11 20:47:22 +0000251 Segment = CurDAG->getRegister(0, MVT::i32);
Evan Chenge5280532005-12-12 21:49:40 +0000252 }
253
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000254 /// getI8Imm - Return a target constant with the specified value, of type
255 /// i8.
Dan Gohman475871a2008-07-27 21:46:04 +0000256 inline SDValue getI8Imm(unsigned Imm) {
Owen Anderson825b72b2009-08-11 20:47:22 +0000257 return CurDAG->getTargetConstant(Imm, MVT::i8);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000258 }
259
Chris Lattnerc961eea2005-11-16 01:54:32 +0000260 /// getI16Imm - Return a target constant with the specified value, of type
261 /// i16.
Dan Gohman475871a2008-07-27 21:46:04 +0000262 inline SDValue getI16Imm(unsigned Imm) {
Owen Anderson825b72b2009-08-11 20:47:22 +0000263 return CurDAG->getTargetConstant(Imm, MVT::i16);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000264 }
265
266 /// getI32Imm - Return a target constant with the specified value, of type
267 /// i32.
Dan Gohman475871a2008-07-27 21:46:04 +0000268 inline SDValue getI32Imm(unsigned Imm) {
Owen Anderson825b72b2009-08-11 20:47:22 +0000269 return CurDAG->getTargetConstant(Imm, MVT::i32);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000270 }
Evan Chengf597dc72006-02-10 22:24:32 +0000271
Dan Gohman8b746962008-09-23 18:22:58 +0000272 /// getGlobalBaseReg - Return an SDNode that returns the value of
273 /// the global base register. Output instructions required to
274 /// initialize the global base register, if necessary.
275 ///
Evan Cheng9ade2182006-08-26 05:34:46 +0000276 SDNode *getGlobalBaseReg();
Evan Cheng7ccced62006-02-18 00:15:05 +0000277
Dan Gohmanc5534622009-06-03 20:20:00 +0000278 /// getTargetMachine - Return a reference to the TargetMachine, casted
279 /// to the target-specific type.
280 const X86TargetMachine &getTargetMachine() {
281 return static_cast<const X86TargetMachine &>(TM);
282 }
283
284 /// getInstrInfo - Return a reference to the TargetInstrInfo, casted
285 /// to the target-specific type.
286 const X86InstrInfo *getInstrInfo() {
287 return getTargetMachine().getInstrInfo();
288 }
289
Evan Cheng23addc02006-02-10 22:46:26 +0000290#ifndef NDEBUG
291 unsigned Indent;
292#endif
Chris Lattnerc961eea2005-11-16 01:54:32 +0000293 };
294}
295
Evan Chengf4b4c412006-08-08 00:31:00 +0000296
Evan Cheng884c70c2008-11-27 00:49:46 +0000297bool X86DAGToDAGISel::IsLegalAndProfitableToFold(SDNode *N, SDNode *U,
298 SDNode *Root) const {
Bill Wendling98a366d2009-04-29 23:29:43 +0000299 if (OptLevel == CodeGenOpt::None) return false;
Evan Cheng27e1fe92006-10-14 08:33:25 +0000300
Evan Cheng884c70c2008-11-27 00:49:46 +0000301 if (U == Root)
302 switch (U->getOpcode()) {
303 default: break;
304 case ISD::ADD:
305 case ISD::ADDC:
306 case ISD::ADDE:
307 case ISD::AND:
308 case ISD::OR:
309 case ISD::XOR: {
Rafael Espindoladbcfb302009-04-10 10:09:34 +0000310 SDValue Op1 = U->getOperand(1);
311
Evan Cheng884c70c2008-11-27 00:49:46 +0000312 // If the other operand is a 8-bit immediate we should fold the immediate
313 // instead. This reduces code size.
314 // e.g.
315 // movl 4(%esp), %eax
316 // addl $4, %eax
317 // vs.
318 // movl $4, %eax
319 // addl 4(%esp), %eax
320 // The former is 2 bytes shorter. In case where the increment is 1, then
321 // the saving can be 4 bytes (by using incl %eax).
Rafael Espindoladbcfb302009-04-10 10:09:34 +0000322 if (ConstantSDNode *Imm = dyn_cast<ConstantSDNode>(Op1))
Dan Gohman9a49d312009-03-14 02:07:16 +0000323 if (Imm->getAPIntValue().isSignedIntN(8))
324 return false;
Rafael Espindoladbcfb302009-04-10 10:09:34 +0000325
326 // If the other operand is a TLS address, we should fold it instead.
327 // This produces
328 // movl %gs:0, %eax
329 // leal i@NTPOFF(%eax), %eax
330 // instead of
331 // movl $i@NTPOFF, %eax
332 // addl %gs:0, %eax
333 // if the block also has an access to a second TLS address this will save
334 // a load.
335 // FIXME: This is probably also true for non TLS addresses.
336 if (Op1.getOpcode() == X86ISD::Wrapper) {
337 SDValue Val = Op1.getOperand(0);
338 if (Val.getOpcode() == ISD::TargetGlobalTLSAddress)
339 return false;
340 }
Evan Cheng884c70c2008-11-27 00:49:46 +0000341 }
342 }
343
Anton Korobeynikovc1c6ef82009-05-08 18:51:58 +0000344 // Proceed to 'generic' cycle finder code
345 return SelectionDAGISel::IsLegalAndProfitableToFold(N, U, Root);
Evan Chenga8df1b42006-07-27 16:44:36 +0000346}
347
Evan Cheng70e674e2006-08-28 20:10:17 +0000348/// MoveBelowTokenFactor - Replace TokenFactor operand with load's chain operand
349/// and move load below the TokenFactor. Replace store's chain operand with
350/// load's chain result.
Dan Gohmanf350b272008-08-23 02:25:05 +0000351static void MoveBelowTokenFactor(SelectionDAG *CurDAG, SDValue Load,
Dan Gohman475871a2008-07-27 21:46:04 +0000352 SDValue Store, SDValue TF) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000353 SmallVector<SDValue, 4> Ops;
Gabor Greifba36cb52008-08-28 21:40:38 +0000354 for (unsigned i = 0, e = TF.getNode()->getNumOperands(); i != e; ++i)
355 if (Load.getNode() == TF.getOperand(i).getNode())
Evan Chengab6c3bb2008-08-25 21:27:18 +0000356 Ops.push_back(Load.getOperand(0));
Evan Cheng70e674e2006-08-28 20:10:17 +0000357 else
Evan Chengab6c3bb2008-08-25 21:27:18 +0000358 Ops.push_back(TF.getOperand(i));
Dan Gohmanaae317a2009-08-06 09:22:57 +0000359 SDValue NewTF = CurDAG->UpdateNodeOperands(TF, &Ops[0], Ops.size());
360 SDValue NewLoad = CurDAG->UpdateNodeOperands(Load, NewTF,
361 Load.getOperand(1),
362 Load.getOperand(2));
363 CurDAG->UpdateNodeOperands(Store, NewLoad.getValue(1), Store.getOperand(1),
Dan Gohmanf350b272008-08-23 02:25:05 +0000364 Store.getOperand(2), Store.getOperand(3));
Evan Cheng70e674e2006-08-28 20:10:17 +0000365}
366
Evan Chengcd0baf22008-05-23 21:23:16 +0000367/// isRMWLoad - Return true if N is a load that's part of RMW sub-DAG.
368///
Dan Gohman475871a2008-07-27 21:46:04 +0000369static bool isRMWLoad(SDValue N, SDValue Chain, SDValue Address,
370 SDValue &Load) {
Evan Chengcd0baf22008-05-23 21:23:16 +0000371 if (N.getOpcode() == ISD::BIT_CONVERT)
372 N = N.getOperand(0);
373
374 LoadSDNode *LD = dyn_cast<LoadSDNode>(N);
375 if (!LD || LD->isVolatile())
376 return false;
377 if (LD->getAddressingMode() != ISD::UNINDEXED)
378 return false;
379
380 ISD::LoadExtType ExtType = LD->getExtensionType();
381 if (ExtType != ISD::NON_EXTLOAD && ExtType != ISD::EXTLOAD)
382 return false;
383
384 if (N.hasOneUse() &&
385 N.getOperand(1) == Address &&
Gabor Greifba36cb52008-08-28 21:40:38 +0000386 N.getNode()->isOperandOf(Chain.getNode())) {
Evan Chengcd0baf22008-05-23 21:23:16 +0000387 Load = N;
388 return true;
389 }
390 return false;
391}
392
Evan Chengab6c3bb2008-08-25 21:27:18 +0000393/// MoveBelowCallSeqStart - Replace CALLSEQ_START operand with load's chain
394/// operand and move load below the call's chain operand.
395static void MoveBelowCallSeqStart(SelectionDAG *CurDAG, SDValue Load,
Evan Cheng5b2e5892009-01-26 18:43:34 +0000396 SDValue Call, SDValue CallSeqStart) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000397 SmallVector<SDValue, 8> Ops;
Evan Cheng5b2e5892009-01-26 18:43:34 +0000398 SDValue Chain = CallSeqStart.getOperand(0);
399 if (Chain.getNode() == Load.getNode())
400 Ops.push_back(Load.getOperand(0));
401 else {
402 assert(Chain.getOpcode() == ISD::TokenFactor &&
403 "Unexpected CallSeqStart chain operand");
404 for (unsigned i = 0, e = Chain.getNumOperands(); i != e; ++i)
405 if (Chain.getOperand(i).getNode() == Load.getNode())
406 Ops.push_back(Load.getOperand(0));
407 else
408 Ops.push_back(Chain.getOperand(i));
409 SDValue NewChain =
Dale Johannesened2eee62009-02-06 01:31:28 +0000410 CurDAG->getNode(ISD::TokenFactor, Load.getDebugLoc(),
Owen Anderson825b72b2009-08-11 20:47:22 +0000411 MVT::Other, &Ops[0], Ops.size());
Evan Cheng5b2e5892009-01-26 18:43:34 +0000412 Ops.clear();
413 Ops.push_back(NewChain);
414 }
415 for (unsigned i = 1, e = CallSeqStart.getNumOperands(); i != e; ++i)
416 Ops.push_back(CallSeqStart.getOperand(i));
417 CurDAG->UpdateNodeOperands(CallSeqStart, &Ops[0], Ops.size());
Evan Chengab6c3bb2008-08-25 21:27:18 +0000418 CurDAG->UpdateNodeOperands(Load, Call.getOperand(0),
419 Load.getOperand(1), Load.getOperand(2));
420 Ops.clear();
Gabor Greifba36cb52008-08-28 21:40:38 +0000421 Ops.push_back(SDValue(Load.getNode(), 1));
422 for (unsigned i = 1, e = Call.getNode()->getNumOperands(); i != e; ++i)
Evan Chengab6c3bb2008-08-25 21:27:18 +0000423 Ops.push_back(Call.getOperand(i));
424 CurDAG->UpdateNodeOperands(Call, &Ops[0], Ops.size());
425}
426
427/// isCalleeLoad - Return true if call address is a load and it can be
428/// moved below CALLSEQ_START and the chains leading up to the call.
429/// Return the CALLSEQ_START by reference as a second output.
430static bool isCalleeLoad(SDValue Callee, SDValue &Chain) {
Gabor Greifba36cb52008-08-28 21:40:38 +0000431 if (Callee.getNode() == Chain.getNode() || !Callee.hasOneUse())
Evan Chengab6c3bb2008-08-25 21:27:18 +0000432 return false;
Gabor Greifba36cb52008-08-28 21:40:38 +0000433 LoadSDNode *LD = dyn_cast<LoadSDNode>(Callee.getNode());
Evan Chengab6c3bb2008-08-25 21:27:18 +0000434 if (!LD ||
435 LD->isVolatile() ||
436 LD->getAddressingMode() != ISD::UNINDEXED ||
437 LD->getExtensionType() != ISD::NON_EXTLOAD)
438 return false;
439
440 // Now let's find the callseq_start.
441 while (Chain.getOpcode() != ISD::CALLSEQ_START) {
442 if (!Chain.hasOneUse())
443 return false;
444 Chain = Chain.getOperand(0);
445 }
Evan Cheng5b2e5892009-01-26 18:43:34 +0000446
447 if (Chain.getOperand(0).getNode() == Callee.getNode())
448 return true;
449 if (Chain.getOperand(0).getOpcode() == ISD::TokenFactor &&
450 Callee.getValue(1).isOperandOf(Chain.getOperand(0).getNode()))
451 return true;
452 return false;
Evan Chengab6c3bb2008-08-25 21:27:18 +0000453}
454
455
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000456/// PreprocessForRMW - Preprocess the DAG to make instruction selection better.
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000457/// This is only run if not in -O0 mode.
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000458/// This allows the instruction selector to pick more read-modify-write
459/// instructions. This is a common case:
Evan Cheng70e674e2006-08-28 20:10:17 +0000460///
461/// [Load chain]
462/// ^
463/// |
464/// [Load]
465/// ^ ^
466/// | |
467/// / \-
468/// / |
469/// [TokenFactor] [Op]
470/// ^ ^
471/// | |
472/// \ /
473/// \ /
474/// [Store]
475///
476/// The fact the store's chain operand != load's chain will prevent the
477/// (store (op (load))) instruction from being selected. We can transform it to:
478///
479/// [Load chain]
480/// ^
481/// |
482/// [TokenFactor]
483/// ^
484/// |
485/// [Load]
486/// ^ ^
487/// | |
488/// | \-
489/// | |
490/// | [Op]
491/// | ^
492/// | |
493/// \ /
494/// \ /
495/// [Store]
Dan Gohmanf350b272008-08-23 02:25:05 +0000496void X86DAGToDAGISel::PreprocessForRMW() {
497 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
498 E = CurDAG->allnodes_end(); I != E; ++I) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000499 if (I->getOpcode() == X86ISD::CALL) {
500 /// Also try moving call address load from outside callseq_start to just
501 /// before the call to allow it to be folded.
502 ///
503 /// [Load chain]
504 /// ^
505 /// |
506 /// [Load]
507 /// ^ ^
508 /// | |
509 /// / \--
510 /// / |
511 ///[CALLSEQ_START] |
512 /// ^ |
513 /// | |
514 /// [LOAD/C2Reg] |
515 /// | |
516 /// \ /
517 /// \ /
518 /// [CALL]
519 SDValue Chain = I->getOperand(0);
520 SDValue Load = I->getOperand(1);
521 if (!isCalleeLoad(Load, Chain))
522 continue;
523 MoveBelowCallSeqStart(CurDAG, Load, SDValue(I, 0), Chain);
524 ++NumLoadMoved;
525 continue;
526 }
527
Evan Cheng8b2794a2006-10-13 21:14:26 +0000528 if (!ISD::isNON_TRUNCStore(I))
Evan Cheng70e674e2006-08-28 20:10:17 +0000529 continue;
Dan Gohman475871a2008-07-27 21:46:04 +0000530 SDValue Chain = I->getOperand(0);
Evan Chengab6c3bb2008-08-25 21:27:18 +0000531
Gabor Greifba36cb52008-08-28 21:40:38 +0000532 if (Chain.getNode()->getOpcode() != ISD::TokenFactor)
Evan Cheng70e674e2006-08-28 20:10:17 +0000533 continue;
534
Dan Gohman475871a2008-07-27 21:46:04 +0000535 SDValue N1 = I->getOperand(1);
536 SDValue N2 = I->getOperand(2);
Duncan Sands83ec4b62008-06-06 12:08:01 +0000537 if ((N1.getValueType().isFloatingPoint() &&
538 !N1.getValueType().isVector()) ||
Evan Cheng780413d2006-08-29 18:37:37 +0000539 !N1.hasOneUse())
Evan Cheng70e674e2006-08-28 20:10:17 +0000540 continue;
541
542 bool RModW = false;
Dan Gohman475871a2008-07-27 21:46:04 +0000543 SDValue Load;
Gabor Greifba36cb52008-08-28 21:40:38 +0000544 unsigned Opcode = N1.getNode()->getOpcode();
Evan Cheng70e674e2006-08-28 20:10:17 +0000545 switch (Opcode) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000546 case ISD::ADD:
547 case ISD::MUL:
548 case ISD::AND:
549 case ISD::OR:
550 case ISD::XOR:
551 case ISD::ADDC:
552 case ISD::ADDE:
553 case ISD::VECTOR_SHUFFLE: {
554 SDValue N10 = N1.getOperand(0);
555 SDValue N11 = N1.getOperand(1);
556 RModW = isRMWLoad(N10, Chain, N2, Load);
557 if (!RModW)
558 RModW = isRMWLoad(N11, Chain, N2, Load);
559 break;
560 }
561 case ISD::SUB:
562 case ISD::SHL:
563 case ISD::SRA:
564 case ISD::SRL:
565 case ISD::ROTL:
566 case ISD::ROTR:
567 case ISD::SUBC:
568 case ISD::SUBE:
569 case X86ISD::SHLD:
570 case X86ISD::SHRD: {
571 SDValue N10 = N1.getOperand(0);
572 RModW = isRMWLoad(N10, Chain, N2, Load);
573 break;
574 }
Evan Cheng70e674e2006-08-28 20:10:17 +0000575 }
576
Evan Cheng82a35b32006-08-29 06:44:17 +0000577 if (RModW) {
Dan Gohmanf350b272008-08-23 02:25:05 +0000578 MoveBelowTokenFactor(CurDAG, Load, SDValue(I, 0), Chain);
Evan Cheng82a35b32006-08-29 06:44:17 +0000579 ++NumLoadMoved;
580 }
Evan Cheng70e674e2006-08-28 20:10:17 +0000581 }
582}
583
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000584
585/// PreprocessForFPConvert - Walk over the dag lowering fpround and fpextend
586/// nodes that target the FP stack to be store and load to the stack. This is a
587/// gross hack. We would like to simply mark these as being illegal, but when
588/// we do that, legalize produces these when it expands calls, then expands
589/// these in the same legalize pass. We would like dag combine to be able to
590/// hack on these between the call expansion and the node legalization. As such
591/// this pass basically does "really late" legalization of these inline with the
592/// X86 isel pass.
Dan Gohmanf350b272008-08-23 02:25:05 +0000593void X86DAGToDAGISel::PreprocessForFPConvert() {
594 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
595 E = CurDAG->allnodes_end(); I != E; ) {
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000596 SDNode *N = I++; // Preincrement iterator to avoid invalidation issues.
597 if (N->getOpcode() != ISD::FP_ROUND && N->getOpcode() != ISD::FP_EXTEND)
598 continue;
599
600 // If the source and destination are SSE registers, then this is a legal
601 // conversion that should not be lowered.
Owen Andersone50ed302009-08-10 22:56:29 +0000602 EVT SrcVT = N->getOperand(0).getValueType();
603 EVT DstVT = N->getValueType(0);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000604 bool SrcIsSSE = X86Lowering.isScalarFPTypeInSSEReg(SrcVT);
605 bool DstIsSSE = X86Lowering.isScalarFPTypeInSSEReg(DstVT);
606 if (SrcIsSSE && DstIsSSE)
607 continue;
608
Chris Lattner6fa2f9c2008-03-09 07:05:32 +0000609 if (!SrcIsSSE && !DstIsSSE) {
610 // If this is an FPStack extension, it is a noop.
611 if (N->getOpcode() == ISD::FP_EXTEND)
612 continue;
613 // If this is a value-preserving FPStack truncation, it is a noop.
614 if (N->getConstantOperandVal(1))
615 continue;
616 }
617
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000618 // Here we could have an FP stack truncation or an FPStack <-> SSE convert.
619 // FPStack has extload and truncstore. SSE can fold direct loads into other
620 // operations. Based on this, decide what we want to do.
Owen Andersone50ed302009-08-10 22:56:29 +0000621 EVT MemVT;
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000622 if (N->getOpcode() == ISD::FP_ROUND)
623 MemVT = DstVT; // FP_ROUND must use DstVT, we can't do a 'trunc load'.
624 else
625 MemVT = SrcIsSSE ? SrcVT : DstVT;
626
Dan Gohmanf350b272008-08-23 02:25:05 +0000627 SDValue MemTmp = CurDAG->CreateStackTemporary(MemVT);
Dale Johannesend8392542009-02-03 21:48:12 +0000628 DebugLoc dl = N->getDebugLoc();
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000629
630 // FIXME: optimize the case where the src/dest is a load or store?
Dale Johannesend8392542009-02-03 21:48:12 +0000631 SDValue Store = CurDAG->getTruncStore(CurDAG->getEntryNode(), dl,
Dan Gohmanf350b272008-08-23 02:25:05 +0000632 N->getOperand(0),
633 MemTmp, NULL, 0, MemVT);
Dale Johannesend8392542009-02-03 21:48:12 +0000634 SDValue Result = CurDAG->getExtLoad(ISD::EXTLOAD, dl, DstVT, Store, MemTmp,
Dan Gohmanf350b272008-08-23 02:25:05 +0000635 NULL, 0, MemVT);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000636
637 // We're about to replace all uses of the FP_ROUND/FP_EXTEND with the
638 // extload we created. This will cause general havok on the dag because
639 // anything below the conversion could be folded into other existing nodes.
640 // To avoid invalidating 'I', back it up to the convert node.
641 --I;
Dan Gohmanf350b272008-08-23 02:25:05 +0000642 CurDAG->ReplaceAllUsesOfValueWith(SDValue(N, 0), Result);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000643
644 // Now that we did that, the node is dead. Increment the iterator to the
645 // next node to process, then delete N.
646 ++I;
Dan Gohmanf350b272008-08-23 02:25:05 +0000647 CurDAG->DeleteNode(N);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000648 }
649}
650
Chris Lattnerc961eea2005-11-16 01:54:32 +0000651/// InstructionSelectBasicBlock - This callback is invoked by SelectionDAGISel
652/// when it has created a SelectionDAG for us to codegen.
Dan Gohmanf350b272008-08-23 02:25:05 +0000653void X86DAGToDAGISel::InstructionSelect() {
Dan Gohman7571eb52009-08-01 03:42:59 +0000654 const Function *F = MF->getFunction();
Devang Patele76225a2008-10-06 18:03:39 +0000655 OptForSize = F->hasFnAttr(Attribute::OptimizeForSize);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000656
Evan Chengdb8d56b2008-06-30 20:45:06 +0000657 DEBUG(BB->dump());
Bill Wendling98a366d2009-04-29 23:29:43 +0000658 if (OptLevel != CodeGenOpt::None)
Dan Gohmanf350b272008-08-23 02:25:05 +0000659 PreprocessForRMW();
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000660
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +0000661 // FIXME: This should only happen when not compiled with -O0.
Dan Gohmanf350b272008-08-23 02:25:05 +0000662 PreprocessForFPConvert();
Evan Cheng70e674e2006-08-28 20:10:17 +0000663
Chris Lattnerc961eea2005-11-16 01:54:32 +0000664 // Codegen the basic block.
Evan Chengf597dc72006-02-10 22:24:32 +0000665#ifndef NDEBUG
Bill Wendling0ea8bf32009-08-03 00:11:34 +0000666 DEBUG(errs() << "===== Instruction selection begins:\n");
Evan Cheng23addc02006-02-10 22:46:26 +0000667 Indent = 0;
Evan Chengf597dc72006-02-10 22:24:32 +0000668#endif
David Greene8ad4c002008-10-27 21:56:29 +0000669 SelectRoot(*CurDAG);
Evan Chengf597dc72006-02-10 22:24:32 +0000670#ifndef NDEBUG
Bill Wendling0ea8bf32009-08-03 00:11:34 +0000671 DEBUG(errs() << "===== Instruction selection ends:\n");
Evan Chengf597dc72006-02-10 22:24:32 +0000672#endif
Evan Cheng63ce5682006-07-28 00:10:59 +0000673
Dan Gohmanf350b272008-08-23 02:25:05 +0000674 CurDAG->RemoveDeadNodes();
Evan Chengdb8d56b2008-06-30 20:45:06 +0000675}
Chris Lattnerc961eea2005-11-16 01:54:32 +0000676
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000677/// EmitSpecialCodeForMain - Emit any code that needs to be executed only in
678/// the main function.
679void X86DAGToDAGISel::EmitSpecialCodeForMain(MachineBasicBlock *BB,
680 MachineFrameInfo *MFI) {
681 const TargetInstrInfo *TII = TM.getInstrInfo();
682 if (Subtarget->isTargetCygMing())
Dale Johannesen8d13f8f2009-02-13 02:33:27 +0000683 BuildMI(BB, DebugLoc::getUnknownLoc(),
684 TII->get(X86::CALLpcrel32)).addExternalSymbol("__main");
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000685}
686
687void X86DAGToDAGISel::EmitFunctionEntryCode(Function &Fn, MachineFunction &MF) {
688 // If this is main, emit special code for main.
689 MachineBasicBlock *BB = MF.begin();
690 if (Fn.hasExternalLinkage() && Fn.getName() == "main")
691 EmitSpecialCodeForMain(BB, MF.getFrameInfo());
692}
693
Rafael Espindola094fad32009-04-08 21:14:34 +0000694
695bool X86DAGToDAGISel::MatchSegmentBaseAddress(SDValue N,
696 X86ISelAddressMode &AM) {
697 assert(N.getOpcode() == X86ISD::SegmentBaseAddress);
698 SDValue Segment = N.getOperand(0);
699
700 if (AM.Segment.getNode() == 0) {
701 AM.Segment = Segment;
702 return false;
703 }
704
705 return true;
706}
707
708bool X86DAGToDAGISel::MatchLoad(SDValue N, X86ISelAddressMode &AM) {
709 // This optimization is valid because the GNU TLS model defines that
710 // gs:0 (or fs:0 on X86-64) contains its own address.
711 // For more information see http://people.redhat.com/drepper/tls.pdf
712
713 SDValue Address = N.getOperand(1);
714 if (Address.getOpcode() == X86ISD::SegmentBaseAddress &&
715 !MatchSegmentBaseAddress (Address, AM))
716 return false;
717
718 return true;
719}
720
Chris Lattner18c59872009-06-27 04:16:01 +0000721/// MatchWrapper - Try to match X86ISD::Wrapper and X86ISD::WrapperRIP nodes
722/// into an addressing mode. These wrap things that will resolve down into a
723/// symbol reference. If no match is possible, this returns true, otherwise it
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000724/// returns false.
Rafael Espindola49a168d2009-04-12 21:55:03 +0000725bool X86DAGToDAGISel::MatchWrapper(SDValue N, X86ISelAddressMode &AM) {
Chris Lattner18c59872009-06-27 04:16:01 +0000726 // If the addressing mode already has a symbol as the displacement, we can
727 // never match another symbol.
Rafael Espindola49a168d2009-04-12 21:55:03 +0000728 if (AM.hasSymbolicDisplacement())
729 return true;
Rafael Espindola49a168d2009-04-12 21:55:03 +0000730
731 SDValue N0 = N.getOperand(0);
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000732 CodeModel::Model M = TM.getCodeModel();
733
Chris Lattner18c59872009-06-27 04:16:01 +0000734 // Handle X86-64 rip-relative addresses. We check this before checking direct
735 // folding because RIP is preferable to non-RIP accesses.
736 if (Subtarget->is64Bit() &&
737 // Under X86-64 non-small code model, GV (and friends) are 64-bits, so
738 // they cannot be folded into immediate fields.
739 // FIXME: This can be improved for kernel and other models?
Anton Korobeynikov25f1aa02009-08-21 15:41:56 +0000740 (M == CodeModel::Small || M == CodeModel::Kernel) &&
Chris Lattner18c59872009-06-27 04:16:01 +0000741 // Base and index reg must be 0 in order to use %rip as base and lowering
742 // must allow RIP.
743 !AM.hasBaseOrIndexReg() && N.getOpcode() == X86ISD::WrapperRIP) {
Chris Lattner18c59872009-06-27 04:16:01 +0000744 if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(N0)) {
745 int64_t Offset = AM.Disp + G->getOffset();
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000746 if (!X86::isOffsetSuitableForCodeModel(Offset, M)) return true;
Chris Lattner18c59872009-06-27 04:16:01 +0000747 AM.GV = G->getGlobal();
748 AM.Disp = Offset;
Chris Lattnerb8afeb92009-06-26 05:51:45 +0000749 AM.SymbolFlags = G->getTargetFlags();
Chris Lattner18c59872009-06-27 04:16:01 +0000750 } else if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N0)) {
751 int64_t Offset = AM.Disp + CP->getOffset();
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000752 if (!X86::isOffsetSuitableForCodeModel(Offset, M)) return true;
Rafael Espindola49a168d2009-04-12 21:55:03 +0000753 AM.CP = CP->getConstVal();
754 AM.Align = CP->getAlignment();
Chris Lattner18c59872009-06-27 04:16:01 +0000755 AM.Disp = Offset;
Chris Lattner0b0deab2009-06-26 05:56:49 +0000756 AM.SymbolFlags = CP->getTargetFlags();
Chris Lattner18c59872009-06-27 04:16:01 +0000757 } else if (ExternalSymbolSDNode *S = dyn_cast<ExternalSymbolSDNode>(N0)) {
758 AM.ES = S->getSymbol();
759 AM.SymbolFlags = S->getTargetFlags();
760 } else {
761 JumpTableSDNode *J = cast<JumpTableSDNode>(N0);
762 AM.JT = J->getIndex();
763 AM.SymbolFlags = J->getTargetFlags();
Rafael Espindola49a168d2009-04-12 21:55:03 +0000764 }
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000765
Chris Lattner18c59872009-06-27 04:16:01 +0000766 if (N.getOpcode() == X86ISD::WrapperRIP)
Owen Anderson825b72b2009-08-11 20:47:22 +0000767 AM.setBaseReg(CurDAG->getRegister(X86::RIP, MVT::i64));
Rafael Espindola49a168d2009-04-12 21:55:03 +0000768 return false;
Chris Lattner18c59872009-06-27 04:16:01 +0000769 }
770
771 // Handle the case when globals fit in our immediate field: This is true for
772 // X86-32 always and X86-64 when in -static -mcmodel=small mode. In 64-bit
773 // mode, this results in a non-RIP-relative computation.
774 if (!Subtarget->is64Bit() ||
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000775 ((M == CodeModel::Small || M == CodeModel::Kernel) &&
Chris Lattner18c59872009-06-27 04:16:01 +0000776 TM.getRelocationModel() == Reloc::Static)) {
777 if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(N0)) {
778 AM.GV = G->getGlobal();
779 AM.Disp += G->getOffset();
780 AM.SymbolFlags = G->getTargetFlags();
781 } else if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N0)) {
782 AM.CP = CP->getConstVal();
783 AM.Align = CP->getAlignment();
784 AM.Disp += CP->getOffset();
785 AM.SymbolFlags = CP->getTargetFlags();
786 } else if (ExternalSymbolSDNode *S = dyn_cast<ExternalSymbolSDNode>(N0)) {
787 AM.ES = S->getSymbol();
788 AM.SymbolFlags = S->getTargetFlags();
789 } else {
790 JumpTableSDNode *J = cast<JumpTableSDNode>(N0);
791 AM.JT = J->getIndex();
792 AM.SymbolFlags = J->getTargetFlags();
793 }
Rafael Espindola49a168d2009-04-12 21:55:03 +0000794 return false;
795 }
796
797 return true;
798}
799
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000800/// MatchAddress - Add the specified node to the specified addressing mode,
801/// returning true if it cannot be done. This just pattern matches for the
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000802/// addressing mode.
Dan Gohman41d0b9d2009-07-22 23:26:55 +0000803bool X86DAGToDAGISel::MatchAddress(SDValue N, X86ISelAddressMode &AM) {
804 if (MatchAddressRecursively(N, AM, 0))
805 return true;
806
807 // Post-processing: Convert lea(,%reg,2) to lea(%reg,%reg), which has
808 // a smaller encoding and avoids a scaled-index.
809 if (AM.Scale == 2 &&
810 AM.BaseType == X86ISelAddressMode::RegBase &&
811 AM.Base.Reg.getNode() == 0) {
812 AM.Base.Reg = AM.IndexReg;
813 AM.Scale = 1;
814 }
815
Dan Gohmanef74e9b2009-08-20 18:23:44 +0000816 // Post-processing: Convert foo to foo(%rip), even in non-PIC mode,
817 // because it has a smaller encoding.
818 // TODO: Which other code models can use this?
819 if (TM.getCodeModel() == CodeModel::Small &&
820 Subtarget->is64Bit() &&
821 AM.Scale == 1 &&
822 AM.BaseType == X86ISelAddressMode::RegBase &&
823 AM.Base.Reg.getNode() == 0 &&
824 AM.IndexReg.getNode() == 0 &&
Dan Gohman79b765d2009-08-25 17:47:44 +0000825 AM.SymbolFlags == X86II::MO_NO_FLAG &&
Dan Gohmanef74e9b2009-08-20 18:23:44 +0000826 AM.hasSymbolicDisplacement())
827 AM.Base.Reg = CurDAG->getRegister(X86::RIP, MVT::i64);
828
Dan Gohman41d0b9d2009-07-22 23:26:55 +0000829 return false;
830}
831
832bool X86DAGToDAGISel::MatchAddressRecursively(SDValue N, X86ISelAddressMode &AM,
833 unsigned Depth) {
Dan Gohman6520e202008-10-18 02:06:02 +0000834 bool is64Bit = Subtarget->is64Bit();
Dale Johannesen6f38cb62009-02-07 19:59:05 +0000835 DebugLoc dl = N.getDebugLoc();
Bill Wendling12321672009-08-07 21:33:25 +0000836 DEBUG({
837 errs() << "MatchAddress: ";
838 AM.dump();
839 });
Dan Gohmanbadb2d22007-08-13 20:03:06 +0000840 // Limit recursion.
841 if (Depth > 5)
Rafael Espindola523249f2009-03-31 16:16:57 +0000842 return MatchAddressBase(N, AM);
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000843
844 CodeModel::Model M = TM.getCodeModel();
845
Chris Lattner18c59872009-06-27 04:16:01 +0000846 // If this is already a %rip relative address, we can only merge immediates
847 // into it. Instead of handling this in every case, we handle it here.
Evan Cheng25ab6902006-09-08 06:48:29 +0000848 // RIP relative addressing: %rip + 32-bit displacement!
Chris Lattner18c59872009-06-27 04:16:01 +0000849 if (AM.isRIPRelative()) {
850 // FIXME: JumpTable and ExternalSymbol address currently don't like
851 // displacements. It isn't very important, but this should be fixed for
852 // consistency.
853 if (!AM.ES && AM.JT != -1) return true;
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000854
Chris Lattner18c59872009-06-27 04:16:01 +0000855 if (ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(N)) {
856 int64_t Val = AM.Disp + Cst->getSExtValue();
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000857 if (X86::isOffsetSuitableForCodeModel(Val, M,
858 AM.hasSymbolicDisplacement())) {
Chris Lattner18c59872009-06-27 04:16:01 +0000859 AM.Disp = Val;
Evan Cheng25ab6902006-09-08 06:48:29 +0000860 return false;
861 }
862 }
863 return true;
864 }
865
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000866 switch (N.getOpcode()) {
867 default: break;
Evan Cheng25ab6902006-09-08 06:48:29 +0000868 case ISD::Constant: {
Dan Gohman27cae7b2008-11-11 15:52:29 +0000869 uint64_t Val = cast<ConstantSDNode>(N)->getSExtValue();
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000870 if (!is64Bit ||
871 X86::isOffsetSuitableForCodeModel(AM.Disp + Val, M,
872 AM.hasSymbolicDisplacement())) {
Evan Cheng25ab6902006-09-08 06:48:29 +0000873 AM.Disp += Val;
874 return false;
875 }
876 break;
877 }
Evan Cheng51a9ed92006-02-25 10:09:08 +0000878
Rafael Espindola094fad32009-04-08 21:14:34 +0000879 case X86ISD::SegmentBaseAddress:
880 if (!MatchSegmentBaseAddress(N, AM))
881 return false;
882 break;
883
Rafael Espindola49a168d2009-04-12 21:55:03 +0000884 case X86ISD::Wrapper:
Chris Lattner18c59872009-06-27 04:16:01 +0000885 case X86ISD::WrapperRIP:
Rafael Espindola49a168d2009-04-12 21:55:03 +0000886 if (!MatchWrapper(N, AM))
887 return false;
Evan Cheng51a9ed92006-02-25 10:09:08 +0000888 break;
889
Rafael Espindola094fad32009-04-08 21:14:34 +0000890 case ISD::LOAD:
891 if (!MatchLoad(N, AM))
892 return false;
893 break;
894
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000895 case ISD::FrameIndex:
Gabor Greif93c53e52008-08-31 15:37:04 +0000896 if (AM.BaseType == X86ISelAddressMode::RegBase
897 && AM.Base.Reg.getNode() == 0) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000898 AM.BaseType = X86ISelAddressMode::FrameIndexBase;
899 AM.Base.FrameIndex = cast<FrameIndexSDNode>(N)->getIndex();
900 return false;
901 }
902 break;
Evan Chengec693f72005-12-08 02:01:35 +0000903
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000904 case ISD::SHL:
Chris Lattner18c59872009-06-27 04:16:01 +0000905 if (AM.IndexReg.getNode() != 0 || AM.Scale != 1)
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000906 break;
907
Gabor Greif93c53e52008-08-31 15:37:04 +0000908 if (ConstantSDNode
909 *CN = dyn_cast<ConstantSDNode>(N.getNode()->getOperand(1))) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000910 unsigned Val = CN->getZExtValue();
Dan Gohman41d0b9d2009-07-22 23:26:55 +0000911 // Note that we handle x<<1 as (,x,2) rather than (x,x) here so
912 // that the base operand remains free for further matching. If
913 // the base doesn't end up getting used, a post-processing step
914 // in MatchAddress turns (,x,2) into (x,x), which is cheaper.
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000915 if (Val == 1 || Val == 2 || Val == 3) {
916 AM.Scale = 1 << Val;
Gabor Greifba36cb52008-08-28 21:40:38 +0000917 SDValue ShVal = N.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000918
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000919 // Okay, we know that we have a scale by now. However, if the scaled
920 // value is an add of something and a constant, we can fold the
921 // constant into the disp field here.
Gabor Greifba36cb52008-08-28 21:40:38 +0000922 if (ShVal.getNode()->getOpcode() == ISD::ADD && ShVal.hasOneUse() &&
923 isa<ConstantSDNode>(ShVal.getNode()->getOperand(1))) {
924 AM.IndexReg = ShVal.getNode()->getOperand(0);
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000925 ConstantSDNode *AddVal =
Gabor Greifba36cb52008-08-28 21:40:38 +0000926 cast<ConstantSDNode>(ShVal.getNode()->getOperand(1));
Evan Cheng8e278262009-01-17 07:09:27 +0000927 uint64_t Disp = AM.Disp + (AddVal->getSExtValue() << Val);
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000928 if (!is64Bit ||
929 X86::isOffsetSuitableForCodeModel(Disp, M,
930 AM.hasSymbolicDisplacement()))
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000931 AM.Disp = Disp;
932 else
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000933 AM.IndexReg = ShVal;
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000934 } else {
935 AM.IndexReg = ShVal;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000936 }
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000937 return false;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000938 }
939 break;
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000940 }
Evan Chengec693f72005-12-08 02:01:35 +0000941
Dan Gohman83688052007-10-22 20:22:24 +0000942 case ISD::SMUL_LOHI:
943 case ISD::UMUL_LOHI:
944 // A mul_lohi where we need the low part can be folded as a plain multiply.
Gabor Greif99a6cb92008-08-26 22:36:50 +0000945 if (N.getResNo() != 0) break;
Dan Gohman83688052007-10-22 20:22:24 +0000946 // FALL THROUGH
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000947 case ISD::MUL:
Evan Cheng73f24c92009-03-30 21:36:47 +0000948 case X86ISD::MUL_IMM:
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000949 // X*[3,5,9] -> X+X*[2,4,8]
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000950 if (AM.BaseType == X86ISelAddressMode::RegBase &&
Gabor Greifba36cb52008-08-28 21:40:38 +0000951 AM.Base.Reg.getNode() == 0 &&
Chris Lattner18c59872009-06-27 04:16:01 +0000952 AM.IndexReg.getNode() == 0) {
Gabor Greif93c53e52008-08-31 15:37:04 +0000953 if (ConstantSDNode
954 *CN = dyn_cast<ConstantSDNode>(N.getNode()->getOperand(1)))
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000955 if (CN->getZExtValue() == 3 || CN->getZExtValue() == 5 ||
956 CN->getZExtValue() == 9) {
957 AM.Scale = unsigned(CN->getZExtValue())-1;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000958
Gabor Greifba36cb52008-08-28 21:40:38 +0000959 SDValue MulVal = N.getNode()->getOperand(0);
Dan Gohman475871a2008-07-27 21:46:04 +0000960 SDValue Reg;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000961
962 // Okay, we know that we have a scale by now. However, if the scaled
963 // value is an add of something and a constant, we can fold the
964 // constant into the disp field here.
Gabor Greifba36cb52008-08-28 21:40:38 +0000965 if (MulVal.getNode()->getOpcode() == ISD::ADD && MulVal.hasOneUse() &&
966 isa<ConstantSDNode>(MulVal.getNode()->getOperand(1))) {
967 Reg = MulVal.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000968 ConstantSDNode *AddVal =
Gabor Greifba36cb52008-08-28 21:40:38 +0000969 cast<ConstantSDNode>(MulVal.getNode()->getOperand(1));
Evan Cheng8e278262009-01-17 07:09:27 +0000970 uint64_t Disp = AM.Disp + AddVal->getSExtValue() *
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000971 CN->getZExtValue();
Anton Korobeynikovb5e01722009-08-05 23:01:26 +0000972 if (!is64Bit ||
973 X86::isOffsetSuitableForCodeModel(Disp, M,
974 AM.hasSymbolicDisplacement()))
Evan Cheng25ab6902006-09-08 06:48:29 +0000975 AM.Disp = Disp;
976 else
Gabor Greifba36cb52008-08-28 21:40:38 +0000977 Reg = N.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000978 } else {
Gabor Greifba36cb52008-08-28 21:40:38 +0000979 Reg = N.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000980 }
981
982 AM.IndexReg = AM.Base.Reg = Reg;
983 return false;
984 }
Chris Lattner62412262007-02-04 20:18:17 +0000985 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000986 break;
987
Dan Gohman3cd90a12009-05-11 18:02:53 +0000988 case ISD::SUB: {
989 // Given A-B, if A can be completely folded into the address and
990 // the index field with the index field unused, use -B as the index.
991 // This is a win if a has multiple parts that can be folded into
992 // the address. Also, this saves a mov if the base register has
993 // other uses, since it avoids a two-address sub instruction, however
994 // it costs an additional mov if the index register has other uses.
995
996 // Test if the LHS of the sub can be folded.
997 X86ISelAddressMode Backup = AM;
Dan Gohman41d0b9d2009-07-22 23:26:55 +0000998 if (MatchAddressRecursively(N.getNode()->getOperand(0), AM, Depth+1)) {
Dan Gohman3cd90a12009-05-11 18:02:53 +0000999 AM = Backup;
1000 break;
1001 }
1002 // Test if the index field is free for use.
Chris Lattner18c59872009-06-27 04:16:01 +00001003 if (AM.IndexReg.getNode() || AM.isRIPRelative()) {
Dan Gohman3cd90a12009-05-11 18:02:53 +00001004 AM = Backup;
1005 break;
1006 }
1007 int Cost = 0;
1008 SDValue RHS = N.getNode()->getOperand(1);
1009 // If the RHS involves a register with multiple uses, this
1010 // transformation incurs an extra mov, due to the neg instruction
1011 // clobbering its operand.
1012 if (!RHS.getNode()->hasOneUse() ||
1013 RHS.getNode()->getOpcode() == ISD::CopyFromReg ||
1014 RHS.getNode()->getOpcode() == ISD::TRUNCATE ||
1015 RHS.getNode()->getOpcode() == ISD::ANY_EXTEND ||
1016 (RHS.getNode()->getOpcode() == ISD::ZERO_EXTEND &&
Owen Anderson825b72b2009-08-11 20:47:22 +00001017 RHS.getNode()->getOperand(0).getValueType() == MVT::i32))
Dan Gohman3cd90a12009-05-11 18:02:53 +00001018 ++Cost;
1019 // If the base is a register with multiple uses, this
1020 // transformation may save a mov.
1021 if ((AM.BaseType == X86ISelAddressMode::RegBase &&
1022 AM.Base.Reg.getNode() &&
1023 !AM.Base.Reg.getNode()->hasOneUse()) ||
1024 AM.BaseType == X86ISelAddressMode::FrameIndexBase)
1025 --Cost;
1026 // If the folded LHS was interesting, this transformation saves
1027 // address arithmetic.
1028 if ((AM.hasSymbolicDisplacement() && !Backup.hasSymbolicDisplacement()) +
1029 ((AM.Disp != 0) && (Backup.Disp == 0)) +
1030 (AM.Segment.getNode() && !Backup.Segment.getNode()) >= 2)
1031 --Cost;
1032 // If it doesn't look like it may be an overall win, don't do it.
1033 if (Cost >= 0) {
1034 AM = Backup;
1035 break;
1036 }
1037
1038 // Ok, the transformation is legal and appears profitable. Go for it.
1039 SDValue Zero = CurDAG->getConstant(0, N.getValueType());
1040 SDValue Neg = CurDAG->getNode(ISD::SUB, dl, N.getValueType(), Zero, RHS);
1041 AM.IndexReg = Neg;
1042 AM.Scale = 1;
1043
1044 // Insert the new nodes into the topological ordering.
1045 if (Zero.getNode()->getNodeId() == -1 ||
1046 Zero.getNode()->getNodeId() > N.getNode()->getNodeId()) {
1047 CurDAG->RepositionNode(N.getNode(), Zero.getNode());
1048 Zero.getNode()->setNodeId(N.getNode()->getNodeId());
1049 }
1050 if (Neg.getNode()->getNodeId() == -1 ||
1051 Neg.getNode()->getNodeId() > N.getNode()->getNodeId()) {
1052 CurDAG->RepositionNode(N.getNode(), Neg.getNode());
1053 Neg.getNode()->setNodeId(N.getNode()->getNodeId());
1054 }
1055 return false;
1056 }
1057
Evan Cheng8e278262009-01-17 07:09:27 +00001058 case ISD::ADD: {
1059 X86ISelAddressMode Backup = AM;
Dan Gohman41d0b9d2009-07-22 23:26:55 +00001060 if (!MatchAddressRecursively(N.getNode()->getOperand(0), AM, Depth+1) &&
1061 !MatchAddressRecursively(N.getNode()->getOperand(1), AM, Depth+1))
Evan Cheng8e278262009-01-17 07:09:27 +00001062 return false;
1063 AM = Backup;
Dan Gohman41d0b9d2009-07-22 23:26:55 +00001064 if (!MatchAddressRecursively(N.getNode()->getOperand(1), AM, Depth+1) &&
1065 !MatchAddressRecursively(N.getNode()->getOperand(0), AM, Depth+1))
Evan Cheng8e278262009-01-17 07:09:27 +00001066 return false;
1067 AM = Backup;
Dan Gohman77502c92009-03-13 02:25:09 +00001068
1069 // If we couldn't fold both operands into the address at the same time,
1070 // see if we can just put each operand into a register and fold at least
1071 // the add.
1072 if (AM.BaseType == X86ISelAddressMode::RegBase &&
1073 !AM.Base.Reg.getNode() &&
Chris Lattner18c59872009-06-27 04:16:01 +00001074 !AM.IndexReg.getNode()) {
Dan Gohman77502c92009-03-13 02:25:09 +00001075 AM.Base.Reg = N.getNode()->getOperand(0);
1076 AM.IndexReg = N.getNode()->getOperand(1);
1077 AM.Scale = 1;
1078 return false;
1079 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001080 break;
Evan Cheng8e278262009-01-17 07:09:27 +00001081 }
Evan Chenge6ad27e2006-05-30 06:59:36 +00001082
Chris Lattner62412262007-02-04 20:18:17 +00001083 case ISD::OR:
1084 // Handle "X | C" as "X + C" iff X is known to have C bits clear.
Chris Lattner5aaddaa2007-12-08 07:22:58 +00001085 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
1086 X86ISelAddressMode Backup = AM;
Dan Gohman27cae7b2008-11-11 15:52:29 +00001087 uint64_t Offset = CN->getSExtValue();
Chris Lattner5aaddaa2007-12-08 07:22:58 +00001088 // Start with the LHS as an addr mode.
Dan Gohman41d0b9d2009-07-22 23:26:55 +00001089 if (!MatchAddressRecursively(N.getOperand(0), AM, Depth+1) &&
Chris Lattner5aaddaa2007-12-08 07:22:58 +00001090 // Address could not have picked a GV address for the displacement.
1091 AM.GV == NULL &&
1092 // On x86-64, the resultant disp must fit in 32-bits.
Anton Korobeynikovb5e01722009-08-05 23:01:26 +00001093 (!is64Bit ||
1094 X86::isOffsetSuitableForCodeModel(AM.Disp + Offset, M,
1095 AM.hasSymbolicDisplacement())) &&
Chris Lattner5aaddaa2007-12-08 07:22:58 +00001096 // Check to see if the LHS & C is zero.
Dan Gohman2e68b6f2008-02-25 21:11:39 +00001097 CurDAG->MaskedValueIsZero(N.getOperand(0), CN->getAPIntValue())) {
Dan Gohman27cae7b2008-11-11 15:52:29 +00001098 AM.Disp += Offset;
Chris Lattner5aaddaa2007-12-08 07:22:58 +00001099 return false;
Evan Chenge6ad27e2006-05-30 06:59:36 +00001100 }
Chris Lattner5aaddaa2007-12-08 07:22:58 +00001101 AM = Backup;
Evan Chenge6ad27e2006-05-30 06:59:36 +00001102 }
1103 break;
Evan Cheng1314b002007-12-13 00:43:27 +00001104
1105 case ISD::AND: {
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001106 // Perform some heroic transforms on an and of a constant-count shift
1107 // with a constant to enable use of the scaled offset field.
1108
Dan Gohman475871a2008-07-27 21:46:04 +00001109 SDValue Shift = N.getOperand(0);
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001110 if (Shift.getNumOperands() != 2) break;
Dan Gohman8be6bbe2008-11-05 04:14:16 +00001111
Evan Cheng1314b002007-12-13 00:43:27 +00001112 // Scale must not be used already.
Gabor Greifba36cb52008-08-28 21:40:38 +00001113 if (AM.IndexReg.getNode() != 0 || AM.Scale != 1) break;
Evan Chengbe3bf422008-02-07 08:53:49 +00001114
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001115 SDValue X = Shift.getOperand(0);
Evan Cheng1314b002007-12-13 00:43:27 +00001116 ConstantSDNode *C2 = dyn_cast<ConstantSDNode>(N.getOperand(1));
1117 ConstantSDNode *C1 = dyn_cast<ConstantSDNode>(Shift.getOperand(1));
1118 if (!C1 || !C2) break;
1119
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001120 // Handle "(X >> (8-C1)) & C2" as "(X >> 8) & 0xff)" if safe. This
1121 // allows us to convert the shift and and into an h-register extract and
1122 // a scaled index.
1123 if (Shift.getOpcode() == ISD::SRL && Shift.hasOneUse()) {
1124 unsigned ScaleLog = 8 - C1->getZExtValue();
Rafael Espindola7c366832009-04-16 12:34:53 +00001125 if (ScaleLog > 0 && ScaleLog < 4 &&
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001126 C2->getZExtValue() == (UINT64_C(0xff) << ScaleLog)) {
Owen Anderson825b72b2009-08-11 20:47:22 +00001127 SDValue Eight = CurDAG->getConstant(8, MVT::i8);
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001128 SDValue Mask = CurDAG->getConstant(0xff, N.getValueType());
1129 SDValue Srl = CurDAG->getNode(ISD::SRL, dl, N.getValueType(),
1130 X, Eight);
1131 SDValue And = CurDAG->getNode(ISD::AND, dl, N.getValueType(),
1132 Srl, Mask);
Owen Anderson825b72b2009-08-11 20:47:22 +00001133 SDValue ShlCount = CurDAG->getConstant(ScaleLog, MVT::i8);
Dan Gohman62ad1382009-04-14 22:45:05 +00001134 SDValue Shl = CurDAG->getNode(ISD::SHL, dl, N.getValueType(),
1135 And, ShlCount);
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001136
1137 // Insert the new nodes into the topological ordering.
1138 if (Eight.getNode()->getNodeId() == -1 ||
1139 Eight.getNode()->getNodeId() > X.getNode()->getNodeId()) {
1140 CurDAG->RepositionNode(X.getNode(), Eight.getNode());
1141 Eight.getNode()->setNodeId(X.getNode()->getNodeId());
1142 }
1143 if (Mask.getNode()->getNodeId() == -1 ||
1144 Mask.getNode()->getNodeId() > X.getNode()->getNodeId()) {
1145 CurDAG->RepositionNode(X.getNode(), Mask.getNode());
1146 Mask.getNode()->setNodeId(X.getNode()->getNodeId());
1147 }
1148 if (Srl.getNode()->getNodeId() == -1 ||
1149 Srl.getNode()->getNodeId() > Shift.getNode()->getNodeId()) {
1150 CurDAG->RepositionNode(Shift.getNode(), Srl.getNode());
1151 Srl.getNode()->setNodeId(Shift.getNode()->getNodeId());
1152 }
1153 if (And.getNode()->getNodeId() == -1 ||
1154 And.getNode()->getNodeId() > N.getNode()->getNodeId()) {
1155 CurDAG->RepositionNode(N.getNode(), And.getNode());
1156 And.getNode()->setNodeId(N.getNode()->getNodeId());
1157 }
Dan Gohman62ad1382009-04-14 22:45:05 +00001158 if (ShlCount.getNode()->getNodeId() == -1 ||
1159 ShlCount.getNode()->getNodeId() > X.getNode()->getNodeId()) {
1160 CurDAG->RepositionNode(X.getNode(), ShlCount.getNode());
1161 ShlCount.getNode()->setNodeId(N.getNode()->getNodeId());
1162 }
1163 if (Shl.getNode()->getNodeId() == -1 ||
1164 Shl.getNode()->getNodeId() > N.getNode()->getNodeId()) {
1165 CurDAG->RepositionNode(N.getNode(), Shl.getNode());
1166 Shl.getNode()->setNodeId(N.getNode()->getNodeId());
1167 }
1168 CurDAG->ReplaceAllUsesWith(N, Shl);
Dan Gohman21e3dfb2009-04-13 16:09:41 +00001169 AM.IndexReg = And;
1170 AM.Scale = (1 << ScaleLog);
1171 return false;
1172 }
1173 }
1174
1175 // Handle "(X << C1) & C2" as "(X & (C2>>C1)) << C1" if safe and if this
1176 // allows us to fold the shift into this addressing mode.
1177 if (Shift.getOpcode() != ISD::SHL) break;
1178
Evan Cheng1314b002007-12-13 00:43:27 +00001179 // Not likely to be profitable if either the AND or SHIFT node has more
1180 // than one use (unless all uses are for address computation). Besides,
1181 // isel mechanism requires their node ids to be reused.
1182 if (!N.hasOneUse() || !Shift.hasOneUse())
1183 break;
1184
1185 // Verify that the shift amount is something we can fold.
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +00001186 unsigned ShiftCst = C1->getZExtValue();
Evan Cheng1314b002007-12-13 00:43:27 +00001187 if (ShiftCst != 1 && ShiftCst != 2 && ShiftCst != 3)
1188 break;
1189
1190 // Get the new AND mask, this folds to a constant.
Dale Johannesend8392542009-02-03 21:48:12 +00001191 SDValue NewANDMask = CurDAG->getNode(ISD::SRL, dl, N.getValueType(),
Evan Cheng552e3be2008-10-14 17:15:39 +00001192 SDValue(C2, 0), SDValue(C1, 0));
Dale Johannesend8392542009-02-03 21:48:12 +00001193 SDValue NewAND = CurDAG->getNode(ISD::AND, dl, N.getValueType(), X,
1194 NewANDMask);
1195 SDValue NewSHIFT = CurDAG->getNode(ISD::SHL, dl, N.getValueType(),
Dan Gohman7b8e9642008-10-13 20:52:04 +00001196 NewAND, SDValue(C1, 0));
Dan Gohman8be6bbe2008-11-05 04:14:16 +00001197
1198 // Insert the new nodes into the topological ordering.
1199 if (C1->getNodeId() > X.getNode()->getNodeId()) {
1200 CurDAG->RepositionNode(X.getNode(), C1);
1201 C1->setNodeId(X.getNode()->getNodeId());
1202 }
1203 if (NewANDMask.getNode()->getNodeId() == -1 ||
1204 NewANDMask.getNode()->getNodeId() > X.getNode()->getNodeId()) {
1205 CurDAG->RepositionNode(X.getNode(), NewANDMask.getNode());
1206 NewANDMask.getNode()->setNodeId(X.getNode()->getNodeId());
1207 }
1208 if (NewAND.getNode()->getNodeId() == -1 ||
1209 NewAND.getNode()->getNodeId() > Shift.getNode()->getNodeId()) {
1210 CurDAG->RepositionNode(Shift.getNode(), NewAND.getNode());
1211 NewAND.getNode()->setNodeId(Shift.getNode()->getNodeId());
1212 }
1213 if (NewSHIFT.getNode()->getNodeId() == -1 ||
1214 NewSHIFT.getNode()->getNodeId() > N.getNode()->getNodeId()) {
1215 CurDAG->RepositionNode(N.getNode(), NewSHIFT.getNode());
1216 NewSHIFT.getNode()->setNodeId(N.getNode()->getNodeId());
1217 }
1218
Dan Gohman7b8e9642008-10-13 20:52:04 +00001219 CurDAG->ReplaceAllUsesWith(N, NewSHIFT);
Evan Cheng1314b002007-12-13 00:43:27 +00001220
1221 AM.Scale = 1 << ShiftCst;
1222 AM.IndexReg = NewAND;
1223 return false;
1224 }
Evan Chenge6ad27e2006-05-30 06:59:36 +00001225 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001226
Rafael Espindola523249f2009-03-31 16:16:57 +00001227 return MatchAddressBase(N, AM);
Dan Gohmanbadb2d22007-08-13 20:03:06 +00001228}
1229
1230/// MatchAddressBase - Helper for MatchAddress. Add the specified node to the
1231/// specified addressing mode without any further recursion.
Rafael Espindola523249f2009-03-31 16:16:57 +00001232bool X86DAGToDAGISel::MatchAddressBase(SDValue N, X86ISelAddressMode &AM) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001233 // Is the base register already occupied?
Gabor Greifba36cb52008-08-28 21:40:38 +00001234 if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base.Reg.getNode()) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001235 // If so, check to see if the scale index register is set.
Chris Lattner18c59872009-06-27 04:16:01 +00001236 if (AM.IndexReg.getNode() == 0) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001237 AM.IndexReg = N;
1238 AM.Scale = 1;
1239 return false;
1240 }
1241
1242 // Otherwise, we cannot select it.
1243 return true;
1244 }
1245
1246 // Default, generate it as a register.
1247 AM.BaseType = X86ISelAddressMode::RegBase;
1248 AM.Base.Reg = N;
1249 return false;
1250}
1251
Evan Chengec693f72005-12-08 02:01:35 +00001252/// SelectAddr - returns true if it is able pattern match an addressing mode.
1253/// It returns the operands which make up the maximal addressing mode it can
1254/// match by reference.
Dan Gohman475871a2008-07-27 21:46:04 +00001255bool X86DAGToDAGISel::SelectAddr(SDValue Op, SDValue N, SDValue &Base,
1256 SDValue &Scale, SDValue &Index,
Rafael Espindola094fad32009-04-08 21:14:34 +00001257 SDValue &Disp, SDValue &Segment) {
Evan Chengec693f72005-12-08 02:01:35 +00001258 X86ISelAddressMode AM;
Evan Cheng4d952322009-03-31 01:13:53 +00001259 bool Done = false;
1260 if (AvoidDupAddrCompute && !N.hasOneUse()) {
1261 unsigned Opcode = N.getOpcode();
1262 if (Opcode != ISD::Constant && Opcode != ISD::FrameIndex &&
Chris Lattner18c59872009-06-27 04:16:01 +00001263 Opcode != X86ISD::Wrapper && Opcode != X86ISD::WrapperRIP) {
Evan Cheng4d952322009-03-31 01:13:53 +00001264 // If we are able to fold N into addressing mode, then we'll allow it even
1265 // if N has multiple uses. In general, addressing computation is used as
1266 // addresses by all of its uses. But watch out for CopyToReg uses, that
1267 // means the address computation is liveout. It will be computed by a LEA
1268 // so we want to avoid computing the address twice.
1269 for (SDNode::use_iterator UI = N.getNode()->use_begin(),
1270 UE = N.getNode()->use_end(); UI != UE; ++UI) {
1271 if (UI->getOpcode() == ISD::CopyToReg) {
Rafael Espindola523249f2009-03-31 16:16:57 +00001272 MatchAddressBase(N, AM);
Evan Cheng4d952322009-03-31 01:13:53 +00001273 Done = true;
1274 break;
1275 }
1276 }
1277 }
1278 }
1279
1280 if (!Done && MatchAddress(N, AM))
Evan Cheng8700e142006-01-11 06:09:51 +00001281 return false;
Evan Chengec693f72005-12-08 02:01:35 +00001282
Owen Andersone50ed302009-08-10 22:56:29 +00001283 EVT VT = N.getValueType();
Evan Cheng8700e142006-01-11 06:09:51 +00001284 if (AM.BaseType == X86ISelAddressMode::RegBase) {
Gabor Greifba36cb52008-08-28 21:40:38 +00001285 if (!AM.Base.Reg.getNode())
Evan Cheng25ab6902006-09-08 06:48:29 +00001286 AM.Base.Reg = CurDAG->getRegister(0, VT);
Evan Chengec693f72005-12-08 02:01:35 +00001287 }
Evan Cheng8700e142006-01-11 06:09:51 +00001288
Gabor Greifba36cb52008-08-28 21:40:38 +00001289 if (!AM.IndexReg.getNode())
Evan Cheng25ab6902006-09-08 06:48:29 +00001290 AM.IndexReg = CurDAG->getRegister(0, VT);
Evan Cheng8700e142006-01-11 06:09:51 +00001291
Rafael Espindola094fad32009-04-08 21:14:34 +00001292 getAddressOperands(AM, Base, Scale, Index, Disp, Segment);
Evan Cheng8700e142006-01-11 06:09:51 +00001293 return true;
Evan Chengec693f72005-12-08 02:01:35 +00001294}
1295
Chris Lattner3a7cd952006-10-07 21:55:32 +00001296/// SelectScalarSSELoad - Match a scalar SSE load. In particular, we want to
1297/// match a load whose top elements are either undef or zeros. The load flavor
1298/// is derived from the type of N, which is either v4f32 or v2f64.
Dan Gohman475871a2008-07-27 21:46:04 +00001299bool X86DAGToDAGISel::SelectScalarSSELoad(SDValue Op, SDValue Pred,
1300 SDValue N, SDValue &Base,
1301 SDValue &Scale, SDValue &Index,
Rafael Espindola094fad32009-04-08 21:14:34 +00001302 SDValue &Disp, SDValue &Segment,
1303 SDValue &InChain,
Dan Gohman475871a2008-07-27 21:46:04 +00001304 SDValue &OutChain) {
Chris Lattner3a7cd952006-10-07 21:55:32 +00001305 if (N.getOpcode() == ISD::SCALAR_TO_VECTOR) {
Chris Lattner4fe4f252006-10-11 22:09:58 +00001306 InChain = N.getOperand(0).getValue(1);
Gabor Greifba36cb52008-08-28 21:40:38 +00001307 if (ISD::isNON_EXTLoad(InChain.getNode()) &&
Evan Cheng07e4b002006-10-16 06:34:55 +00001308 InChain.getValue(0).hasOneUse() &&
Evan Chengd6373bc2006-11-10 21:23:04 +00001309 N.hasOneUse() &&
Evan Cheng884c70c2008-11-27 00:49:46 +00001310 IsLegalAndProfitableToFold(N.getNode(), Pred.getNode(), Op.getNode())) {
Evan Cheng82a91642006-10-11 21:06:01 +00001311 LoadSDNode *LD = cast<LoadSDNode>(InChain);
Rafael Espindola094fad32009-04-08 21:14:34 +00001312 if (!SelectAddr(Op, LD->getBasePtr(), Base, Scale, Index, Disp, Segment))
Chris Lattner3a7cd952006-10-07 21:55:32 +00001313 return false;
Evan Cheng82a91642006-10-11 21:06:01 +00001314 OutChain = LD->getChain();
Chris Lattner3a7cd952006-10-07 21:55:32 +00001315 return true;
1316 }
1317 }
Chris Lattner4fe4f252006-10-11 22:09:58 +00001318
1319 // Also handle the case where we explicitly require zeros in the top
Chris Lattner3a7cd952006-10-07 21:55:32 +00001320 // elements. This is a vector shuffle from the zero vector.
Gabor Greifba36cb52008-08-28 21:40:38 +00001321 if (N.getOpcode() == X86ISD::VZEXT_MOVL && N.getNode()->hasOneUse() &&
Chris Lattner8a594482007-11-25 00:24:49 +00001322 // Check to see if the top elements are all zeros (or bitcast of zeros).
Evan Cheng7e2ff772008-05-08 00:57:18 +00001323 N.getOperand(0).getOpcode() == ISD::SCALAR_TO_VECTOR &&
Gabor Greifba36cb52008-08-28 21:40:38 +00001324 N.getOperand(0).getNode()->hasOneUse() &&
1325 ISD::isNON_EXTLoad(N.getOperand(0).getOperand(0).getNode()) &&
Evan Cheng7e2ff772008-05-08 00:57:18 +00001326 N.getOperand(0).getOperand(0).hasOneUse()) {
1327 // Okay, this is a zero extending load. Fold it.
1328 LoadSDNode *LD = cast<LoadSDNode>(N.getOperand(0).getOperand(0));
Rafael Espindola094fad32009-04-08 21:14:34 +00001329 if (!SelectAddr(Op, LD->getBasePtr(), Base, Scale, Index, Disp, Segment))
Evan Cheng7e2ff772008-05-08 00:57:18 +00001330 return false;
1331 OutChain = LD->getChain();
Dan Gohman475871a2008-07-27 21:46:04 +00001332 InChain = SDValue(LD, 1);
Evan Cheng7e2ff772008-05-08 00:57:18 +00001333 return true;
Chris Lattner4fe4f252006-10-11 22:09:58 +00001334 }
Chris Lattner3a7cd952006-10-07 21:55:32 +00001335 return false;
1336}
1337
1338
Evan Cheng51a9ed92006-02-25 10:09:08 +00001339/// SelectLEAAddr - it calls SelectAddr and determines if the maximal addressing
1340/// mode it matches can be cost effectively emitted as an LEA instruction.
Dan Gohman475871a2008-07-27 21:46:04 +00001341bool X86DAGToDAGISel::SelectLEAAddr(SDValue Op, SDValue N,
1342 SDValue &Base, SDValue &Scale,
1343 SDValue &Index, SDValue &Disp) {
Evan Cheng51a9ed92006-02-25 10:09:08 +00001344 X86ISelAddressMode AM;
Rafael Espindoladbcfb302009-04-10 10:09:34 +00001345
1346 // Set AM.Segment to prevent MatchAddress from using one. LEA doesn't support
1347 // segments.
1348 SDValue Copy = AM.Segment;
Owen Anderson825b72b2009-08-11 20:47:22 +00001349 SDValue T = CurDAG->getRegister(0, MVT::i32);
Rafael Espindoladbcfb302009-04-10 10:09:34 +00001350 AM.Segment = T;
Evan Cheng51a9ed92006-02-25 10:09:08 +00001351 if (MatchAddress(N, AM))
1352 return false;
Rafael Espindoladbcfb302009-04-10 10:09:34 +00001353 assert (T == AM.Segment);
1354 AM.Segment = Copy;
Rafael Espindola094fad32009-04-08 21:14:34 +00001355
Owen Andersone50ed302009-08-10 22:56:29 +00001356 EVT VT = N.getValueType();
Evan Cheng51a9ed92006-02-25 10:09:08 +00001357 unsigned Complexity = 0;
1358 if (AM.BaseType == X86ISelAddressMode::RegBase)
Gabor Greifba36cb52008-08-28 21:40:38 +00001359 if (AM.Base.Reg.getNode())
Evan Cheng51a9ed92006-02-25 10:09:08 +00001360 Complexity = 1;
1361 else
Evan Cheng25ab6902006-09-08 06:48:29 +00001362 AM.Base.Reg = CurDAG->getRegister(0, VT);
Evan Cheng51a9ed92006-02-25 10:09:08 +00001363 else if (AM.BaseType == X86ISelAddressMode::FrameIndexBase)
1364 Complexity = 4;
1365
Gabor Greifba36cb52008-08-28 21:40:38 +00001366 if (AM.IndexReg.getNode())
Evan Cheng51a9ed92006-02-25 10:09:08 +00001367 Complexity++;
1368 else
Evan Cheng25ab6902006-09-08 06:48:29 +00001369 AM.IndexReg = CurDAG->getRegister(0, VT);
Evan Cheng51a9ed92006-02-25 10:09:08 +00001370
Chris Lattnera16b7cb2007-03-20 06:08:29 +00001371 // Don't match just leal(,%reg,2). It's cheaper to do addl %reg, %reg, or with
1372 // a simple shift.
1373 if (AM.Scale > 1)
Evan Cheng8c03fe42006-02-28 21:13:57 +00001374 Complexity++;
Evan Cheng51a9ed92006-02-25 10:09:08 +00001375
1376 // FIXME: We are artificially lowering the criteria to turn ADD %reg, $GA
1377 // to a LEA. This is determined with some expermentation but is by no means
1378 // optimal (especially for code size consideration). LEA is nice because of
1379 // its three-address nature. Tweak the cost function again when we can run
1380 // convertToThreeAddress() at register allocation time.
Dan Gohman2d0a1cc2009-02-07 00:43:41 +00001381 if (AM.hasSymbolicDisplacement()) {
Evan Cheng25ab6902006-09-08 06:48:29 +00001382 // For X86-64, we should always use lea to materialize RIP relative
1383 // addresses.
Evan Cheng953fa042006-12-05 22:03:40 +00001384 if (Subtarget->is64Bit())
Evan Cheng25ab6902006-09-08 06:48:29 +00001385 Complexity = 4;
1386 else
1387 Complexity += 2;
1388 }
Evan Cheng51a9ed92006-02-25 10:09:08 +00001389
Gabor Greifba36cb52008-08-28 21:40:38 +00001390 if (AM.Disp && (AM.Base.Reg.getNode() || AM.IndexReg.getNode()))
Evan Cheng51a9ed92006-02-25 10:09:08 +00001391 Complexity++;
1392
Chris Lattner25142782009-07-11 22:50:33 +00001393 // If it isn't worth using an LEA, reject it.
Chris Lattner14f75112009-07-11 23:07:30 +00001394 if (Complexity <= 2)
Chris Lattner25142782009-07-11 22:50:33 +00001395 return false;
1396
1397 SDValue Segment;
1398 getAddressOperands(AM, Base, Scale, Index, Disp, Segment);
1399 return true;
Evan Cheng51a9ed92006-02-25 10:09:08 +00001400}
1401
Chris Lattner5c0b16d2009-06-20 20:38:48 +00001402/// SelectTLSADDRAddr - This is only run on TargetGlobalTLSAddress nodes.
1403bool X86DAGToDAGISel::SelectTLSADDRAddr(SDValue Op, SDValue N, SDValue &Base,
1404 SDValue &Scale, SDValue &Index,
1405 SDValue &Disp) {
1406 assert(Op.getOpcode() == X86ISD::TLSADDR);
1407 assert(N.getOpcode() == ISD::TargetGlobalTLSAddress);
1408 const GlobalAddressSDNode *GA = cast<GlobalAddressSDNode>(N);
1409
1410 X86ISelAddressMode AM;
1411 AM.GV = GA->getGlobal();
1412 AM.Disp += GA->getOffset();
1413 AM.Base.Reg = CurDAG->getRegister(0, N.getValueType());
Chris Lattnerba8ef452009-06-26 21:18:37 +00001414 AM.SymbolFlags = GA->getTargetFlags();
1415
Owen Anderson825b72b2009-08-11 20:47:22 +00001416 if (N.getValueType() == MVT::i32) {
Chris Lattner5c0b16d2009-06-20 20:38:48 +00001417 AM.Scale = 1;
Owen Anderson825b72b2009-08-11 20:47:22 +00001418 AM.IndexReg = CurDAG->getRegister(X86::EBX, MVT::i32);
Chris Lattner5c0b16d2009-06-20 20:38:48 +00001419 } else {
Owen Anderson825b72b2009-08-11 20:47:22 +00001420 AM.IndexReg = CurDAG->getRegister(0, MVT::i64);
Chris Lattner5c0b16d2009-06-20 20:38:48 +00001421 }
1422
1423 SDValue Segment;
1424 getAddressOperands(AM, Base, Scale, Index, Disp, Segment);
1425 return true;
1426}
1427
1428
Dan Gohman475871a2008-07-27 21:46:04 +00001429bool X86DAGToDAGISel::TryFoldLoad(SDValue P, SDValue N,
1430 SDValue &Base, SDValue &Scale,
Rafael Espindola094fad32009-04-08 21:14:34 +00001431 SDValue &Index, SDValue &Disp,
1432 SDValue &Segment) {
Gabor Greifba36cb52008-08-28 21:40:38 +00001433 if (ISD::isNON_EXTLoad(N.getNode()) &&
Evan Cheng5e351682006-02-06 06:02:33 +00001434 N.hasOneUse() &&
Evan Cheng884c70c2008-11-27 00:49:46 +00001435 IsLegalAndProfitableToFold(N.getNode(), P.getNode(), P.getNode()))
Rafael Espindola094fad32009-04-08 21:14:34 +00001436 return SelectAddr(P, N.getOperand(1), Base, Scale, Index, Disp, Segment);
Evan Cheng0114e942006-01-06 20:36:21 +00001437 return false;
1438}
1439
Dan Gohman8b746962008-09-23 18:22:58 +00001440/// getGlobalBaseReg - Return an SDNode that returns the value of
1441/// the global base register. Output instructions required to
1442/// initialize the global base register, if necessary.
Evan Cheng7ccced62006-02-18 00:15:05 +00001443///
Evan Cheng9ade2182006-08-26 05:34:46 +00001444SDNode *X86DAGToDAGISel::getGlobalBaseReg() {
Dan Gohmanc5534622009-06-03 20:20:00 +00001445 unsigned GlobalBaseReg = getInstrInfo()->getGlobalBaseReg(MF);
Gabor Greifba36cb52008-08-28 21:40:38 +00001446 return CurDAG->getRegister(GlobalBaseReg, TLI.getPointerTy()).getNode();
Evan Cheng7ccced62006-02-18 00:15:05 +00001447}
1448
Evan Chengb245d922006-05-20 01:36:52 +00001449static SDNode *FindCallStartFromCall(SDNode *Node) {
1450 if (Node->getOpcode() == ISD::CALLSEQ_START) return Node;
Owen Anderson825b72b2009-08-11 20:47:22 +00001451 assert(Node->getOperand(0).getValueType() == MVT::Other &&
Evan Chengb245d922006-05-20 01:36:52 +00001452 "Node doesn't have a token chain argument!");
Gabor Greifba36cb52008-08-28 21:40:38 +00001453 return FindCallStartFromCall(Node->getOperand(0).getNode());
Evan Chengb245d922006-05-20 01:36:52 +00001454}
1455
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001456SDNode *X86DAGToDAGISel::SelectAtomic64(SDNode *Node, unsigned Opc) {
1457 SDValue Chain = Node->getOperand(0);
1458 SDValue In1 = Node->getOperand(1);
1459 SDValue In2L = Node->getOperand(2);
1460 SDValue In2H = Node->getOperand(3);
Rafael Espindola094fad32009-04-08 21:14:34 +00001461 SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4;
1462 if (!SelectAddr(In1, In1, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4))
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001463 return NULL;
Dale Johannesen1b54c7f2008-10-03 19:41:08 +00001464 SDValue LSI = Node->getOperand(4); // MemOperand
Rafael Espindola094fad32009-04-08 21:14:34 +00001465 const SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, In2L, In2H, LSI, Chain};
Rafael Espindolae4d5d342009-03-27 15:45:05 +00001466 return CurDAG->getTargetNode(Opc, Node->getDebugLoc(),
Owen Anderson825b72b2009-08-11 20:47:22 +00001467 MVT::i32, MVT::i32, MVT::Other, Ops,
Rafael Espindolaa0a4f072009-03-28 19:02:18 +00001468 array_lengthof(Ops));
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001469}
Christopher Lambc59e5212007-08-10 21:48:46 +00001470
Owen Andersone50ed302009-08-10 22:56:29 +00001471SDNode *X86DAGToDAGISel::SelectAtomicLoadAdd(SDNode *Node, EVT NVT) {
Evan Cheng37b73872009-07-30 08:33:02 +00001472 if (Node->hasAnyUseOfValue(0))
1473 return 0;
1474
1475 // Optimize common patterns for __sync_add_and_fetch and
1476 // __sync_sub_and_fetch where the result is not used. This allows us
1477 // to use "lock" version of add, sub, inc, dec instructions.
1478 // FIXME: Do not use special instructions but instead add the "lock"
1479 // prefix to the target node somehow. The extra information will then be
1480 // transferred to machine instruction and it denotes the prefix.
1481 SDValue Chain = Node->getOperand(0);
1482 SDValue Ptr = Node->getOperand(1);
1483 SDValue Val = Node->getOperand(2);
1484 SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4;
1485 if (!SelectAddr(Ptr, Ptr, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4))
1486 return 0;
1487
1488 bool isInc = false, isDec = false, isSub = false, isCN = false;
1489 ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Val);
1490 if (CN) {
1491 isCN = true;
1492 int64_t CNVal = CN->getSExtValue();
1493 if (CNVal == 1)
1494 isInc = true;
1495 else if (CNVal == -1)
1496 isDec = true;
1497 else if (CNVal >= 0)
1498 Val = CurDAG->getTargetConstant(CNVal, NVT);
1499 else {
1500 isSub = true;
1501 Val = CurDAG->getTargetConstant(-CNVal, NVT);
1502 }
1503 } else if (Val.hasOneUse() &&
1504 Val.getOpcode() == ISD::SUB &&
1505 X86::isZeroNode(Val.getOperand(0))) {
1506 isSub = true;
1507 Val = Val.getOperand(1);
1508 }
1509
1510 unsigned Opc = 0;
Owen Anderson825b72b2009-08-11 20:47:22 +00001511 switch (NVT.getSimpleVT().SimpleTy) {
Evan Cheng37b73872009-07-30 08:33:02 +00001512 default: return 0;
Owen Anderson825b72b2009-08-11 20:47:22 +00001513 case MVT::i8:
Evan Cheng37b73872009-07-30 08:33:02 +00001514 if (isInc)
1515 Opc = X86::LOCK_INC8m;
1516 else if (isDec)
1517 Opc = X86::LOCK_DEC8m;
1518 else if (isSub) {
1519 if (isCN)
1520 Opc = X86::LOCK_SUB8mi;
1521 else
1522 Opc = X86::LOCK_SUB8mr;
1523 } else {
1524 if (isCN)
1525 Opc = X86::LOCK_ADD8mi;
1526 else
1527 Opc = X86::LOCK_ADD8mr;
1528 }
1529 break;
Owen Anderson825b72b2009-08-11 20:47:22 +00001530 case MVT::i16:
Evan Cheng37b73872009-07-30 08:33:02 +00001531 if (isInc)
1532 Opc = X86::LOCK_INC16m;
1533 else if (isDec)
1534 Opc = X86::LOCK_DEC16m;
1535 else if (isSub) {
1536 if (isCN) {
1537 if (Predicate_i16immSExt8(Val.getNode()))
1538 Opc = X86::LOCK_SUB16mi8;
1539 else
1540 Opc = X86::LOCK_SUB16mi;
1541 } else
1542 Opc = X86::LOCK_SUB16mr;
1543 } else {
1544 if (isCN) {
1545 if (Predicate_i16immSExt8(Val.getNode()))
1546 Opc = X86::LOCK_ADD16mi8;
1547 else
1548 Opc = X86::LOCK_ADD16mi;
1549 } else
1550 Opc = X86::LOCK_ADD16mr;
1551 }
1552 break;
Owen Anderson825b72b2009-08-11 20:47:22 +00001553 case MVT::i32:
Evan Cheng37b73872009-07-30 08:33:02 +00001554 if (isInc)
1555 Opc = X86::LOCK_INC32m;
1556 else if (isDec)
1557 Opc = X86::LOCK_DEC32m;
1558 else if (isSub) {
1559 if (isCN) {
1560 if (Predicate_i32immSExt8(Val.getNode()))
1561 Opc = X86::LOCK_SUB32mi8;
1562 else
1563 Opc = X86::LOCK_SUB32mi;
1564 } else
1565 Opc = X86::LOCK_SUB32mr;
1566 } else {
1567 if (isCN) {
1568 if (Predicate_i32immSExt8(Val.getNode()))
1569 Opc = X86::LOCK_ADD32mi8;
1570 else
1571 Opc = X86::LOCK_ADD32mi;
1572 } else
1573 Opc = X86::LOCK_ADD32mr;
1574 }
1575 break;
Owen Anderson825b72b2009-08-11 20:47:22 +00001576 case MVT::i64:
Evan Cheng37b73872009-07-30 08:33:02 +00001577 if (isInc)
1578 Opc = X86::LOCK_INC64m;
1579 else if (isDec)
1580 Opc = X86::LOCK_DEC64m;
1581 else if (isSub) {
1582 Opc = X86::LOCK_SUB64mr;
1583 if (isCN) {
1584 if (Predicate_i64immSExt8(Val.getNode()))
1585 Opc = X86::LOCK_SUB64mi8;
1586 else if (Predicate_i64immSExt32(Val.getNode()))
1587 Opc = X86::LOCK_SUB64mi32;
1588 }
1589 } else {
1590 Opc = X86::LOCK_ADD64mr;
1591 if (isCN) {
1592 if (Predicate_i64immSExt8(Val.getNode()))
1593 Opc = X86::LOCK_ADD64mi8;
1594 else if (Predicate_i64immSExt32(Val.getNode()))
1595 Opc = X86::LOCK_ADD64mi32;
1596 }
1597 }
1598 break;
1599 }
1600
1601 DebugLoc dl = Node->getDebugLoc();
1602 SDValue Undef = SDValue(CurDAG->getTargetNode(TargetInstrInfo::IMPLICIT_DEF,
1603 dl, NVT), 0);
1604 SDValue MemOp = CurDAG->getMemOperand(cast<MemSDNode>(Node)->getMemOperand());
1605 if (isInc || isDec) {
1606 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, MemOp, Chain };
Owen Anderson825b72b2009-08-11 20:47:22 +00001607 SDValue Ret = SDValue(CurDAG->getTargetNode(Opc, dl, MVT::Other, Ops, 7), 0);
Evan Cheng37b73872009-07-30 08:33:02 +00001608 SDValue RetVals[] = { Undef, Ret };
1609 return CurDAG->getMergeValues(RetVals, 2, dl).getNode();
1610 } else {
1611 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, Val, MemOp, Chain };
Owen Anderson825b72b2009-08-11 20:47:22 +00001612 SDValue Ret = SDValue(CurDAG->getTargetNode(Opc, dl, MVT::Other, Ops, 8), 0);
Evan Cheng37b73872009-07-30 08:33:02 +00001613 SDValue RetVals[] = { Undef, Ret };
1614 return CurDAG->getMergeValues(RetVals, 2, dl).getNode();
1615 }
1616}
1617
Dan Gohman475871a2008-07-27 21:46:04 +00001618SDNode *X86DAGToDAGISel::Select(SDValue N) {
Gabor Greifba36cb52008-08-28 21:40:38 +00001619 SDNode *Node = N.getNode();
Owen Andersone50ed302009-08-10 22:56:29 +00001620 EVT NVT = Node->getValueType(0);
Evan Cheng0114e942006-01-06 20:36:21 +00001621 unsigned Opc, MOpc;
1622 unsigned Opcode = Node->getOpcode();
Dale Johannesend8392542009-02-03 21:48:12 +00001623 DebugLoc dl = Node->getDebugLoc();
1624
Evan Chengf597dc72006-02-10 22:24:32 +00001625#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00001626 DEBUG({
1627 errs() << std::string(Indent, ' ') << "Selecting: ";
1628 Node->dump(CurDAG);
1629 errs() << '\n';
1630 });
Evan Cheng23addc02006-02-10 22:46:26 +00001631 Indent += 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001632#endif
1633
Dan Gohmane8be6c62008-07-17 19:10:17 +00001634 if (Node->isMachineOpcode()) {
Evan Chengf597dc72006-02-10 22:24:32 +00001635#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00001636 DEBUG({
1637 errs() << std::string(Indent-2, ' ') << "== ";
1638 Node->dump(CurDAG);
1639 errs() << '\n';
1640 });
Evan Cheng23addc02006-02-10 22:46:26 +00001641 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001642#endif
Evan Cheng64a752f2006-08-11 09:08:15 +00001643 return NULL; // Already selected.
Evan Cheng34167212006-02-09 00:37:58 +00001644 }
Evan Cheng38262ca2006-01-11 22:15:18 +00001645
Evan Cheng0114e942006-01-06 20:36:21 +00001646 switch (Opcode) {
Dan Gohman72677342009-08-02 16:10:52 +00001647 default: break;
1648 case X86ISD::GlobalBaseReg:
1649 return getGlobalBaseReg();
Evan Cheng020d2e82006-02-23 20:41:18 +00001650
Dan Gohman72677342009-08-02 16:10:52 +00001651 case X86ISD::ATOMOR64_DAG:
1652 return SelectAtomic64(Node, X86::ATOMOR6432);
1653 case X86ISD::ATOMXOR64_DAG:
1654 return SelectAtomic64(Node, X86::ATOMXOR6432);
1655 case X86ISD::ATOMADD64_DAG:
1656 return SelectAtomic64(Node, X86::ATOMADD6432);
1657 case X86ISD::ATOMSUB64_DAG:
1658 return SelectAtomic64(Node, X86::ATOMSUB6432);
1659 case X86ISD::ATOMNAND64_DAG:
1660 return SelectAtomic64(Node, X86::ATOMNAND6432);
1661 case X86ISD::ATOMAND64_DAG:
1662 return SelectAtomic64(Node, X86::ATOMAND6432);
1663 case X86ISD::ATOMSWAP64_DAG:
1664 return SelectAtomic64(Node, X86::ATOMSWAP6432);
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001665
Dan Gohman72677342009-08-02 16:10:52 +00001666 case ISD::ATOMIC_LOAD_ADD: {
1667 SDNode *RetVal = SelectAtomicLoadAdd(Node, NVT);
1668 if (RetVal)
1669 return RetVal;
1670 break;
1671 }
1672
1673 case ISD::SMUL_LOHI:
1674 case ISD::UMUL_LOHI: {
1675 SDValue N0 = Node->getOperand(0);
1676 SDValue N1 = Node->getOperand(1);
1677
1678 bool isSigned = Opcode == ISD::SMUL_LOHI;
Bill Wendling12321672009-08-07 21:33:25 +00001679 if (!isSigned) {
Owen Anderson825b72b2009-08-11 20:47:22 +00001680 switch (NVT.getSimpleVT().SimpleTy) {
Dan Gohman72677342009-08-02 16:10:52 +00001681 default: llvm_unreachable("Unsupported VT!");
Owen Anderson825b72b2009-08-11 20:47:22 +00001682 case MVT::i8: Opc = X86::MUL8r; MOpc = X86::MUL8m; break;
1683 case MVT::i16: Opc = X86::MUL16r; MOpc = X86::MUL16m; break;
1684 case MVT::i32: Opc = X86::MUL32r; MOpc = X86::MUL32m; break;
1685 case MVT::i64: Opc = X86::MUL64r; MOpc = X86::MUL64m; break;
Dan Gohman72677342009-08-02 16:10:52 +00001686 }
Bill Wendling12321672009-08-07 21:33:25 +00001687 } else {
Owen Anderson825b72b2009-08-11 20:47:22 +00001688 switch (NVT.getSimpleVT().SimpleTy) {
Dan Gohman72677342009-08-02 16:10:52 +00001689 default: llvm_unreachable("Unsupported VT!");
Owen Anderson825b72b2009-08-11 20:47:22 +00001690 case MVT::i8: Opc = X86::IMUL8r; MOpc = X86::IMUL8m; break;
1691 case MVT::i16: Opc = X86::IMUL16r; MOpc = X86::IMUL16m; break;
1692 case MVT::i32: Opc = X86::IMUL32r; MOpc = X86::IMUL32m; break;
1693 case MVT::i64: Opc = X86::IMUL64r; MOpc = X86::IMUL64m; break;
Dan Gohman72677342009-08-02 16:10:52 +00001694 }
Bill Wendling12321672009-08-07 21:33:25 +00001695 }
Dan Gohman72677342009-08-02 16:10:52 +00001696
1697 unsigned LoReg, HiReg;
Owen Anderson825b72b2009-08-11 20:47:22 +00001698 switch (NVT.getSimpleVT().SimpleTy) {
Dan Gohman72677342009-08-02 16:10:52 +00001699 default: llvm_unreachable("Unsupported VT!");
Owen Anderson825b72b2009-08-11 20:47:22 +00001700 case MVT::i8: LoReg = X86::AL; HiReg = X86::AH; break;
1701 case MVT::i16: LoReg = X86::AX; HiReg = X86::DX; break;
1702 case MVT::i32: LoReg = X86::EAX; HiReg = X86::EDX; break;
1703 case MVT::i64: LoReg = X86::RAX; HiReg = X86::RDX; break;
Dan Gohman72677342009-08-02 16:10:52 +00001704 }
1705
1706 SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4;
1707 bool foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4);
Bill Wendling12321672009-08-07 21:33:25 +00001708 // Multiply is commmutative.
Dan Gohman72677342009-08-02 16:10:52 +00001709 if (!foldedLoad) {
1710 foldedLoad = TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4);
1711 if (foldedLoad)
1712 std::swap(N0, N1);
1713 }
1714
1715 SDValue InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, LoReg,
1716 N0, SDValue()).getValue(1);
1717
1718 if (foldedLoad) {
1719 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, N1.getOperand(0),
1720 InFlag };
1721 SDNode *CNode =
Owen Anderson825b72b2009-08-11 20:47:22 +00001722 CurDAG->getTargetNode(MOpc, dl, MVT::Other, MVT::Flag, Ops,
Dan Gohman72677342009-08-02 16:10:52 +00001723 array_lengthof(Ops));
1724 InFlag = SDValue(CNode, 1);
1725 // Update the chain.
1726 ReplaceUses(N1.getValue(1), SDValue(CNode, 0));
1727 } else {
1728 InFlag =
Owen Anderson825b72b2009-08-11 20:47:22 +00001729 SDValue(CurDAG->getTargetNode(Opc, dl, MVT::Flag, N1, InFlag), 0);
Dan Gohman72677342009-08-02 16:10:52 +00001730 }
1731
1732 // Copy the low half of the result, if it is needed.
1733 if (!N.getValue(0).use_empty()) {
1734 SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
1735 LoReg, NVT, InFlag);
1736 InFlag = Result.getValue(2);
1737 ReplaceUses(N.getValue(0), Result);
1738#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00001739 DEBUG({
1740 errs() << std::string(Indent-2, ' ') << "=> ";
1741 Result.getNode()->dump(CurDAG);
1742 errs() << '\n';
1743 });
Dan Gohman72677342009-08-02 16:10:52 +00001744#endif
1745 }
1746 // Copy the high half of the result, if it is needed.
1747 if (!N.getValue(1).use_empty()) {
1748 SDValue Result;
1749 if (HiReg == X86::AH && Subtarget->is64Bit()) {
1750 // Prevent use of AH in a REX instruction by referencing AX instead.
1751 // Shift it down 8 bits.
1752 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
Owen Anderson825b72b2009-08-11 20:47:22 +00001753 X86::AX, MVT::i16, InFlag);
Dan Gohman72677342009-08-02 16:10:52 +00001754 InFlag = Result.getValue(2);
Owen Anderson825b72b2009-08-11 20:47:22 +00001755 Result = SDValue(CurDAG->getTargetNode(X86::SHR16ri, dl, MVT::i16,
Dan Gohman72677342009-08-02 16:10:52 +00001756 Result,
Owen Anderson825b72b2009-08-11 20:47:22 +00001757 CurDAG->getTargetConstant(8, MVT::i8)), 0);
Dan Gohman72677342009-08-02 16:10:52 +00001758 // Then truncate it down to i8.
Dan Gohman6a402dc2009-08-19 18:16:17 +00001759 Result = CurDAG->getTargetExtractSubreg(X86::SUBREG_8BIT, dl,
1760 MVT::i8, Result);
Dan Gohman72677342009-08-02 16:10:52 +00001761 } else {
1762 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
1763 HiReg, NVT, InFlag);
1764 InFlag = Result.getValue(2);
1765 }
1766 ReplaceUses(N.getValue(1), Result);
1767#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00001768 DEBUG({
1769 errs() << std::string(Indent-2, ' ') << "=> ";
1770 Result.getNode()->dump(CurDAG);
1771 errs() << '\n';
1772 });
Dan Gohman72677342009-08-02 16:10:52 +00001773#endif
1774 }
1775
1776#ifndef NDEBUG
1777 Indent -= 2;
1778#endif
1779
1780 return NULL;
1781 }
1782
1783 case ISD::SDIVREM:
1784 case ISD::UDIVREM: {
1785 SDValue N0 = Node->getOperand(0);
1786 SDValue N1 = Node->getOperand(1);
1787
1788 bool isSigned = Opcode == ISD::SDIVREM;
Bill Wendling12321672009-08-07 21:33:25 +00001789 if (!isSigned) {
Owen Anderson825b72b2009-08-11 20:47:22 +00001790 switch (NVT.getSimpleVT().SimpleTy) {
Dan Gohman72677342009-08-02 16:10:52 +00001791 default: llvm_unreachable("Unsupported VT!");
Owen Anderson825b72b2009-08-11 20:47:22 +00001792 case MVT::i8: Opc = X86::DIV8r; MOpc = X86::DIV8m; break;
1793 case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break;
1794 case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break;
1795 case MVT::i64: Opc = X86::DIV64r; MOpc = X86::DIV64m; break;
Dan Gohman72677342009-08-02 16:10:52 +00001796 }
Bill Wendling12321672009-08-07 21:33:25 +00001797 } else {
Owen Anderson825b72b2009-08-11 20:47:22 +00001798 switch (NVT.getSimpleVT().SimpleTy) {
Dan Gohman72677342009-08-02 16:10:52 +00001799 default: llvm_unreachable("Unsupported VT!");
Owen Anderson825b72b2009-08-11 20:47:22 +00001800 case MVT::i8: Opc = X86::IDIV8r; MOpc = X86::IDIV8m; break;
1801 case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break;
1802 case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break;
1803 case MVT::i64: Opc = X86::IDIV64r; MOpc = X86::IDIV64m; break;
Dan Gohman72677342009-08-02 16:10:52 +00001804 }
Bill Wendling12321672009-08-07 21:33:25 +00001805 }
Dan Gohman72677342009-08-02 16:10:52 +00001806
1807 unsigned LoReg, HiReg;
1808 unsigned ClrOpcode, SExtOpcode;
Owen Anderson825b72b2009-08-11 20:47:22 +00001809 switch (NVT.getSimpleVT().SimpleTy) {
Dan Gohman72677342009-08-02 16:10:52 +00001810 default: llvm_unreachable("Unsupported VT!");
Owen Anderson825b72b2009-08-11 20:47:22 +00001811 case MVT::i8:
Dan Gohman72677342009-08-02 16:10:52 +00001812 LoReg = X86::AL; HiReg = X86::AH;
1813 ClrOpcode = 0;
1814 SExtOpcode = X86::CBW;
1815 break;
Owen Anderson825b72b2009-08-11 20:47:22 +00001816 case MVT::i16:
Dan Gohman72677342009-08-02 16:10:52 +00001817 LoReg = X86::AX; HiReg = X86::DX;
1818 ClrOpcode = X86::MOV16r0;
1819 SExtOpcode = X86::CWD;
1820 break;
Owen Anderson825b72b2009-08-11 20:47:22 +00001821 case MVT::i32:
Dan Gohman72677342009-08-02 16:10:52 +00001822 LoReg = X86::EAX; HiReg = X86::EDX;
1823 ClrOpcode = X86::MOV32r0;
1824 SExtOpcode = X86::CDQ;
1825 break;
Owen Anderson825b72b2009-08-11 20:47:22 +00001826 case MVT::i64:
Dan Gohman72677342009-08-02 16:10:52 +00001827 LoReg = X86::RAX; HiReg = X86::RDX;
1828 ClrOpcode = ~0U; // NOT USED.
1829 SExtOpcode = X86::CQO;
Evan Cheng37b73872009-07-30 08:33:02 +00001830 break;
1831 }
1832
Dan Gohman72677342009-08-02 16:10:52 +00001833 SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4;
1834 bool foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4);
1835 bool signBitIsZero = CurDAG->SignBitIsZero(N0);
Dan Gohman525178c2007-10-08 18:33:35 +00001836
Dan Gohman72677342009-08-02 16:10:52 +00001837 SDValue InFlag;
Owen Anderson825b72b2009-08-11 20:47:22 +00001838 if (NVT == MVT::i8 && (!isSigned || signBitIsZero)) {
Dan Gohman72677342009-08-02 16:10:52 +00001839 // Special case for div8, just use a move with zero extension to AX to
1840 // clear the upper 8 bits (AH).
1841 SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, Move, Chain;
1842 if (TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4)) {
1843 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, N0.getOperand(0) };
1844 Move =
Owen Anderson825b72b2009-08-11 20:47:22 +00001845 SDValue(CurDAG->getTargetNode(X86::MOVZX16rm8, dl, MVT::i16,
1846 MVT::Other, Ops,
Dan Gohman72677342009-08-02 16:10:52 +00001847 array_lengthof(Ops)), 0);
1848 Chain = Move.getValue(1);
1849 ReplaceUses(N0.getValue(1), Chain);
Evan Cheng0114e942006-01-06 20:36:21 +00001850 } else {
Dan Gohman72677342009-08-02 16:10:52 +00001851 Move =
Owen Anderson825b72b2009-08-11 20:47:22 +00001852 SDValue(CurDAG->getTargetNode(X86::MOVZX16rr8, dl, MVT::i16, N0),0);
Dan Gohman72677342009-08-02 16:10:52 +00001853 Chain = CurDAG->getEntryNode();
1854 }
1855 Chain = CurDAG->getCopyToReg(Chain, dl, X86::AX, Move, SDValue());
1856 InFlag = Chain.getValue(1);
1857 } else {
1858 InFlag =
1859 CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl,
1860 LoReg, N0, SDValue()).getValue(1);
1861 if (isSigned && !signBitIsZero) {
1862 // Sign extend the low part into the high part.
Evan Cheng7e9b26f2006-02-09 07:17:49 +00001863 InFlag =
Owen Anderson825b72b2009-08-11 20:47:22 +00001864 SDValue(CurDAG->getTargetNode(SExtOpcode, dl, MVT::Flag, InFlag),0);
Dan Gohman72677342009-08-02 16:10:52 +00001865 } else {
1866 // Zero out the high part, effectively zero extending the input.
1867 SDValue ClrNode;
Evan Cheng0114e942006-01-06 20:36:21 +00001868
Owen Anderson825b72b2009-08-11 20:47:22 +00001869 if (NVT.getSimpleVT() == MVT::i64) {
1870 ClrNode = SDValue(CurDAG->getTargetNode(X86::MOV32r0, dl, MVT::i32),
Dan Gohman72677342009-08-02 16:10:52 +00001871 0);
1872 // We just did a 32-bit clear, insert it into a 64-bit register to
1873 // clear the whole 64-bit reg.
1874 SDValue Undef =
1875 SDValue(CurDAG->getTargetNode(TargetInstrInfo::IMPLICIT_DEF,
Owen Anderson825b72b2009-08-11 20:47:22 +00001876 dl, MVT::i64), 0);
Dan Gohman72677342009-08-02 16:10:52 +00001877 SDValue SubRegNo =
Owen Anderson825b72b2009-08-11 20:47:22 +00001878 CurDAG->getTargetConstant(X86::SUBREG_32BIT, MVT::i32);
Dan Gohman72677342009-08-02 16:10:52 +00001879 ClrNode =
1880 SDValue(CurDAG->getTargetNode(TargetInstrInfo::INSERT_SUBREG, dl,
Owen Anderson825b72b2009-08-11 20:47:22 +00001881 MVT::i64, Undef, ClrNode, SubRegNo),
Dan Gohman72677342009-08-02 16:10:52 +00001882 0);
Dan Gohman525178c2007-10-08 18:33:35 +00001883 } else {
Dan Gohman72677342009-08-02 16:10:52 +00001884 ClrNode = SDValue(CurDAG->getTargetNode(ClrOpcode, dl, NVT), 0);
Dan Gohman525178c2007-10-08 18:33:35 +00001885 }
Dan Gohman72677342009-08-02 16:10:52 +00001886
1887 InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, HiReg,
1888 ClrNode, InFlag).getValue(1);
Dan Gohman525178c2007-10-08 18:33:35 +00001889 }
Evan Cheng948f3432006-01-06 23:19:29 +00001890 }
Dan Gohman525178c2007-10-08 18:33:35 +00001891
Dan Gohman72677342009-08-02 16:10:52 +00001892 if (foldedLoad) {
1893 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, N1.getOperand(0),
1894 InFlag };
1895 SDNode *CNode =
Owen Anderson825b72b2009-08-11 20:47:22 +00001896 CurDAG->getTargetNode(MOpc, dl, MVT::Other, MVT::Flag, Ops,
Dan Gohman72677342009-08-02 16:10:52 +00001897 array_lengthof(Ops));
1898 InFlag = SDValue(CNode, 1);
1899 // Update the chain.
1900 ReplaceUses(N1.getValue(1), SDValue(CNode, 0));
1901 } else {
1902 InFlag =
Owen Anderson825b72b2009-08-11 20:47:22 +00001903 SDValue(CurDAG->getTargetNode(Opc, dl, MVT::Flag, N1, InFlag), 0);
Dan Gohman72677342009-08-02 16:10:52 +00001904 }
Evan Cheng948f3432006-01-06 23:19:29 +00001905
Dan Gohman72677342009-08-02 16:10:52 +00001906 // Copy the division (low) result, if it is needed.
1907 if (!N.getValue(0).use_empty()) {
1908 SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
1909 LoReg, NVT, InFlag);
1910 InFlag = Result.getValue(2);
1911 ReplaceUses(N.getValue(0), Result);
1912#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00001913 DEBUG({
1914 errs() << std::string(Indent-2, ' ') << "=> ";
1915 Result.getNode()->dump(CurDAG);
1916 errs() << '\n';
1917 });
Dan Gohman72677342009-08-02 16:10:52 +00001918#endif
1919 }
1920 // Copy the remainder (high) result, if it is needed.
1921 if (!N.getValue(1).use_empty()) {
1922 SDValue Result;
1923 if (HiReg == X86::AH && Subtarget->is64Bit()) {
1924 // Prevent use of AH in a REX instruction by referencing AX instead.
1925 // Shift it down 8 bits.
1926 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
Owen Anderson825b72b2009-08-11 20:47:22 +00001927 X86::AX, MVT::i16, InFlag);
Dan Gohmana37c9f72007-09-25 18:23:27 +00001928 InFlag = Result.getValue(2);
Owen Anderson825b72b2009-08-11 20:47:22 +00001929 Result = SDValue(CurDAG->getTargetNode(X86::SHR16ri, dl, MVT::i16,
Dan Gohman72677342009-08-02 16:10:52 +00001930 Result,
Owen Anderson825b72b2009-08-11 20:47:22 +00001931 CurDAG->getTargetConstant(8, MVT::i8)),
Dan Gohman72677342009-08-02 16:10:52 +00001932 0);
1933 // Then truncate it down to i8.
Dan Gohman6a402dc2009-08-19 18:16:17 +00001934 Result = CurDAG->getTargetExtractSubreg(X86::SUBREG_8BIT, dl,
1935 MVT::i8, Result);
Dan Gohman72677342009-08-02 16:10:52 +00001936 } else {
1937 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl,
1938 HiReg, NVT, InFlag);
1939 InFlag = Result.getValue(2);
Evan Chengf7ef26e2007-08-09 21:59:35 +00001940 }
Dan Gohman72677342009-08-02 16:10:52 +00001941 ReplaceUses(N.getValue(1), Result);
Dan Gohmana37c9f72007-09-25 18:23:27 +00001942#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00001943 DEBUG({
1944 errs() << std::string(Indent-2, ' ') << "=> ";
1945 Result.getNode()->dump(CurDAG);
1946 errs() << '\n';
1947 });
Dan Gohmana37c9f72007-09-25 18:23:27 +00001948#endif
Dan Gohman72677342009-08-02 16:10:52 +00001949 }
Evan Chengf597dc72006-02-10 22:24:32 +00001950
1951#ifndef NDEBUG
Dan Gohman72677342009-08-02 16:10:52 +00001952 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001953#endif
Evan Cheng64a752f2006-08-11 09:08:15 +00001954
Dan Gohman72677342009-08-02 16:10:52 +00001955 return NULL;
1956 }
1957
Dan Gohman6a402dc2009-08-19 18:16:17 +00001958 case X86ISD::CMP: {
Dan Gohman6a402dc2009-08-19 18:16:17 +00001959 SDValue N0 = Node->getOperand(0);
1960 SDValue N1 = Node->getOperand(1);
1961
1962 // Look for (X86cmp (and $op, $imm), 0) and see if we can convert it to
1963 // use a smaller encoding.
1964 if (N0.getNode()->getOpcode() == ISD::AND && N0.getNode()->hasOneUse() &&
1965 N0.getValueType() != MVT::i8 &&
1966 X86::isZeroNode(N1)) {
1967 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N0.getNode()->getOperand(1));
1968 if (!C) break;
1969
1970 // For example, convert "testl %eax, $8" to "testb %al, $8"
1971 if ((C->getZExtValue() & ~UINT64_C(0xff)) == 0) {
1972 SDValue Imm = CurDAG->getTargetConstant(C->getZExtValue(), MVT::i8);
1973 SDValue Reg = N0.getNode()->getOperand(0);
1974
1975 // On x86-32, only the ABCD registers have 8-bit subregisters.
1976 if (!Subtarget->is64Bit()) {
1977 TargetRegisterClass *TRC = 0;
1978 switch (N0.getValueType().getSimpleVT().SimpleTy) {
1979 case MVT::i32: TRC = &X86::GR32_ABCDRegClass; break;
1980 case MVT::i16: TRC = &X86::GR16_ABCDRegClass; break;
1981 default: llvm_unreachable("Unsupported TEST operand type!");
1982 }
1983 SDValue RC = CurDAG->getTargetConstant(TRC->getID(), MVT::i32);
1984 Reg = SDValue(CurDAG->getTargetNode(X86::COPY_TO_REGCLASS, dl,
1985 Reg.getValueType(), Reg, RC), 0);
1986 }
1987
1988 // Extract the l-register.
1989 SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::SUBREG_8BIT, dl,
1990 MVT::i8, Reg);
1991
1992 // Emit a testb.
1993 return CurDAG->getTargetNode(X86::TEST8ri, dl, MVT::i32, Subreg, Imm);
1994 }
1995
1996 // For example, "testl %eax, $2048" to "testb %ah, $8".
1997 if ((C->getZExtValue() & ~UINT64_C(0xff00)) == 0) {
1998 // Shift the immediate right by 8 bits.
1999 SDValue ShiftedImm = CurDAG->getTargetConstant(C->getZExtValue() >> 8,
2000 MVT::i8);
2001 SDValue Reg = N0.getNode()->getOperand(0);
2002
2003 // Put the value in an ABCD register.
2004 TargetRegisterClass *TRC = 0;
2005 switch (N0.getValueType().getSimpleVT().SimpleTy) {
2006 case MVT::i64: TRC = &X86::GR64_ABCDRegClass; break;
2007 case MVT::i32: TRC = &X86::GR32_ABCDRegClass; break;
2008 case MVT::i16: TRC = &X86::GR16_ABCDRegClass; break;
2009 default: llvm_unreachable("Unsupported TEST operand type!");
2010 }
2011 SDValue RC = CurDAG->getTargetConstant(TRC->getID(), MVT::i32);
2012 Reg = SDValue(CurDAG->getTargetNode(X86::COPY_TO_REGCLASS, dl,
2013 Reg.getValueType(), Reg, RC), 0);
2014
2015 // Extract the h-register.
2016 SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::SUBREG_8BIT_HI, dl,
2017 MVT::i8, Reg);
2018
2019 // Emit a testb. No special NOREX tricks are needed since there's
2020 // only one GPR operand!
2021 return CurDAG->getTargetNode(X86::TEST8ri, dl, MVT::i32,
2022 Subreg, ShiftedImm);
2023 }
2024
2025 // For example, "testl %eax, $32776" to "testw %ax, $32776".
2026 if ((C->getZExtValue() & ~UINT64_C(0xffff)) == 0 &&
2027 N0.getValueType() != MVT::i16) {
2028 SDValue Imm = CurDAG->getTargetConstant(C->getZExtValue(), MVT::i16);
2029 SDValue Reg = N0.getNode()->getOperand(0);
2030
2031 // Extract the 16-bit subregister.
2032 SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::SUBREG_16BIT, dl,
2033 MVT::i16, Reg);
2034
2035 // Emit a testw.
2036 return CurDAG->getTargetNode(X86::TEST16ri, dl, MVT::i32, Subreg, Imm);
2037 }
2038
2039 // For example, "testq %rax, $268468232" to "testl %eax, $268468232".
2040 if ((C->getZExtValue() & ~UINT64_C(0xffffffff)) == 0 &&
2041 N0.getValueType() == MVT::i64) {
2042 SDValue Imm = CurDAG->getTargetConstant(C->getZExtValue(), MVT::i32);
2043 SDValue Reg = N0.getNode()->getOperand(0);
2044
2045 // Extract the 32-bit subregister.
2046 SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::SUBREG_32BIT, dl,
2047 MVT::i32, Reg);
2048
2049 // Emit a testl.
2050 return CurDAG->getTargetNode(X86::TEST32ri, dl, MVT::i32, Subreg, Imm);
2051 }
2052 }
2053 break;
2054 }
Chris Lattnerc961eea2005-11-16 01:54:32 +00002055 }
2056
Evan Cheng9ade2182006-08-26 05:34:46 +00002057 SDNode *ResNode = SelectCode(N);
Evan Cheng64a752f2006-08-11 09:08:15 +00002058
Evan Chengf597dc72006-02-10 22:24:32 +00002059#ifndef NDEBUG
Bill Wendling12321672009-08-07 21:33:25 +00002060 DEBUG({
2061 errs() << std::string(Indent-2, ' ') << "=> ";
2062 if (ResNode == NULL || ResNode == N.getNode())
2063 N.getNode()->dump(CurDAG);
2064 else
2065 ResNode->dump(CurDAG);
2066 errs() << '\n';
2067 });
Evan Cheng23addc02006-02-10 22:46:26 +00002068 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00002069#endif
Evan Cheng64a752f2006-08-11 09:08:15 +00002070
2071 return ResNode;
Chris Lattnerc961eea2005-11-16 01:54:32 +00002072}
2073
Chris Lattnerc0bad572006-06-08 18:03:49 +00002074bool X86DAGToDAGISel::
Dan Gohman475871a2008-07-27 21:46:04 +00002075SelectInlineAsmMemoryOperand(const SDValue &Op, char ConstraintCode,
Dan Gohmanf350b272008-08-23 02:25:05 +00002076 std::vector<SDValue> &OutOps) {
Rafael Espindola094fad32009-04-08 21:14:34 +00002077 SDValue Op0, Op1, Op2, Op3, Op4;
Chris Lattnerc0bad572006-06-08 18:03:49 +00002078 switch (ConstraintCode) {
2079 case 'o': // offsetable ??
2080 case 'v': // not offsetable ??
2081 default: return true;
2082 case 'm': // memory
Rafael Espindola094fad32009-04-08 21:14:34 +00002083 if (!SelectAddr(Op, Op, Op0, Op1, Op2, Op3, Op4))
Chris Lattnerc0bad572006-06-08 18:03:49 +00002084 return true;
2085 break;
2086 }
2087
Evan Cheng04699902006-08-26 01:05:16 +00002088 OutOps.push_back(Op0);
2089 OutOps.push_back(Op1);
2090 OutOps.push_back(Op2);
2091 OutOps.push_back(Op3);
Rafael Espindola094fad32009-04-08 21:14:34 +00002092 OutOps.push_back(Op4);
Chris Lattnerc0bad572006-06-08 18:03:49 +00002093 return false;
2094}
2095
Chris Lattnerc961eea2005-11-16 01:54:32 +00002096/// createX86ISelDag - This pass converts a legalized DAG into a
2097/// X86-specific DAG, ready for instruction scheduling.
2098///
Bill Wendling98a366d2009-04-29 23:29:43 +00002099FunctionPass *llvm::createX86ISelDag(X86TargetMachine &TM,
2100 llvm::CodeGenOpt::Level OptLevel) {
Bill Wendlingbe8cc2a2009-04-29 00:15:41 +00002101 return new X86DAGToDAGISel(TM, OptLevel);
Chris Lattnerc961eea2005-11-16 01:54:32 +00002102}