blob: 26486e019bf6669c1c102a1b76578b6efec03a3f [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"
38#include "llvm/Support/MathExtras.h"
Dale Johannesen50dd1d02008-08-11 23:46:25 +000039#include "llvm/Support/Streams.h"
Evan Chengcdda25d2008-04-25 08:22:20 +000040#include "llvm/ADT/SmallPtrSet.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000041#include "llvm/ADT/Statistic.h"
42using namespace llvm;
43
Chris Lattner95b2c7d2006-12-19 22:59:26 +000044STATISTIC(NumLoadMoved, "Number of loads moved below TokenFactor");
45
Chris Lattnerc961eea2005-11-16 01:54:32 +000046//===----------------------------------------------------------------------===//
47// Pattern Matcher Implementation
48//===----------------------------------------------------------------------===//
49
50namespace {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000051 /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses
Dan Gohman475871a2008-07-27 21:46:04 +000052 /// SDValue's instead of register numbers for the leaves of the matched
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000053 /// tree.
54 struct X86ISelAddressMode {
55 enum {
56 RegBase,
Chris Lattnerd74ea2b2006-05-24 17:04:05 +000057 FrameIndexBase
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000058 } BaseType;
59
60 struct { // This is really a union, discriminated by BaseType!
Dan Gohman475871a2008-07-27 21:46:04 +000061 SDValue Reg;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000062 int FrameIndex;
63 } Base;
64
Evan Chengbe3bf422008-02-07 08:53:49 +000065 bool isRIPRel; // RIP as base?
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000066 unsigned Scale;
Dan Gohman475871a2008-07-27 21:46:04 +000067 SDValue IndexReg;
Dan Gohman27cae7b2008-11-11 15:52:29 +000068 int32_t Disp;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000069 GlobalValue *GV;
Evan Cheng51a9ed92006-02-25 10:09:08 +000070 Constant *CP;
Evan Cheng25ab6902006-09-08 06:48:29 +000071 const char *ES;
72 int JT;
Evan Cheng51a9ed92006-02-25 10:09:08 +000073 unsigned Align; // CP alignment.
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000074
75 X86ISelAddressMode()
Evan Cheng25ab6902006-09-08 06:48:29 +000076 : BaseType(RegBase), isRIPRel(false), Scale(1), IndexReg(), Disp(0),
77 GV(0), CP(0), ES(0), JT(-1), Align(0) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000078 }
Dale Johannesen50dd1d02008-08-11 23:46:25 +000079 void dump() {
80 cerr << "X86ISelAddressMode " << this << "\n";
Gabor Greif93c53e52008-08-31 15:37:04 +000081 cerr << "Base.Reg ";
82 if (Base.Reg.getNode() != 0) Base.Reg.getNode()->dump();
83 else cerr << "nul";
Dale Johannesen50dd1d02008-08-11 23:46:25 +000084 cerr << " Base.FrameIndex " << Base.FrameIndex << "\n";
85 cerr << "isRIPRel " << isRIPRel << " Scale" << Scale << "\n";
Gabor Greif93c53e52008-08-31 15:37:04 +000086 cerr << "IndexReg ";
87 if (IndexReg.getNode() != 0) IndexReg.getNode()->dump();
88 else cerr << "nul";
Dale Johannesen50dd1d02008-08-11 23:46:25 +000089 cerr << " Disp " << Disp << "\n";
90 cerr << "GV "; if (GV) GV->dump();
91 else cerr << "nul";
92 cerr << " CP "; if (CP) CP->dump();
93 else cerr << "nul";
94 cerr << "\n";
95 cerr << "ES "; if (ES) cerr << ES; else cerr << "nul";
96 cerr << " JT" << JT << " Align" << Align << "\n";
97 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000098 };
99}
100
101namespace {
Chris Lattnerc961eea2005-11-16 01:54:32 +0000102 //===--------------------------------------------------------------------===//
103 /// ISel - X86 specific code to select X86 machine instructions for
104 /// SelectionDAG operations.
105 ///
Chris Lattner2c79de82006-06-28 23:27:49 +0000106 class VISIBILITY_HIDDEN X86DAGToDAGISel : public SelectionDAGISel {
Evan Cheng25ab6902006-09-08 06:48:29 +0000107 /// TM - Keep a reference to X86TargetMachine.
108 ///
109 X86TargetMachine &TM;
110
Chris Lattnerc961eea2005-11-16 01:54:32 +0000111 /// X86Lowering - This object fully describes how to lower LLVM code to an
112 /// X86-specific SelectionDAG.
Dan Gohmanda8ac5f2008-10-03 16:55:19 +0000113 X86TargetLowering &X86Lowering;
Chris Lattnerc961eea2005-11-16 01:54:32 +0000114
115 /// Subtarget - Keep a pointer to the X86Subtarget around so that we can
116 /// make the right decision when generating code for different targets.
117 const X86Subtarget *Subtarget;
Evan Cheng7ccced62006-02-18 00:15:05 +0000118
Evan Chengdb8d56b2008-06-30 20:45:06 +0000119 /// CurBB - Current BB being isel'd.
120 ///
121 MachineBasicBlock *CurBB;
122
Evan Chengb7a75a52008-09-26 23:41:32 +0000123 /// OptForSize - If true, selector should try to optimize for code size
124 /// instead of performance.
125 bool OptForSize;
126
Chris Lattnerc961eea2005-11-16 01:54:32 +0000127 public:
Evan Cheng25ab6902006-09-08 06:48:29 +0000128 X86DAGToDAGISel(X86TargetMachine &tm, bool fast)
Dan Gohman79ce2762009-01-15 19:20:50 +0000129 : SelectionDAGISel(tm, fast),
Dan Gohman38217fe2008-10-03 16:17:33 +0000130 TM(tm), X86Lowering(*TM.getTargetLowering()),
Evan Chengb7a75a52008-09-26 23:41:32 +0000131 Subtarget(&TM.getSubtarget<X86Subtarget>()),
Devang Patel4ae641f2008-10-01 23:18:38 +0000132 OptForSize(false) {}
Chris Lattnerc961eea2005-11-16 01:54:32 +0000133
134 virtual const char *getPassName() const {
135 return "X86 DAG->DAG Instruction Selection";
136 }
137
Evan Chengdb8d56b2008-06-30 20:45:06 +0000138 /// InstructionSelect - This callback is invoked by
Chris Lattnerc961eea2005-11-16 01:54:32 +0000139 /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
Dan Gohmanf350b272008-08-23 02:25:05 +0000140 virtual void InstructionSelect();
Evan Chengdb8d56b2008-06-30 20:45:06 +0000141
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000142 virtual void EmitFunctionEntryCode(Function &Fn, MachineFunction &MF);
143
Evan Cheng884c70c2008-11-27 00:49:46 +0000144 virtual
145 bool IsLegalAndProfitableToFold(SDNode *N, SDNode *U, SDNode *Root) const;
Evan Chenga8df1b42006-07-27 16:44:36 +0000146
Chris Lattnerc961eea2005-11-16 01:54:32 +0000147// Include the pieces autogenerated from the target description.
148#include "X86GenDAGISel.inc"
149
150 private:
Dan Gohman475871a2008-07-27 21:46:04 +0000151 SDNode *Select(SDValue N);
Dale Johannesen48c1bc22008-10-02 18:53:47 +0000152 SDNode *SelectAtomic64(SDNode *Node, unsigned Opc);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000153
Dan Gohman475871a2008-07-27 21:46:04 +0000154 bool MatchAddress(SDValue N, X86ISelAddressMode &AM,
Anton Korobeynikovf6e93532007-03-28 18:38:33 +0000155 bool isRoot = true, unsigned Depth = 0);
Dan Gohman475871a2008-07-27 21:46:04 +0000156 bool MatchAddressBase(SDValue N, X86ISelAddressMode &AM,
Dan Gohmanbadb2d22007-08-13 20:03:06 +0000157 bool isRoot, unsigned Depth);
Dan Gohman475871a2008-07-27 21:46:04 +0000158 bool SelectAddr(SDValue Op, SDValue N, SDValue &Base,
159 SDValue &Scale, SDValue &Index, SDValue &Disp);
160 bool SelectLEAAddr(SDValue Op, SDValue N, SDValue &Base,
161 SDValue &Scale, SDValue &Index, SDValue &Disp);
162 bool SelectScalarSSELoad(SDValue Op, SDValue Pred,
163 SDValue N, SDValue &Base, SDValue &Scale,
164 SDValue &Index, SDValue &Disp,
165 SDValue &InChain, SDValue &OutChain);
166 bool TryFoldLoad(SDValue P, SDValue N,
167 SDValue &Base, SDValue &Scale,
168 SDValue &Index, SDValue &Disp);
Dan Gohmanf350b272008-08-23 02:25:05 +0000169 void PreprocessForRMW();
170 void PreprocessForFPConvert();
Evan Cheng2ef88a02006-08-07 22:28:20 +0000171
Chris Lattnerc0bad572006-06-08 18:03:49 +0000172 /// SelectInlineAsmMemoryOperand - Implement addressing mode selection for
173 /// inline asm expressions.
Dan Gohman475871a2008-07-27 21:46:04 +0000174 virtual bool SelectInlineAsmMemoryOperand(const SDValue &Op,
Chris Lattnerc0bad572006-06-08 18:03:49 +0000175 char ConstraintCode,
Dan Gohmanf350b272008-08-23 02:25:05 +0000176 std::vector<SDValue> &OutOps);
Chris Lattnerc0bad572006-06-08 18:03:49 +0000177
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000178 void EmitSpecialCodeForMain(MachineBasicBlock *BB, MachineFrameInfo *MFI);
179
Dan Gohman475871a2008-07-27 21:46:04 +0000180 inline void getAddressOperands(X86ISelAddressMode &AM, SDValue &Base,
181 SDValue &Scale, SDValue &Index,
182 SDValue &Disp) {
Evan Chenge5280532005-12-12 21:49:40 +0000183 Base = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) ?
Evan Cheng25ab6902006-09-08 06:48:29 +0000184 CurDAG->getTargetFrameIndex(AM.Base.FrameIndex, TLI.getPointerTy()) :
185 AM.Base.Reg;
Evan Chengbdce7b42005-12-17 09:13:43 +0000186 Scale = getI8Imm(AM.Scale);
Evan Chenge5280532005-12-12 21:49:40 +0000187 Index = AM.IndexReg;
Evan Cheng25ab6902006-09-08 06:48:29 +0000188 // These are 32-bit even in 64-bit mode since RIP relative offset
189 // is 32-bit.
190 if (AM.GV)
191 Disp = CurDAG->getTargetGlobalAddress(AM.GV, MVT::i32, AM.Disp);
192 else if (AM.CP)
Gabor Greif93c53e52008-08-31 15:37:04 +0000193 Disp = CurDAG->getTargetConstantPool(AM.CP, MVT::i32,
194 AM.Align, AM.Disp);
Evan Cheng25ab6902006-09-08 06:48:29 +0000195 else if (AM.ES)
Bill Wendling056292f2008-09-16 21:48:12 +0000196 Disp = CurDAG->getTargetExternalSymbol(AM.ES, MVT::i32);
Evan Cheng25ab6902006-09-08 06:48:29 +0000197 else if (AM.JT != -1)
198 Disp = CurDAG->getTargetJumpTable(AM.JT, MVT::i32);
199 else
Dan Gohman27cae7b2008-11-11 15:52:29 +0000200 Disp = CurDAG->getTargetConstant(AM.Disp, MVT::i32);
Evan Chenge5280532005-12-12 21:49:40 +0000201 }
202
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000203 /// getI8Imm - Return a target constant with the specified value, of type
204 /// i8.
Dan Gohman475871a2008-07-27 21:46:04 +0000205 inline SDValue getI8Imm(unsigned Imm) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000206 return CurDAG->getTargetConstant(Imm, MVT::i8);
207 }
208
Chris Lattnerc961eea2005-11-16 01:54:32 +0000209 /// getI16Imm - Return a target constant with the specified value, of type
210 /// i16.
Dan Gohman475871a2008-07-27 21:46:04 +0000211 inline SDValue getI16Imm(unsigned Imm) {
Chris Lattnerc961eea2005-11-16 01:54:32 +0000212 return CurDAG->getTargetConstant(Imm, MVT::i16);
213 }
214
215 /// getI32Imm - Return a target constant with the specified value, of type
216 /// i32.
Dan Gohman475871a2008-07-27 21:46:04 +0000217 inline SDValue getI32Imm(unsigned Imm) {
Chris Lattnerc961eea2005-11-16 01:54:32 +0000218 return CurDAG->getTargetConstant(Imm, MVT::i32);
219 }
Evan Chengf597dc72006-02-10 22:24:32 +0000220
Dan Gohman8b746962008-09-23 18:22:58 +0000221 /// getGlobalBaseReg - Return an SDNode that returns the value of
222 /// the global base register. Output instructions required to
223 /// initialize the global base register, if necessary.
224 ///
Evan Cheng9ade2182006-08-26 05:34:46 +0000225 SDNode *getGlobalBaseReg();
Evan Cheng7ccced62006-02-18 00:15:05 +0000226
Dan Gohman0bfa1bf2008-08-20 21:27:32 +0000227 /// getTruncateTo8Bit - return an SDNode that implements a subreg based
228 /// truncate of the specified operand to i8. This can be done with tablegen,
229 /// except that this code uses MVT::Flag in a tricky way that happens to
230 /// improve scheduling in some cases.
231 SDNode *getTruncateTo8Bit(SDValue N0);
Christopher Lambc59e5212007-08-10 21:48:46 +0000232
Evan Cheng23addc02006-02-10 22:46:26 +0000233#ifndef NDEBUG
234 unsigned Indent;
235#endif
Chris Lattnerc961eea2005-11-16 01:54:32 +0000236 };
237}
238
Gabor Greif93c53e52008-08-31 15:37:04 +0000239/// findFlagUse - Return use of MVT::Flag value produced by the specified
240/// SDNode.
Evan Chengcdda25d2008-04-25 08:22:20 +0000241///
Evan Chenga275ecb2006-10-10 01:46:56 +0000242static SDNode *findFlagUse(SDNode *N) {
243 unsigned FlagResNo = N->getNumValues()-1;
244 for (SDNode::use_iterator I = N->use_begin(), E = N->use_end(); I != E; ++I) {
Dan Gohmane8ecf482009-01-27 02:37:43 +0000245 SDUse &Use = I.getUse();
246 if (Use.getResNo() == FlagResNo)
247 return Use.getUser();
Evan Chenga275ecb2006-10-10 01:46:56 +0000248 }
249 return NULL;
250}
251
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000252/// findNonImmUse - Return true if "Use" is a non-immediate use of "Def".
253/// This function recursively traverses up the operand chain, ignoring
254/// certain nodes.
255static bool findNonImmUse(SDNode *Use, SDNode* Def, SDNode *ImmedUse,
256 SDNode *Root,
Evan Chengcdda25d2008-04-25 08:22:20 +0000257 SmallPtrSet<SDNode*, 16> &Visited) {
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000258 if (Use->getNodeId() < Def->getNodeId() ||
Evan Chengcdda25d2008-04-25 08:22:20 +0000259 !Visited.insert(Use))
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000260 return false;
261
262 for (unsigned i = 0, e = Use->getNumOperands(); i != e; ++i) {
Gabor Greifba36cb52008-08-28 21:40:38 +0000263 SDNode *N = Use->getOperand(i).getNode();
Evan Cheng27e1fe92006-10-14 08:33:25 +0000264 if (N == Def) {
Dan Gohman682d5a82008-09-17 01:39:10 +0000265 if (Use == ImmedUse || Use == Root)
Evan Cheng419ace92008-04-25 08:55:28 +0000266 continue; // We are not looking for immediate use.
Dan Gohman682d5a82008-09-17 01:39:10 +0000267 assert(N != Root);
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000268 return true;
Evan Chengf4b4c412006-08-08 00:31:00 +0000269 }
Evan Chengcdda25d2008-04-25 08:22:20 +0000270
271 // Traverse up the operand chain.
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000272 if (findNonImmUse(N, Def, ImmedUse, Root, Visited))
273 return true;
Evan Chengf4b4c412006-08-08 00:31:00 +0000274 }
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000275 return false;
Evan Chengf4b4c412006-08-08 00:31:00 +0000276}
277
Evan Cheng27e1fe92006-10-14 08:33:25 +0000278/// isNonImmUse - Start searching from Root up the DAG to check is Def can
279/// be reached. Return true if that's the case. However, ignore direct uses
280/// by ImmedUse (which would be U in the example illustrated in
Evan Cheng884c70c2008-11-27 00:49:46 +0000281/// IsLegalAndProfitableToFold) and by Root (which can happen in the store
282/// case).
Evan Cheng27e1fe92006-10-14 08:33:25 +0000283/// FIXME: to be really generic, we should allow direct use by any node
284/// that is being folded. But realisticly since we only fold loads which
285/// have one non-chain use, we only need to watch out for load/op/store
286/// and load/op/cmp case where the root (store / cmp) may reach the load via
287/// its chain operand.
Dan Gohman682d5a82008-09-17 01:39:10 +0000288static inline bool isNonImmUse(SDNode *Root, SDNode *Def, SDNode *ImmedUse) {
Evan Chengcdda25d2008-04-25 08:22:20 +0000289 SmallPtrSet<SDNode*, 16> Visited;
Dan Gohmanc03e9a12009-01-27 19:04:30 +0000290 return findNonImmUse(Root, Def, ImmedUse, Root, Visited);
Evan Chengf4b4c412006-08-08 00:31:00 +0000291}
292
293
Evan Cheng884c70c2008-11-27 00:49:46 +0000294bool X86DAGToDAGISel::IsLegalAndProfitableToFold(SDNode *N, SDNode *U,
295 SDNode *Root) const {
Dan Gohmanea9587b2008-08-13 19:55:00 +0000296 if (Fast) return false;
Evan Cheng27e1fe92006-10-14 08:33:25 +0000297
Evan Cheng884c70c2008-11-27 00:49:46 +0000298 if (U == Root)
299 switch (U->getOpcode()) {
300 default: break;
301 case ISD::ADD:
302 case ISD::ADDC:
303 case ISD::ADDE:
304 case ISD::AND:
305 case ISD::OR:
306 case ISD::XOR: {
307 // If the other operand is a 8-bit immediate we should fold the immediate
308 // instead. This reduces code size.
309 // e.g.
310 // movl 4(%esp), %eax
311 // addl $4, %eax
312 // vs.
313 // movl $4, %eax
314 // addl 4(%esp), %eax
315 // The former is 2 bytes shorter. In case where the increment is 1, then
316 // the saving can be 4 bytes (by using incl %eax).
317 ConstantSDNode *Imm = dyn_cast<ConstantSDNode>(U->getOperand(1));
318 if (Imm) {
319 if (U->getValueType(0) == MVT::i64) {
320 if ((int32_t)Imm->getZExtValue() == (int64_t)Imm->getZExtValue())
321 return false;
322 } else {
323 if ((int8_t)Imm->getZExtValue() == (int64_t)Imm->getZExtValue())
324 return false;
325 }
326 }
327 }
328 }
329
Dan Gohman682d5a82008-09-17 01:39:10 +0000330 // If Root use can somehow reach N through a path that that doesn't contain
331 // U then folding N would create a cycle. e.g. In the following
332 // diagram, Root can reach N through X. If N is folded into into Root, then
333 // X is both a predecessor and a successor of U.
Evan Chenga8df1b42006-07-27 16:44:36 +0000334 //
Dan Gohman682d5a82008-09-17 01:39:10 +0000335 // [N*] //
336 // ^ ^ //
337 // / \ //
338 // [U*] [X]? //
339 // ^ ^ //
340 // \ / //
341 // \ / //
342 // [Root*] //
343 //
344 // * indicates nodes to be folded together.
345 //
346 // If Root produces a flag, then it gets (even more) interesting. Since it
347 // will be "glued" together with its flag use in the scheduler, we need to
348 // check if it might reach N.
349 //
350 // [N*] //
351 // ^ ^ //
352 // / \ //
353 // [U*] [X]? //
354 // ^ ^ //
355 // \ \ //
356 // \ | //
357 // [Root*] | //
358 // ^ | //
359 // f | //
360 // | / //
361 // [Y] / //
362 // ^ / //
363 // f / //
364 // | / //
365 // [FU] //
366 //
367 // If FU (flag use) indirectly reaches N (the load), and Root folds N
368 // (call it Fold), then X is a predecessor of FU and a successor of
369 // Fold. But since Fold and FU are flagged together, this will create
370 // a cycle in the scheduling graph.
Evan Cheng27e1fe92006-10-14 08:33:25 +0000371
Duncan Sands83ec4b62008-06-06 12:08:01 +0000372 MVT VT = Root->getValueType(Root->getNumValues()-1);
Dan Gohman682d5a82008-09-17 01:39:10 +0000373 while (VT == MVT::Flag) {
Evan Cheng27e1fe92006-10-14 08:33:25 +0000374 SDNode *FU = findFlagUse(Root);
375 if (FU == NULL)
376 break;
Dan Gohman682d5a82008-09-17 01:39:10 +0000377 Root = FU;
Evan Cheng27e1fe92006-10-14 08:33:25 +0000378 VT = Root->getValueType(Root->getNumValues()-1);
Evan Chenga275ecb2006-10-10 01:46:56 +0000379 }
Evan Cheng27e1fe92006-10-14 08:33:25 +0000380
Dan Gohman682d5a82008-09-17 01:39:10 +0000381 return !isNonImmUse(Root, N, U);
Evan Chenga8df1b42006-07-27 16:44:36 +0000382}
383
Evan Cheng70e674e2006-08-28 20:10:17 +0000384/// MoveBelowTokenFactor - Replace TokenFactor operand with load's chain operand
385/// and move load below the TokenFactor. Replace store's chain operand with
386/// load's chain result.
Dan Gohmanf350b272008-08-23 02:25:05 +0000387static void MoveBelowTokenFactor(SelectionDAG *CurDAG, SDValue Load,
Dan Gohman475871a2008-07-27 21:46:04 +0000388 SDValue Store, SDValue TF) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000389 SmallVector<SDValue, 4> Ops;
Gabor Greifba36cb52008-08-28 21:40:38 +0000390 for (unsigned i = 0, e = TF.getNode()->getNumOperands(); i != e; ++i)
391 if (Load.getNode() == TF.getOperand(i).getNode())
Evan Chengab6c3bb2008-08-25 21:27:18 +0000392 Ops.push_back(Load.getOperand(0));
Evan Cheng70e674e2006-08-28 20:10:17 +0000393 else
Evan Chengab6c3bb2008-08-25 21:27:18 +0000394 Ops.push_back(TF.getOperand(i));
Dan Gohmanf350b272008-08-23 02:25:05 +0000395 CurDAG->UpdateNodeOperands(TF, &Ops[0], Ops.size());
396 CurDAG->UpdateNodeOperands(Load, TF, Load.getOperand(1), Load.getOperand(2));
397 CurDAG->UpdateNodeOperands(Store, Load.getValue(1), Store.getOperand(1),
398 Store.getOperand(2), Store.getOperand(3));
Evan Cheng70e674e2006-08-28 20:10:17 +0000399}
400
Evan Chengcd0baf22008-05-23 21:23:16 +0000401/// isRMWLoad - Return true if N is a load that's part of RMW sub-DAG.
402///
Dan Gohman475871a2008-07-27 21:46:04 +0000403static bool isRMWLoad(SDValue N, SDValue Chain, SDValue Address,
404 SDValue &Load) {
Evan Chengcd0baf22008-05-23 21:23:16 +0000405 if (N.getOpcode() == ISD::BIT_CONVERT)
406 N = N.getOperand(0);
407
408 LoadSDNode *LD = dyn_cast<LoadSDNode>(N);
409 if (!LD || LD->isVolatile())
410 return false;
411 if (LD->getAddressingMode() != ISD::UNINDEXED)
412 return false;
413
414 ISD::LoadExtType ExtType = LD->getExtensionType();
415 if (ExtType != ISD::NON_EXTLOAD && ExtType != ISD::EXTLOAD)
416 return false;
417
418 if (N.hasOneUse() &&
419 N.getOperand(1) == Address &&
Gabor Greifba36cb52008-08-28 21:40:38 +0000420 N.getNode()->isOperandOf(Chain.getNode())) {
Evan Chengcd0baf22008-05-23 21:23:16 +0000421 Load = N;
422 return true;
423 }
424 return false;
425}
426
Evan Chengab6c3bb2008-08-25 21:27:18 +0000427/// MoveBelowCallSeqStart - Replace CALLSEQ_START operand with load's chain
428/// operand and move load below the call's chain operand.
429static void MoveBelowCallSeqStart(SelectionDAG *CurDAG, SDValue Load,
Evan Cheng5b2e5892009-01-26 18:43:34 +0000430 SDValue Call, SDValue CallSeqStart) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000431 SmallVector<SDValue, 8> Ops;
Evan Cheng5b2e5892009-01-26 18:43:34 +0000432 SDValue Chain = CallSeqStart.getOperand(0);
433 if (Chain.getNode() == Load.getNode())
434 Ops.push_back(Load.getOperand(0));
435 else {
436 assert(Chain.getOpcode() == ISD::TokenFactor &&
437 "Unexpected CallSeqStart chain operand");
438 for (unsigned i = 0, e = Chain.getNumOperands(); i != e; ++i)
439 if (Chain.getOperand(i).getNode() == Load.getNode())
440 Ops.push_back(Load.getOperand(0));
441 else
442 Ops.push_back(Chain.getOperand(i));
443 SDValue NewChain =
444 CurDAG->getNode(ISD::TokenFactor, MVT::Other, &Ops[0], Ops.size());
445 Ops.clear();
446 Ops.push_back(NewChain);
447 }
448 for (unsigned i = 1, e = CallSeqStart.getNumOperands(); i != e; ++i)
449 Ops.push_back(CallSeqStart.getOperand(i));
450 CurDAG->UpdateNodeOperands(CallSeqStart, &Ops[0], Ops.size());
Evan Chengab6c3bb2008-08-25 21:27:18 +0000451 CurDAG->UpdateNodeOperands(Load, Call.getOperand(0),
452 Load.getOperand(1), Load.getOperand(2));
453 Ops.clear();
Gabor Greifba36cb52008-08-28 21:40:38 +0000454 Ops.push_back(SDValue(Load.getNode(), 1));
455 for (unsigned i = 1, e = Call.getNode()->getNumOperands(); i != e; ++i)
Evan Chengab6c3bb2008-08-25 21:27:18 +0000456 Ops.push_back(Call.getOperand(i));
457 CurDAG->UpdateNodeOperands(Call, &Ops[0], Ops.size());
458}
459
460/// isCalleeLoad - Return true if call address is a load and it can be
461/// moved below CALLSEQ_START and the chains leading up to the call.
462/// Return the CALLSEQ_START by reference as a second output.
463static bool isCalleeLoad(SDValue Callee, SDValue &Chain) {
Gabor Greifba36cb52008-08-28 21:40:38 +0000464 if (Callee.getNode() == Chain.getNode() || !Callee.hasOneUse())
Evan Chengab6c3bb2008-08-25 21:27:18 +0000465 return false;
Gabor Greifba36cb52008-08-28 21:40:38 +0000466 LoadSDNode *LD = dyn_cast<LoadSDNode>(Callee.getNode());
Evan Chengab6c3bb2008-08-25 21:27:18 +0000467 if (!LD ||
468 LD->isVolatile() ||
469 LD->getAddressingMode() != ISD::UNINDEXED ||
470 LD->getExtensionType() != ISD::NON_EXTLOAD)
471 return false;
472
473 // Now let's find the callseq_start.
474 while (Chain.getOpcode() != ISD::CALLSEQ_START) {
475 if (!Chain.hasOneUse())
476 return false;
477 Chain = Chain.getOperand(0);
478 }
Evan Cheng5b2e5892009-01-26 18:43:34 +0000479
480 if (Chain.getOperand(0).getNode() == Callee.getNode())
481 return true;
482 if (Chain.getOperand(0).getOpcode() == ISD::TokenFactor &&
483 Callee.getValue(1).isOperandOf(Chain.getOperand(0).getNode()))
484 return true;
485 return false;
Evan Chengab6c3bb2008-08-25 21:27:18 +0000486}
487
488
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000489/// PreprocessForRMW - Preprocess the DAG to make instruction selection better.
490/// This is only run if not in -fast mode (aka -O0).
491/// This allows the instruction selector to pick more read-modify-write
492/// instructions. This is a common case:
Evan Cheng70e674e2006-08-28 20:10:17 +0000493///
494/// [Load chain]
495/// ^
496/// |
497/// [Load]
498/// ^ ^
499/// | |
500/// / \-
501/// / |
502/// [TokenFactor] [Op]
503/// ^ ^
504/// | |
505/// \ /
506/// \ /
507/// [Store]
508///
509/// The fact the store's chain operand != load's chain will prevent the
510/// (store (op (load))) instruction from being selected. We can transform it to:
511///
512/// [Load chain]
513/// ^
514/// |
515/// [TokenFactor]
516/// ^
517/// |
518/// [Load]
519/// ^ ^
520/// | |
521/// | \-
522/// | |
523/// | [Op]
524/// | ^
525/// | |
526/// \ /
527/// \ /
528/// [Store]
Dan Gohmanf350b272008-08-23 02:25:05 +0000529void X86DAGToDAGISel::PreprocessForRMW() {
530 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
531 E = CurDAG->allnodes_end(); I != E; ++I) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000532 if (I->getOpcode() == X86ISD::CALL) {
533 /// Also try moving call address load from outside callseq_start to just
534 /// before the call to allow it to be folded.
535 ///
536 /// [Load chain]
537 /// ^
538 /// |
539 /// [Load]
540 /// ^ ^
541 /// | |
542 /// / \--
543 /// / |
544 ///[CALLSEQ_START] |
545 /// ^ |
546 /// | |
547 /// [LOAD/C2Reg] |
548 /// | |
549 /// \ /
550 /// \ /
551 /// [CALL]
552 SDValue Chain = I->getOperand(0);
553 SDValue Load = I->getOperand(1);
554 if (!isCalleeLoad(Load, Chain))
555 continue;
556 MoveBelowCallSeqStart(CurDAG, Load, SDValue(I, 0), Chain);
557 ++NumLoadMoved;
558 continue;
559 }
560
Evan Cheng8b2794a2006-10-13 21:14:26 +0000561 if (!ISD::isNON_TRUNCStore(I))
Evan Cheng70e674e2006-08-28 20:10:17 +0000562 continue;
Dan Gohman475871a2008-07-27 21:46:04 +0000563 SDValue Chain = I->getOperand(0);
Evan Chengab6c3bb2008-08-25 21:27:18 +0000564
Gabor Greifba36cb52008-08-28 21:40:38 +0000565 if (Chain.getNode()->getOpcode() != ISD::TokenFactor)
Evan Cheng70e674e2006-08-28 20:10:17 +0000566 continue;
567
Dan Gohman475871a2008-07-27 21:46:04 +0000568 SDValue N1 = I->getOperand(1);
569 SDValue N2 = I->getOperand(2);
Duncan Sands83ec4b62008-06-06 12:08:01 +0000570 if ((N1.getValueType().isFloatingPoint() &&
571 !N1.getValueType().isVector()) ||
Evan Cheng780413d2006-08-29 18:37:37 +0000572 !N1.hasOneUse())
Evan Cheng70e674e2006-08-28 20:10:17 +0000573 continue;
574
575 bool RModW = false;
Dan Gohman475871a2008-07-27 21:46:04 +0000576 SDValue Load;
Gabor Greifba36cb52008-08-28 21:40:38 +0000577 unsigned Opcode = N1.getNode()->getOpcode();
Evan Cheng70e674e2006-08-28 20:10:17 +0000578 switch (Opcode) {
Evan Chengab6c3bb2008-08-25 21:27:18 +0000579 case ISD::ADD:
580 case ISD::MUL:
581 case ISD::AND:
582 case ISD::OR:
583 case ISD::XOR:
584 case ISD::ADDC:
585 case ISD::ADDE:
586 case ISD::VECTOR_SHUFFLE: {
587 SDValue N10 = N1.getOperand(0);
588 SDValue N11 = N1.getOperand(1);
589 RModW = isRMWLoad(N10, Chain, N2, Load);
590 if (!RModW)
591 RModW = isRMWLoad(N11, Chain, N2, Load);
592 break;
593 }
594 case ISD::SUB:
595 case ISD::SHL:
596 case ISD::SRA:
597 case ISD::SRL:
598 case ISD::ROTL:
599 case ISD::ROTR:
600 case ISD::SUBC:
601 case ISD::SUBE:
602 case X86ISD::SHLD:
603 case X86ISD::SHRD: {
604 SDValue N10 = N1.getOperand(0);
605 RModW = isRMWLoad(N10, Chain, N2, Load);
606 break;
607 }
Evan Cheng70e674e2006-08-28 20:10:17 +0000608 }
609
Evan Cheng82a35b32006-08-29 06:44:17 +0000610 if (RModW) {
Dan Gohmanf350b272008-08-23 02:25:05 +0000611 MoveBelowTokenFactor(CurDAG, Load, SDValue(I, 0), Chain);
Evan Cheng82a35b32006-08-29 06:44:17 +0000612 ++NumLoadMoved;
613 }
Evan Cheng70e674e2006-08-28 20:10:17 +0000614 }
615}
616
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000617
618/// PreprocessForFPConvert - Walk over the dag lowering fpround and fpextend
619/// nodes that target the FP stack to be store and load to the stack. This is a
620/// gross hack. We would like to simply mark these as being illegal, but when
621/// we do that, legalize produces these when it expands calls, then expands
622/// these in the same legalize pass. We would like dag combine to be able to
623/// hack on these between the call expansion and the node legalization. As such
624/// this pass basically does "really late" legalization of these inline with the
625/// X86 isel pass.
Dan Gohmanf350b272008-08-23 02:25:05 +0000626void X86DAGToDAGISel::PreprocessForFPConvert() {
627 for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(),
628 E = CurDAG->allnodes_end(); I != E; ) {
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000629 SDNode *N = I++; // Preincrement iterator to avoid invalidation issues.
630 if (N->getOpcode() != ISD::FP_ROUND && N->getOpcode() != ISD::FP_EXTEND)
631 continue;
632
633 // If the source and destination are SSE registers, then this is a legal
634 // conversion that should not be lowered.
Duncan Sands83ec4b62008-06-06 12:08:01 +0000635 MVT SrcVT = N->getOperand(0).getValueType();
636 MVT DstVT = N->getValueType(0);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000637 bool SrcIsSSE = X86Lowering.isScalarFPTypeInSSEReg(SrcVT);
638 bool DstIsSSE = X86Lowering.isScalarFPTypeInSSEReg(DstVT);
639 if (SrcIsSSE && DstIsSSE)
640 continue;
641
Chris Lattner6fa2f9c2008-03-09 07:05:32 +0000642 if (!SrcIsSSE && !DstIsSSE) {
643 // If this is an FPStack extension, it is a noop.
644 if (N->getOpcode() == ISD::FP_EXTEND)
645 continue;
646 // If this is a value-preserving FPStack truncation, it is a noop.
647 if (N->getConstantOperandVal(1))
648 continue;
649 }
650
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000651 // Here we could have an FP stack truncation or an FPStack <-> SSE convert.
652 // FPStack has extload and truncstore. SSE can fold direct loads into other
653 // operations. Based on this, decide what we want to do.
Duncan Sands83ec4b62008-06-06 12:08:01 +0000654 MVT MemVT;
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000655 if (N->getOpcode() == ISD::FP_ROUND)
656 MemVT = DstVT; // FP_ROUND must use DstVT, we can't do a 'trunc load'.
657 else
658 MemVT = SrcIsSSE ? SrcVT : DstVT;
659
Dan Gohmanf350b272008-08-23 02:25:05 +0000660 SDValue MemTmp = CurDAG->CreateStackTemporary(MemVT);
Dale Johannesend8392542009-02-03 21:48:12 +0000661 DebugLoc dl = N->getDebugLoc();
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000662
663 // FIXME: optimize the case where the src/dest is a load or store?
Dale Johannesend8392542009-02-03 21:48:12 +0000664 SDValue Store = CurDAG->getTruncStore(CurDAG->getEntryNode(), dl,
Dan Gohmanf350b272008-08-23 02:25:05 +0000665 N->getOperand(0),
666 MemTmp, NULL, 0, MemVT);
Dale Johannesend8392542009-02-03 21:48:12 +0000667 SDValue Result = CurDAG->getExtLoad(ISD::EXTLOAD, dl, DstVT, Store, MemTmp,
Dan Gohmanf350b272008-08-23 02:25:05 +0000668 NULL, 0, MemVT);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000669
670 // We're about to replace all uses of the FP_ROUND/FP_EXTEND with the
671 // extload we created. This will cause general havok on the dag because
672 // anything below the conversion could be folded into other existing nodes.
673 // To avoid invalidating 'I', back it up to the convert node.
674 --I;
Dan Gohmanf350b272008-08-23 02:25:05 +0000675 CurDAG->ReplaceAllUsesOfValueWith(SDValue(N, 0), Result);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000676
677 // Now that we did that, the node is dead. Increment the iterator to the
678 // next node to process, then delete N.
679 ++I;
Dan Gohmanf350b272008-08-23 02:25:05 +0000680 CurDAG->DeleteNode(N);
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000681 }
682}
683
Chris Lattnerc961eea2005-11-16 01:54:32 +0000684/// InstructionSelectBasicBlock - This callback is invoked by SelectionDAGISel
685/// when it has created a SelectionDAG for us to codegen.
Dan Gohmanf350b272008-08-23 02:25:05 +0000686void X86DAGToDAGISel::InstructionSelect() {
Evan Chengdb8d56b2008-06-30 20:45:06 +0000687 CurBB = BB; // BB can change as result of isel.
Devang Patele76225a2008-10-06 18:03:39 +0000688 const Function *F = CurDAG->getMachineFunction().getFunction();
689 OptForSize = F->hasFnAttr(Attribute::OptimizeForSize);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000690
Evan Chengdb8d56b2008-06-30 20:45:06 +0000691 DEBUG(BB->dump());
Dan Gohmanea9587b2008-08-13 19:55:00 +0000692 if (!Fast)
Dan Gohmanf350b272008-08-23 02:25:05 +0000693 PreprocessForRMW();
Chris Lattnerd43d00c2008-01-24 08:07:48 +0000694
695 // FIXME: This should only happen when not -fast.
Dan Gohmanf350b272008-08-23 02:25:05 +0000696 PreprocessForFPConvert();
Evan Cheng70e674e2006-08-28 20:10:17 +0000697
Chris Lattnerc961eea2005-11-16 01:54:32 +0000698 // Codegen the basic block.
Evan Chengf597dc72006-02-10 22:24:32 +0000699#ifndef NDEBUG
Bill Wendling6345d752006-11-17 07:52:03 +0000700 DOUT << "===== Instruction selection begins:\n";
Evan Cheng23addc02006-02-10 22:46:26 +0000701 Indent = 0;
Evan Chengf597dc72006-02-10 22:24:32 +0000702#endif
David Greene8ad4c002008-10-27 21:56:29 +0000703 SelectRoot(*CurDAG);
Evan Chengf597dc72006-02-10 22:24:32 +0000704#ifndef NDEBUG
Bill Wendling6345d752006-11-17 07:52:03 +0000705 DOUT << "===== Instruction selection ends:\n";
Evan Chengf597dc72006-02-10 22:24:32 +0000706#endif
Evan Cheng63ce5682006-07-28 00:10:59 +0000707
Dan Gohmanf350b272008-08-23 02:25:05 +0000708 CurDAG->RemoveDeadNodes();
Evan Chengdb8d56b2008-06-30 20:45:06 +0000709}
Chris Lattnerc961eea2005-11-16 01:54:32 +0000710
Anton Korobeynikov2fe12592007-09-25 21:52:30 +0000711/// EmitSpecialCodeForMain - Emit any code that needs to be executed only in
712/// the main function.
713void X86DAGToDAGISel::EmitSpecialCodeForMain(MachineBasicBlock *BB,
714 MachineFrameInfo *MFI) {
715 const TargetInstrInfo *TII = TM.getInstrInfo();
716 if (Subtarget->isTargetCygMing())
717 BuildMI(BB, TII->get(X86::CALLpcrel32)).addExternalSymbol("__main");
718}
719
720void X86DAGToDAGISel::EmitFunctionEntryCode(Function &Fn, MachineFunction &MF) {
721 // If this is main, emit special code for main.
722 MachineBasicBlock *BB = MF.begin();
723 if (Fn.hasExternalLinkage() && Fn.getName() == "main")
724 EmitSpecialCodeForMain(BB, MF.getFrameInfo());
725}
726
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000727/// MatchAddress - Add the specified node to the specified addressing mode,
728/// returning true if it cannot be done. This just pattern matches for the
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000729/// addressing mode.
Dan Gohman475871a2008-07-27 21:46:04 +0000730bool X86DAGToDAGISel::MatchAddress(SDValue N, X86ISelAddressMode &AM,
Anton Korobeynikov33bf8c42007-03-28 18:36:33 +0000731 bool isRoot, unsigned Depth) {
Dan Gohman6520e202008-10-18 02:06:02 +0000732 bool is64Bit = Subtarget->is64Bit();
Dale Johannesend8392542009-02-03 21:48:12 +0000733 DebugLoc dl = N.getNode()->getDebugLoc();
Evan Chengda43bcf2008-09-24 00:05:32 +0000734 DOUT << "MatchAddress: "; DEBUG(AM.dump());
Dan Gohmanbadb2d22007-08-13 20:03:06 +0000735 // Limit recursion.
736 if (Depth > 5)
737 return MatchAddressBase(N, AM, isRoot, Depth);
Anton Korobeynikov33bf8c42007-03-28 18:36:33 +0000738
Evan Cheng25ab6902006-09-08 06:48:29 +0000739 // RIP relative addressing: %rip + 32-bit displacement!
740 if (AM.isRIPRel) {
741 if (!AM.ES && AM.JT != -1 && N.getOpcode() == ISD::Constant) {
Dan Gohman27cae7b2008-11-11 15:52:29 +0000742 uint64_t Val = cast<ConstantSDNode>(N)->getSExtValue();
Dan Gohman6520e202008-10-18 02:06:02 +0000743 if (!is64Bit || isInt32(AM.Disp + Val)) {
Evan Cheng25ab6902006-09-08 06:48:29 +0000744 AM.Disp += Val;
745 return false;
746 }
747 }
748 return true;
749 }
750
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000751 switch (N.getOpcode()) {
752 default: break;
Evan Cheng25ab6902006-09-08 06:48:29 +0000753 case ISD::Constant: {
Dan Gohman27cae7b2008-11-11 15:52:29 +0000754 uint64_t Val = cast<ConstantSDNode>(N)->getSExtValue();
Dan Gohman6520e202008-10-18 02:06:02 +0000755 if (!is64Bit || isInt32(AM.Disp + Val)) {
Evan Cheng25ab6902006-09-08 06:48:29 +0000756 AM.Disp += Val;
757 return false;
758 }
759 break;
760 }
Evan Cheng51a9ed92006-02-25 10:09:08 +0000761
Evan Cheng19f2ffc2006-12-05 04:01:03 +0000762 case X86ISD::Wrapper: {
Dan Gohman6520e202008-10-18 02:06:02 +0000763 DOUT << "Wrapper: 64bit " << is64Bit;
764 DOUT << " AM "; DEBUG(AM.dump()); DOUT << "\n";
Evan Cheng0085a282006-11-30 21:55:46 +0000765 // Under X86-64 non-small code model, GV (and friends) are 64-bits.
Evan Chengbe3bf422008-02-07 08:53:49 +0000766 // Also, base and index reg must be 0 in order to use rip as base.
767 if (is64Bit && (TM.getCodeModel() != CodeModel::Small ||
Gabor Greifba36cb52008-08-28 21:40:38 +0000768 AM.Base.Reg.getNode() || AM.IndexReg.getNode()))
Evan Cheng0085a282006-11-30 21:55:46 +0000769 break;
Evan Cheng28b514392006-12-05 19:50:18 +0000770 if (AM.GV != 0 || AM.CP != 0 || AM.ES != 0 || AM.JT != -1)
771 break;
Evan Cheng25ab6902006-09-08 06:48:29 +0000772 // If value is available in a register both base and index components have
773 // been picked, we can't fit the result available in the register in the
774 // addressing mode. Duplicate GlobalAddress or ConstantPool as displacement.
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000775 {
Dan Gohman475871a2008-07-27 21:46:04 +0000776 SDValue N0 = N.getOperand(0);
Evan Cheng28b514392006-12-05 19:50:18 +0000777 if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(N0)) {
Dan Gohman27cae7b2008-11-11 15:52:29 +0000778 uint64_t Offset = G->getOffset();
779 if (!is64Bit || isInt32(AM.Disp + Offset)) {
Dan Gohman6520e202008-10-18 02:06:02 +0000780 GlobalValue *GV = G->getGlobal();
781 AM.GV = GV;
Dan Gohman27cae7b2008-11-11 15:52:29 +0000782 AM.Disp += Offset;
Dan Gohman6520e202008-10-18 02:06:02 +0000783 AM.isRIPRel = TM.symbolicAddressesAreRIPRel();
784 return false;
785 }
Evan Cheng28b514392006-12-05 19:50:18 +0000786 } else if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N0)) {
Dan Gohman27cae7b2008-11-11 15:52:29 +0000787 uint64_t Offset = CP->getOffset();
788 if (!is64Bit || isInt32(AM.Disp + Offset)) {
Dan Gohman6520e202008-10-18 02:06:02 +0000789 AM.CP = CP->getConstVal();
790 AM.Align = CP->getAlignment();
Dan Gohman27cae7b2008-11-11 15:52:29 +0000791 AM.Disp += Offset;
Dan Gohman6520e202008-10-18 02:06:02 +0000792 AM.isRIPRel = TM.symbolicAddressesAreRIPRel();
793 return false;
794 }
Bill Wendling056292f2008-09-16 21:48:12 +0000795 } else if (ExternalSymbolSDNode *S =dyn_cast<ExternalSymbolSDNode>(N0)) {
Evan Chengbe3bf422008-02-07 08:53:49 +0000796 AM.ES = S->getSymbol();
Dan Gohman97135e12008-09-26 19:15:30 +0000797 AM.isRIPRel = TM.symbolicAddressesAreRIPRel();
Evan Chengbe3bf422008-02-07 08:53:49 +0000798 return false;
Evan Cheng28b514392006-12-05 19:50:18 +0000799 } else if (JumpTableSDNode *J = dyn_cast<JumpTableSDNode>(N0)) {
Evan Chengbe3bf422008-02-07 08:53:49 +0000800 AM.JT = J->getIndex();
Dan Gohman97135e12008-09-26 19:15:30 +0000801 AM.isRIPRel = TM.symbolicAddressesAreRIPRel();
Evan Chengbe3bf422008-02-07 08:53:49 +0000802 return false;
Evan Cheng51a9ed92006-02-25 10:09:08 +0000803 }
804 }
805 break;
Evan Cheng0085a282006-11-30 21:55:46 +0000806 }
Evan Cheng51a9ed92006-02-25 10:09:08 +0000807
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000808 case ISD::FrameIndex:
Gabor Greif93c53e52008-08-31 15:37:04 +0000809 if (AM.BaseType == X86ISelAddressMode::RegBase
810 && AM.Base.Reg.getNode() == 0) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000811 AM.BaseType = X86ISelAddressMode::FrameIndexBase;
812 AM.Base.FrameIndex = cast<FrameIndexSDNode>(N)->getIndex();
813 return false;
814 }
815 break;
Evan Chengec693f72005-12-08 02:01:35 +0000816
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000817 case ISD::SHL:
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000818 if (AM.IndexReg.getNode() != 0 || AM.Scale != 1 || AM.isRIPRel)
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000819 break;
820
Gabor Greif93c53e52008-08-31 15:37:04 +0000821 if (ConstantSDNode
822 *CN = dyn_cast<ConstantSDNode>(N.getNode()->getOperand(1))) {
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000823 unsigned Val = CN->getZExtValue();
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000824 if (Val == 1 || Val == 2 || Val == 3) {
825 AM.Scale = 1 << Val;
Gabor Greifba36cb52008-08-28 21:40:38 +0000826 SDValue ShVal = N.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000827
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000828 // Okay, we know that we have a scale by now. However, if the scaled
829 // value is an add of something and a constant, we can fold the
830 // constant into the disp field here.
Gabor Greifba36cb52008-08-28 21:40:38 +0000831 if (ShVal.getNode()->getOpcode() == ISD::ADD && ShVal.hasOneUse() &&
832 isa<ConstantSDNode>(ShVal.getNode()->getOperand(1))) {
833 AM.IndexReg = ShVal.getNode()->getOperand(0);
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000834 ConstantSDNode *AddVal =
Gabor Greifba36cb52008-08-28 21:40:38 +0000835 cast<ConstantSDNode>(ShVal.getNode()->getOperand(1));
Evan Cheng8e278262009-01-17 07:09:27 +0000836 uint64_t Disp = AM.Disp + (AddVal->getSExtValue() << Val);
Dan Gohman6520e202008-10-18 02:06:02 +0000837 if (!is64Bit || isInt32(Disp))
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000838 AM.Disp = Disp;
839 else
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000840 AM.IndexReg = ShVal;
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000841 } else {
842 AM.IndexReg = ShVal;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000843 }
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000844 return false;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000845 }
846 break;
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000847 }
Evan Chengec693f72005-12-08 02:01:35 +0000848
Dan Gohman83688052007-10-22 20:22:24 +0000849 case ISD::SMUL_LOHI:
850 case ISD::UMUL_LOHI:
851 // A mul_lohi where we need the low part can be folded as a plain multiply.
Gabor Greif99a6cb92008-08-26 22:36:50 +0000852 if (N.getResNo() != 0) break;
Dan Gohman83688052007-10-22 20:22:24 +0000853 // FALL THROUGH
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000854 case ISD::MUL:
855 // X*[3,5,9] -> X+X*[2,4,8]
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000856 if (AM.BaseType == X86ISelAddressMode::RegBase &&
Gabor Greifba36cb52008-08-28 21:40:38 +0000857 AM.Base.Reg.getNode() == 0 &&
858 AM.IndexReg.getNode() == 0 &&
Evan Chengbe3bf422008-02-07 08:53:49 +0000859 !AM.isRIPRel) {
Gabor Greif93c53e52008-08-31 15:37:04 +0000860 if (ConstantSDNode
861 *CN = dyn_cast<ConstantSDNode>(N.getNode()->getOperand(1)))
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000862 if (CN->getZExtValue() == 3 || CN->getZExtValue() == 5 ||
863 CN->getZExtValue() == 9) {
864 AM.Scale = unsigned(CN->getZExtValue())-1;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000865
Gabor Greifba36cb52008-08-28 21:40:38 +0000866 SDValue MulVal = N.getNode()->getOperand(0);
Dan Gohman475871a2008-07-27 21:46:04 +0000867 SDValue Reg;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000868
869 // Okay, we know that we have a scale by now. However, if the scaled
870 // value is an add of something and a constant, we can fold the
871 // constant into the disp field here.
Gabor Greifba36cb52008-08-28 21:40:38 +0000872 if (MulVal.getNode()->getOpcode() == ISD::ADD && MulVal.hasOneUse() &&
873 isa<ConstantSDNode>(MulVal.getNode()->getOperand(1))) {
874 Reg = MulVal.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000875 ConstantSDNode *AddVal =
Gabor Greifba36cb52008-08-28 21:40:38 +0000876 cast<ConstantSDNode>(MulVal.getNode()->getOperand(1));
Evan Cheng8e278262009-01-17 07:09:27 +0000877 uint64_t Disp = AM.Disp + AddVal->getSExtValue() *
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000878 CN->getZExtValue();
Dan Gohman6520e202008-10-18 02:06:02 +0000879 if (!is64Bit || isInt32(Disp))
Evan Cheng25ab6902006-09-08 06:48:29 +0000880 AM.Disp = Disp;
881 else
Gabor Greifba36cb52008-08-28 21:40:38 +0000882 Reg = N.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000883 } else {
Gabor Greifba36cb52008-08-28 21:40:38 +0000884 Reg = N.getNode()->getOperand(0);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000885 }
886
887 AM.IndexReg = AM.Base.Reg = Reg;
888 return false;
889 }
Chris Lattner62412262007-02-04 20:18:17 +0000890 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000891 break;
892
Evan Cheng8e278262009-01-17 07:09:27 +0000893 case ISD::ADD: {
894 X86ISelAddressMode Backup = AM;
895 if (!MatchAddress(N.getNode()->getOperand(0), AM, false, Depth+1) &&
896 !MatchAddress(N.getNode()->getOperand(1), AM, false, Depth+1))
897 return false;
898 AM = Backup;
899 if (!MatchAddress(N.getNode()->getOperand(1), AM, false, Depth+1) &&
900 !MatchAddress(N.getNode()->getOperand(0), AM, false, Depth+1))
901 return false;
902 AM = Backup;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000903 break;
Evan Cheng8e278262009-01-17 07:09:27 +0000904 }
Evan Chenge6ad27e2006-05-30 06:59:36 +0000905
Chris Lattner62412262007-02-04 20:18:17 +0000906 case ISD::OR:
907 // Handle "X | C" as "X + C" iff X is known to have C bits clear.
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000908 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.getOperand(1))) {
909 X86ISelAddressMode Backup = AM;
Dan Gohman27cae7b2008-11-11 15:52:29 +0000910 uint64_t Offset = CN->getSExtValue();
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000911 // Start with the LHS as an addr mode.
912 if (!MatchAddress(N.getOperand(0), AM, false) &&
913 // Address could not have picked a GV address for the displacement.
914 AM.GV == NULL &&
915 // On x86-64, the resultant disp must fit in 32-bits.
Dan Gohman27cae7b2008-11-11 15:52:29 +0000916 (!is64Bit || isInt32(AM.Disp + Offset)) &&
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000917 // Check to see if the LHS & C is zero.
Dan Gohman2e68b6f2008-02-25 21:11:39 +0000918 CurDAG->MaskedValueIsZero(N.getOperand(0), CN->getAPIntValue())) {
Dan Gohman27cae7b2008-11-11 15:52:29 +0000919 AM.Disp += Offset;
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000920 return false;
Evan Chenge6ad27e2006-05-30 06:59:36 +0000921 }
Chris Lattner5aaddaa2007-12-08 07:22:58 +0000922 AM = Backup;
Evan Chenge6ad27e2006-05-30 06:59:36 +0000923 }
924 break;
Evan Cheng1314b002007-12-13 00:43:27 +0000925
926 case ISD::AND: {
927 // Handle "(x << C1) & C2" as "(X & (C2>>C1)) << C1" if safe and if this
928 // allows us to fold the shift into this addressing mode.
Dan Gohman475871a2008-07-27 21:46:04 +0000929 SDValue Shift = N.getOperand(0);
Evan Cheng1314b002007-12-13 00:43:27 +0000930 if (Shift.getOpcode() != ISD::SHL) break;
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000931
Evan Cheng1314b002007-12-13 00:43:27 +0000932 // Scale must not be used already.
Gabor Greifba36cb52008-08-28 21:40:38 +0000933 if (AM.IndexReg.getNode() != 0 || AM.Scale != 1) break;
Evan Chengbe3bf422008-02-07 08:53:49 +0000934
935 // Not when RIP is used as the base.
936 if (AM.isRIPRel) break;
Evan Cheng1314b002007-12-13 00:43:27 +0000937
938 ConstantSDNode *C2 = dyn_cast<ConstantSDNode>(N.getOperand(1));
939 ConstantSDNode *C1 = dyn_cast<ConstantSDNode>(Shift.getOperand(1));
940 if (!C1 || !C2) break;
941
942 // Not likely to be profitable if either the AND or SHIFT node has more
943 // than one use (unless all uses are for address computation). Besides,
944 // isel mechanism requires their node ids to be reused.
945 if (!N.hasOneUse() || !Shift.hasOneUse())
946 break;
947
948 // Verify that the shift amount is something we can fold.
Dan Gohmanf5aeb1a2008-09-12 16:56:44 +0000949 unsigned ShiftCst = C1->getZExtValue();
Evan Cheng1314b002007-12-13 00:43:27 +0000950 if (ShiftCst != 1 && ShiftCst != 2 && ShiftCst != 3)
951 break;
952
953 // Get the new AND mask, this folds to a constant.
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000954 SDValue X = Shift.getOperand(0);
Dale Johannesend8392542009-02-03 21:48:12 +0000955 SDValue NewANDMask = CurDAG->getNode(ISD::SRL, dl, N.getValueType(),
Evan Cheng552e3be2008-10-14 17:15:39 +0000956 SDValue(C2, 0), SDValue(C1, 0));
Dale Johannesend8392542009-02-03 21:48:12 +0000957 SDValue NewAND = CurDAG->getNode(ISD::AND, dl, N.getValueType(), X,
958 NewANDMask);
959 SDValue NewSHIFT = CurDAG->getNode(ISD::SHL, dl, N.getValueType(),
Dan Gohman7b8e9642008-10-13 20:52:04 +0000960 NewAND, SDValue(C1, 0));
Dan Gohman8be6bbe2008-11-05 04:14:16 +0000961
962 // Insert the new nodes into the topological ordering.
963 if (C1->getNodeId() > X.getNode()->getNodeId()) {
964 CurDAG->RepositionNode(X.getNode(), C1);
965 C1->setNodeId(X.getNode()->getNodeId());
966 }
967 if (NewANDMask.getNode()->getNodeId() == -1 ||
968 NewANDMask.getNode()->getNodeId() > X.getNode()->getNodeId()) {
969 CurDAG->RepositionNode(X.getNode(), NewANDMask.getNode());
970 NewANDMask.getNode()->setNodeId(X.getNode()->getNodeId());
971 }
972 if (NewAND.getNode()->getNodeId() == -1 ||
973 NewAND.getNode()->getNodeId() > Shift.getNode()->getNodeId()) {
974 CurDAG->RepositionNode(Shift.getNode(), NewAND.getNode());
975 NewAND.getNode()->setNodeId(Shift.getNode()->getNodeId());
976 }
977 if (NewSHIFT.getNode()->getNodeId() == -1 ||
978 NewSHIFT.getNode()->getNodeId() > N.getNode()->getNodeId()) {
979 CurDAG->RepositionNode(N.getNode(), NewSHIFT.getNode());
980 NewSHIFT.getNode()->setNodeId(N.getNode()->getNodeId());
981 }
982
Dan Gohman7b8e9642008-10-13 20:52:04 +0000983 CurDAG->ReplaceAllUsesWith(N, NewSHIFT);
Evan Cheng1314b002007-12-13 00:43:27 +0000984
985 AM.Scale = 1 << ShiftCst;
986 AM.IndexReg = NewAND;
987 return false;
988 }
Evan Chenge6ad27e2006-05-30 06:59:36 +0000989 }
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000990
Dan Gohmanbadb2d22007-08-13 20:03:06 +0000991 return MatchAddressBase(N, AM, isRoot, Depth);
992}
993
994/// MatchAddressBase - Helper for MatchAddress. Add the specified node to the
995/// specified addressing mode without any further recursion.
Dan Gohman475871a2008-07-27 21:46:04 +0000996bool X86DAGToDAGISel::MatchAddressBase(SDValue N, X86ISelAddressMode &AM,
Dan Gohmanbadb2d22007-08-13 20:03:06 +0000997 bool isRoot, unsigned Depth) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000998 // Is the base register already occupied?
Gabor Greifba36cb52008-08-28 21:40:38 +0000999 if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base.Reg.getNode()) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001000 // If so, check to see if the scale index register is set.
Gabor Greifba36cb52008-08-28 21:40:38 +00001001 if (AM.IndexReg.getNode() == 0 && !AM.isRIPRel) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +00001002 AM.IndexReg = N;
1003 AM.Scale = 1;
1004 return false;
1005 }
1006
1007 // Otherwise, we cannot select it.
1008 return true;
1009 }
1010
1011 // Default, generate it as a register.
1012 AM.BaseType = X86ISelAddressMode::RegBase;
1013 AM.Base.Reg = N;
1014 return false;
1015}
1016
Evan Chengec693f72005-12-08 02:01:35 +00001017/// SelectAddr - returns true if it is able pattern match an addressing mode.
1018/// It returns the operands which make up the maximal addressing mode it can
1019/// match by reference.
Dan Gohman475871a2008-07-27 21:46:04 +00001020bool X86DAGToDAGISel::SelectAddr(SDValue Op, SDValue N, SDValue &Base,
1021 SDValue &Scale, SDValue &Index,
1022 SDValue &Disp) {
Evan Chengec693f72005-12-08 02:01:35 +00001023 X86ISelAddressMode AM;
Evan Cheng8700e142006-01-11 06:09:51 +00001024 if (MatchAddress(N, AM))
1025 return false;
Evan Chengec693f72005-12-08 02:01:35 +00001026
Duncan Sands83ec4b62008-06-06 12:08:01 +00001027 MVT VT = N.getValueType();
Evan Cheng8700e142006-01-11 06:09:51 +00001028 if (AM.BaseType == X86ISelAddressMode::RegBase) {
Gabor Greifba36cb52008-08-28 21:40:38 +00001029 if (!AM.Base.Reg.getNode())
Evan Cheng25ab6902006-09-08 06:48:29 +00001030 AM.Base.Reg = CurDAG->getRegister(0, VT);
Evan Chengec693f72005-12-08 02:01:35 +00001031 }
Evan Cheng8700e142006-01-11 06:09:51 +00001032
Gabor Greifba36cb52008-08-28 21:40:38 +00001033 if (!AM.IndexReg.getNode())
Evan Cheng25ab6902006-09-08 06:48:29 +00001034 AM.IndexReg = CurDAG->getRegister(0, VT);
Evan Cheng8700e142006-01-11 06:09:51 +00001035
1036 getAddressOperands(AM, Base, Scale, Index, Disp);
1037 return true;
Evan Chengec693f72005-12-08 02:01:35 +00001038}
1039
Chris Lattner3a7cd952006-10-07 21:55:32 +00001040/// SelectScalarSSELoad - Match a scalar SSE load. In particular, we want to
1041/// match a load whose top elements are either undef or zeros. The load flavor
1042/// is derived from the type of N, which is either v4f32 or v2f64.
Dan Gohman475871a2008-07-27 21:46:04 +00001043bool X86DAGToDAGISel::SelectScalarSSELoad(SDValue Op, SDValue Pred,
1044 SDValue N, SDValue &Base,
1045 SDValue &Scale, SDValue &Index,
1046 SDValue &Disp, SDValue &InChain,
1047 SDValue &OutChain) {
Chris Lattner3a7cd952006-10-07 21:55:32 +00001048 if (N.getOpcode() == ISD::SCALAR_TO_VECTOR) {
Chris Lattner4fe4f252006-10-11 22:09:58 +00001049 InChain = N.getOperand(0).getValue(1);
Gabor Greifba36cb52008-08-28 21:40:38 +00001050 if (ISD::isNON_EXTLoad(InChain.getNode()) &&
Evan Cheng07e4b002006-10-16 06:34:55 +00001051 InChain.getValue(0).hasOneUse() &&
Evan Chengd6373bc2006-11-10 21:23:04 +00001052 N.hasOneUse() &&
Evan Cheng884c70c2008-11-27 00:49:46 +00001053 IsLegalAndProfitableToFold(N.getNode(), Pred.getNode(), Op.getNode())) {
Evan Cheng82a91642006-10-11 21:06:01 +00001054 LoadSDNode *LD = cast<LoadSDNode>(InChain);
Evan Cheng0d538262006-11-08 20:34:28 +00001055 if (!SelectAddr(Op, LD->getBasePtr(), Base, Scale, Index, Disp))
Chris Lattner3a7cd952006-10-07 21:55:32 +00001056 return false;
Evan Cheng82a91642006-10-11 21:06:01 +00001057 OutChain = LD->getChain();
Chris Lattner3a7cd952006-10-07 21:55:32 +00001058 return true;
1059 }
1060 }
Chris Lattner4fe4f252006-10-11 22:09:58 +00001061
1062 // Also handle the case where we explicitly require zeros in the top
Chris Lattner3a7cd952006-10-07 21:55:32 +00001063 // elements. This is a vector shuffle from the zero vector.
Gabor Greifba36cb52008-08-28 21:40:38 +00001064 if (N.getOpcode() == X86ISD::VZEXT_MOVL && N.getNode()->hasOneUse() &&
Chris Lattner8a594482007-11-25 00:24:49 +00001065 // Check to see if the top elements are all zeros (or bitcast of zeros).
Evan Cheng7e2ff772008-05-08 00:57:18 +00001066 N.getOperand(0).getOpcode() == ISD::SCALAR_TO_VECTOR &&
Gabor Greifba36cb52008-08-28 21:40:38 +00001067 N.getOperand(0).getNode()->hasOneUse() &&
1068 ISD::isNON_EXTLoad(N.getOperand(0).getOperand(0).getNode()) &&
Evan Cheng7e2ff772008-05-08 00:57:18 +00001069 N.getOperand(0).getOperand(0).hasOneUse()) {
1070 // Okay, this is a zero extending load. Fold it.
1071 LoadSDNode *LD = cast<LoadSDNode>(N.getOperand(0).getOperand(0));
1072 if (!SelectAddr(Op, LD->getBasePtr(), Base, Scale, Index, Disp))
1073 return false;
1074 OutChain = LD->getChain();
Dan Gohman475871a2008-07-27 21:46:04 +00001075 InChain = SDValue(LD, 1);
Evan Cheng7e2ff772008-05-08 00:57:18 +00001076 return true;
Chris Lattner4fe4f252006-10-11 22:09:58 +00001077 }
Chris Lattner3a7cd952006-10-07 21:55:32 +00001078 return false;
1079}
1080
1081
Evan Cheng51a9ed92006-02-25 10:09:08 +00001082/// SelectLEAAddr - it calls SelectAddr and determines if the maximal addressing
1083/// mode it matches can be cost effectively emitted as an LEA instruction.
Dan Gohman475871a2008-07-27 21:46:04 +00001084bool X86DAGToDAGISel::SelectLEAAddr(SDValue Op, SDValue N,
1085 SDValue &Base, SDValue &Scale,
1086 SDValue &Index, SDValue &Disp) {
Evan Cheng51a9ed92006-02-25 10:09:08 +00001087 X86ISelAddressMode AM;
1088 if (MatchAddress(N, AM))
1089 return false;
1090
Duncan Sands83ec4b62008-06-06 12:08:01 +00001091 MVT VT = N.getValueType();
Evan Cheng51a9ed92006-02-25 10:09:08 +00001092 unsigned Complexity = 0;
1093 if (AM.BaseType == X86ISelAddressMode::RegBase)
Gabor Greifba36cb52008-08-28 21:40:38 +00001094 if (AM.Base.Reg.getNode())
Evan Cheng51a9ed92006-02-25 10:09:08 +00001095 Complexity = 1;
1096 else
Evan Cheng25ab6902006-09-08 06:48:29 +00001097 AM.Base.Reg = CurDAG->getRegister(0, VT);
Evan Cheng51a9ed92006-02-25 10:09:08 +00001098 else if (AM.BaseType == X86ISelAddressMode::FrameIndexBase)
1099 Complexity = 4;
1100
Gabor Greifba36cb52008-08-28 21:40:38 +00001101 if (AM.IndexReg.getNode())
Evan Cheng51a9ed92006-02-25 10:09:08 +00001102 Complexity++;
1103 else
Evan Cheng25ab6902006-09-08 06:48:29 +00001104 AM.IndexReg = CurDAG->getRegister(0, VT);
Evan Cheng51a9ed92006-02-25 10:09:08 +00001105
Chris Lattnera16b7cb2007-03-20 06:08:29 +00001106 // Don't match just leal(,%reg,2). It's cheaper to do addl %reg, %reg, or with
1107 // a simple shift.
1108 if (AM.Scale > 1)
Evan Cheng8c03fe42006-02-28 21:13:57 +00001109 Complexity++;
Evan Cheng51a9ed92006-02-25 10:09:08 +00001110
1111 // FIXME: We are artificially lowering the criteria to turn ADD %reg, $GA
1112 // to a LEA. This is determined with some expermentation but is by no means
1113 // optimal (especially for code size consideration). LEA is nice because of
1114 // its three-address nature. Tweak the cost function again when we can run
1115 // convertToThreeAddress() at register allocation time.
Evan Cheng25ab6902006-09-08 06:48:29 +00001116 if (AM.GV || AM.CP || AM.ES || AM.JT != -1) {
1117 // For X86-64, we should always use lea to materialize RIP relative
1118 // addresses.
Evan Cheng953fa042006-12-05 22:03:40 +00001119 if (Subtarget->is64Bit())
Evan Cheng25ab6902006-09-08 06:48:29 +00001120 Complexity = 4;
1121 else
1122 Complexity += 2;
1123 }
Evan Cheng51a9ed92006-02-25 10:09:08 +00001124
Gabor Greifba36cb52008-08-28 21:40:38 +00001125 if (AM.Disp && (AM.Base.Reg.getNode() || AM.IndexReg.getNode()))
Evan Cheng51a9ed92006-02-25 10:09:08 +00001126 Complexity++;
1127
1128 if (Complexity > 2) {
1129 getAddressOperands(AM, Base, Scale, Index, Disp);
1130 return true;
1131 }
Evan Cheng51a9ed92006-02-25 10:09:08 +00001132 return false;
1133}
1134
Dan Gohman475871a2008-07-27 21:46:04 +00001135bool X86DAGToDAGISel::TryFoldLoad(SDValue P, SDValue N,
1136 SDValue &Base, SDValue &Scale,
1137 SDValue &Index, SDValue &Disp) {
Gabor Greifba36cb52008-08-28 21:40:38 +00001138 if (ISD::isNON_EXTLoad(N.getNode()) &&
Evan Cheng5e351682006-02-06 06:02:33 +00001139 N.hasOneUse() &&
Evan Cheng884c70c2008-11-27 00:49:46 +00001140 IsLegalAndProfitableToFold(N.getNode(), P.getNode(), P.getNode()))
Evan Cheng0d538262006-11-08 20:34:28 +00001141 return SelectAddr(P, N.getOperand(1), Base, Scale, Index, Disp);
Evan Cheng0114e942006-01-06 20:36:21 +00001142 return false;
1143}
1144
Dan Gohman8b746962008-09-23 18:22:58 +00001145/// getGlobalBaseReg - Return an SDNode that returns the value of
1146/// the global base register. Output instructions required to
1147/// initialize the global base register, if necessary.
Evan Cheng7ccced62006-02-18 00:15:05 +00001148///
Evan Cheng9ade2182006-08-26 05:34:46 +00001149SDNode *X86DAGToDAGISel::getGlobalBaseReg() {
Dan Gohman57c3dac2008-09-30 00:58:23 +00001150 MachineFunction *MF = CurBB->getParent();
1151 unsigned GlobalBaseReg = TM.getInstrInfo()->getGlobalBaseReg(MF);
Gabor Greifba36cb52008-08-28 21:40:38 +00001152 return CurDAG->getRegister(GlobalBaseReg, TLI.getPointerTy()).getNode();
Evan Cheng7ccced62006-02-18 00:15:05 +00001153}
1154
Evan Chengb245d922006-05-20 01:36:52 +00001155static SDNode *FindCallStartFromCall(SDNode *Node) {
1156 if (Node->getOpcode() == ISD::CALLSEQ_START) return Node;
1157 assert(Node->getOperand(0).getValueType() == MVT::Other &&
1158 "Node doesn't have a token chain argument!");
Gabor Greifba36cb52008-08-28 21:40:38 +00001159 return FindCallStartFromCall(Node->getOperand(0).getNode());
Evan Chengb245d922006-05-20 01:36:52 +00001160}
1161
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001162/// getTruncateTo8Bit - return an SDNode that implements a subreg based
1163/// truncate of the specified operand to i8. This can be done with tablegen,
1164/// except that this code uses MVT::Flag in a tricky way that happens to
1165/// improve scheduling in some cases.
1166SDNode *X86DAGToDAGISel::getTruncateTo8Bit(SDValue N0) {
1167 assert(!Subtarget->is64Bit() &&
1168 "getTruncateTo8Bit is only needed on x86-32!");
1169 SDValue SRIdx = CurDAG->getTargetConstant(1, MVT::i32); // SubRegSet 1
Dale Johannesend8392542009-02-03 21:48:12 +00001170 DebugLoc dl = N0.getNode()->getDebugLoc();
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001171
1172 // Ensure that the source register has an 8-bit subreg on 32-bit targets
1173 unsigned Opc;
1174 MVT N0VT = N0.getValueType();
1175 switch (N0VT.getSimpleVT()) {
1176 default: assert(0 && "Unknown truncate!");
1177 case MVT::i16:
1178 Opc = X86::MOV16to16_;
1179 break;
1180 case MVT::i32:
1181 Opc = X86::MOV32to32_;
1182 break;
1183 }
1184
1185 // The use of MVT::Flag here is not strictly accurate, but it helps
1186 // scheduling in some cases.
Dale Johannesend8392542009-02-03 21:48:12 +00001187 N0 = SDValue(CurDAG->getTargetNode(Opc, dl, N0VT, MVT::Flag, N0), 0);
1188 return CurDAG->getTargetNode(X86::EXTRACT_SUBREG, dl,
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001189 MVT::i8, N0, SRIdx, N0.getValue(1));
Christopher Lambc59e5212007-08-10 21:48:46 +00001190}
1191
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001192SDNode *X86DAGToDAGISel::SelectAtomic64(SDNode *Node, unsigned Opc) {
1193 SDValue Chain = Node->getOperand(0);
1194 SDValue In1 = Node->getOperand(1);
1195 SDValue In2L = Node->getOperand(2);
1196 SDValue In2H = Node->getOperand(3);
1197 SDValue Tmp0, Tmp1, Tmp2, Tmp3;
1198 if (!SelectAddr(In1, In1, Tmp0, Tmp1, Tmp2, Tmp3))
1199 return NULL;
Dale Johannesen1b54c7f2008-10-03 19:41:08 +00001200 SDValue LSI = Node->getOperand(4); // MemOperand
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001201 const SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, In2L, In2H, LSI, Chain };
Dale Johannesend8392542009-02-03 21:48:12 +00001202 return CurDAG->getTargetNode(Opc, Node->getDebugLoc(),
1203 MVT::i32, MVT::i32, MVT::Other, Ops, 8);
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001204}
Christopher Lambc59e5212007-08-10 21:48:46 +00001205
Dan Gohman475871a2008-07-27 21:46:04 +00001206SDNode *X86DAGToDAGISel::Select(SDValue N) {
Gabor Greifba36cb52008-08-28 21:40:38 +00001207 SDNode *Node = N.getNode();
Duncan Sands83ec4b62008-06-06 12:08:01 +00001208 MVT NVT = Node->getValueType(0);
Evan Cheng0114e942006-01-06 20:36:21 +00001209 unsigned Opc, MOpc;
1210 unsigned Opcode = Node->getOpcode();
Dale Johannesend8392542009-02-03 21:48:12 +00001211 DebugLoc dl = Node->getDebugLoc();
1212
Evan Chengf597dc72006-02-10 22:24:32 +00001213#ifndef NDEBUG
Bill Wendling6345d752006-11-17 07:52:03 +00001214 DOUT << std::string(Indent, ' ') << "Selecting: ";
Evan Chengf597dc72006-02-10 22:24:32 +00001215 DEBUG(Node->dump(CurDAG));
Bill Wendling6345d752006-11-17 07:52:03 +00001216 DOUT << "\n";
Evan Cheng23addc02006-02-10 22:46:26 +00001217 Indent += 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001218#endif
1219
Dan Gohmane8be6c62008-07-17 19:10:17 +00001220 if (Node->isMachineOpcode()) {
Evan Chengf597dc72006-02-10 22:24:32 +00001221#ifndef NDEBUG
Bill Wendling6345d752006-11-17 07:52:03 +00001222 DOUT << std::string(Indent-2, ' ') << "== ";
Evan Chengf597dc72006-02-10 22:24:32 +00001223 DEBUG(Node->dump(CurDAG));
Bill Wendling6345d752006-11-17 07:52:03 +00001224 DOUT << "\n";
Evan Cheng23addc02006-02-10 22:46:26 +00001225 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001226#endif
Evan Cheng64a752f2006-08-11 09:08:15 +00001227 return NULL; // Already selected.
Evan Cheng34167212006-02-09 00:37:58 +00001228 }
Evan Cheng38262ca2006-01-11 22:15:18 +00001229
Evan Cheng0114e942006-01-06 20:36:21 +00001230 switch (Opcode) {
Chris Lattnerc961eea2005-11-16 01:54:32 +00001231 default: break;
Evan Cheng020d2e82006-02-23 20:41:18 +00001232 case X86ISD::GlobalBaseReg:
Evan Cheng9ade2182006-08-26 05:34:46 +00001233 return getGlobalBaseReg();
Evan Cheng020d2e82006-02-23 20:41:18 +00001234
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001235 case X86ISD::ATOMOR64_DAG:
1236 return SelectAtomic64(Node, X86::ATOMOR6432);
1237 case X86ISD::ATOMXOR64_DAG:
1238 return SelectAtomic64(Node, X86::ATOMXOR6432);
1239 case X86ISD::ATOMADD64_DAG:
1240 return SelectAtomic64(Node, X86::ATOMADD6432);
1241 case X86ISD::ATOMSUB64_DAG:
1242 return SelectAtomic64(Node, X86::ATOMSUB6432);
1243 case X86ISD::ATOMNAND64_DAG:
1244 return SelectAtomic64(Node, X86::ATOMNAND6432);
1245 case X86ISD::ATOMAND64_DAG:
1246 return SelectAtomic64(Node, X86::ATOMAND6432);
Dale Johannesen880ae362008-10-03 22:25:52 +00001247 case X86ISD::ATOMSWAP64_DAG:
1248 return SelectAtomic64(Node, X86::ATOMSWAP6432);
Dale Johannesen48c1bc22008-10-02 18:53:47 +00001249
Dan Gohman525178c2007-10-08 18:33:35 +00001250 case ISD::SMUL_LOHI:
1251 case ISD::UMUL_LOHI: {
Dan Gohman475871a2008-07-27 21:46:04 +00001252 SDValue N0 = Node->getOperand(0);
1253 SDValue N1 = Node->getOperand(1);
Dan Gohman525178c2007-10-08 18:33:35 +00001254
Dan Gohman525178c2007-10-08 18:33:35 +00001255 bool isSigned = Opcode == ISD::SMUL_LOHI;
1256 if (!isSigned)
Duncan Sands83ec4b62008-06-06 12:08:01 +00001257 switch (NVT.getSimpleVT()) {
Evan Cheng0114e942006-01-06 20:36:21 +00001258 default: assert(0 && "Unsupported VT!");
1259 case MVT::i8: Opc = X86::MUL8r; MOpc = X86::MUL8m; break;
1260 case MVT::i16: Opc = X86::MUL16r; MOpc = X86::MUL16m; break;
1261 case MVT::i32: Opc = X86::MUL32r; MOpc = X86::MUL32m; break;
Evan Cheng25ab6902006-09-08 06:48:29 +00001262 case MVT::i64: Opc = X86::MUL64r; MOpc = X86::MUL64m; break;
Evan Cheng0114e942006-01-06 20:36:21 +00001263 }
1264 else
Duncan Sands83ec4b62008-06-06 12:08:01 +00001265 switch (NVT.getSimpleVT()) {
Evan Cheng0114e942006-01-06 20:36:21 +00001266 default: assert(0 && "Unsupported VT!");
1267 case MVT::i8: Opc = X86::IMUL8r; MOpc = X86::IMUL8m; break;
1268 case MVT::i16: Opc = X86::IMUL16r; MOpc = X86::IMUL16m; break;
1269 case MVT::i32: Opc = X86::IMUL32r; MOpc = X86::IMUL32m; break;
Evan Cheng25ab6902006-09-08 06:48:29 +00001270 case MVT::i64: Opc = X86::IMUL64r; MOpc = X86::IMUL64m; break;
Evan Cheng0114e942006-01-06 20:36:21 +00001271 }
1272
1273 unsigned LoReg, HiReg;
Duncan Sands83ec4b62008-06-06 12:08:01 +00001274 switch (NVT.getSimpleVT()) {
Evan Cheng0114e942006-01-06 20:36:21 +00001275 default: assert(0 && "Unsupported VT!");
1276 case MVT::i8: LoReg = X86::AL; HiReg = X86::AH; break;
1277 case MVT::i16: LoReg = X86::AX; HiReg = X86::DX; break;
1278 case MVT::i32: LoReg = X86::EAX; HiReg = X86::EDX; break;
Evan Cheng25ab6902006-09-08 06:48:29 +00001279 case MVT::i64: LoReg = X86::RAX; HiReg = X86::RDX; break;
Evan Cheng0114e942006-01-06 20:36:21 +00001280 }
1281
Dan Gohman475871a2008-07-27 21:46:04 +00001282 SDValue Tmp0, Tmp1, Tmp2, Tmp3;
Evan Cheng7afa1662007-08-02 05:48:35 +00001283 bool foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
Dan Gohman525178c2007-10-08 18:33:35 +00001284 // multiplty is commmutative
Evan Cheng948f3432006-01-06 23:19:29 +00001285 if (!foldedLoad) {
Evan Cheng5e351682006-02-06 06:02:33 +00001286 foldedLoad = TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng7afa1662007-08-02 05:48:35 +00001287 if (foldedLoad)
1288 std::swap(N0, N1);
Evan Cheng948f3432006-01-06 23:19:29 +00001289 }
1290
Dan Gohman475871a2008-07-27 21:46:04 +00001291 SDValue InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), LoReg,
1292 N0, SDValue()).getValue(1);
Evan Cheng0114e942006-01-06 20:36:21 +00001293
1294 if (foldedLoad) {
Dan Gohman475871a2008-07-27 21:46:04 +00001295 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, N1.getOperand(0), InFlag };
Evan Cheng7e9b26f2006-02-09 07:17:49 +00001296 SDNode *CNode =
Dale Johannesend8392542009-02-03 21:48:12 +00001297 CurDAG->getTargetNode(MOpc, dl, MVT::Other, MVT::Flag, Ops, 6);
Dan Gohman475871a2008-07-27 21:46:04 +00001298 InFlag = SDValue(CNode, 1);
Dan Gohman525178c2007-10-08 18:33:35 +00001299 // Update the chain.
Dan Gohman475871a2008-07-27 21:46:04 +00001300 ReplaceUses(N1.getValue(1), SDValue(CNode, 0));
Evan Cheng0114e942006-01-06 20:36:21 +00001301 } else {
Evan Cheng7e9b26f2006-02-09 07:17:49 +00001302 InFlag =
Dale Johannesend8392542009-02-03 21:48:12 +00001303 SDValue(CurDAG->getTargetNode(Opc, dl, MVT::Flag, N1, InFlag), 0);
Evan Cheng0114e942006-01-06 20:36:21 +00001304 }
1305
Dan Gohman525178c2007-10-08 18:33:35 +00001306 // Copy the low half of the result, if it is needed.
1307 if (!N.getValue(0).use_empty()) {
Dan Gohman475871a2008-07-27 21:46:04 +00001308 SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
Dan Gohman525178c2007-10-08 18:33:35 +00001309 LoReg, NVT, InFlag);
1310 InFlag = Result.getValue(2);
1311 ReplaceUses(N.getValue(0), Result);
1312#ifndef NDEBUG
1313 DOUT << std::string(Indent-2, ' ') << "=> ";
Gabor Greifba36cb52008-08-28 21:40:38 +00001314 DEBUG(Result.getNode()->dump(CurDAG));
Dan Gohman525178c2007-10-08 18:33:35 +00001315 DOUT << "\n";
1316#endif
Evan Chengf7ef26e2007-08-09 21:59:35 +00001317 }
Dan Gohman525178c2007-10-08 18:33:35 +00001318 // Copy the high half of the result, if it is needed.
1319 if (!N.getValue(1).use_empty()) {
Dan Gohman475871a2008-07-27 21:46:04 +00001320 SDValue Result;
Dan Gohman525178c2007-10-08 18:33:35 +00001321 if (HiReg == X86::AH && Subtarget->is64Bit()) {
1322 // Prevent use of AH in a REX instruction by referencing AX instead.
1323 // Shift it down 8 bits.
1324 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
1325 X86::AX, MVT::i16, InFlag);
1326 InFlag = Result.getValue(2);
Dale Johannesend8392542009-02-03 21:48:12 +00001327 Result = SDValue(CurDAG->getTargetNode(X86::SHR16ri, dl, MVT::i16,
1328 Result,
Gabor Greif93c53e52008-08-31 15:37:04 +00001329 CurDAG->getTargetConstant(8, MVT::i8)), 0);
Dan Gohman525178c2007-10-08 18:33:35 +00001330 // Then truncate it down to i8.
Dan Gohman475871a2008-07-27 21:46:04 +00001331 SDValue SRIdx = CurDAG->getTargetConstant(1, MVT::i32); // SubRegSet 1
Dale Johannesend8392542009-02-03 21:48:12 +00001332 Result = SDValue(CurDAG->getTargetNode(X86::EXTRACT_SUBREG, dl,
Dan Gohman525178c2007-10-08 18:33:35 +00001333 MVT::i8, Result, SRIdx), 0);
1334 } else {
1335 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
1336 HiReg, NVT, InFlag);
1337 InFlag = Result.getValue(2);
1338 }
1339 ReplaceUses(N.getValue(1), Result);
1340#ifndef NDEBUG
1341 DOUT << std::string(Indent-2, ' ') << "=> ";
Gabor Greifba36cb52008-08-28 21:40:38 +00001342 DEBUG(Result.getNode()->dump(CurDAG));
Dan Gohman525178c2007-10-08 18:33:35 +00001343 DOUT << "\n";
1344#endif
1345 }
Evan Cheng34167212006-02-09 00:37:58 +00001346
Evan Chengf597dc72006-02-10 22:24:32 +00001347#ifndef NDEBUG
Evan Cheng23addc02006-02-10 22:46:26 +00001348 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001349#endif
Dan Gohman525178c2007-10-08 18:33:35 +00001350
Evan Cheng64a752f2006-08-11 09:08:15 +00001351 return NULL;
Evan Cheng948f3432006-01-06 23:19:29 +00001352 }
Evan Cheng7ccced62006-02-18 00:15:05 +00001353
Dan Gohman525178c2007-10-08 18:33:35 +00001354 case ISD::SDIVREM:
1355 case ISD::UDIVREM: {
Dan Gohman475871a2008-07-27 21:46:04 +00001356 SDValue N0 = Node->getOperand(0);
1357 SDValue N1 = Node->getOperand(1);
Dan Gohman525178c2007-10-08 18:33:35 +00001358
1359 bool isSigned = Opcode == ISD::SDIVREM;
Evan Cheng948f3432006-01-06 23:19:29 +00001360 if (!isSigned)
Duncan Sands83ec4b62008-06-06 12:08:01 +00001361 switch (NVT.getSimpleVT()) {
Evan Cheng948f3432006-01-06 23:19:29 +00001362 default: assert(0 && "Unsupported VT!");
1363 case MVT::i8: Opc = X86::DIV8r; MOpc = X86::DIV8m; break;
1364 case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break;
1365 case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break;
Evan Cheng25ab6902006-09-08 06:48:29 +00001366 case MVT::i64: Opc = X86::DIV64r; MOpc = X86::DIV64m; break;
Evan Cheng948f3432006-01-06 23:19:29 +00001367 }
1368 else
Duncan Sands83ec4b62008-06-06 12:08:01 +00001369 switch (NVT.getSimpleVT()) {
Evan Cheng948f3432006-01-06 23:19:29 +00001370 default: assert(0 && "Unsupported VT!");
1371 case MVT::i8: Opc = X86::IDIV8r; MOpc = X86::IDIV8m; break;
1372 case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break;
1373 case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break;
Evan Cheng25ab6902006-09-08 06:48:29 +00001374 case MVT::i64: Opc = X86::IDIV64r; MOpc = X86::IDIV64m; break;
Evan Cheng948f3432006-01-06 23:19:29 +00001375 }
1376
1377 unsigned LoReg, HiReg;
1378 unsigned ClrOpcode, SExtOpcode;
Duncan Sands83ec4b62008-06-06 12:08:01 +00001379 switch (NVT.getSimpleVT()) {
Evan Cheng948f3432006-01-06 23:19:29 +00001380 default: assert(0 && "Unsupported VT!");
1381 case MVT::i8:
1382 LoReg = X86::AL; HiReg = X86::AH;
Evan Chengb1409ce2006-11-17 22:10:14 +00001383 ClrOpcode = 0;
Evan Cheng948f3432006-01-06 23:19:29 +00001384 SExtOpcode = X86::CBW;
1385 break;
1386 case MVT::i16:
1387 LoReg = X86::AX; HiReg = X86::DX;
Evan Chengaede9b92006-06-02 21:20:34 +00001388 ClrOpcode = X86::MOV16r0;
Evan Cheng948f3432006-01-06 23:19:29 +00001389 SExtOpcode = X86::CWD;
1390 break;
1391 case MVT::i32:
1392 LoReg = X86::EAX; HiReg = X86::EDX;
Evan Chengaede9b92006-06-02 21:20:34 +00001393 ClrOpcode = X86::MOV32r0;
Evan Cheng948f3432006-01-06 23:19:29 +00001394 SExtOpcode = X86::CDQ;
1395 break;
Evan Cheng25ab6902006-09-08 06:48:29 +00001396 case MVT::i64:
1397 LoReg = X86::RAX; HiReg = X86::RDX;
1398 ClrOpcode = X86::MOV64r0;
1399 SExtOpcode = X86::CQO;
1400 break;
Evan Cheng948f3432006-01-06 23:19:29 +00001401 }
1402
Dan Gohman475871a2008-07-27 21:46:04 +00001403 SDValue Tmp0, Tmp1, Tmp2, Tmp3;
Dan Gohman525178c2007-10-08 18:33:35 +00001404 bool foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
Dan Gohman1ef4d8f2009-01-21 14:50:16 +00001405 bool signBitIsZero = CurDAG->SignBitIsZero(N0);
Dan Gohman525178c2007-10-08 18:33:35 +00001406
Dan Gohman475871a2008-07-27 21:46:04 +00001407 SDValue InFlag;
Dan Gohman1ef4d8f2009-01-21 14:50:16 +00001408 if (NVT == MVT::i8 && (!isSigned || signBitIsZero)) {
Evan Chengb1409ce2006-11-17 22:10:14 +00001409 // Special case for div8, just use a move with zero extension to AX to
1410 // clear the upper 8 bits (AH).
Dan Gohman475871a2008-07-27 21:46:04 +00001411 SDValue Tmp0, Tmp1, Tmp2, Tmp3, Move, Chain;
Evan Chengb1409ce2006-11-17 22:10:14 +00001412 if (TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3)) {
Dan Gohman475871a2008-07-27 21:46:04 +00001413 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, N0.getOperand(0) };
Evan Chengb1409ce2006-11-17 22:10:14 +00001414 Move =
Dale Johannesend8392542009-02-03 21:48:12 +00001415 SDValue(CurDAG->getTargetNode(X86::MOVZX16rm8, dl, MVT::i16,
1416 MVT::Other, Ops, 5), 0);
Evan Chengb1409ce2006-11-17 22:10:14 +00001417 Chain = Move.getValue(1);
1418 ReplaceUses(N0.getValue(1), Chain);
1419 } else {
Evan Chengb1409ce2006-11-17 22:10:14 +00001420 Move =
Dale Johannesend8392542009-02-03 21:48:12 +00001421 SDValue(CurDAG->getTargetNode(X86::MOVZX16rr8, dl, MVT::i16, N0),0);
Evan Chengb1409ce2006-11-17 22:10:14 +00001422 Chain = CurDAG->getEntryNode();
1423 }
Dan Gohman475871a2008-07-27 21:46:04 +00001424 Chain = CurDAG->getCopyToReg(Chain, X86::AX, Move, SDValue());
Evan Cheng948f3432006-01-06 23:19:29 +00001425 InFlag = Chain.getValue(1);
Evan Chengb1409ce2006-11-17 22:10:14 +00001426 } else {
Evan Chengb1409ce2006-11-17 22:10:14 +00001427 InFlag =
Dan Gohman525178c2007-10-08 18:33:35 +00001428 CurDAG->getCopyToReg(CurDAG->getEntryNode(),
Dan Gohman475871a2008-07-27 21:46:04 +00001429 LoReg, N0, SDValue()).getValue(1);
Dan Gohman1ef4d8f2009-01-21 14:50:16 +00001430 if (isSigned && !signBitIsZero) {
Evan Chengb1409ce2006-11-17 22:10:14 +00001431 // Sign extend the low part into the high part.
1432 InFlag =
Dale Johannesend8392542009-02-03 21:48:12 +00001433 SDValue(CurDAG->getTargetNode(SExtOpcode, dl, MVT::Flag, InFlag),0);
Evan Chengb1409ce2006-11-17 22:10:14 +00001434 } else {
1435 // Zero out the high part, effectively zero extending the input.
Dale Johannesend8392542009-02-03 21:48:12 +00001436 SDValue ClrNode = SDValue(CurDAG->getTargetNode(ClrOpcode, dl, NVT),
1437 0);
Dan Gohman525178c2007-10-08 18:33:35 +00001438 InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), HiReg,
1439 ClrNode, InFlag).getValue(1);
Evan Chengb1409ce2006-11-17 22:10:14 +00001440 }
Evan Cheng948f3432006-01-06 23:19:29 +00001441 }
1442
1443 if (foldedLoad) {
Dan Gohman475871a2008-07-27 21:46:04 +00001444 SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, N1.getOperand(0), InFlag };
Evan Cheng7e9b26f2006-02-09 07:17:49 +00001445 SDNode *CNode =
Dale Johannesend8392542009-02-03 21:48:12 +00001446 CurDAG->getTargetNode(MOpc, dl, MVT::Other, MVT::Flag, Ops, 6);
Dan Gohman475871a2008-07-27 21:46:04 +00001447 InFlag = SDValue(CNode, 1);
Dan Gohman525178c2007-10-08 18:33:35 +00001448 // Update the chain.
Dan Gohman475871a2008-07-27 21:46:04 +00001449 ReplaceUses(N1.getValue(1), SDValue(CNode, 0));
Evan Cheng948f3432006-01-06 23:19:29 +00001450 } else {
Evan Cheng7e9b26f2006-02-09 07:17:49 +00001451 InFlag =
Dale Johannesend8392542009-02-03 21:48:12 +00001452 SDValue(CurDAG->getTargetNode(Opc, dl, MVT::Flag, N1, InFlag), 0);
Evan Cheng948f3432006-01-06 23:19:29 +00001453 }
1454
Dan Gohmana37c9f72007-09-25 18:23:27 +00001455 // Copy the division (low) result, if it is needed.
1456 if (!N.getValue(0).use_empty()) {
Dan Gohman475871a2008-07-27 21:46:04 +00001457 SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
Dan Gohman525178c2007-10-08 18:33:35 +00001458 LoReg, NVT, InFlag);
Dan Gohmana37c9f72007-09-25 18:23:27 +00001459 InFlag = Result.getValue(2);
1460 ReplaceUses(N.getValue(0), Result);
1461#ifndef NDEBUG
1462 DOUT << std::string(Indent-2, ' ') << "=> ";
Gabor Greifba36cb52008-08-28 21:40:38 +00001463 DEBUG(Result.getNode()->dump(CurDAG));
Dan Gohmana37c9f72007-09-25 18:23:27 +00001464 DOUT << "\n";
1465#endif
Evan Chengf7ef26e2007-08-09 21:59:35 +00001466 }
Dan Gohmana37c9f72007-09-25 18:23:27 +00001467 // Copy the remainder (high) result, if it is needed.
1468 if (!N.getValue(1).use_empty()) {
Dan Gohman475871a2008-07-27 21:46:04 +00001469 SDValue Result;
Dan Gohmana37c9f72007-09-25 18:23:27 +00001470 if (HiReg == X86::AH && Subtarget->is64Bit()) {
1471 // Prevent use of AH in a REX instruction by referencing AX instead.
1472 // Shift it down 8 bits.
Dan Gohman525178c2007-10-08 18:33:35 +00001473 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
1474 X86::AX, MVT::i16, InFlag);
Dan Gohmana37c9f72007-09-25 18:23:27 +00001475 InFlag = Result.getValue(2);
Dale Johannesend8392542009-02-03 21:48:12 +00001476 Result = SDValue(CurDAG->getTargetNode(X86::SHR16ri, dl, MVT::i16,
1477 Result,
1478 CurDAG->getTargetConstant(8, MVT::i8)),
1479 0);
Dan Gohmana37c9f72007-09-25 18:23:27 +00001480 // Then truncate it down to i8.
Dan Gohman475871a2008-07-27 21:46:04 +00001481 SDValue SRIdx = CurDAG->getTargetConstant(1, MVT::i32); // SubRegSet 1
Dale Johannesend8392542009-02-03 21:48:12 +00001482 Result = SDValue(CurDAG->getTargetNode(X86::EXTRACT_SUBREG, dl,
Dan Gohmana37c9f72007-09-25 18:23:27 +00001483 MVT::i8, Result, SRIdx), 0);
1484 } else {
Dan Gohman525178c2007-10-08 18:33:35 +00001485 Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
1486 HiReg, NVT, InFlag);
Dan Gohmana37c9f72007-09-25 18:23:27 +00001487 InFlag = Result.getValue(2);
1488 }
1489 ReplaceUses(N.getValue(1), Result);
1490#ifndef NDEBUG
1491 DOUT << std::string(Indent-2, ' ') << "=> ";
Gabor Greifba36cb52008-08-28 21:40:38 +00001492 DEBUG(Result.getNode()->dump(CurDAG));
Dan Gohmana37c9f72007-09-25 18:23:27 +00001493 DOUT << "\n";
1494#endif
1495 }
Evan Chengf597dc72006-02-10 22:24:32 +00001496
1497#ifndef NDEBUG
Evan Cheng23addc02006-02-10 22:46:26 +00001498 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001499#endif
Evan Cheng64a752f2006-08-11 09:08:15 +00001500
1501 return NULL;
Evan Cheng0114e942006-01-06 20:36:21 +00001502 }
Christopher Lamba1eb1552007-08-10 22:22:41 +00001503
Christopher Lambc59e5212007-08-10 21:48:46 +00001504 case ISD::SIGN_EXTEND_INREG: {
Duncan Sands83ec4b62008-06-06 12:08:01 +00001505 MVT SVT = cast<VTSDNode>(Node->getOperand(1))->getVT();
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001506 if (SVT == MVT::i8 && !Subtarget->is64Bit()) {
1507 SDValue N0 = Node->getOperand(0);
Christopher Lambc59e5212007-08-10 21:48:46 +00001508
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001509 SDValue TruncOp = SDValue(getTruncateTo8Bit(N0), 0);
1510 unsigned Opc = 0;
1511 switch (NVT.getSimpleVT()) {
1512 default: assert(0 && "Unknown sign_extend_inreg!");
1513 case MVT::i16:
1514 Opc = X86::MOVSX16rr8;
1515 break;
1516 case MVT::i32:
1517 Opc = X86::MOVSX32rr8;
1518 break;
1519 }
1520
Dale Johannesend8392542009-02-03 21:48:12 +00001521 SDNode *ResNode = CurDAG->getTargetNode(Opc, dl, NVT, TruncOp);
Christopher Lambc59e5212007-08-10 21:48:46 +00001522
1523#ifndef NDEBUG
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001524 DOUT << std::string(Indent-2, ' ') << "=> ";
Gabor Greifba36cb52008-08-28 21:40:38 +00001525 DEBUG(TruncOp.getNode()->dump(CurDAG));
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001526 DOUT << "\n";
1527 DOUT << std::string(Indent-2, ' ') << "=> ";
1528 DEBUG(ResNode->dump(CurDAG));
1529 DOUT << "\n";
1530 Indent -= 2;
Christopher Lambc59e5212007-08-10 21:48:46 +00001531#endif
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001532 return ResNode;
1533 }
Christopher Lambc59e5212007-08-10 21:48:46 +00001534 break;
1535 }
1536
1537 case ISD::TRUNCATE: {
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001538 if (NVT == MVT::i8 && !Subtarget->is64Bit()) {
1539 SDValue Input = Node->getOperand(0);
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001540 SDNode *ResNode = getTruncateTo8Bit(Input);
Christopher Lambc59e5212007-08-10 21:48:46 +00001541
Evan Cheng403be7e2006-05-08 08:01:26 +00001542#ifndef NDEBUG
Bill Wendling6345d752006-11-17 07:52:03 +00001543 DOUT << std::string(Indent-2, ' ') << "=> ";
Evan Cheng9ade2182006-08-26 05:34:46 +00001544 DEBUG(ResNode->dump(CurDAG));
Bill Wendling6345d752006-11-17 07:52:03 +00001545 DOUT << "\n";
Evan Cheng403be7e2006-05-08 08:01:26 +00001546 Indent -= 2;
1547#endif
Dan Gohman0bfa1bf2008-08-20 21:27:32 +00001548 return ResNode;
1549 }
Evan Cheng6b2e2542006-05-20 07:44:28 +00001550 break;
Evan Cheng403be7e2006-05-08 08:01:26 +00001551 }
Evan Cheng851bc042008-06-17 02:01:22 +00001552
1553 case ISD::DECLARE: {
1554 // Handle DECLARE nodes here because the second operand may have been
1555 // wrapped in X86ISD::Wrapper.
Dan Gohman475871a2008-07-27 21:46:04 +00001556 SDValue Chain = Node->getOperand(0);
1557 SDValue N1 = Node->getOperand(1);
1558 SDValue N2 = Node->getOperand(2);
Evan Cheng6bb14ca2008-12-10 21:49:05 +00001559 FrameIndexSDNode *FINode = dyn_cast<FrameIndexSDNode>(N1);
1560 if (!FINode)
Evan Chengfab83872008-06-18 02:48:27 +00001561 break;
Evan Chengfab83872008-06-18 02:48:27 +00001562 if (N2.getOpcode() == ISD::ADD &&
1563 N2.getOperand(0).getOpcode() == X86ISD::GlobalBaseReg)
1564 N2 = N2.getOperand(1);
Evan Cheng6bb14ca2008-12-10 21:49:05 +00001565 if (N2.getOpcode() != X86ISD::Wrapper)
1566 break;
Evan Chengf2accb52009-01-10 03:33:22 +00001567 GlobalAddressSDNode *GVNode =
1568 dyn_cast<GlobalAddressSDNode>(N2.getOperand(0));
Evan Cheng6bb14ca2008-12-10 21:49:05 +00001569 if (!GVNode)
1570 break;
1571 SDValue Tmp1 = CurDAG->getTargetFrameIndex(FINode->getIndex(),
1572 TLI.getPointerTy());
1573 SDValue Tmp2 = CurDAG->getTargetGlobalAddress(GVNode->getGlobal(),
1574 TLI.getPointerTy());
1575 SDValue Ops[] = { Tmp1, Tmp2, Chain };
Dale Johannesend8392542009-02-03 21:48:12 +00001576 return CurDAG->getTargetNode(TargetInstrInfo::DECLARE, dl,
Evan Cheng6bb14ca2008-12-10 21:49:05 +00001577 MVT::Other, Ops, 3);
Evan Cheng851bc042008-06-17 02:01:22 +00001578 break;
1579 }
Chris Lattnerc961eea2005-11-16 01:54:32 +00001580 }
1581
Evan Cheng9ade2182006-08-26 05:34:46 +00001582 SDNode *ResNode = SelectCode(N);
Evan Cheng64a752f2006-08-11 09:08:15 +00001583
Evan Chengf597dc72006-02-10 22:24:32 +00001584#ifndef NDEBUG
Bill Wendling6345d752006-11-17 07:52:03 +00001585 DOUT << std::string(Indent-2, ' ') << "=> ";
Gabor Greifba36cb52008-08-28 21:40:38 +00001586 if (ResNode == NULL || ResNode == N.getNode())
1587 DEBUG(N.getNode()->dump(CurDAG));
Evan Cheng9ade2182006-08-26 05:34:46 +00001588 else
1589 DEBUG(ResNode->dump(CurDAG));
Bill Wendling6345d752006-11-17 07:52:03 +00001590 DOUT << "\n";
Evan Cheng23addc02006-02-10 22:46:26 +00001591 Indent -= 2;
Evan Chengf597dc72006-02-10 22:24:32 +00001592#endif
Evan Cheng64a752f2006-08-11 09:08:15 +00001593
1594 return ResNode;
Chris Lattnerc961eea2005-11-16 01:54:32 +00001595}
1596
Chris Lattnerc0bad572006-06-08 18:03:49 +00001597bool X86DAGToDAGISel::
Dan Gohman475871a2008-07-27 21:46:04 +00001598SelectInlineAsmMemoryOperand(const SDValue &Op, char ConstraintCode,
Dan Gohmanf350b272008-08-23 02:25:05 +00001599 std::vector<SDValue> &OutOps) {
Dan Gohman475871a2008-07-27 21:46:04 +00001600 SDValue Op0, Op1, Op2, Op3;
Chris Lattnerc0bad572006-06-08 18:03:49 +00001601 switch (ConstraintCode) {
1602 case 'o': // offsetable ??
1603 case 'v': // not offsetable ??
1604 default: return true;
1605 case 'm': // memory
Evan Cheng0d538262006-11-08 20:34:28 +00001606 if (!SelectAddr(Op, Op, Op0, Op1, Op2, Op3))
Chris Lattnerc0bad572006-06-08 18:03:49 +00001607 return true;
1608 break;
1609 }
1610
Evan Cheng04699902006-08-26 01:05:16 +00001611 OutOps.push_back(Op0);
1612 OutOps.push_back(Op1);
1613 OutOps.push_back(Op2);
1614 OutOps.push_back(Op3);
Chris Lattnerc0bad572006-06-08 18:03:49 +00001615 return false;
1616}
1617
Chris Lattnerc961eea2005-11-16 01:54:32 +00001618/// createX86ISelDag - This pass converts a legalized DAG into a
1619/// X86-specific DAG, ready for instruction scheduling.
1620///
Evan Chenge50794a2006-08-29 18:28:33 +00001621FunctionPass *llvm::createX86ISelDag(X86TargetMachine &TM, bool Fast) {
1622 return new X86DAGToDAGISel(TM, Fast);
Chris Lattnerc961eea2005-11-16 01:54:32 +00001623}