blob: d2bfc58813f9fb59fe7b5078cddae5f61e4d71ce [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
15#include "X86.h"
Evan Chengbc7a0f442006-01-11 06:09:51 +000016#include "X86InstrBuilder.h"
Chris Lattner7c551262006-01-11 01:15:34 +000017#include "X86RegisterInfo.h"
Chris Lattner655e7df2005-11-16 01:54:32 +000018#include "X86Subtarget.h"
19#include "X86ISelLowering.h"
Chris Lattner3f0f71b2005-11-19 02:11:08 +000020#include "llvm/GlobalValue.h"
Chris Lattner7c551262006-01-11 01:15:34 +000021#include "llvm/Instructions.h"
22#include "llvm/Support/CFG.h"
Chris Lattner3f0f71b2005-11-19 02:11:08 +000023#include "llvm/CodeGen/MachineConstantPool.h"
Chris Lattner655e7df2005-11-16 01:54:32 +000024#include "llvm/CodeGen/MachineFunction.h"
Evan Cheng73a1ad92006-01-10 20:26:56 +000025#include "llvm/CodeGen/MachineFrameInfo.h"
Chris Lattner7c551262006-01-11 01:15:34 +000026#include "llvm/CodeGen/MachineInstrBuilder.h"
27#include "llvm/CodeGen/SSARegMap.h"
Chris Lattner655e7df2005-11-16 01:54:32 +000028#include "llvm/CodeGen/SelectionDAGISel.h"
29#include "llvm/Target/TargetMachine.h"
30#include "llvm/Support/Debug.h"
31#include "llvm/ADT/Statistic.h"
Chris Lattnerde02d772006-01-22 23:41:00 +000032#include <iostream>
Evan Cheng54cb1832006-02-05 06:46:41 +000033#include <set>
Chris Lattner655e7df2005-11-16 01:54:32 +000034using namespace llvm;
35
36//===----------------------------------------------------------------------===//
37// Pattern Matcher Implementation
38//===----------------------------------------------------------------------===//
39
40namespace {
Chris Lattner3f0f71b2005-11-19 02:11:08 +000041 /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses
42 /// SDOperand's instead of register numbers for the leaves of the matched
43 /// tree.
44 struct X86ISelAddressMode {
45 enum {
46 RegBase,
47 FrameIndexBase,
Evan Chengc9fab312005-12-08 02:01:35 +000048 ConstantPoolBase
Chris Lattner3f0f71b2005-11-19 02:11:08 +000049 } BaseType;
50
51 struct { // This is really a union, discriminated by BaseType!
52 SDOperand Reg;
53 int FrameIndex;
54 } Base;
55
56 unsigned Scale;
57 SDOperand IndexReg;
58 unsigned Disp;
59 GlobalValue *GV;
60
61 X86ISelAddressMode()
Evan Cheng4eb7af92005-11-30 02:51:20 +000062 : BaseType(RegBase), Scale(1), IndexReg(), Disp(0), GV(0) {
Chris Lattner3f0f71b2005-11-19 02:11:08 +000063 }
64 };
65}
66
67namespace {
Chris Lattner655e7df2005-11-16 01:54:32 +000068 Statistic<>
69 NumFPKill("x86-codegen", "Number of FP_REG_KILL instructions added");
70
71 //===--------------------------------------------------------------------===//
72 /// ISel - X86 specific code to select X86 machine instructions for
73 /// SelectionDAG operations.
74 ///
75 class X86DAGToDAGISel : public SelectionDAGISel {
76 /// ContainsFPCode - Every instruction we select that uses or defines a FP
77 /// register should set this to true.
78 bool ContainsFPCode;
79
80 /// X86Lowering - This object fully describes how to lower LLVM code to an
81 /// X86-specific SelectionDAG.
82 X86TargetLowering X86Lowering;
83
84 /// Subtarget - Keep a pointer to the X86Subtarget around so that we can
85 /// make the right decision when generating code for different targets.
86 const X86Subtarget *Subtarget;
87 public:
88 X86DAGToDAGISel(TargetMachine &TM)
89 : SelectionDAGISel(X86Lowering), X86Lowering(TM) {
90 Subtarget = &TM.getSubtarget<X86Subtarget>();
91 }
92
93 virtual const char *getPassName() const {
94 return "X86 DAG->DAG Instruction Selection";
95 }
96
97 /// InstructionSelectBasicBlock - This callback is invoked by
98 /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
99 virtual void InstructionSelectBasicBlock(SelectionDAG &DAG);
100
Evan Chengbc7a0f442006-01-11 06:09:51 +0000101 virtual void EmitFunctionEntryCode(Function &Fn, MachineFunction &MF);
102
Chris Lattner655e7df2005-11-16 01:54:32 +0000103// Include the pieces autogenerated from the target description.
104#include "X86GenDAGISel.inc"
105
106 private:
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000107 void Select(SDOperand &Result, SDOperand N);
Chris Lattner655e7df2005-11-16 01:54:32 +0000108
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000109 bool MatchAddress(SDOperand N, X86ISelAddressMode &AM);
Evan Chengc9fab312005-12-08 02:01:35 +0000110 bool SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
111 SDOperand &Index, SDOperand &Disp);
112 bool SelectLEAAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
113 SDOperand &Index, SDOperand &Disp);
Evan Chengd5f2ba02006-02-06 06:02:33 +0000114 bool TryFoldLoad(SDOperand P, SDOperand N,
115 SDOperand &Base, SDOperand &Scale,
Evan Cheng10d27902006-01-06 20:36:21 +0000116 SDOperand &Index, SDOperand &Disp);
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000117
Evan Cheng67ed58e2005-12-12 21:49:40 +0000118 inline void getAddressOperands(X86ISelAddressMode &AM, SDOperand &Base,
119 SDOperand &Scale, SDOperand &Index,
120 SDOperand &Disp) {
121 Base = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) ?
122 CurDAG->getTargetFrameIndex(AM.Base.FrameIndex, MVT::i32) : AM.Base.Reg;
Evan Cheng1d712482005-12-17 09:13:43 +0000123 Scale = getI8Imm(AM.Scale);
Evan Cheng67ed58e2005-12-12 21:49:40 +0000124 Index = AM.IndexReg;
125 Disp = AM.GV ? CurDAG->getTargetGlobalAddress(AM.GV, MVT::i32, AM.Disp)
126 : getI32Imm(AM.Disp);
127 }
128
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000129 /// getI8Imm - Return a target constant with the specified value, of type
130 /// i8.
131 inline SDOperand getI8Imm(unsigned Imm) {
132 return CurDAG->getTargetConstant(Imm, MVT::i8);
133 }
134
Chris Lattner655e7df2005-11-16 01:54:32 +0000135 /// getI16Imm - Return a target constant with the specified value, of type
136 /// i16.
137 inline SDOperand getI16Imm(unsigned Imm) {
138 return CurDAG->getTargetConstant(Imm, MVT::i16);
139 }
140
141 /// getI32Imm - Return a target constant with the specified value, of type
142 /// i32.
143 inline SDOperand getI32Imm(unsigned Imm) {
144 return CurDAG->getTargetConstant(Imm, MVT::i32);
145 }
146 };
147}
148
149/// InstructionSelectBasicBlock - This callback is invoked by SelectionDAGISel
150/// when it has created a SelectionDAG for us to codegen.
151void X86DAGToDAGISel::InstructionSelectBasicBlock(SelectionDAG &DAG) {
152 DEBUG(BB->dump());
Chris Lattner7c551262006-01-11 01:15:34 +0000153 MachineFunction::iterator FirstMBB = BB;
Chris Lattner655e7df2005-11-16 01:54:32 +0000154
155 // Codegen the basic block.
Evan Cheng54cb1832006-02-05 06:46:41 +0000156 DAG.setRoot(SelectRoot(DAG.getRoot()));
Evan Cheng1d9b6712005-12-19 22:36:02 +0000157 CodeGenMap.clear();
Chris Lattner655e7df2005-11-16 01:54:32 +0000158 DAG.RemoveDeadNodes();
159
160 // Emit machine code to BB.
161 ScheduleAndEmitDAG(DAG);
Chris Lattner7c551262006-01-11 01:15:34 +0000162
163 // If we are emitting FP stack code, scan the basic block to determine if this
164 // block defines any FP values. If so, put an FP_REG_KILL instruction before
165 // the terminator of the block.
Evan Chengcde9e302006-01-27 08:10:46 +0000166 if (!Subtarget->hasSSE2()) {
Chris Lattner7c551262006-01-11 01:15:34 +0000167 // Note that FP stack instructions *are* used in SSE code when returning
168 // values, but these are not live out of the basic block, so we don't need
169 // an FP_REG_KILL in this case either.
170 bool ContainsFPCode = false;
171
172 // Scan all of the machine instructions in these MBBs, checking for FP
173 // stores.
174 MachineFunction::iterator MBBI = FirstMBB;
175 do {
176 for (MachineBasicBlock::iterator I = MBBI->begin(), E = MBBI->end();
177 !ContainsFPCode && I != E; ++I) {
178 for (unsigned op = 0, e = I->getNumOperands(); op != e; ++op) {
179 if (I->getOperand(op).isRegister() && I->getOperand(op).isDef() &&
180 MRegisterInfo::isVirtualRegister(I->getOperand(op).getReg()) &&
181 RegMap->getRegClass(I->getOperand(0).getReg()) ==
182 X86::RFPRegisterClass) {
183 ContainsFPCode = true;
184 break;
185 }
186 }
187 }
188 } while (!ContainsFPCode && &*(MBBI++) != BB);
189
190 // Check PHI nodes in successor blocks. These PHI's will be lowered to have
191 // a copy of the input value in this block.
192 if (!ContainsFPCode) {
193 // Final check, check LLVM BB's that are successors to the LLVM BB
194 // corresponding to BB for FP PHI nodes.
195 const BasicBlock *LLVMBB = BB->getBasicBlock();
196 const PHINode *PN;
197 for (succ_const_iterator SI = succ_begin(LLVMBB), E = succ_end(LLVMBB);
198 !ContainsFPCode && SI != E; ++SI) {
199 for (BasicBlock::const_iterator II = SI->begin();
200 (PN = dyn_cast<PHINode>(II)); ++II) {
201 if (PN->getType()->isFloatingPoint()) {
202 ContainsFPCode = true;
203 break;
204 }
205 }
206 }
207 }
208
209 // Finally, if we found any FP code, emit the FP_REG_KILL instruction.
210 if (ContainsFPCode) {
211 BuildMI(*BB, BB->getFirstTerminator(), X86::FP_REG_KILL, 0);
212 ++NumFPKill;
213 }
214 }
Chris Lattner655e7df2005-11-16 01:54:32 +0000215}
216
Evan Chengbc7a0f442006-01-11 06:09:51 +0000217/// EmitSpecialCodeForMain - Emit any code that needs to be executed only in
218/// the main function.
219static void EmitSpecialCodeForMain(MachineBasicBlock *BB,
220 MachineFrameInfo *MFI) {
221 // Switch the FPU to 64-bit precision mode for better compatibility and speed.
222 int CWFrameIdx = MFI->CreateStackObject(2, 2);
223 addFrameReference(BuildMI(BB, X86::FNSTCW16m, 4), CWFrameIdx);
224
225 // Set the high part to be 64-bit precision.
226 addFrameReference(BuildMI(BB, X86::MOV8mi, 5),
227 CWFrameIdx, 1).addImm(2);
228
229 // Reload the modified control word now.
230 addFrameReference(BuildMI(BB, X86::FLDCW16m, 4), CWFrameIdx);
231}
232
233void X86DAGToDAGISel::EmitFunctionEntryCode(Function &Fn, MachineFunction &MF) {
234 // If this is main, emit special code for main.
235 MachineBasicBlock *BB = MF.begin();
236 if (Fn.hasExternalLinkage() && Fn.getName() == "main")
237 EmitSpecialCodeForMain(BB, MF.getFrameInfo());
238}
239
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000240/// MatchAddress - Add the specified node to the specified addressing mode,
241/// returning true if it cannot be done. This just pattern matches for the
242/// addressing mode
243bool X86DAGToDAGISel::MatchAddress(SDOperand N, X86ISelAddressMode &AM) {
244 switch (N.getOpcode()) {
245 default: break;
246 case ISD::FrameIndex:
247 if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
248 AM.BaseType = X86ISelAddressMode::FrameIndexBase;
249 AM.Base.FrameIndex = cast<FrameIndexSDNode>(N)->getIndex();
250 return false;
251 }
252 break;
Evan Chengc9fab312005-12-08 02:01:35 +0000253
254 case ISD::ConstantPool:
255 if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
256 if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N)) {
257 AM.BaseType = X86ISelAddressMode::ConstantPoolBase;
Evan Cheng72d5c252006-01-31 22:28:30 +0000258 AM.Base.Reg = CurDAG->getTargetConstantPool(CP->get(), MVT::i32,
259 CP->getAlignment());
Evan Chengc9fab312005-12-08 02:01:35 +0000260 return false;
261 }
262 }
263 break;
264
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000265 case ISD::GlobalAddress:
Evan Cheng9cdc16c2005-12-21 23:05:39 +0000266 case ISD::TargetGlobalAddress:
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000267 if (AM.GV == 0) {
Evan Chenga74ce622005-12-21 02:39:21 +0000268 AM.GV = cast<GlobalAddressSDNode>(N)->getGlobal();
Evan Cheng1d712482005-12-17 09:13:43 +0000269 return false;
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000270 }
271 break;
Evan Chengc9fab312005-12-08 02:01:35 +0000272
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000273 case ISD::Constant:
274 AM.Disp += cast<ConstantSDNode>(N)->getValue();
275 return false;
Evan Chengc9fab312005-12-08 02:01:35 +0000276
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000277 case ISD::SHL:
278 if (AM.IndexReg.Val == 0 && AM.Scale == 1)
279 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1))) {
280 unsigned Val = CN->getValue();
281 if (Val == 1 || Val == 2 || Val == 3) {
282 AM.Scale = 1 << Val;
283 SDOperand ShVal = N.Val->getOperand(0);
284
285 // Okay, we know that we have a scale by now. However, if the scaled
286 // value is an add of something and a constant, we can fold the
287 // constant into the disp field here.
288 if (ShVal.Val->getOpcode() == ISD::ADD && ShVal.hasOneUse() &&
289 isa<ConstantSDNode>(ShVal.Val->getOperand(1))) {
290 AM.IndexReg = ShVal.Val->getOperand(0);
291 ConstantSDNode *AddVal =
292 cast<ConstantSDNode>(ShVal.Val->getOperand(1));
293 AM.Disp += AddVal->getValue() << Val;
294 } else {
295 AM.IndexReg = ShVal;
296 }
297 return false;
298 }
299 }
300 break;
Evan Chengc9fab312005-12-08 02:01:35 +0000301
Chris Lattner3f0f71b2005-11-19 02:11:08 +0000302 case ISD::MUL:
303 // X*[3,5,9] -> X+X*[2,4,8]
304 if (AM.IndexReg.Val == 0 && AM.BaseType == X86ISelAddressMode::RegBase &&
305 AM.Base.Reg.Val == 0)
306 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1)))
307 if (CN->getValue() == 3 || CN->getValue() == 5 || CN->getValue() == 9) {
308 AM.Scale = unsigned(CN->getValue())-1;
309
310 SDOperand MulVal = N.Val->getOperand(0);
311 SDOperand Reg;
312
313 // Okay, we know that we have a scale by now. However, if the scaled
314 // value is an add of something and a constant, we can fold the
315 // constant into the disp field here.
316 if (MulVal.Val->getOpcode() == ISD::ADD && MulVal.hasOneUse() &&
317 isa<ConstantSDNode>(MulVal.Val->getOperand(1))) {
318 Reg = MulVal.Val->getOperand(0);
319 ConstantSDNode *AddVal =
320 cast<ConstantSDNode>(MulVal.Val->getOperand(1));
321 AM.Disp += AddVal->getValue() * CN->getValue();
322 } else {
323 Reg = N.Val->getOperand(0);
324 }
325
326 AM.IndexReg = AM.Base.Reg = Reg;
327 return false;
328 }
329 break;
330
331 case ISD::ADD: {
332 X86ISelAddressMode Backup = AM;
333 if (!MatchAddress(N.Val->getOperand(0), AM) &&
334 !MatchAddress(N.Val->getOperand(1), AM))
335 return false;
336 AM = Backup;
337 if (!MatchAddress(N.Val->getOperand(1), AM) &&
338 !MatchAddress(N.Val->getOperand(0), AM))
339 return false;
340 AM = Backup;
341 break;
342 }
343 }
344
345 // Is the base register already occupied?
346 if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base.Reg.Val) {
347 // If so, check to see if the scale index register is set.
348 if (AM.IndexReg.Val == 0) {
349 AM.IndexReg = N;
350 AM.Scale = 1;
351 return false;
352 }
353
354 // Otherwise, we cannot select it.
355 return true;
356 }
357
358 // Default, generate it as a register.
359 AM.BaseType = X86ISelAddressMode::RegBase;
360 AM.Base.Reg = N;
361 return false;
362}
363
Evan Chengc9fab312005-12-08 02:01:35 +0000364/// SelectAddr - returns true if it is able pattern match an addressing mode.
365/// It returns the operands which make up the maximal addressing mode it can
366/// match by reference.
367bool X86DAGToDAGISel::SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
368 SDOperand &Index, SDOperand &Disp) {
369 X86ISelAddressMode AM;
Evan Chengbc7a0f442006-01-11 06:09:51 +0000370 if (MatchAddress(N, AM))
371 return false;
Evan Chengc9fab312005-12-08 02:01:35 +0000372
Evan Chengbc7a0f442006-01-11 06:09:51 +0000373 if (AM.BaseType == X86ISelAddressMode::RegBase) {
Evan Chengd19d51f2006-02-05 05:25:07 +0000374 if (!AM.Base.Reg.Val)
Evan Chengbc7a0f442006-01-11 06:09:51 +0000375 AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
Evan Chengc9fab312005-12-08 02:01:35 +0000376 }
Evan Chengbc7a0f442006-01-11 06:09:51 +0000377
Evan Chengd19d51f2006-02-05 05:25:07 +0000378 if (!AM.IndexReg.Val)
Evan Chengbc7a0f442006-01-11 06:09:51 +0000379 AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
380
381 getAddressOperands(AM, Base, Scale, Index, Disp);
382 return true;
Evan Chengc9fab312005-12-08 02:01:35 +0000383}
384
Evan Chengd5f2ba02006-02-06 06:02:33 +0000385bool X86DAGToDAGISel::TryFoldLoad(SDOperand P, SDOperand N,
386 SDOperand &Base, SDOperand &Scale,
387 SDOperand &Index, SDOperand &Disp) {
388 if (N.getOpcode() == ISD::LOAD &&
389 N.hasOneUse() &&
390 !CodeGenMap.count(N.getValue(0)) &&
391 (P.getNumOperands() == 1 || !isNonImmUse(P.Val, N.Val)))
Evan Cheng10d27902006-01-06 20:36:21 +0000392 return SelectAddr(N.getOperand(1), Base, Scale, Index, Disp);
393 return false;
394}
395
396static bool isRegister0(SDOperand Op) {
Evan Chengc9fab312005-12-08 02:01:35 +0000397 if (RegisterSDNode *R = dyn_cast<RegisterSDNode>(Op))
398 return (R->getReg() == 0);
399 return false;
400}
401
402/// SelectLEAAddr - it calls SelectAddr and determines if the maximal addressing
403/// mode it matches can be cost effectively emitted as an LEA instruction.
404/// For X86, it always is unless it's just a (Reg + const).
Chris Lattner29852a582006-01-11 00:46:55 +0000405bool X86DAGToDAGISel::SelectLEAAddr(SDOperand N, SDOperand &Base,
406 SDOperand &Scale,
Evan Chengc9fab312005-12-08 02:01:35 +0000407 SDOperand &Index, SDOperand &Disp) {
Evan Cheng67ed58e2005-12-12 21:49:40 +0000408 X86ISelAddressMode AM;
409 if (!MatchAddress(N, AM)) {
410 bool SelectBase = false;
411 bool SelectIndex = false;
412 bool Check = false;
413 if (AM.BaseType == X86ISelAddressMode::RegBase) {
414 if (AM.Base.Reg.Val) {
415 Check = true;
416 SelectBase = true;
Evan Chengc9fab312005-12-08 02:01:35 +0000417 } else {
Evan Cheng67ed58e2005-12-12 21:49:40 +0000418 AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
Evan Chengc9fab312005-12-08 02:01:35 +0000419 }
Evan Chengc9fab312005-12-08 02:01:35 +0000420 }
Evan Cheng67ed58e2005-12-12 21:49:40 +0000421
422 if (AM.IndexReg.Val) {
423 SelectIndex = true;
424 } else {
425 AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
426 }
427
428 if (Check) {
429 unsigned Complexity = 0;
430 if (AM.Scale > 1)
431 Complexity++;
432 if (SelectIndex)
433 Complexity++;
434 if (AM.GV)
435 Complexity++;
436 else if (AM.Disp > 1)
437 Complexity++;
438 if (Complexity <= 1)
439 return false;
440 }
441
Evan Cheng67ed58e2005-12-12 21:49:40 +0000442 getAddressOperands(AM, Base, Scale, Index, Disp);
Evan Chengc9fab312005-12-08 02:01:35 +0000443 return true;
Evan Chengc9fab312005-12-08 02:01:35 +0000444 }
Evan Cheng67ed58e2005-12-12 21:49:40 +0000445 return false;
Evan Chengc9fab312005-12-08 02:01:35 +0000446}
447
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000448void X86DAGToDAGISel::Select(SDOperand &Result, SDOperand N) {
Evan Cheng00fcb002005-12-15 01:02:48 +0000449 SDNode *Node = N.Val;
450 MVT::ValueType NVT = Node->getValueType(0);
Evan Cheng10d27902006-01-06 20:36:21 +0000451 unsigned Opc, MOpc;
452 unsigned Opcode = Node->getOpcode();
Chris Lattner655e7df2005-11-16 01:54:32 +0000453
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000454 if (Opcode >= ISD::BUILTIN_OP_END && Opcode < X86ISD::FIRST_NUMBER) {
455 Result = N;
456 return; // Already selected.
457 }
Evan Cheng2ae799a2006-01-11 22:15:18 +0000458
459 std::map<SDOperand, SDOperand>::iterator CGMI = CodeGenMap.find(N);
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000460 if (CGMI != CodeGenMap.end()) {
461 Result = CGMI->second;
462 return;
463 }
Chris Lattner655e7df2005-11-16 01:54:32 +0000464
Evan Cheng10d27902006-01-06 20:36:21 +0000465 switch (Opcode) {
Chris Lattner655e7df2005-11-16 01:54:32 +0000466 default: break;
Evan Cheng10d27902006-01-06 20:36:21 +0000467 case ISD::MULHU:
468 case ISD::MULHS: {
469 if (Opcode == ISD::MULHU)
470 switch (NVT) {
471 default: assert(0 && "Unsupported VT!");
472 case MVT::i8: Opc = X86::MUL8r; MOpc = X86::MUL8m; break;
473 case MVT::i16: Opc = X86::MUL16r; MOpc = X86::MUL16m; break;
474 case MVT::i32: Opc = X86::MUL32r; MOpc = X86::MUL32m; break;
475 }
476 else
477 switch (NVT) {
478 default: assert(0 && "Unsupported VT!");
479 case MVT::i8: Opc = X86::IMUL8r; MOpc = X86::IMUL8m; break;
480 case MVT::i16: Opc = X86::IMUL16r; MOpc = X86::IMUL16m; break;
481 case MVT::i32: Opc = X86::IMUL32r; MOpc = X86::IMUL32m; break;
482 }
483
484 unsigned LoReg, HiReg;
485 switch (NVT) {
486 default: assert(0 && "Unsupported VT!");
487 case MVT::i8: LoReg = X86::AL; HiReg = X86::AH; break;
488 case MVT::i16: LoReg = X86::AX; HiReg = X86::DX; break;
489 case MVT::i32: LoReg = X86::EAX; HiReg = X86::EDX; break;
490 }
491
492 SDOperand N0 = Node->getOperand(0);
493 SDOperand N1 = Node->getOperand(1);
494
495 bool foldedLoad = false;
496 SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000497 foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng92e27972006-01-06 23:19:29 +0000498 // MULHU and MULHS are commmutative
499 if (!foldedLoad) {
Evan Chengd5f2ba02006-02-06 06:02:33 +0000500 foldedLoad = TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng92e27972006-01-06 23:19:29 +0000501 if (foldedLoad) {
502 N0 = Node->getOperand(1);
503 N1 = Node->getOperand(0);
504 }
505 }
506
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000507 SDOperand Chain;
508 if (foldedLoad)
509 Select(Chain, N1.getOperand(0));
510 else
511 Chain = CurDAG->getEntryNode();
Evan Cheng10d27902006-01-06 20:36:21 +0000512
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000513 SDOperand InFlag(0, 0);
514 Select(N0, N0);
Evan Cheng10d27902006-01-06 20:36:21 +0000515 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000516 N0, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000517 InFlag = Chain.getValue(1);
518
519 if (foldedLoad) {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000520 Select(Tmp0, Tmp0);
521 Select(Tmp1, Tmp1);
522 Select(Tmp2, Tmp2);
523 Select(Tmp3, Tmp3);
Evan Cheng10d27902006-01-06 20:36:21 +0000524 Chain = CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
525 Tmp2, Tmp3, Chain, InFlag);
526 InFlag = Chain.getValue(1);
527 } else {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000528 Select(N1, N1);
529 InFlag = CurDAG->getTargetNode(Opc, MVT::Flag, N1, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000530 }
531
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000532 Result = CurDAG->getCopyFromReg(Chain, HiReg, NVT, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000533 CodeGenMap[N.getValue(0)] = Result;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000534 if (foldedLoad) {
Evan Cheng92e27972006-01-06 23:19:29 +0000535 CodeGenMap[N1.getValue(1)] = Result.getValue(1);
Evan Chengd5f2ba02006-02-06 06:02:33 +0000536 AddHandleReplacement(N1.getValue(1), Result.getValue(1));
537 }
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000538
539 return;
Evan Cheng92e27972006-01-06 23:19:29 +0000540 }
541
542 case ISD::SDIV:
543 case ISD::UDIV:
544 case ISD::SREM:
545 case ISD::UREM: {
546 bool isSigned = Opcode == ISD::SDIV || Opcode == ISD::SREM;
547 bool isDiv = Opcode == ISD::SDIV || Opcode == ISD::UDIV;
548 if (!isSigned)
549 switch (NVT) {
550 default: assert(0 && "Unsupported VT!");
551 case MVT::i8: Opc = X86::DIV8r; MOpc = X86::DIV8m; break;
552 case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break;
553 case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break;
554 }
555 else
556 switch (NVT) {
557 default: assert(0 && "Unsupported VT!");
558 case MVT::i8: Opc = X86::IDIV8r; MOpc = X86::IDIV8m; break;
559 case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break;
560 case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break;
561 }
562
563 unsigned LoReg, HiReg;
564 unsigned ClrOpcode, SExtOpcode;
565 switch (NVT) {
566 default: assert(0 && "Unsupported VT!");
567 case MVT::i8:
568 LoReg = X86::AL; HiReg = X86::AH;
569 ClrOpcode = X86::MOV8ri;
570 SExtOpcode = X86::CBW;
571 break;
572 case MVT::i16:
573 LoReg = X86::AX; HiReg = X86::DX;
574 ClrOpcode = X86::MOV16ri;
575 SExtOpcode = X86::CWD;
576 break;
577 case MVT::i32:
578 LoReg = X86::EAX; HiReg = X86::EDX;
579 ClrOpcode = X86::MOV32ri;
580 SExtOpcode = X86::CDQ;
581 break;
582 }
583
584 SDOperand N0 = Node->getOperand(0);
585 SDOperand N1 = Node->getOperand(1);
586
587 bool foldedLoad = false;
588 SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000589 foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000590 SDOperand Chain;
591 if (foldedLoad)
592 Select(Chain, N1.getOperand(0));
593 else
594 Chain = CurDAG->getEntryNode();
Evan Cheng92e27972006-01-06 23:19:29 +0000595
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000596 SDOperand InFlag(0, 0);
597 Select(N0, N0);
Evan Cheng92e27972006-01-06 23:19:29 +0000598 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000599 N0, InFlag);
Evan Cheng92e27972006-01-06 23:19:29 +0000600 InFlag = Chain.getValue(1);
601
602 if (isSigned) {
603 // Sign extend the low part into the high part.
604 InFlag = CurDAG->getTargetNode(SExtOpcode, MVT::Flag, InFlag);
605 } else {
606 // Zero out the high part, effectively zero extending the input.
607 SDOperand ClrNode =
608 CurDAG->getTargetNode(ClrOpcode, NVT,
609 CurDAG->getTargetConstant(0, NVT));
610 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(HiReg, NVT),
611 ClrNode, InFlag);
612 InFlag = Chain.getValue(1);
613 }
614
615 if (foldedLoad) {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000616 Select(Tmp0, Tmp0);
617 Select(Tmp1, Tmp1);
618 Select(Tmp2, Tmp2);
619 Select(Tmp3, Tmp3);
Evan Cheng92e27972006-01-06 23:19:29 +0000620 Chain = CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
621 Tmp2, Tmp3, Chain, InFlag);
622 InFlag = Chain.getValue(1);
623 } else {
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000624 Select(N1, N1);
625 InFlag = CurDAG->getTargetNode(Opc, MVT::Flag, N1, InFlag);
Evan Cheng92e27972006-01-06 23:19:29 +0000626 }
627
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000628 Result = CurDAG->getCopyFromReg(Chain, isDiv ? LoReg : HiReg,
629 NVT, InFlag);
Evan Cheng92e27972006-01-06 23:19:29 +0000630 CodeGenMap[N.getValue(0)] = Result;
Evan Chengd5f2ba02006-02-06 06:02:33 +0000631 if (foldedLoad) {
Evan Cheng92e27972006-01-06 23:19:29 +0000632 CodeGenMap[N1.getValue(1)] = Result.getValue(1);
Evan Chengd5f2ba02006-02-06 06:02:33 +0000633 AddHandleReplacement(N1.getValue(1), Result.getValue(1));
634 }
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000635 return;
Evan Cheng10d27902006-01-06 20:36:21 +0000636 }
Evan Cheng4eb7af92005-11-30 02:51:20 +0000637
Evan Chengbc7708c2005-12-17 02:02:50 +0000638 case ISD::TRUNCATE: {
639 unsigned Reg;
640 MVT::ValueType VT;
641 switch (Node->getOperand(0).getValueType()) {
642 default: assert(0 && "Unknown truncate!");
643 case MVT::i16: Reg = X86::AX; Opc = X86::MOV16rr; VT = MVT::i16; break;
644 case MVT::i32: Reg = X86::EAX; Opc = X86::MOV32rr; VT = MVT::i32; break;
645 }
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000646 SDOperand Tmp0, Tmp1;
647 Select(Tmp0, Node->getOperand(0));
648 Select(Tmp1, CurDAG->getTargetNode(Opc, VT, Tmp0));
Evan Chengbc7708c2005-12-17 02:02:50 +0000649 SDOperand InFlag = SDOperand(0,0);
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000650 Result = CurDAG->getCopyToReg(CurDAG->getEntryNode(), Reg, Tmp1, InFlag);
Evan Chengbc7708c2005-12-17 02:02:50 +0000651 SDOperand Chain = Result.getValue(0);
652 InFlag = Result.getValue(1);
653
654 switch (NVT) {
655 default: assert(0 && "Unknown truncate!");
656 case MVT::i8: Reg = X86::AL; Opc = X86::MOV8rr; VT = MVT::i8; break;
657 case MVT::i16: Reg = X86::AX; Opc = X86::MOV16rr; VT = MVT::i16; break;
658 }
659
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000660 Result = CurDAG->getCopyFromReg(Chain, Reg, VT, InFlag);
Evan Cheng10d27902006-01-06 20:36:21 +0000661 if (N.Val->hasOneUse())
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000662 Result =CurDAG->SelectNodeTo(N.Val, Opc, VT, Result);
Evan Cheng10d27902006-01-06 20:36:21 +0000663 else
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000664 Result = CodeGenMap[N] = CurDAG->getTargetNode(Opc, VT, Result);
665 return;
Evan Chengbc7708c2005-12-17 02:02:50 +0000666 }
Chris Lattner655e7df2005-11-16 01:54:32 +0000667 }
668
Evan Cheng6dc90ca2006-02-09 00:37:58 +0000669 SelectCode(Result, N);
Chris Lattner655e7df2005-11-16 01:54:32 +0000670}
671
672/// createX86ISelDag - This pass converts a legalized DAG into a
673/// X86-specific DAG, ready for instruction scheduling.
674///
675FunctionPass *llvm::createX86ISelDag(TargetMachine &TM) {
676 return new X86DAGToDAGISel(TM);
677}