blob: 84e90b72a80ef5de7d49ccb685c85afe19b3874a [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//
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 Cheng8700e142006-01-11 06:09:51 +000016#include "X86InstrBuilder.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000017#include "X86RegisterInfo.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000018#include "X86Subtarget.h"
19#include "X86ISelLowering.h"
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000020#include "llvm/GlobalValue.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000021#include "llvm/Instructions.h"
22#include "llvm/Support/CFG.h"
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000023#include "llvm/CodeGen/MachineConstantPool.h"
Chris Lattnerc961eea2005-11-16 01:54:32 +000024#include "llvm/CodeGen/MachineFunction.h"
Evan Chengaaca22c2006-01-10 20:26:56 +000025#include "llvm/CodeGen/MachineFrameInfo.h"
Chris Lattner92cb0af2006-01-11 01:15:34 +000026#include "llvm/CodeGen/MachineInstrBuilder.h"
27#include "llvm/CodeGen/SSARegMap.h"
Chris Lattnerc961eea2005-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"
32using namespace llvm;
33
34//===----------------------------------------------------------------------===//
35// Pattern Matcher Implementation
36//===----------------------------------------------------------------------===//
37
38namespace {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000039 /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses
40 /// SDOperand's instead of register numbers for the leaves of the matched
41 /// tree.
42 struct X86ISelAddressMode {
43 enum {
44 RegBase,
45 FrameIndexBase,
Evan Chengec693f72005-12-08 02:01:35 +000046 ConstantPoolBase
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000047 } BaseType;
48
49 struct { // This is really a union, discriminated by BaseType!
50 SDOperand Reg;
51 int FrameIndex;
52 } Base;
53
54 unsigned Scale;
55 SDOperand IndexReg;
56 unsigned Disp;
57 GlobalValue *GV;
58
59 X86ISelAddressMode()
Evan Chengbd3d25c2005-11-30 02:51:20 +000060 : BaseType(RegBase), Scale(1), IndexReg(), Disp(0), GV(0) {
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +000061 }
62 };
63}
64
65namespace {
Chris Lattnerc961eea2005-11-16 01:54:32 +000066 Statistic<>
67 NumFPKill("x86-codegen", "Number of FP_REG_KILL instructions added");
68
69 //===--------------------------------------------------------------------===//
70 /// ISel - X86 specific code to select X86 machine instructions for
71 /// SelectionDAG operations.
72 ///
73 class X86DAGToDAGISel : public SelectionDAGISel {
74 /// ContainsFPCode - Every instruction we select that uses or defines a FP
75 /// register should set this to true.
76 bool ContainsFPCode;
77
78 /// X86Lowering - This object fully describes how to lower LLVM code to an
79 /// X86-specific SelectionDAG.
80 X86TargetLowering X86Lowering;
81
82 /// Subtarget - Keep a pointer to the X86Subtarget around so that we can
83 /// make the right decision when generating code for different targets.
84 const X86Subtarget *Subtarget;
85 public:
86 X86DAGToDAGISel(TargetMachine &TM)
87 : SelectionDAGISel(X86Lowering), X86Lowering(TM) {
88 Subtarget = &TM.getSubtarget<X86Subtarget>();
89 }
90
91 virtual const char *getPassName() const {
92 return "X86 DAG->DAG Instruction Selection";
93 }
94
95 /// InstructionSelectBasicBlock - This callback is invoked by
96 /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
97 virtual void InstructionSelectBasicBlock(SelectionDAG &DAG);
98
Evan Cheng8700e142006-01-11 06:09:51 +000099 virtual void EmitFunctionEntryCode(Function &Fn, MachineFunction &MF);
100
Chris Lattnerc961eea2005-11-16 01:54:32 +0000101// Include the pieces autogenerated from the target description.
102#include "X86GenDAGISel.inc"
103
104 private:
105 SDOperand Select(SDOperand N);
106
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000107 bool MatchAddress(SDOperand N, X86ISelAddressMode &AM);
Evan Chengec693f72005-12-08 02:01:35 +0000108 bool SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
109 SDOperand &Index, SDOperand &Disp);
110 bool SelectLEAAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
111 SDOperand &Index, SDOperand &Disp);
Evan Cheng0114e942006-01-06 20:36:21 +0000112 bool TryFoldLoad(SDOperand N, SDOperand &Base, SDOperand &Scale,
113 SDOperand &Index, SDOperand &Disp);
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000114
Evan Chenge5280532005-12-12 21:49:40 +0000115 inline void getAddressOperands(X86ISelAddressMode &AM, SDOperand &Base,
116 SDOperand &Scale, SDOperand &Index,
117 SDOperand &Disp) {
118 Base = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) ?
119 CurDAG->getTargetFrameIndex(AM.Base.FrameIndex, MVT::i32) : AM.Base.Reg;
Evan Chengbdce7b42005-12-17 09:13:43 +0000120 Scale = getI8Imm(AM.Scale);
Evan Chenge5280532005-12-12 21:49:40 +0000121 Index = AM.IndexReg;
122 Disp = AM.GV ? CurDAG->getTargetGlobalAddress(AM.GV, MVT::i32, AM.Disp)
123 : getI32Imm(AM.Disp);
124 }
125
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000126 /// getI8Imm - Return a target constant with the specified value, of type
127 /// i8.
128 inline SDOperand getI8Imm(unsigned Imm) {
129 return CurDAG->getTargetConstant(Imm, MVT::i8);
130 }
131
Chris Lattnerc961eea2005-11-16 01:54:32 +0000132 /// getI16Imm - Return a target constant with the specified value, of type
133 /// i16.
134 inline SDOperand getI16Imm(unsigned Imm) {
135 return CurDAG->getTargetConstant(Imm, MVT::i16);
136 }
137
138 /// getI32Imm - Return a target constant with the specified value, of type
139 /// i32.
140 inline SDOperand getI32Imm(unsigned Imm) {
141 return CurDAG->getTargetConstant(Imm, MVT::i32);
142 }
143 };
144}
145
146/// InstructionSelectBasicBlock - This callback is invoked by SelectionDAGISel
147/// when it has created a SelectionDAG for us to codegen.
148void X86DAGToDAGISel::InstructionSelectBasicBlock(SelectionDAG &DAG) {
149 DEBUG(BB->dump());
Chris Lattner92cb0af2006-01-11 01:15:34 +0000150 MachineFunction::iterator FirstMBB = BB;
Chris Lattnerc961eea2005-11-16 01:54:32 +0000151
152 // Codegen the basic block.
153 DAG.setRoot(Select(DAG.getRoot()));
Evan Chengfcaa9952005-12-19 22:36:02 +0000154 CodeGenMap.clear();
Chris Lattnerc961eea2005-11-16 01:54:32 +0000155 DAG.RemoveDeadNodes();
156
157 // Emit machine code to BB.
158 ScheduleAndEmitDAG(DAG);
Chris Lattner92cb0af2006-01-11 01:15:34 +0000159
160 // If we are emitting FP stack code, scan the basic block to determine if this
161 // block defines any FP values. If so, put an FP_REG_KILL instruction before
162 // the terminator of the block.
163 if (X86Vector < SSE2) {
164 // Note that FP stack instructions *are* used in SSE code when returning
165 // values, but these are not live out of the basic block, so we don't need
166 // an FP_REG_KILL in this case either.
167 bool ContainsFPCode = false;
168
169 // Scan all of the machine instructions in these MBBs, checking for FP
170 // stores.
171 MachineFunction::iterator MBBI = FirstMBB;
172 do {
173 for (MachineBasicBlock::iterator I = MBBI->begin(), E = MBBI->end();
174 !ContainsFPCode && I != E; ++I) {
175 for (unsigned op = 0, e = I->getNumOperands(); op != e; ++op) {
176 if (I->getOperand(op).isRegister() && I->getOperand(op).isDef() &&
177 MRegisterInfo::isVirtualRegister(I->getOperand(op).getReg()) &&
178 RegMap->getRegClass(I->getOperand(0).getReg()) ==
179 X86::RFPRegisterClass) {
180 ContainsFPCode = true;
181 break;
182 }
183 }
184 }
185 } while (!ContainsFPCode && &*(MBBI++) != BB);
186
187 // Check PHI nodes in successor blocks. These PHI's will be lowered to have
188 // a copy of the input value in this block.
189 if (!ContainsFPCode) {
190 // Final check, check LLVM BB's that are successors to the LLVM BB
191 // corresponding to BB for FP PHI nodes.
192 const BasicBlock *LLVMBB = BB->getBasicBlock();
193 const PHINode *PN;
194 for (succ_const_iterator SI = succ_begin(LLVMBB), E = succ_end(LLVMBB);
195 !ContainsFPCode && SI != E; ++SI) {
196 for (BasicBlock::const_iterator II = SI->begin();
197 (PN = dyn_cast<PHINode>(II)); ++II) {
198 if (PN->getType()->isFloatingPoint()) {
199 ContainsFPCode = true;
200 break;
201 }
202 }
203 }
204 }
205
206 // Finally, if we found any FP code, emit the FP_REG_KILL instruction.
207 if (ContainsFPCode) {
208 BuildMI(*BB, BB->getFirstTerminator(), X86::FP_REG_KILL, 0);
209 ++NumFPKill;
210 }
211 }
Chris Lattnerc961eea2005-11-16 01:54:32 +0000212}
213
Evan Cheng8700e142006-01-11 06:09:51 +0000214/// EmitSpecialCodeForMain - Emit any code that needs to be executed only in
215/// the main function.
216static void EmitSpecialCodeForMain(MachineBasicBlock *BB,
217 MachineFrameInfo *MFI) {
218 // Switch the FPU to 64-bit precision mode for better compatibility and speed.
219 int CWFrameIdx = MFI->CreateStackObject(2, 2);
220 addFrameReference(BuildMI(BB, X86::FNSTCW16m, 4), CWFrameIdx);
221
222 // Set the high part to be 64-bit precision.
223 addFrameReference(BuildMI(BB, X86::MOV8mi, 5),
224 CWFrameIdx, 1).addImm(2);
225
226 // Reload the modified control word now.
227 addFrameReference(BuildMI(BB, X86::FLDCW16m, 4), CWFrameIdx);
228}
229
230void X86DAGToDAGISel::EmitFunctionEntryCode(Function &Fn, MachineFunction &MF) {
231 // If this is main, emit special code for main.
232 MachineBasicBlock *BB = MF.begin();
233 if (Fn.hasExternalLinkage() && Fn.getName() == "main")
234 EmitSpecialCodeForMain(BB, MF.getFrameInfo());
235}
236
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000237/// MatchAddress - Add the specified node to the specified addressing mode,
238/// returning true if it cannot be done. This just pattern matches for the
239/// addressing mode
240bool X86DAGToDAGISel::MatchAddress(SDOperand N, X86ISelAddressMode &AM) {
241 switch (N.getOpcode()) {
242 default: break;
243 case ISD::FrameIndex:
244 if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
245 AM.BaseType = X86ISelAddressMode::FrameIndexBase;
246 AM.Base.FrameIndex = cast<FrameIndexSDNode>(N)->getIndex();
247 return false;
248 }
249 break;
Evan Chengec693f72005-12-08 02:01:35 +0000250
251 case ISD::ConstantPool:
252 if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
253 if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N)) {
254 AM.BaseType = X86ISelAddressMode::ConstantPoolBase;
255 AM.Base.Reg = CurDAG->getTargetConstantPool(CP->get(), MVT::i32);
256 return false;
257 }
258 }
259 break;
260
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000261 case ISD::GlobalAddress:
Evan Cheng3a03ebb2005-12-21 23:05:39 +0000262 case ISD::TargetGlobalAddress:
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000263 if (AM.GV == 0) {
Evan Chengb077b842005-12-21 02:39:21 +0000264 AM.GV = cast<GlobalAddressSDNode>(N)->getGlobal();
Evan Chengbdce7b42005-12-17 09:13:43 +0000265 return false;
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000266 }
267 break;
Evan Chengec693f72005-12-08 02:01:35 +0000268
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000269 case ISD::Constant:
270 AM.Disp += cast<ConstantSDNode>(N)->getValue();
271 return false;
Evan Chengec693f72005-12-08 02:01:35 +0000272
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000273 case ISD::SHL:
274 if (AM.IndexReg.Val == 0 && AM.Scale == 1)
275 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1))) {
276 unsigned Val = CN->getValue();
277 if (Val == 1 || Val == 2 || Val == 3) {
278 AM.Scale = 1 << Val;
279 SDOperand ShVal = N.Val->getOperand(0);
280
281 // Okay, we know that we have a scale by now. However, if the scaled
282 // value is an add of something and a constant, we can fold the
283 // constant into the disp field here.
284 if (ShVal.Val->getOpcode() == ISD::ADD && ShVal.hasOneUse() &&
285 isa<ConstantSDNode>(ShVal.Val->getOperand(1))) {
286 AM.IndexReg = ShVal.Val->getOperand(0);
287 ConstantSDNode *AddVal =
288 cast<ConstantSDNode>(ShVal.Val->getOperand(1));
289 AM.Disp += AddVal->getValue() << Val;
290 } else {
291 AM.IndexReg = ShVal;
292 }
293 return false;
294 }
295 }
296 break;
Evan Chengec693f72005-12-08 02:01:35 +0000297
Chris Lattnerf9ce9fb2005-11-19 02:11:08 +0000298 case ISD::MUL:
299 // X*[3,5,9] -> X+X*[2,4,8]
300 if (AM.IndexReg.Val == 0 && AM.BaseType == X86ISelAddressMode::RegBase &&
301 AM.Base.Reg.Val == 0)
302 if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1)))
303 if (CN->getValue() == 3 || CN->getValue() == 5 || CN->getValue() == 9) {
304 AM.Scale = unsigned(CN->getValue())-1;
305
306 SDOperand MulVal = N.Val->getOperand(0);
307 SDOperand Reg;
308
309 // Okay, we know that we have a scale by now. However, if the scaled
310 // value is an add of something and a constant, we can fold the
311 // constant into the disp field here.
312 if (MulVal.Val->getOpcode() == ISD::ADD && MulVal.hasOneUse() &&
313 isa<ConstantSDNode>(MulVal.Val->getOperand(1))) {
314 Reg = MulVal.Val->getOperand(0);
315 ConstantSDNode *AddVal =
316 cast<ConstantSDNode>(MulVal.Val->getOperand(1));
317 AM.Disp += AddVal->getValue() * CN->getValue();
318 } else {
319 Reg = N.Val->getOperand(0);
320 }
321
322 AM.IndexReg = AM.Base.Reg = Reg;
323 return false;
324 }
325 break;
326
327 case ISD::ADD: {
328 X86ISelAddressMode Backup = AM;
329 if (!MatchAddress(N.Val->getOperand(0), AM) &&
330 !MatchAddress(N.Val->getOperand(1), AM))
331 return false;
332 AM = Backup;
333 if (!MatchAddress(N.Val->getOperand(1), AM) &&
334 !MatchAddress(N.Val->getOperand(0), AM))
335 return false;
336 AM = Backup;
337 break;
338 }
339 }
340
341 // Is the base register already occupied?
342 if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base.Reg.Val) {
343 // If so, check to see if the scale index register is set.
344 if (AM.IndexReg.Val == 0) {
345 AM.IndexReg = N;
346 AM.Scale = 1;
347 return false;
348 }
349
350 // Otherwise, we cannot select it.
351 return true;
352 }
353
354 // Default, generate it as a register.
355 AM.BaseType = X86ISelAddressMode::RegBase;
356 AM.Base.Reg = N;
357 return false;
358}
359
Evan Chengec693f72005-12-08 02:01:35 +0000360/// SelectAddr - returns true if it is able pattern match an addressing mode.
361/// It returns the operands which make up the maximal addressing mode it can
362/// match by reference.
363bool X86DAGToDAGISel::SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
364 SDOperand &Index, SDOperand &Disp) {
365 X86ISelAddressMode AM;
Evan Cheng8700e142006-01-11 06:09:51 +0000366 if (MatchAddress(N, AM))
367 return false;
Evan Chengec693f72005-12-08 02:01:35 +0000368
Evan Cheng8700e142006-01-11 06:09:51 +0000369 if (AM.BaseType == X86ISelAddressMode::RegBase) {
370 if (AM.Base.Reg.Val) {
371 if (AM.Base.Reg.getOpcode() != ISD::Register)
372 AM.Base.Reg = Select(AM.Base.Reg);
373 } else {
374 AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
375 }
Evan Chengec693f72005-12-08 02:01:35 +0000376 }
Evan Cheng8700e142006-01-11 06:09:51 +0000377
378 if (AM.IndexReg.Val)
379 AM.IndexReg = Select(AM.IndexReg);
380 else
381 AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
382
383 getAddressOperands(AM, Base, Scale, Index, Disp);
384 return true;
Evan Chengec693f72005-12-08 02:01:35 +0000385}
386
Evan Cheng0114e942006-01-06 20:36:21 +0000387bool X86DAGToDAGISel::TryFoldLoad(SDOperand N, SDOperand &Base,
388 SDOperand &Scale, SDOperand &Index,
389 SDOperand &Disp) {
390 if (N.getOpcode() == ISD::LOAD && N.hasOneUse() &&
Evan Cheng948f3432006-01-06 23:19:29 +0000391 CodeGenMap.count(N.getValue(1)) == 0)
Evan Cheng0114e942006-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 Chengec693f72005-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 Lattnera2b694c2006-01-11 00:46:55 +0000405bool X86DAGToDAGISel::SelectLEAAddr(SDOperand N, SDOperand &Base,
406 SDOperand &Scale,
Evan Chengec693f72005-12-08 02:01:35 +0000407 SDOperand &Index, SDOperand &Disp) {
Evan Chenge5280532005-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 Chengec693f72005-12-08 02:01:35 +0000417 } else {
Evan Chenge5280532005-12-12 21:49:40 +0000418 AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
Evan Chengec693f72005-12-08 02:01:35 +0000419 }
Evan Chengec693f72005-12-08 02:01:35 +0000420 }
Evan Chenge5280532005-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
442 if (SelectBase)
443 AM.Base.Reg = Select(AM.Base.Reg);
444 if (SelectIndex)
445 AM.IndexReg = Select(AM.IndexReg);
446
447 getAddressOperands(AM, Base, Scale, Index, Disp);
Evan Chengec693f72005-12-08 02:01:35 +0000448 return true;
Evan Chengec693f72005-12-08 02:01:35 +0000449 }
Evan Chenge5280532005-12-12 21:49:40 +0000450 return false;
Evan Chengec693f72005-12-08 02:01:35 +0000451}
452
Evan Chengdef941b2005-12-15 01:02:48 +0000453SDOperand X86DAGToDAGISel::Select(SDOperand N) {
454 SDNode *Node = N.Val;
455 MVT::ValueType NVT = Node->getValueType(0);
Evan Cheng0114e942006-01-06 20:36:21 +0000456 unsigned Opc, MOpc;
457 unsigned Opcode = Node->getOpcode();
Chris Lattnerc961eea2005-11-16 01:54:32 +0000458
Evan Cheng0114e942006-01-06 20:36:21 +0000459 if (Opcode >= ISD::BUILTIN_OP_END && Opcode < X86ISD::FIRST_NUMBER)
Evan Chengdef941b2005-12-15 01:02:48 +0000460 return N; // Already selected.
Evan Cheng38262ca2006-01-11 22:15:18 +0000461
462 std::map<SDOperand, SDOperand>::iterator CGMI = CodeGenMap.find(N);
463 if (CGMI != CodeGenMap.end()) return CGMI->second;
Chris Lattnerc961eea2005-11-16 01:54:32 +0000464
Evan Cheng0114e942006-01-06 20:36:21 +0000465 switch (Opcode) {
Chris Lattnerc961eea2005-11-16 01:54:32 +0000466 default: break;
Evan Cheng0114e942006-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;
497 foldedLoad = TryFoldLoad(N1, Tmp0, Tmp1, Tmp2, Tmp3);
Evan Cheng948f3432006-01-06 23:19:29 +0000498 // MULHU and MULHS are commmutative
499 if (!foldedLoad) {
500 foldedLoad = TryFoldLoad(N0, Tmp0, Tmp1, Tmp2, Tmp3);
501 if (foldedLoad) {
502 N0 = Node->getOperand(1);
503 N1 = Node->getOperand(0);
504 }
505 }
506
Evan Cheng0114e942006-01-06 20:36:21 +0000507 SDOperand Chain = foldedLoad ? Select(N1.getOperand(0))
508 : CurDAG->getEntryNode();
509
510 SDOperand InFlag;
511 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
512 Select(N0), InFlag);
513 InFlag = Chain.getValue(1);
514
515 if (foldedLoad) {
516 Chain = CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
517 Tmp2, Tmp3, Chain, InFlag);
518 InFlag = Chain.getValue(1);
519 } else {
520 InFlag = CurDAG->getTargetNode(Opc, MVT::Flag, Select(N1), InFlag);
521 }
522
523 SDOperand Result = CurDAG->getCopyFromReg(Chain, HiReg, NVT, InFlag);
524 CodeGenMap[N.getValue(0)] = Result;
Evan Cheng948f3432006-01-06 23:19:29 +0000525 if (foldedLoad)
526 CodeGenMap[N1.getValue(1)] = Result.getValue(1);
527 return Result;
528 }
529
530 case ISD::SDIV:
531 case ISD::UDIV:
532 case ISD::SREM:
533 case ISD::UREM: {
534 bool isSigned = Opcode == ISD::SDIV || Opcode == ISD::SREM;
535 bool isDiv = Opcode == ISD::SDIV || Opcode == ISD::UDIV;
536 if (!isSigned)
537 switch (NVT) {
538 default: assert(0 && "Unsupported VT!");
539 case MVT::i8: Opc = X86::DIV8r; MOpc = X86::DIV8m; break;
540 case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break;
541 case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break;
542 }
543 else
544 switch (NVT) {
545 default: assert(0 && "Unsupported VT!");
546 case MVT::i8: Opc = X86::IDIV8r; MOpc = X86::IDIV8m; break;
547 case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break;
548 case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break;
549 }
550
551 unsigned LoReg, HiReg;
552 unsigned ClrOpcode, SExtOpcode;
553 switch (NVT) {
554 default: assert(0 && "Unsupported VT!");
555 case MVT::i8:
556 LoReg = X86::AL; HiReg = X86::AH;
557 ClrOpcode = X86::MOV8ri;
558 SExtOpcode = X86::CBW;
559 break;
560 case MVT::i16:
561 LoReg = X86::AX; HiReg = X86::DX;
562 ClrOpcode = X86::MOV16ri;
563 SExtOpcode = X86::CWD;
564 break;
565 case MVT::i32:
566 LoReg = X86::EAX; HiReg = X86::EDX;
567 ClrOpcode = X86::MOV32ri;
568 SExtOpcode = X86::CDQ;
569 break;
570 }
571
572 SDOperand N0 = Node->getOperand(0);
573 SDOperand N1 = Node->getOperand(1);
574
575 bool foldedLoad = false;
576 SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
577 foldedLoad = TryFoldLoad(N1, Tmp0, Tmp1, Tmp2, Tmp3);
578 SDOperand Chain = foldedLoad ? Select(N1.getOperand(0))
579 : CurDAG->getEntryNode();
580
581 SDOperand InFlag;
582 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
583 Select(N0), InFlag);
584 InFlag = Chain.getValue(1);
585
586 if (isSigned) {
587 // Sign extend the low part into the high part.
588 InFlag = CurDAG->getTargetNode(SExtOpcode, MVT::Flag, InFlag);
589 } else {
590 // Zero out the high part, effectively zero extending the input.
591 SDOperand ClrNode =
592 CurDAG->getTargetNode(ClrOpcode, NVT,
593 CurDAG->getTargetConstant(0, NVT));
594 Chain = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(HiReg, NVT),
595 ClrNode, InFlag);
596 InFlag = Chain.getValue(1);
597 }
598
599 if (foldedLoad) {
600 Chain = CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
601 Tmp2, Tmp3, Chain, InFlag);
602 InFlag = Chain.getValue(1);
603 } else {
604 InFlag = CurDAG->getTargetNode(Opc, MVT::Flag, Select(N1), InFlag);
605 }
606
607 SDOperand Result = CurDAG->getCopyFromReg(Chain, isDiv ? LoReg : HiReg,
608 NVT, InFlag);
609 CodeGenMap[N.getValue(0)] = Result;
610 if (foldedLoad)
611 CodeGenMap[N1.getValue(1)] = Result.getValue(1);
612 return Result;
Evan Cheng0114e942006-01-06 20:36:21 +0000613 }
Evan Chengbd3d25c2005-11-30 02:51:20 +0000614
Evan Cheng45f37bc2005-12-17 02:02:50 +0000615 case ISD::TRUNCATE: {
616 unsigned Reg;
617 MVT::ValueType VT;
618 switch (Node->getOperand(0).getValueType()) {
619 default: assert(0 && "Unknown truncate!");
620 case MVT::i16: Reg = X86::AX; Opc = X86::MOV16rr; VT = MVT::i16; break;
621 case MVT::i32: Reg = X86::EAX; Opc = X86::MOV32rr; VT = MVT::i32; break;
622 }
623 SDOperand Tmp0 = Select(Node->getOperand(0));
624 SDOperand Tmp1 = CurDAG->getTargetNode(Opc, VT, Tmp0);
625 SDOperand InFlag = SDOperand(0,0);
626 SDOperand Result = CurDAG->getCopyToReg(CurDAG->getEntryNode(),
Evan Cheng38262ca2006-01-11 22:15:18 +0000627 Reg, Tmp1, InFlag);
Evan Cheng45f37bc2005-12-17 02:02:50 +0000628 SDOperand Chain = Result.getValue(0);
629 InFlag = Result.getValue(1);
630
631 switch (NVT) {
632 default: assert(0 && "Unknown truncate!");
633 case MVT::i8: Reg = X86::AL; Opc = X86::MOV8rr; VT = MVT::i8; break;
634 case MVT::i16: Reg = X86::AX; Opc = X86::MOV16rr; VT = MVT::i16; break;
635 }
636
637 Result = CurDAG->getCopyFromReg(Chain,
638 Reg, VT, InFlag);
Evan Cheng0114e942006-01-06 20:36:21 +0000639 if (N.Val->hasOneUse())
640 return CurDAG->SelectNodeTo(N.Val, Opc, VT, Result);
641 else
642 return CodeGenMap[N] = CurDAG->getTargetNode(Opc, VT, Result);
Evan Cheng45f37bc2005-12-17 02:02:50 +0000643 break;
644 }
Chris Lattnerc961eea2005-11-16 01:54:32 +0000645 }
646
Evan Chengdef941b2005-12-15 01:02:48 +0000647 return SelectCode(N);
Chris Lattnerc961eea2005-11-16 01:54:32 +0000648}
649
650/// createX86ISelDag - This pass converts a legalized DAG into a
651/// X86-specific DAG, ready for instruction scheduling.
652///
653FunctionPass *llvm::createX86ISelDag(TargetMachine &TM) {
654 return new X86DAGToDAGISel(TM);
655}