lib/Target/X86/X86ISelDAGToDAG.cpp

   1 //===- X86ISelDAGToDAG.cpp - A DAG pattern matching inst selector for X86 -===//
   2 //
   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 #define DEBUG_TYPE "isel"
  16 #include "X86.h"
  17 #include "X86InstrBuilder.h"
  18 #include "X86ISelLowering.h"
  19 #include "X86RegisterInfo.h"
  20 #include "X86Subtarget.h"
  21 #include "X86TargetMachine.h"
  22 #include "llvm/GlobalValue.h"
  23 #include "llvm/Instructions.h"
  24 #include "llvm/Intrinsics.h"
  25 #include "llvm/Support/CFG.h"
  26 #include "llvm/CodeGen/MachineConstantPool.h"
  27 #include "llvm/CodeGen/MachineFunction.h"
  28 #include "llvm/CodeGen/MachineFrameInfo.h"
  29 #include "llvm/CodeGen/MachineInstrBuilder.h"
  30 #include "llvm/CodeGen/SSARegMap.h"
  31 #include "llvm/CodeGen/SelectionDAGISel.h"
  32 #include "llvm/Target/TargetMachine.h"
  33 #include "llvm/Support/Debug.h"
  34 #include "llvm/ADT/Statistic.h"
  35 #include <iostream>
  36 #include <set>
  37 using namespace llvm;
  38
  39 //===----------------------------------------------------------------------===//
  40 //                      Pattern Matcher Implementation
  41 //===----------------------------------------------------------------------===//
  42
  43 namespace {
  44   /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses
  45   /// SDOperand's instead of register numbers for the leaves of the matched
  46   /// tree.
  47   struct X86ISelAddressMode {
  48     enum {
  49       RegBase,
  50       FrameIndexBase,
  51     } BaseType;
  52
  53     struct {            // This is really a union, discriminated by BaseType!
  54       SDOperand Reg;
  55       int FrameIndex;
  56     } Base;
  57
  58     unsigned Scale;
  59     SDOperand IndexReg;
  60     unsigned Disp;
  61     GlobalValue *GV;
  62     Constant *CP;
  63     unsigned Align;    // CP alignment.
  64
  65     X86ISelAddressMode()
  66       : BaseType(RegBase), Scale(1), IndexReg(), Disp(0), GV(0),
  67         CP(0), Align(0) {
  68     }
  69   };
  70 }
  71
  72 namespace {
  73   Statistic<>
  74   NumFPKill("x86-codegen", "Number of FP_REG_KILL instructions added");
  75
  76   //===--------------------------------------------------------------------===//
  77   /// ISel - X86 specific code to select X86 machine instructions for
  78   /// SelectionDAG operations.
  79   ///
  80   class X86DAGToDAGISel : public SelectionDAGISel {
  81     /// ContainsFPCode - Every instruction we select that uses or defines a FP
  82     /// register should set this to true.
  83     bool ContainsFPCode;
  84
  85     /// X86Lowering - This object fully describes how to lower LLVM code to an
  86     /// X86-specific SelectionDAG.
  87     X86TargetLowering X86Lowering;
  88
  89     /// Subtarget - Keep a pointer to the X86Subtarget around so that we can
  90     /// make the right decision when generating code for different targets.
  91     const X86Subtarget *Subtarget;
  92
  93     unsigned GlobalBaseReg;
  94   public:
  95     X86DAGToDAGISel(X86TargetMachine &TM)
  96       : SelectionDAGISel(X86Lowering),
  97         X86Lowering(*TM.getTargetLowering()) {
  98       Subtarget = &TM.getSubtarget<X86Subtarget>();
  99     }
 100
 101     virtual bool runOnFunction(Function &Fn) {
 102       // Make sure we re-emit a set of the global base reg if necessary
 103       GlobalBaseReg = 0;
 104       return SelectionDAGISel::runOnFunction(Fn);
 105     }
 106
 107     virtual const char *getPassName() const {
 108       return "X86 DAG->DAG Instruction Selection";
 109     }
 110
 111     /// InstructionSelectBasicBlock - This callback is invoked by
 112     /// SelectionDAGISel when it has created a SelectionDAG for us to codegen.
 113     virtual void InstructionSelectBasicBlock(SelectionDAG &DAG);
 114
 115     virtual void EmitFunctionEntryCode(Function &Fn, MachineFunction &MF);
 116
 117 // Include the pieces autogenerated from the target description.
 118 #include "X86GenDAGISel.inc"
 119
 120   private:
 121     void Select(SDOperand &Result, SDOperand N);
 122
 123     bool MatchAddress(SDOperand N, X86ISelAddressMode &AM, bool isRoot = true);
 124     bool SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
 125                     SDOperand &Index, SDOperand &Disp);
 126     bool SelectLEAAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
 127                        SDOperand &Index, SDOperand &Disp);
 128     bool TryFoldLoad(SDOperand P, SDOperand N,
 129                      SDOperand &Base, SDOperand &Scale,
 130                      SDOperand &Index, SDOperand &Disp);
 131
 132     inline void getAddressOperands(X86ISelAddressMode &AM, SDOperand &Base,
 133                                    SDOperand &Scale, SDOperand &Index,
 134                                    SDOperand &Disp) {
 135       Base  = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) ?
 136         CurDAG->getTargetFrameIndex(AM.Base.FrameIndex, MVT::i32) : AM.Base.Reg;
 137       Scale = getI8Imm(AM.Scale);
 138       Index = AM.IndexReg;
 139       Disp  = AM.GV ? CurDAG->getTargetGlobalAddress(AM.GV, MVT::i32, AM.Disp)
 140         : (AM.CP ?
 141            CurDAG->getTargetConstantPool(AM.CP, MVT::i32, AM.Align, AM.Disp)
 142            : getI32Imm(AM.Disp));
 143     }
 144
 145     /// getI8Imm - Return a target constant with the specified value, of type
 146     /// i8.
 147     inline SDOperand getI8Imm(unsigned Imm) {
 148       return CurDAG->getTargetConstant(Imm, MVT::i8);
 149     }
 150
 151     /// getI16Imm - Return a target constant with the specified value, of type
 152     /// i16.
 153     inline SDOperand getI16Imm(unsigned Imm) {
 154       return CurDAG->getTargetConstant(Imm, MVT::i16);
 155     }
 156
 157     /// getI32Imm - Return a target constant with the specified value, of type
 158     /// i32.
 159     inline SDOperand getI32Imm(unsigned Imm) {
 160       return CurDAG->getTargetConstant(Imm, MVT::i32);
 161     }
 162
 163     /// getGlobalBaseReg - insert code into the entry mbb to materialize the PIC
 164     /// base register.  Return the virtual register that holds this value.
 165     SDOperand getGlobalBaseReg();
 166
 167 #ifndef NDEBUG
 168     unsigned Indent;
 169 #endif
 170   };
 171 }
 172
 173 /// InstructionSelectBasicBlock - This callback is invoked by SelectionDAGISel
 174 /// when it has created a SelectionDAG for us to codegen.
 175 void X86DAGToDAGISel::InstructionSelectBasicBlock(SelectionDAG &DAG) {
 176   DEBUG(BB->dump());
 177   MachineFunction::iterator FirstMBB = BB;
 178
 179   // Codegen the basic block.
 180 #ifndef NDEBUG
 181   DEBUG(std::cerr << "===== Instruction selection begins:\n");
 182   Indent = 0;
 183 #endif
 184   DAG.setRoot(SelectRoot(DAG.getRoot()));
 185 #ifndef NDEBUG
 186   DEBUG(std::cerr << "===== Instruction selection ends:\n");
 187 #endif
 188   CodeGenMap.clear();
 189   DAG.RemoveDeadNodes();
 190
 191   // Emit machine code to BB.
 192   ScheduleAndEmitDAG(DAG);
 193
 194   // If we are emitting FP stack code, scan the basic block to determine if this
 195   // block defines any FP values.  If so, put an FP_REG_KILL instruction before
 196   // the terminator of the block.
 197   if (!Subtarget->hasSSE2()) {
 198     // Note that FP stack instructions *are* used in SSE code when returning
 199     // values, but these are not live out of the basic block, so we don't need
 200     // an FP_REG_KILL in this case either.
 201     bool ContainsFPCode = false;
 202
 203     // Scan all of the machine instructions in these MBBs, checking for FP
 204     // stores.
 205     MachineFunction::iterator MBBI = FirstMBB;
 206     do {
 207       for (MachineBasicBlock::iterator I = MBBI->begin(), E = MBBI->end();
 208            !ContainsFPCode && I != E; ++I) {
 209         for (unsigned op = 0, e = I->getNumOperands(); op != e; ++op) {
 210           if (I->getOperand(op).isRegister() && I->getOperand(op).isDef() &&
 211               MRegisterInfo::isVirtualRegister(I->getOperand(op).getReg()) &&
 212               RegMap->getRegClass(I->getOperand(0).getReg()) ==
 213                 X86::RFPRegisterClass) {
 214             ContainsFPCode = true;
 215             break;
 216           }
 217         }
 218       }
 219     } while (!ContainsFPCode && &*(MBBI++) != BB);
 220
 221     // Check PHI nodes in successor blocks.  These PHI's will be lowered to have
 222     // a copy of the input value in this block.
 223     if (!ContainsFPCode) {
 224       // Final check, check LLVM BB's that are successors to the LLVM BB
 225       // corresponding to BB for FP PHI nodes.
 226       const BasicBlock *LLVMBB = BB->getBasicBlock();
 227       const PHINode *PN;
 228       for (succ_const_iterator SI = succ_begin(LLVMBB), E = succ_end(LLVMBB);
 229            !ContainsFPCode && SI != E; ++SI) {
 230         for (BasicBlock::const_iterator II = SI->begin();
 231              (PN = dyn_cast<PHINode>(II)); ++II) {
 232           if (PN->getType()->isFloatingPoint()) {
 233             ContainsFPCode = true;
 234             break;
 235           }
 236         }
 237       }
 238     }
 239
 240     // Finally, if we found any FP code, emit the FP_REG_KILL instruction.
 241     if (ContainsFPCode) {
 242       BuildMI(*BB, BB->getFirstTerminator(), X86::FP_REG_KILL, 0);
 243       ++NumFPKill;
 244     }
 245   }
 246 }
 247
 248 /// EmitSpecialCodeForMain - Emit any code that needs to be executed only in
 249 /// the main function.
 250 static void EmitSpecialCodeForMain(MachineBasicBlock *BB,
 251                                    MachineFrameInfo *MFI) {
 252   // Switch the FPU to 64-bit precision mode for better compatibility and speed.
 253   int CWFrameIdx = MFI->CreateStackObject(2, 2);
 254   addFrameReference(BuildMI(BB, X86::FNSTCW16m, 4), CWFrameIdx);
 255
 256   // Set the high part to be 64-bit precision.
 257   addFrameReference(BuildMI(BB, X86::MOV8mi, 5),
 258                     CWFrameIdx, 1).addImm(2);
 259
 260   // Reload the modified control word now.
 261   addFrameReference(BuildMI(BB, X86::FLDCW16m, 4), CWFrameIdx);
 262 }
 263
 264 void X86DAGToDAGISel::EmitFunctionEntryCode(Function &Fn, MachineFunction &MF) {
 265   // If this is main, emit special code for main.
 266   MachineBasicBlock *BB = MF.begin();
 267   if (Fn.hasExternalLinkage() && Fn.getName() == "main")
 268     EmitSpecialCodeForMain(BB, MF.getFrameInfo());
 269 }
 270
 271 /// MatchAddress - Add the specified node to the specified addressing mode,
 272 /// returning true if it cannot be done.  This just pattern matches for the
 273 /// addressing mode
 274 bool X86DAGToDAGISel::MatchAddress(SDOperand N, X86ISelAddressMode &AM,
 275                                    bool isRoot) {
 276   bool Available = false;
 277   // If N has already been selected, reuse the result unless in some very
 278   // specific cases.
 279   std::map<SDOperand, SDOperand>::iterator CGMI= CodeGenMap.find(N.getValue(0));
 280   if (CGMI != CodeGenMap.end()) {
 281     Available = true;
 282   }
 283
 284   switch (N.getOpcode()) {
 285   default: break;
 286   case ISD::Constant:
 287     AM.Disp += cast<ConstantSDNode>(N)->getValue();
 288     return false;
 289
 290   case X86ISD::Wrapper:
 291     // If both base and index components have been picked, we can't fit
 292     // the result available in the register in the addressing mode. Duplicate
 293     // GlobalAddress or ConstantPool as displacement.
 294     if (!Available || (AM.Base.Reg.Val && AM.IndexReg.Val)) {
 295       if (ConstantPoolSDNode *CP =
 296           dyn_cast<ConstantPoolSDNode>(N.getOperand(0))) {
 297         if (AM.CP == 0) {
 298           AM.CP = CP->get();
 299           AM.Align = CP->getAlignment();
 300           AM.Disp += CP->getOffset();
 301           return false;
 302         }
 303       } else if (GlobalAddressSDNode *G =
 304                  dyn_cast<GlobalAddressSDNode>(N.getOperand(0))) {
 305         if (AM.GV == 0) {
 306           AM.GV = G->getGlobal();
 307           AM.Disp += G->getOffset();
 308           return false;
 309         }
 310       }
 311     }
 312     break;
 313
 314   case ISD::FrameIndex:
 315     if (AM.BaseType == X86ISelAddressMode::RegBase && AM.Base.Reg.Val == 0) {
 316       AM.BaseType = X86ISelAddressMode::FrameIndexBase;
 317       AM.Base.FrameIndex = cast<FrameIndexSDNode>(N)->getIndex();
 318       return false;
 319     }
 320     break;
 321
 322   case ISD::SHL:
 323     if (!Available && AM.IndexReg.Val == 0 && AM.Scale == 1)
 324       if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1))) {
 325         unsigned Val = CN->getValue();
 326         if (Val == 1 || Val == 2 || Val == 3) {
 327           AM.Scale = 1 << Val;
 328           SDOperand ShVal = N.Val->getOperand(0);
 329
 330           // Okay, we know that we have a scale by now.  However, if the scaled
 331           // value is an add of something and a constant, we can fold the
 332           // constant into the disp field here.
 333           if (ShVal.Val->getOpcode() == ISD::ADD && ShVal.hasOneUse() &&
 334               isa<ConstantSDNode>(ShVal.Val->getOperand(1))) {
 335             AM.IndexReg = ShVal.Val->getOperand(0);
 336             ConstantSDNode *AddVal =
 337               cast<ConstantSDNode>(ShVal.Val->getOperand(1));
 338             AM.Disp += AddVal->getValue() << Val;
 339           } else {
 340             AM.IndexReg = ShVal;
 341           }
 342           return false;
 343         }
 344       }
 345     break;
 346
 347   case ISD::MUL:
 348     // X*[3,5,9] -> X+X*[2,4,8]
 349     if (!Available &&
 350         AM.BaseType == X86ISelAddressMode::RegBase &&
 351         AM.Base.Reg.Val == 0 &&
 352         AM.IndexReg.Val == 0)
 353       if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N.Val->getOperand(1)))
 354         if (CN->getValue() == 3 || CN->getValue() == 5 || CN->getValue() == 9) {
 355           AM.Scale = unsigned(CN->getValue())-1;
 356
 357           SDOperand MulVal = N.Val->getOperand(0);
 358           SDOperand Reg;
 359
 360           // Okay, we know that we have a scale by now.  However, if the scaled
 361           // value is an add of something and a constant, we can fold the
 362           // constant into the disp field here.
 363           if (MulVal.Val->getOpcode() == ISD::ADD && MulVal.hasOneUse() &&
 364               isa<ConstantSDNode>(MulVal.Val->getOperand(1))) {
 365             Reg = MulVal.Val->getOperand(0);
 366             ConstantSDNode *AddVal =
 367               cast<ConstantSDNode>(MulVal.Val->getOperand(1));
 368             AM.Disp += AddVal->getValue() * CN->getValue();
 369           } else {
 370             Reg = N.Val->getOperand(0);
 371           }
 372
 373           AM.IndexReg = AM.Base.Reg = Reg;
 374           return false;
 375         }
 376     break;
 377
 378   case ISD::ADD: {
 379     if (!Available) {
 380       X86ISelAddressMode Backup = AM;
 381       if (!MatchAddress(N.Val->getOperand(0), AM, false) &&
 382           !MatchAddress(N.Val->getOperand(1), AM, false))
 383         return false;
 384       AM = Backup;
 385       if (!MatchAddress(N.Val->getOperand(1), AM, false) &&
 386           !MatchAddress(N.Val->getOperand(0), AM, false))
 387         return false;
 388       AM = Backup;
 389     }
 390     break;
 391   }
 392   }
 393
 394   // Is the base register already occupied?
 395   if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base.Reg.Val) {
 396     // If so, check to see if the scale index register is set.
 397     if (AM.IndexReg.Val == 0) {
 398       AM.IndexReg = N;
 399       AM.Scale = 1;
 400       return false;
 401     }
 402
 403     // Otherwise, we cannot select it.
 404     return true;
 405   }
 406
 407   // Default, generate it as a register.
 408   AM.BaseType = X86ISelAddressMode::RegBase;
 409   AM.Base.Reg = N;
 410   return false;
 411 }
 412
 413 /// SelectAddr - returns true if it is able pattern match an addressing mode.
 414 /// It returns the operands which make up the maximal addressing mode it can
 415 /// match by reference.
 416 bool X86DAGToDAGISel::SelectAddr(SDOperand N, SDOperand &Base, SDOperand &Scale,
 417                                  SDOperand &Index, SDOperand &Disp) {
 418   X86ISelAddressMode AM;
 419   if (MatchAddress(N, AM))
 420     return false;
 421
 422   if (AM.BaseType == X86ISelAddressMode::RegBase) {
 423     if (!AM.Base.Reg.Val)
 424       AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
 425   }
 426
 427   if (!AM.IndexReg.Val)
 428     AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
 429
 430   getAddressOperands(AM, Base, Scale, Index, Disp);
 431
 432   return true;
 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).
 438 bool X86DAGToDAGISel::SelectLEAAddr(SDOperand N, SDOperand &Base,
 439                                     SDOperand &Scale,
 440                                     SDOperand &Index, SDOperand &Disp) {
 441   X86ISelAddressMode AM;
 442   if (MatchAddress(N, AM))
 443     return false;
 444
 445   unsigned Complexity = 0;
 446   if (AM.BaseType == X86ISelAddressMode::RegBase)
 447     if (AM.Base.Reg.Val)
 448       Complexity = 1;
 449     else
 450       AM.Base.Reg = CurDAG->getRegister(0, MVT::i32);
 451   else if (AM.BaseType == X86ISelAddressMode::FrameIndexBase)
 452     Complexity = 4;
 453
 454   if (AM.IndexReg.Val)
 455     Complexity++;
 456   else
 457     AM.IndexReg = CurDAG->getRegister(0, MVT::i32);
 458
 459   if (AM.Scale > 2)
 460     Complexity += 2;
 461   // Don't match just leal(,%reg,2). It's cheaper to do addl %reg, %reg
 462   else if (AM.Scale > 1)
 463     Complexity++;
 464
 465   // FIXME: We are artificially lowering the criteria to turn ADD %reg, $GA
 466   // to a LEA. This is determined with some expermentation but is by no means
 467   // optimal (especially for code size consideration). LEA is nice because of
 468   // its three-address nature. Tweak the cost function again when we can run
 469   // convertToThreeAddress() at register allocation time.
 470   if (AM.GV || AM.CP)
 471     Complexity += 2;
 472
 473   if (AM.Disp && (AM.Base.Reg.Val || AM.IndexReg.Val))
 474     Complexity++;
 475
 476   if (Complexity > 2) {
 477     getAddressOperands(AM, Base, Scale, Index, Disp);
 478     return true;
 479   }
 480
 481   return false;
 482 }
 483
 484 bool X86DAGToDAGISel::TryFoldLoad(SDOperand P, SDOperand N,
 485                                   SDOperand &Base, SDOperand &Scale,
 486                                   SDOperand &Index, SDOperand &Disp) {
 487   if (N.getOpcode() == ISD::LOAD &&
 488       N.hasOneUse() &&
 489       !CodeGenMap.count(N.getValue(0)) &&
 490       (P.getNumOperands() == 1 || !isNonImmUse(P.Val, N.Val)))
 491     return SelectAddr(N.getOperand(1), Base, Scale, Index, Disp);
 492   return false;
 493 }
 494
 495 static bool isRegister0(SDOperand Op) {
 496   if (RegisterSDNode *R = dyn_cast<RegisterSDNode>(Op))
 497     return (R->getReg() == 0);
 498   return false;
 499 }
 500
 501 /// getGlobalBaseReg - Output the instructions required to put the
 502 /// base address to use for accessing globals into a register.
 503 ///
 504 SDOperand X86DAGToDAGISel::getGlobalBaseReg() {
 505   if (!GlobalBaseReg) {
 506     // Insert the set of GlobalBaseReg into the first MBB of the function
 507     MachineBasicBlock &FirstMBB = BB->getParent()->front();
 508     MachineBasicBlock::iterator MBBI = FirstMBB.begin();
 509     SSARegMap *RegMap = BB->getParent()->getSSARegMap();
 510     // FIXME: when we get to LP64, we will need to create the appropriate
 511     // type of register here.
 512     GlobalBaseReg = RegMap->createVirtualRegister(X86::GR32RegisterClass);
 513     BuildMI(FirstMBB, MBBI, X86::MovePCtoStack, 0);
 514     BuildMI(FirstMBB, MBBI, X86::POP32r, 1, GlobalBaseReg);
 515   }
 516   return CurDAG->getRegister(GlobalBaseReg, MVT::i32);
 517 }
 518
 519 static SDNode *FindCallStartFromCall(SDNode *Node) {
 520   if (Node->getOpcode() == ISD::CALLSEQ_START) return Node;
 521     assert(Node->getOperand(0).getValueType() == MVT::Other &&
 522          "Node doesn't have a token chain argument!");
 523   return FindCallStartFromCall(Node->getOperand(0).Val);
 524 }
 525
 526 void X86DAGToDAGISel::Select(SDOperand &Result, SDOperand N) {
 527   SDNode *Node = N.Val;
 528   MVT::ValueType NVT = Node->getValueType(0);
 529   unsigned Opc, MOpc;
 530   unsigned Opcode = Node->getOpcode();
 531
 532 #ifndef NDEBUG
 533   DEBUG(std::cerr << std::string(Indent, ' '));
 534   DEBUG(std::cerr << "Selecting: ");
 535   DEBUG(Node->dump(CurDAG));
 536   DEBUG(std::cerr << "\n");
 537   Indent += 2;
 538 #endif
 539
 540   if (Opcode >= ISD::BUILTIN_OP_END && Opcode < X86ISD::FIRST_NUMBER) {
 541     Result = N;
 542 #ifndef NDEBUG
 543     DEBUG(std::cerr << std::string(Indent-2, ' '));
 544     DEBUG(std::cerr << "== ");
 545     DEBUG(Node->dump(CurDAG));
 546     DEBUG(std::cerr << "\n");
 547     Indent -= 2;
 548 #endif
 549     return;   // Already selected.
 550   }
 551
 552   std::map<SDOperand, SDOperand>::iterator CGMI = CodeGenMap.find(N);
 553   if (CGMI != CodeGenMap.end()) {
 554     Result = CGMI->second;
 555 #ifndef NDEBUG
 556     DEBUG(std::cerr << std::string(Indent-2, ' '));
 557     DEBUG(std::cerr << "== ");
 558     DEBUG(Result.Val->dump(CurDAG));
 559     DEBUG(std::cerr << "\n");
 560     Indent -= 2;
 561 #endif
 562     return;
 563   }
 564
 565   switch (Opcode) {
 566     default: break;
 567     case X86ISD::GlobalBaseReg:
 568       Result = getGlobalBaseReg();
 569       return;
 570
 571     case ISD::ADD: {
 572       // Turn ADD X, c to MOV32ri X+c. This cannot be done with tblgen'd
 573       // code and is matched first so to prevent it from being turned into
 574       // LEA32r X+c.
 575       SDOperand N0 = N.getOperand(0);
 576       SDOperand N1 = N.getOperand(1);
 577       if (N.Val->getValueType(0) == MVT::i32 &&
 578           N0.getOpcode() == X86ISD::Wrapper &&
 579           N1.getOpcode() == ISD::Constant) {
 580         unsigned Offset = (unsigned)cast<ConstantSDNode>(N1)->getValue();
 581         SDOperand C(0, 0);
 582         // TODO: handle ExternalSymbolSDNode.
 583         if (GlobalAddressSDNode *G =
 584             dyn_cast<GlobalAddressSDNode>(N0.getOperand(0))) {
 585           C = CurDAG->getTargetGlobalAddress(G->getGlobal(), MVT::i32,
 586                                              G->getOffset() + Offset);
 587         } else if (ConstantPoolSDNode *CP =
 588                    dyn_cast<ConstantPoolSDNode>(N0.getOperand(0))) {
 589           C = CurDAG->getTargetConstantPool(CP->get(), MVT::i32,
 590                                             CP->getAlignment(),
 591                                             CP->getOffset()+Offset);
 592         }
 593
 594         if (C.Val) {
 595           if (N.Val->hasOneUse()) {
 596             Result = CurDAG->SelectNodeTo(N.Val, X86::MOV32ri, MVT::i32, C);
 597           } else {
 598             SDNode *ResNode = CurDAG->getTargetNode(X86::MOV32ri, MVT::i32, C);
 599             Result = CodeGenMap[N] = SDOperand(ResNode, 0);
 600           }
 601           return;
 602         }
 603       }
 604
 605       // Other cases are handled by auto-generated code.
 606       break;
 607     }
 608
 609     case ISD::MULHU:
 610     case ISD::MULHS: {
 611       if (Opcode == ISD::MULHU)
 612         switch (NVT) {
 613         default: assert(0 && "Unsupported VT!");
 614         case MVT::i8:  Opc = X86::MUL8r;  MOpc = X86::MUL8m;  break;
 615         case MVT::i16: Opc = X86::MUL16r; MOpc = X86::MUL16m; break;
 616         case MVT::i32: Opc = X86::MUL32r; MOpc = X86::MUL32m; break;
 617         }
 618       else
 619         switch (NVT) {
 620         default: assert(0 && "Unsupported VT!");
 621         case MVT::i8:  Opc = X86::IMUL8r;  MOpc = X86::IMUL8m;  break;
 622         case MVT::i16: Opc = X86::IMUL16r; MOpc = X86::IMUL16m; break;
 623         case MVT::i32: Opc = X86::IMUL32r; MOpc = X86::IMUL32m; break;
 624         }
 625
 626       unsigned LoReg, HiReg;
 627       switch (NVT) {
 628       default: assert(0 && "Unsupported VT!");
 629       case MVT::i8:  LoReg = X86::AL;  HiReg = X86::AH;  break;
 630       case MVT::i16: LoReg = X86::AX;  HiReg = X86::DX;  break;
 631       case MVT::i32: LoReg = X86::EAX; HiReg = X86::EDX; break;
 632       }
 633
 634       SDOperand N0 = Node->getOperand(0);
 635       SDOperand N1 = Node->getOperand(1);
 636
 637       bool foldedLoad = false;
 638       SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
 639       foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
 640       // MULHU and MULHS are commmutative
 641       if (!foldedLoad) {
 642         foldedLoad = TryFoldLoad(N, N0, Tmp0, Tmp1, Tmp2, Tmp3);
 643         if (foldedLoad) {
 644           N0 = Node->getOperand(1);
 645           N1 = Node->getOperand(0);
 646         }
 647       }
 648
 649       SDOperand Chain;
 650       if (foldedLoad)
 651         Select(Chain, N1.getOperand(0));
 652       else
 653         Chain = CurDAG->getEntryNode();
 654
 655       SDOperand InFlag(0, 0);
 656       Select(N0, N0);
 657       Chain  = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
 658                                     N0, InFlag);
 659       InFlag = Chain.getValue(1);
 660
 661       if (foldedLoad) {
 662         Select(Tmp0, Tmp0);
 663         Select(Tmp1, Tmp1);
 664         Select(Tmp2, Tmp2);
 665         Select(Tmp3, Tmp3);
 666         SDNode *CNode =
 667           CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
 668                                 Tmp2, Tmp3, Chain, InFlag);
 669         Chain  = SDOperand(CNode, 0);
 670         InFlag = SDOperand(CNode, 1);
 671       } else {
 672         Select(N1, N1);
 673         InFlag =
 674           SDOperand(CurDAG->getTargetNode(Opc, MVT::Flag, N1, InFlag), 0);
 675       }
 676
 677       Result = CurDAG->getCopyFromReg(Chain, HiReg, NVT, InFlag);
 678       CodeGenMap[N.getValue(0)] = Result;
 679       if (foldedLoad) {
 680         CodeGenMap[N1.getValue(1)] = Result.getValue(1);
 681         AddHandleReplacement(N1.Val, 1, Result.Val, 1);
 682       }
 683
 684 #ifndef NDEBUG
 685       DEBUG(std::cerr << std::string(Indent-2, ' '));
 686       DEBUG(std::cerr << "== ");
 687       DEBUG(Result.Val->dump(CurDAG));
 688       DEBUG(std::cerr << "\n");
 689       Indent -= 2;
 690 #endif
 691       return;
 692     }
 693
 694     case ISD::SDIV:
 695     case ISD::UDIV:
 696     case ISD::SREM:
 697     case ISD::UREM: {
 698       bool isSigned = Opcode == ISD::SDIV || Opcode == ISD::SREM;
 699       bool isDiv    = Opcode == ISD::SDIV || Opcode == ISD::UDIV;
 700       if (!isSigned)
 701         switch (NVT) {
 702         default: assert(0 && "Unsupported VT!");
 703         case MVT::i8:  Opc = X86::DIV8r;  MOpc = X86::DIV8m;  break;
 704         case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break;
 705         case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break;
 706         }
 707       else
 708         switch (NVT) {
 709         default: assert(0 && "Unsupported VT!");
 710         case MVT::i8:  Opc = X86::IDIV8r;  MOpc = X86::IDIV8m;  break;
 711         case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break;
 712         case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break;
 713         }
 714
 715       unsigned LoReg, HiReg;
 716       unsigned ClrOpcode, SExtOpcode;
 717       switch (NVT) {
 718       default: assert(0 && "Unsupported VT!");
 719       case MVT::i8:
 720         LoReg = X86::AL;  HiReg = X86::AH;
 721         ClrOpcode  = X86::MOV8ri;
 722         SExtOpcode = X86::CBW;
 723         break;
 724       case MVT::i16:
 725         LoReg = X86::AX;  HiReg = X86::DX;
 726         ClrOpcode  = X86::MOV16ri;
 727         SExtOpcode = X86::CWD;
 728         break;
 729       case MVT::i32:
 730         LoReg = X86::EAX; HiReg = X86::EDX;
 731         ClrOpcode  = X86::MOV32ri;
 732         SExtOpcode = X86::CDQ;
 733         break;
 734       }
 735
 736       SDOperand N0 = Node->getOperand(0);
 737       SDOperand N1 = Node->getOperand(1);
 738
 739       bool foldedLoad = false;
 740       SDOperand Tmp0, Tmp1, Tmp2, Tmp3;
 741       foldedLoad = TryFoldLoad(N, N1, Tmp0, Tmp1, Tmp2, Tmp3);
 742       SDOperand Chain;
 743       if (foldedLoad)
 744         Select(Chain, N1.getOperand(0));
 745       else
 746         Chain = CurDAG->getEntryNode();
 747
 748       SDOperand InFlag(0, 0);
 749       Select(N0, N0);
 750       Chain  = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(LoReg, NVT),
 751                                     N0, InFlag);
 752       InFlag = Chain.getValue(1);
 753
 754       if (isSigned) {
 755         // Sign extend the low part into the high part.
 756         InFlag =
 757           SDOperand(CurDAG->getTargetNode(SExtOpcode, MVT::Flag, InFlag), 0);
 758       } else {
 759         // Zero out the high part, effectively zero extending the input.
 760         SDOperand ClrNode =
 761           SDOperand(CurDAG->getTargetNode(ClrOpcode, NVT,
 762                                          CurDAG->getTargetConstant(0, NVT)), 0);
 763         Chain  = CurDAG->getCopyToReg(Chain, CurDAG->getRegister(HiReg, NVT),
 764                                       ClrNode, InFlag);
 765         InFlag = Chain.getValue(1);
 766       }
 767
 768       if (foldedLoad) {
 769         Select(Tmp0, Tmp0);
 770         Select(Tmp1, Tmp1);
 771         Select(Tmp2, Tmp2);
 772         Select(Tmp3, Tmp3);
 773         SDNode *CNode =
 774           CurDAG->getTargetNode(MOpc, MVT::Other, MVT::Flag, Tmp0, Tmp1,
 775                                 Tmp2, Tmp3, Chain, InFlag);
 776         Chain  = SDOperand(CNode, 0);
 777         InFlag = SDOperand(CNode, 1);
 778       } else {
 779         Select(N1, N1);
 780         InFlag =
 781           SDOperand(CurDAG->getTargetNode(Opc, MVT::Flag, N1, InFlag), 0);
 782       }
 783
 784       Result = CurDAG->getCopyFromReg(Chain, isDiv ? LoReg : HiReg,
 785                                       NVT, InFlag);
 786       CodeGenMap[N.getValue(0)] = Result;
 787       if (foldedLoad) {
 788         CodeGenMap[N1.getValue(1)] = Result.getValue(1);
 789         AddHandleReplacement(N1.Val, 1, Result.Val, 1);
 790       }
 791
 792 #ifndef NDEBUG
 793       DEBUG(std::cerr << std::string(Indent-2, ' '));
 794       DEBUG(std::cerr << "== ");
 795       DEBUG(Result.Val->dump(CurDAG));
 796       DEBUG(std::cerr << "\n");
 797       Indent -= 2;
 798 #endif
 799       return;
 800     }
 801
 802     case ISD::TRUNCATE: {
 803       if (NVT == MVT::i8) {
 804         unsigned Opc2;
 805         MVT::ValueType VT;
 806         switch (Node->getOperand(0).getValueType()) {
 807         default: assert(0 && "Unknown truncate!");
 808         case MVT::i16:
 809           Opc = X86::MOV16to16_;
 810           VT = MVT::i16;
 811           Opc2 = X86::TRUNC_GR16_GR8;
 812           break;
 813         case MVT::i32:
 814           Opc = X86::MOV32to32_;
 815           VT = MVT::i32;
 816           Opc2 = X86::TRUNC_GR32_GR8;
 817           break;
 818         }
 819
 820         SDOperand Tmp0, Tmp1;
 821         Select(Tmp0, Node->getOperand(0));
 822         Tmp1 = SDOperand(CurDAG->getTargetNode(Opc, VT, Tmp0), 0);
 823         Result = CodeGenMap[N] =
 824           SDOperand(CurDAG->getTargetNode(Opc2, NVT, Tmp1), 0);
 825
 826 #ifndef NDEBUG
 827         DEBUG(std::cerr << std::string(Indent-2, ' '));
 828         DEBUG(std::cerr << "== ");
 829         DEBUG(Result.Val->dump(CurDAG));
 830         DEBUG(std::cerr << "\n");
 831         Indent -= 2;
 832 #endif
 833         return;
 834       }
 835
 836       break;
 837     }
 838   }
 839
 840   SelectCode(Result, N);
 841 #ifndef NDEBUG
 842   DEBUG(std::cerr << std::string(Indent-2, ' '));
 843   DEBUG(std::cerr << "=> ");
 844   DEBUG(Result.Val->dump(CurDAG));
 845   DEBUG(std::cerr << "\n");
 846   Indent -= 2;
 847 #endif
 848 }
 849
 850 /// createX86ISelDag - This pass converts a legalized DAG into a
 851 /// X86-specific DAG, ready for instruction scheduling.
 852 ///
 853 FunctionPass *llvm::createX86ISelDag(X86TargetMachine &TM) {
 854   return new X86DAGToDAGISel(TM);
 855 }