blob: a01a85cd0e164fb468c2add7d065708d9f653c38 [file] [log] [blame]
Chris Lattner5930d3d2005-11-16 22:59:19 +00001//===- X86ISelDAGToDAG.cpp - A DAG pattern matching inst selector for X86 -===//
Chris Lattner655e7df2005-11-16 01:54:32 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by the Evan Cheng and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
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 Chengd49cc362006-02-10 22:24:32 +000015#define DEBUG_TYPE "isel"
Chris Lattner655e7df2005-11-16 01:54:32 +000016#include "X86.h"
Evan Chengbc7a0f442006-01-11 06:09:51 +000017#include "X86InstrBuilder.h"
Chris Lattner7c551262006-01-11 01:15:34 +000018#include "X86RegisterInfo.h"
Chris Lattner655e7df2005-11-16 01:54:32 +000019#include "X86Subtarget.h"
20#include "X86ISelLowering.h"
Chris Lattner3f0f71b2005-11-19 02:11:08 +000021#include "llvm/GlobalValue.h"
Chris Lattner7c551262006-01-11 01:15:34 +000022#include "llvm/Instructions.h"
23#include "llvm/Support/CFG.h"
Chris Lattner3f0f71b2005-11-19 02:11:08 +000024#include "llvm/CodeGen/MachineConstantPool.h"
Chris Lattner655e7df2005-11-16 01:54:32 +000025#include "llvm/CodeGen/MachineFunction.h"
Evan Cheng73a1ad92006-01-10 20:26:56 +000026#include "llvm/CodeGen/MachineFrameInfo.h"
Chris Lattner7c551262006-01-11 01:15:34 +000027#include "llvm/CodeGen/MachineInstrBuilder.h"
28#include "llvm/CodeGen/SSARegMap.h"
Chris Lattner655e7df2005-11-16 01:54:32 +000029#include "llvm/CodeGen/SelectionDAGISel.h"
30#include "llvm/Target/TargetMachine.h"
31#include "llvm/Support/Debug.h"
32#include "llvm/ADT/Statistic.h"
Chris Lattnerde02d772006-01-22 23:41:00 +000033#include <iostream>
Evan Cheng54cb1832006-02-05 06:46:41 +000034#include <set>
Chris Lattner655e7df2005-11-16 01:54:32 +000035using namespace llvm;
36
37//===----------------------------------------------------------------------===//
38// Pattern Matcher Implementation
39//===----------------------------------------------------------------------===//
40
41namespace {
Chris Lattner3f0f71b2005-11-19 02:11:08 +000042 /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses
43 /// SDOperand's instead of register numbers for the leaves of the matched
44 /// tree.
45 struct X86ISelAddressMode {
46 enum {
47 RegBase,
48 FrameIndexBase,
Evan Chengc9fab312005-12-08 02:01:35 +000049 ConstantPoolBase
Chris Lattner3f0f71b2005-11-19 02:11:08 +000050 } BaseType;
51
52 struct { // This is really a union, discriminated by BaseType!
53 SDOperand Reg;
54 int FrameIndex;
55 } Base;
56
57 unsigned Scale;
58 SDOperand IndexReg;
59 unsigned Disp;
60 GlobalValue *GV;
61
62 X86ISelAddressMode()
Evan Cheng4eb7af92005-11-30 02:51:20 +000063 : BaseType(RegBase), Scale(1), IndexReg(), Disp(0), GV(0) {
Chris Lattner3f0f71b2005-11-19 02:11:08 +000064 }
65 };
66}
67
68namespace {
Chris Lattner655e7df2005-11-16 01:54:32 +000069 Statistic<>
70 NumFPKill("x86-codegen", "Number of FP_REG_KILL instructions added");
71
72 //===--------------------------------------------------------------------===//
73 /// ISel - X86 specific code to select X86 machine instructions for
74 /// SelectionDAG operations.
75 ///
76 class X86DAGToDAGISel : public SelectionDAGISel {
77 /// ContainsFPCode - Every instruction we select that uses or defines a FP
78 /// register should set this to true.
79 bool ContainsFPCode;
80
81 /// X86Lowering - This object fully describes how to lower LLVM code to an
82 /// X86-specific SelectionDAG.
83 X86TargetLowering X86Lowering;
84
85 /// Subtarget - Keep a pointer to the X86Subtarget around so that we can
86 /// make the right decision when generating code for different targets.
87 const X86Subtarget *Subtarget;
88 public:
89 X86DAGToDAGISel(TargetMachine &TM)
90 : SelectionDAGISel(X86Lowering), X86Lowering(TM) {
91 Subtarget = &TM.getSubtarget<X86Subtarget>();
92 }
93
94 virtual const char *getPassName() const {
95 return "X86 DAG->DAG Instruction Selection";
96 }
97
98 /// InstructionSelectBasicBlock - This callback is invoked by
99 /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
100 virtual void InstructionSelectBasicBlock(SelectionDAG &DAG);
101
Evan Chengbc7a0f442006-01-11 06:09:51 +0000102 virtual void EmitFunctionEntryCode(Function &Fn, MachineFunction &MF);
103
Chris Lattner655e7df2005-11-16 01:54:32 +0000104// Include the pieces autogenerated from the target description.
105#include "X86GenDAGISel.inc"
106
107 private:
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000108 void Select(SDOperand &Result, SDOperand N);
Chris Lattner655e7df2005-11-16 01:54:32 +0000109
Evan Chenga86ba852006-02-11 02:05:36 +0000110 bool MatchAddress(SDOperand N, X86ISelAddressMode &AM, bool isRoot = true);
Evan Chengc9fab312005-12-08 02:01:35 +0000111 bool SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
112 SDOperand &Index, SDOperand &Disp);
113 bool SelectLEAAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
114 SDOperand &Index, SDOperand &Disp);
Evan Chengd5f2ba02006-02-06 06:02:33 +0000115 bool TryFoldLoad(SDOperand P, SDOperand N,
116 SDOperand &Base, SDOperand &Scale,
Evan Cheng10d27902006-01-06 20:36:21 +0000117 SDOperand &Index, SDOperand &Disp);
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000118
Evan Cheng67ed58e2005-12-12 21:49:40 +0000119 inline void getAddressOperands(X86ISelAddressMode &AM, SDOperand &Base,
120 SDOperand &Scale, SDOperand &Index,
121 SDOperand &Disp) {
122 Base = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) ?
123 CurDAG->getTargetFrameIndex(AM.Base.FrameIndex, MVT::i32) : AM.Base.Reg;
Evan Cheng1d712482005-12-17 09:13:43 +0000124 Scale = getI8Imm(AM.Scale);
Evan Cheng67ed58e2005-12-12 21:49:40 +0000125 Index = AM.IndexReg;
126 Disp = AM.GV ? CurDAG->getTargetGlobalAddress(AM.GV, MVT::i32, AM.Disp)
127 : getI32Imm(AM.Disp);
128 }
129
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000130 /// getI8Imm - Return a target constant with the specified value, of type
131 /// i8.
132 inline SDOperand getI8Imm(unsigned Imm) {
133 return CurDAG->getTargetConstant(Imm, MVT::i8);
134 }
135
Chris Lattner655e7df2005-11-16 01:54:32 +0000136 /// getI16Imm - Return a target constant with the specified value, of type
137 /// i16.
138 inline SDOperand getI16Imm(unsigned Imm) {
139 return CurDAG->getTargetConstant(Imm, MVT::i16);
140 }
141
142 /// getI32Imm - Return a target constant with the specified value, of type
143 /// i32.
144 inline SDOperand getI32Imm(unsigned Imm) {
145 return CurDAG->getTargetConstant(Imm, MVT::i32);
146 }
Evan Chengd49cc362006-02-10 22:24:32 +0000147
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000148#ifndef NDEBUG
149 unsigned Indent;
150#endif
Chris Lattner655e7df2005-11-16 01:54:32 +0000151 };
152}
153
154/// InstructionSelectBasicBlock - This callback is invoked by SelectionDAGISel
155/// when it has created a SelectionDAG for us to codegen.
156void X86DAGToDAGISel::InstructionSelectBasicBlock(SelectionDAG &DAG) {
157 DEBUG(BB->dump());
Chris Lattner7c551262006-01-11 01:15:34 +0000158 MachineFunction::iterator FirstMBB = BB;
Chris Lattner655e7df2005-11-16 01:54:32 +0000159
160 // Codegen the basic block.
Evan Chengd49cc362006-02-10 22:24:32 +0000161#ifndef NDEBUG
162 DEBUG(std::cerr << "===== Instruction selection begins:\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000163 Indent = 0;
Evan Chengd49cc362006-02-10 22:24:32 +0000164#endif
Evan Cheng54cb1832006-02-05 06:46:41 +0000165 DAG.setRoot(SelectRoot(DAG.getRoot()));
Evan Chengd49cc362006-02-10 22:24:32 +0000166#ifndef NDEBUG
167 DEBUG(std::cerr << "===== Instruction selection ends:\n");
168#endif
Evan Cheng1d9b6712005-12-19 22:36:02 +0000169 CodeGenMap.clear();
Chris Lattner655e7df2005-11-16 01:54:32 +0000170 DAG.RemoveDeadNodes();
171
172 // Emit machine code to BB.
173 ScheduleAndEmitDAG(DAG);
Chris Lattner7c551262006-01-11 01:15:34 +0000174
175 // If we are emitting FP stack code, scan the basic block to determine if this
176 // block defines any FP values. If so, put an FP_REG_KILL instruction before
177 // the terminator of the block.
Evan Chengcde9e302006-01-27 08:10:46 +0000178 if (!Subtarget->hasSSE2()) {
Chris Lattner7c551262006-01-11 01:15:34 +0000179 // Note that FP stack instructions *are* used in SSE code when returning
180 // values, but these are not live out of the basic block, so we don't need
181 // an FP_REG_KILL in this case either.
182 bool ContainsFPCode = false;
183
184 // Scan all of the machine instructions in these MBBs, checking for FP
185 // stores.
186 MachineFunction::iterator MBBI = FirstMBB;
187 do {
188 for (MachineBasicBlock::iterator I = MBBI->begin(), E = MBBI->end();
189 !ContainsFPCode && I != E; ++I) {
190 for (unsigned op = 0, e = I->getNumOperands(); op != e; ++op) {
191 if (I->getOperand(op).isRegister() && I->getOperand(op).isDef() &&
192 MRegisterInfo::isVirtualRegister(I->getOperand(op).getReg()) &&
193 RegMap->getRegClass(I->getOperand(0).getReg()) ==
194 X86::RFPRegisterClass) {
195 ContainsFPCode = true;
196 break;
197 }
198 }
199 }
200 } while (!ContainsFPCode && &*(MBBI++) != BB);
201
202 // Check PHI nodes in successor blocks. These PHI's will be lowered to have
203 // a copy of the input value in this block.
204 if (!ContainsFPCode) {
205 // Final check, check LLVM BB's that are successors to the LLVM BB
206 // corresponding to BB for FP PHI nodes.
207 const BasicBlock *LLVMBB = BB->getBasicBlock();
208 const PHINode *PN;
209 for (succ_const_iterator SI = succ_begin(LLVMBB), E = succ_end(LLVMBB);
210 !ContainsFPCode && SI != E; ++SI) {
211 for (BasicBlock::const_iterator II = SI->begin();
212 (PN = dyn_cast<PHINode>(II)); ++II) {
213 if (PN->getType()->isFloatingPoint()) {
214 ContainsFPCode = true;
215 break;
216 }
217 }
218 }
219 }
220
221 // Finally, if we found any FP code, emit the FP_REG_KILL instruction.
222 if (ContainsFPCode) {
223 BuildMI(*BB, BB->getFirstTerminator(), X86::FP_REG_KILL, 0);
224 ++NumFPKill;
225 }
226 }
Chris Lattner655e7df2005-11-16 01:54:32 +0000227}
228
Evan Chengbc7a0f442006-01-11 06:09:51 +0000229/// EmitSpecialCodeForMain - Emit any code that needs to be executed only in
230/// the main function.
231static void EmitSpecialCodeForMain(MachineBasicBlock *BB,
232 MachineFrameInfo *MFI) {
233 // Switch the FPU to 64-bit precision mode for better compatibility and speed.
234 int CWFrameIdx = MFI->CreateStackObject(2, 2);
235 addFrameReference(BuildMI(BB, X86::FNSTCW16m, 4), CWFrameIdx);
236
237 // Set the high part to be 64-bit precision.
238 addFrameReference(BuildMI(BB, X86::MOV8mi, 5),
239 CWFrameIdx, 1).addImm(2);
240
241 // Reload the modified control word now.
242 addFrameReference(BuildMI(BB, X86::FLDCW16m, 4), CWFrameIdx);
243}
244
245void X86DAGToDAGISel::EmitFunctionEntryCode(Function &Fn, MachineFunction &MF) {
246 // If this is main, emit special code for main.
247 MachineBasicBlock *BB = MF.begin();
248 if (Fn.hasExternalLinkage() && Fn.getName() == "main")
249 EmitSpecialCodeForMain(BB, MF.getFrameInfo());
250}
251
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000252/// MatchAddress - Add the specified node to the specified addressing mode,
253/// returning true if it cannot be done. This just pattern matches for the
254/// addressing mode
Evan Chenga86ba852006-02-11 02:05:36 +0000255bool X86DAGToDAGISel::MatchAddress(SDOperand N, X86ISelAddressMode &AM,
256 bool isRoot) {
257 bool StopHere = false;
258 // If N has already been selected, we may or may not want to fold its
259 // operands into the addressing mode. It will result in code duplication!
260 // FIXME: Right now we do. That is, as long as the selected target node
261 // does not produce a chain. This may require a more sophisticated heuristics.
262 std::map<SDOperand, SDOperand>::iterator CGMI= CodeGenMap.find(N.getValue(0));
263 if (CGMI != CodeGenMap.end()) {
264 if (isRoot)
265 // Stop here if it is a root. It's probably not profitable to go deeper.
266 StopHere = true;
267 else {
268 for (unsigned i = 0, e = CGMI->second.Val->getNumValues(); i != e; ++i) {
269 if (CGMI->second.Val->getValueType(i) == MVT::Other)
270 StopHere = true;
271 }
272 }
273 }
274
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000275 switch (N.getOpcode()) {
276 default: break;
277 case ISD::FrameIndex:
278 if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
279 AM.BaseType = X86ISelAddressMode::FrameIndexBase;
280 AM.Base.FrameIndex = cast<FrameIndexSDNode>(N)->getIndex();
281 return false;
282 }
283 break;
Evan Chengc9fab312005-12-08 02:01:35 +0000284
285 case ISD::ConstantPool:
286 if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
287 if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N)) {
288 AM.BaseType = X86ISelAddressMode::ConstantPoolBase;
Evan Cheng72d5c252006-01-31 22:28:30 +0000289 AM.Base.Reg = CurDAG->getTargetConstantPool(CP->get(), MVT::i32,
290 CP->getAlignment());
Evan Chengc9fab312005-12-08 02:01:35 +0000291 return false;
292 }
293 }
294 break;
295
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000296 case ISD::GlobalAddress:
Evan Cheng9cdc16c2005-12-21 23:05:39 +0000297 case ISD::TargetGlobalAddress:
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000298 if (AM.GV == 0) {
Evan Chenga74ce622005-12-21 02:39:21 +0000299 AM.GV = cast<GlobalAddressSDNode>(N)->getGlobal();
Evan Cheng1d712482005-12-17 09:13:43 +0000300 return false;
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000301 }
302 break;
Evan Chengc9fab312005-12-08 02:01:35 +0000303
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000304 case ISD::Constant:
305 AM.Disp += cast<ConstantSDNode>(N)->getValue();
306 return false;
Evan Chengc9fab312005-12-08 02:01:35 +0000307
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000308 case ISD::SHL:
Evan Chenga86ba852006-02-11 02:05:36 +0000309 if (!StopHere && AM.IndexReg.Val == 0 && AM.Scale == 1)
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000310 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1))) {
311 unsigned Val = CN->getValue();
312 if (Val == 1 || Val == 2 || Val == 3) {
313 AM.Scale = 1 << Val;
314 SDOperand ShVal = N.Val->getOperand(0);
315
316 // Okay, we know that we have a scale by now. However, if the scaled
317 // value is an add of something and a constant, we can fold the
318 // constant into the disp field here.
319 if (ShVal.Val->getOpcode() == ISD::ADD && ShVal.hasOneUse() &&
320 isa<ConstantSDNode>(ShVal.Val->getOperand(1))) {
321 AM.IndexReg = ShVal.Val->getOperand(0);
322 ConstantSDNode *AddVal =
323 cast<ConstantSDNode>(ShVal.Val->getOperand(1));
324 AM.Disp += AddVal->getValue() << Val;
325 } else {
326 AM.IndexReg = ShVal;
327 }
328 return false;
329 }
330 }
331 break;
Evan Chengc9fab312005-12-08 02:01:35 +0000332
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000333 case ISD::MUL:
334 // X*[3,5,9] -> X+X*[2,4,8]
Evan Chenga86ba852006-02-11 02:05:36 +0000335 if (!StopHere && AM.IndexReg.Val == 0 && AM.BaseType == X86ISelAddressMode::RegBase &&
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000336 AM.Base.Reg.Val == 0)
337 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1)))
338 if (CN->getValue() == 3 || CN->getValue() == 5 || CN->getValue() == 9) {
339 AM.Scale = unsigned(CN->getValue())-1;
340
341 SDOperand MulVal = N.Val->getOperand(0);
342 SDOperand Reg;
343
344 // Okay, we know that we have a scale by now. However, if the scaled
345 // value is an add of something and a constant, we can fold the
346 // constant into the disp field here.
347 if (MulVal.Val->getOpcode() == ISD::ADD && MulVal.hasOneUse() &&
348 isa<ConstantSDNode>(MulVal.Val->getOperand(1))) {
349 Reg = MulVal.Val->getOperand(0);
350 ConstantSDNode *AddVal =
351 cast<ConstantSDNode>(MulVal.Val->getOperand(1));
352 AM.Disp += AddVal->getValue() * CN->getValue();
353 } else {
354 Reg = N.Val->getOperand(0);
355 }
356
357 AM.IndexReg = AM.Base.Reg = Reg;
358 return false;
359 }
360 break;
361
362 case ISD::ADD: {
Evan Chenga86ba852006-02-11 02:05:36 +0000363 if (!StopHere) {
364 X86ISelAddressMode Backup = AM;
365 if (!MatchAddress(N.Val->getOperand(0), AM, false) &&
366 !MatchAddress(N.Val->getOperand(1), AM, false))
367 return false;
368 AM = Backup;
369 if (!MatchAddress(N.Val->getOperand(1), AM, false) &&
370 !MatchAddress(N.Val->getOperand(0), AM, false))
371 return false;
372 AM = Backup;
373 }
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000374 break;
375 }
376 }
377
378 // Is the base register already occupied?
379 if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base.Reg.Val) {
380 // If so, check to see if the scale index register is set.
381 if (AM.IndexReg.Val == 0) {
382 AM.IndexReg = N;
383 AM.Scale = 1;
384 return false;
385 }
386
387 // Otherwise, we cannot select it.
388 return true;
389 }
390
391 // Default, generate it as a register.
392 AM.BaseType = X86ISelAddressMode::RegBase;
393 AM.Base.Reg = N;
394 return false;
395}
396
Evan Chengc9fab312005-12-08 02:01:35 +0000397/// SelectAddr - returns true if it is able pattern match an addressing mode.
398/// It returns the operands which make up the maximal addressing mode it can
399/// match by reference.
400bool X86DAGToDAGISel::SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
401 SDOperand &Index, SDOperand &Disp) {
402 X86ISelAddressMode AM;
Evan Chengbc7a0f442006-01-11 06:09:51 +0000403 if (MatchAddress(N, AM))
404 return false;
Evan Chengc9fab312005-12-08 02:01:35 +0000405
Evan Chengbc7a0f442006-01-11 06:09:51 +0000406 if (AM.BaseType == X86ISelAddressMode::RegBase) {
Evan Chengd19d51f2006-02-05 05:25:07 +0000407 if (!AM.Base.Reg.Val)
Evan Chengbc7a0f442006-01-11 06:09:51 +0000408 AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
Evan Chengc9fab312005-12-08 02:01:35 +0000409 }
Evan Chengbc7a0f442006-01-11 06:09:51 +0000410
Evan Chengd19d51f2006-02-05 05:25:07 +0000411 if (!AM.IndexReg.Val)
Evan Chengbc7a0f442006-01-11 06:09:51 +0000412 AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
413
414 getAddressOperands(AM, Base, Scale, Index, Disp);
415 return true;
Evan Chengc9fab312005-12-08 02:01:35 +0000416}
417
Evan Chengd5f2ba02006-02-06 06:02:33 +0000418bool X86DAGToDAGISel::TryFoldLoad(SDOperand P, SDOperand N,
419 SDOperand &Base, SDOperand &Scale,
420 SDOperand &Index, SDOperand &Disp) {
421 if (N.getOpcode() == ISD::LOAD &&
422 N.hasOneUse() &&
423 !CodeGenMap.count(N.getValue(0)) &&
424 (P.getNumOperands() == 1 || !isNonImmUse(P.Val, N.Val)))
Evan Cheng10d27902006-01-06 20:36:21 +0000425 return SelectAddr(N.getOperand(1), Base, Scale, Index, Disp);
426 return false;
427}
428
429static bool isRegister0(SDOperand Op) {
Evan Chengc9fab312005-12-08 02:01:35 +0000430 if (RegisterSDNode *R = dyn_cast<RegisterSDNode>(Op))
431 return (R->getReg() == 0);
432 return false;
433}
434
435/// SelectLEAAddr - it calls SelectAddr and determines if the maximal addressing
436/// mode it matches can be cost effectively emitted as an LEA instruction.
437/// For X86, it always is unless it's just a (Reg + const).
Chris Lattner29852a582006-01-11 00:46:55 +0000438bool X86DAGToDAGISel::SelectLEAAddr(SDOperand N, SDOperand &Base,
439 SDOperand &Scale,
Evan Chengc9fab312005-12-08 02:01:35 +0000440 SDOperand &Index, SDOperand &Disp) {
Evan Cheng67ed58e2005-12-12 21:49:40 +0000441 X86ISelAddressMode AM;
442 if (!MatchAddress(N, AM)) {
443 bool SelectBase = false;
444 bool SelectIndex = false;
445 bool Check = false;
446 if (AM.BaseType == X86ISelAddressMode::RegBase) {
447 if (AM.Base.Reg.Val) {
448 Check = true;
449 SelectBase = true;
Evan Chengc9fab312005-12-08 02:01:35 +0000450 } else {
Evan Cheng67ed58e2005-12-12 21:49:40 +0000451 AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
Evan Chengc9fab312005-12-08 02:01:35 +0000452 }
Evan Chengc9fab312005-12-08 02:01:35 +0000453 }
Evan Cheng67ed58e2005-12-12 21:49:40 +0000454
455 if (AM.IndexReg.Val) {
456 SelectIndex = true;
457 } else {
458 AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
459 }
460
461 if (Check) {
462 unsigned Complexity = 0;
463 if (AM.Scale > 1)
464 Complexity++;
465 if (SelectIndex)
466 Complexity++;
467 if (AM.GV)
468 Complexity++;
469 else if (AM.Disp > 1)
470 Complexity++;
471 if (Complexity <= 1)
472 return false;
473 }
474
Evan Cheng67ed58e2005-12-12 21:49:40 +0000475 getAddressOperands(AM, Base, Scale, Index, Disp);
Evan Chengc9fab312005-12-08 02:01:35 +0000476 return true;
Evan Chengc9fab312005-12-08 02:01:35 +0000477 }
Evan Cheng67ed58e2005-12-12 21:49:40 +0000478 return false;
Evan Chengc9fab312005-12-08 02:01:35 +0000479}
480
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000481void X86DAGToDAGISel::Select(SDOperand &Result, SDOperand N) {
Evan Cheng00fcb002005-12-15 01:02:48 +0000482 SDNode *Node = N.Val;
483 MVT::ValueType NVT = Node->getValueType(0);
Evan Cheng10d27902006-01-06 20:36:21 +0000484 unsigned Opc, MOpc;
485 unsigned Opcode = Node->getOpcode();
Chris Lattner655e7df2005-11-16 01:54:32 +0000486
Evan Chengd49cc362006-02-10 22:24:32 +0000487#ifndef NDEBUG
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000488 DEBUG(std::cerr << std::string(Indent, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000489 DEBUG(std::cerr << "Selecting: ");
490 DEBUG(Node->dump(CurDAG));
491 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000492 Indent += 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000493#endif
494
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000495 if (Opcode >= ISD::BUILTIN_OP_END && Opcode < X86ISD::FIRST_NUMBER) {
496 Result = N;
Evan Chengd49cc362006-02-10 22:24:32 +0000497#ifndef NDEBUG
Evan Chenga86ba852006-02-11 02:05:36 +0000498 DEBUG(std::cerr << std::string(Indent-2, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000499 DEBUG(std::cerr << "== ");
500 DEBUG(Node->dump(CurDAG));
501 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000502 Indent -= 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000503#endif
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000504 return; // Already selected.
505 }
Evan Cheng2ae799a2006-01-11 22:15:18 +0000506
507 std::map<SDOperand, SDOperand>::iterator CGMI = CodeGenMap.find(N);
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000508 if (CGMI != CodeGenMap.end()) {
509 Result = CGMI->second;
Evan Chengd49cc362006-02-10 22:24:32 +0000510#ifndef NDEBUG
Evan Chenga86ba852006-02-11 02:05:36 +0000511 DEBUG(std::cerr << std::string(Indent-2, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000512 DEBUG(std::cerr << "== ");
513 DEBUG(Result.Val->dump(CurDAG));
514 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000515 Indent -= 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000516#endif
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000517 return;
518 }
Chris Lattner655e7df2005-11-16 01:54:32 +0000519
Evan Cheng10d27902006-01-06 20:36:21 +0000520 switch (Opcode) {
Chris Lattner655e7df2005-11-16 01:54:32 +0000521 default: break;
Evan Cheng10d27902006-01-06 20:36:21 +0000522 case ISD::MULHU:
523 case ISD::MULHS: {
524 if (Opcode == ISD::MULHU)
525 switch (NVT) {
526 default: assert(0 && "Unsupported VT!");
527 case MVT::i8: Opc = X86::MUL8r; MOpc = X86::MUL8m; break;
528 case MVT::i16: Opc = X86::MUL16r; MOpc = X86::MUL16m; break;
529 case MVT::i32: Opc = X86::MUL32r; MOpc = X86::MUL32m; break;
530 }
531 else
532 switch (NVT) {
533 default: assert(0 && "Unsupported VT!");
534 case MVT::i8: Opc = X86::IMUL8r; MOpc = X86::IMUL8m; break;
535 case MVT::i16: Opc = X86::IMUL16r; MOpc = X86::IMUL16m; break;
536 case MVT::i32: Opc = X86::IMUL32r; MOpc = X86::IMUL32m; break;
537 }
538
539 unsigned LoReg, HiReg;
540 switch (NVT) {
541 default: assert(0 && "Unsupported VT!");
542 case MVT::i8: LoReg = X86::AL; HiReg = X86::AH; break;
543 case MVT::i16: LoReg = X86::AX; HiReg = X86::DX; break;
544 case MVT::i32: LoReg = X86::EAX; HiReg = X86::EDX; break;
545 }
546
547 SDOperand N0 = Node->getOperand(0);
548 SDOperand N1 = Node->getOperand(1);
549
550 bool foldedLoad = false;
551 SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000552 foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng92e27972006-01-06 23:19:29 +0000553 // MULHU and MULHS are commmutative
554 if (!foldedLoad) {
Evan Chengd5f2ba02006-02-06 06:02:33 +0000555 foldedLoad = TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng92e27972006-01-06 23:19:29 +0000556 if (foldedLoad) {
557 N0 = Node->getOperand(1);
558 N1 = Node->getOperand(0);
559 }
560 }
561
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000562 SDOperand Chain;
563 if (foldedLoad)
564 Select(Chain, N1.getOperand(0));
565 else
566 Chain = CurDAG->getEntryNode();
Evan Cheng10d27902006-01-06 20:36:21 +0000567
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000568 SDOperand InFlag(0, 0);
569 Select(N0, N0);
Evan Cheng10d27902006-01-06 20:36:21 +0000570 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000571 N0, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000572 InFlag = Chain.getValue(1);
573
574 if (foldedLoad) {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000575 Select(Tmp0, Tmp0);
576 Select(Tmp1, Tmp1);
577 Select(Tmp2, Tmp2);
578 Select(Tmp3, Tmp3);
Evan Chengd1b82d82006-02-09 07:17:49 +0000579 SDNode *CNode =
580 CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
581 Tmp2, Tmp3, Chain, InFlag);
582 Chain = SDOperand(CNode, 0);
583 InFlag = SDOperand(CNode, 1);
Evan Cheng10d27902006-01-06 20:36:21 +0000584 } else {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000585 Select(N1, N1);
Evan Chengd1b82d82006-02-09 07:17:49 +0000586 InFlag =
587 SDOperand(CurDAG->getTargetNode(Opc, MVT::Flag, N1, InFlag), 0);
Evan Cheng10d27902006-01-06 20:36:21 +0000588 }
589
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000590 Result = CurDAG->getCopyFromReg(Chain, HiReg, NVT, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000591 CodeGenMap[N.getValue(0)] = Result;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000592 if (foldedLoad) {
Evan Cheng92e27972006-01-06 23:19:29 +0000593 CodeGenMap[N1.getValue(1)] = Result.getValue(1);
Evan Cheng101e4b92006-02-09 22:12:53 +0000594 AddHandleReplacement(N1.Val, 1, Result.Val, 1);
Evan Chengd5f2ba02006-02-06 06:02:33 +0000595 }
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000596
Evan Chengd49cc362006-02-10 22:24:32 +0000597#ifndef NDEBUG
Evan Chenga86ba852006-02-11 02:05:36 +0000598 DEBUG(std::cerr << std::string(Indent-2, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000599 DEBUG(std::cerr << "== ");
600 DEBUG(Result.Val->dump(CurDAG));
601 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000602 Indent -= 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000603#endif
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000604 return;
Evan Cheng92e27972006-01-06 23:19:29 +0000605 }
606
607 case ISD::SDIV:
608 case ISD::UDIV:
609 case ISD::SREM:
610 case ISD::UREM: {
611 bool isSigned = Opcode == ISD::SDIV || Opcode == ISD::SREM;
612 bool isDiv = Opcode == ISD::SDIV || Opcode == ISD::UDIV;
613 if (!isSigned)
614 switch (NVT) {
615 default: assert(0 && "Unsupported VT!");
616 case MVT::i8: Opc = X86::DIV8r; MOpc = X86::DIV8m; break;
617 case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break;
618 case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break;
619 }
620 else
621 switch (NVT) {
622 default: assert(0 && "Unsupported VT!");
623 case MVT::i8: Opc = X86::IDIV8r; MOpc = X86::IDIV8m; break;
624 case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break;
625 case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break;
626 }
627
628 unsigned LoReg, HiReg;
629 unsigned ClrOpcode, SExtOpcode;
630 switch (NVT) {
631 default: assert(0 && "Unsupported VT!");
632 case MVT::i8:
633 LoReg = X86::AL; HiReg = X86::AH;
634 ClrOpcode = X86::MOV8ri;
635 SExtOpcode = X86::CBW;
636 break;
637 case MVT::i16:
638 LoReg = X86::AX; HiReg = X86::DX;
639 ClrOpcode = X86::MOV16ri;
640 SExtOpcode = X86::CWD;
641 break;
642 case MVT::i32:
643 LoReg = X86::EAX; HiReg = X86::EDX;
644 ClrOpcode = X86::MOV32ri;
645 SExtOpcode = X86::CDQ;
646 break;
647 }
648
649 SDOperand N0 = Node->getOperand(0);
650 SDOperand N1 = Node->getOperand(1);
651
652 bool foldedLoad = false;
653 SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000654 foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000655 SDOperand Chain;
656 if (foldedLoad)
657 Select(Chain, N1.getOperand(0));
658 else
659 Chain = CurDAG->getEntryNode();
Evan Cheng92e27972006-01-06 23:19:29 +0000660
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000661 SDOperand InFlag(0, 0);
662 Select(N0, N0);
Evan Cheng92e27972006-01-06 23:19:29 +0000663 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000664 N0, InFlag);
Evan Cheng92e27972006-01-06 23:19:29 +0000665 InFlag = Chain.getValue(1);
666
667 if (isSigned) {
668 // Sign extend the low part into the high part.
Evan Chengd1b82d82006-02-09 07:17:49 +0000669 InFlag =
670 SDOperand(CurDAG->getTargetNode(SExtOpcode, MVT::Flag, InFlag), 0);
Evan Cheng92e27972006-01-06 23:19:29 +0000671 } else {
672 // Zero out the high part, effectively zero extending the input.
673 SDOperand ClrNode =
Evan Chengd1b82d82006-02-09 07:17:49 +0000674 SDOperand(CurDAG->getTargetNode(ClrOpcode, NVT,
675 CurDAG->getTargetConstant(0, NVT)), 0);
Evan Cheng92e27972006-01-06 23:19:29 +0000676 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(HiReg, NVT),
677 ClrNode, InFlag);
678 InFlag = Chain.getValue(1);
679 }
680
681 if (foldedLoad) {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000682 Select(Tmp0, Tmp0);
683 Select(Tmp1, Tmp1);
684 Select(Tmp2, Tmp2);
685 Select(Tmp3, Tmp3);
Evan Chengd1b82d82006-02-09 07:17:49 +0000686 SDNode *CNode =
687 CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
688 Tmp2, Tmp3, Chain, InFlag);
689 Chain = SDOperand(CNode, 0);
690 InFlag = SDOperand(CNode, 1);
Evan Cheng92e27972006-01-06 23:19:29 +0000691 } else {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000692 Select(N1, N1);
Evan Chengd1b82d82006-02-09 07:17:49 +0000693 InFlag =
694 SDOperand(CurDAG->getTargetNode(Opc, MVT::Flag, N1, InFlag), 0);
Evan Cheng92e27972006-01-06 23:19:29 +0000695 }
696
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000697 Result = CurDAG->getCopyFromReg(Chain, isDiv ? LoReg : HiReg,
698 NVT, InFlag);
Evan Cheng92e27972006-01-06 23:19:29 +0000699 CodeGenMap[N.getValue(0)] = Result;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000700 if (foldedLoad) {
Evan Cheng92e27972006-01-06 23:19:29 +0000701 CodeGenMap[N1.getValue(1)] = Result.getValue(1);
Evan Cheng101e4b92006-02-09 22:12:53 +0000702 AddHandleReplacement(N1.Val, 1, Result.Val, 1);
Evan Chengd5f2ba02006-02-06 06:02:33 +0000703 }
Evan Chengd49cc362006-02-10 22:24:32 +0000704
705#ifndef NDEBUG
Evan Chenga86ba852006-02-11 02:05:36 +0000706 DEBUG(std::cerr << std::string(Indent-2, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000707 DEBUG(std::cerr << "== ");
708 DEBUG(Result.Val->dump(CurDAG));
709 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000710 Indent -= 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000711#endif
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000712 return;
Evan Cheng10d27902006-01-06 20:36:21 +0000713 }
Evan Cheng4eb7af92005-11-30 02:51:20 +0000714
Evan Chengbc7708c2005-12-17 02:02:50 +0000715 case ISD::TRUNCATE: {
716 unsigned Reg;
717 MVT::ValueType VT;
718 switch (Node->getOperand(0).getValueType()) {
719 default: assert(0 && "Unknown truncate!");
720 case MVT::i16: Reg = X86::AX; Opc = X86::MOV16rr; VT = MVT::i16; break;
721 case MVT::i32: Reg = X86::EAX; Opc = X86::MOV32rr; VT = MVT::i32; break;
722 }
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000723 SDOperand Tmp0, Tmp1;
724 Select(Tmp0, Node->getOperand(0));
Evan Chengd1b82d82006-02-09 07:17:49 +0000725 Select(Tmp1, SDOperand(CurDAG->getTargetNode(Opc, VT, Tmp0), 0));
Evan Chengbc7708c2005-12-17 02:02:50 +0000726 SDOperand InFlag = SDOperand(0,0);
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000727 Result = CurDAG->getCopyToReg(CurDAG->getEntryNode(), Reg, Tmp1, InFlag);
Evan Chengbc7708c2005-12-17 02:02:50 +0000728 SDOperand Chain = Result.getValue(0);
729 InFlag = Result.getValue(1);
730
731 switch (NVT) {
732 default: assert(0 && "Unknown truncate!");
733 case MVT::i8: Reg = X86::AL; Opc = X86::MOV8rr; VT = MVT::i8; break;
734 case MVT::i16: Reg = X86::AX; Opc = X86::MOV16rr; VT = MVT::i16; break;
735 }
736
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000737 Result = CurDAG->getCopyFromReg(Chain, Reg, VT, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000738 if (N.Val->hasOneUse())
Evan Chengd1b82d82006-02-09 07:17:49 +0000739 Result = CurDAG->SelectNodeTo(N.Val, Opc, VT, Result);
Evan Cheng10d27902006-01-06 20:36:21 +0000740 else
Evan Chengd1b82d82006-02-09 07:17:49 +0000741 Result = CodeGenMap[N] =
742 SDOperand(CurDAG->getTargetNode(Opc, VT, Result), 0);
Evan Chengd49cc362006-02-10 22:24:32 +0000743
744#ifndef NDEBUG
Evan Chenga86ba852006-02-11 02:05:36 +0000745 DEBUG(std::cerr << std::string(Indent-2, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000746 DEBUG(std::cerr << "== ");
747 DEBUG(Result.Val->dump(CurDAG));
748 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000749 Indent -= 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000750#endif
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000751 return;
Evan Chengbc7708c2005-12-17 02:02:50 +0000752 }
Chris Lattner655e7df2005-11-16 01:54:32 +0000753 }
754
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000755 SelectCode(Result, N);
Evan Chengd49cc362006-02-10 22:24:32 +0000756#ifndef NDEBUG
Evan Chenga86ba852006-02-11 02:05:36 +0000757 DEBUG(std::cerr << std::string(Indent-2, ' '));
Evan Chengd49cc362006-02-10 22:24:32 +0000758 DEBUG(std::cerr << "=> ");
759 DEBUG(Result.Val->dump(CurDAG));
760 DEBUG(std::cerr << "\n");
Evan Cheng2b6f78b2006-02-10 22:46:26 +0000761 Indent -= 2;
Evan Chengd49cc362006-02-10 22:24:32 +0000762#endif
Chris Lattner655e7df2005-11-16 01:54:32 +0000763}
764
765/// createX86ISelDag - This pass converts a legalized DAG into a
766/// X86-specific DAG, ready for instruction scheduling.
767///
768FunctionPass *llvm::createX86ISelDag(TargetMachine &TM) {
769 return new X86DAGToDAGISel(TM);
770}