X86FloatingPoint.cpp revision 36b56886974eae4f9c5ebc96befd3e7bfe5de338
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//===-- X86FloatingPoint.cpp - Floating point Reg -> Stack converter ------===//
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//                     The LLVM Compiler Infrastructure
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This file is distributed under the University of Illinois Open Source
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// License. See LICENSE.TXT for details.
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//===----------------------------------------------------------------------===//
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This file defines the pass which converts floating point instructions from
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// pseudo registers into register stack instructions.  This pass uses live
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// variable information to indicate where the FPn registers are used and their
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// lifetimes.
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The x87 hardware tracks liveness of the stack registers, so it is necessary
16ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch// to implement exact liveness tracking between basic blocks. The CFG edges are
17eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch// partitioned into bundles where the same FP registers must be live in
18d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)// identical stack positions. Instructions are inserted at the end of each basic
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// block to rearrange the live registers to match the outgoing bundle.
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// This approach avoids splitting critical edges at the potential cost of more
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// live register shuffling instructions when critical edges are present.
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//===----------------------------------------------------------------------===//
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define DEBUG_TYPE "x86-codegen"
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "X86.h"
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "X86InstrInfo.h"
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/ADT/DepthFirstIterator.h"
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/ADT/STLExtras.h"
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/ADT/SmallPtrSet.h"
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/ADT/SmallVector.h"
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/ADT/Statistic.h"
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/CodeGen/EdgeBundles.h"
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/CodeGen/MachineFunctionPass.h"
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/CodeGen/MachineInstrBuilder.h"
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/CodeGen/MachineRegisterInfo.h"
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/CodeGen/Passes.h"
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/IR/InlineAsm.h"
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/Support/Debug.h"
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/Support/ErrorHandling.h"
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/Support/raw_ostream.h"
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/Target/TargetInstrInfo.h"
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "llvm/Target/TargetMachine.h"
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <algorithm>
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using namespace llvm;
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
48c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)STATISTIC(NumFXCH, "Number of fxch instructions inserted");
492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)STATISTIC(NumFP  , "Number of floating point instructions");
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  struct FPS : public MachineFunctionPass {
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    static char ID;
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    FPS() : MachineFunctionPass(ID) {
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      initializeEdgeBundlesPass(*PassRegistry::getPassRegistry());
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // This is really only to keep valgrind quiet.
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // The logic in isLive() is too much for it.
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      memset(Stack, 0, sizeof(Stack));
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      memset(RegMap, 0, sizeof(RegMap));
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void getAnalysisUsage(AnalysisUsage &AU) const override {
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      AU.setPreservesCFG();
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      AU.addRequired<EdgeBundles>();
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      AU.addPreservedID(MachineLoopInfoID);
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      AU.addPreservedID(MachineDominatorsID);
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      MachineFunctionPass::getAnalysisUsage(AU);
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool runOnMachineFunction(MachineFunction &MF) override;
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const char *getPassName() const override { return "X86 FP Stackifier"; }
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  private:
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const TargetInstrInfo *TII; // Machine instruction info.
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Two CFG edges are related if they leave the same block, or enter the same
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // block. The transitive closure of an edge under this relation is a
792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // LiveBundle. It represents a set of CFG edges where the live FP stack
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // registers must be allocated identically in the x87 stack.
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // A LiveBundle is usually all the edges leaving a block, or all the edges
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // entering a block, but it can contain more edges if critical edges are
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // present.
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // The set of live FP registers in a LiveBundle is calculated by bundleCFG,
872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // but the exact mapping of FP registers to stack slots is fixed later.
882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    struct LiveBundle {
892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Bit mask of live FP registers. Bit 0 = FP0, bit 1 = FP1, &c.
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned Mask;
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Number of pre-assigned live registers in FixStack. This is 0 when the
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // stack order has not yet been fixed.
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned FixCount;
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Assigned stack order for live-in registers.
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // FixStack[i] == getStackEntry(i) for all i < FixCount.
985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned char FixStack[8];
995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      LiveBundle() : Mask(0), FixCount(0) {}
1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Have the live registers been assigned a stack order yet?
1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      bool isFixed() const { return !Mask || FixCount; }
1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    };
1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Numbered LiveBundle structs. LiveBundles[0] is used for all CFG edges
1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // with no live FP registers.
1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    SmallVector<LiveBundle, 8> LiveBundles;
1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // The edge bundle analysis provides indices into the LiveBundles vector.
1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    EdgeBundles *Bundles;
1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Return a bitmask of FP registers in block's live-in list.
1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    static unsigned calcLiveInMask(MachineBasicBlock *MBB) {
1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned Mask = 0;
1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      for (MachineBasicBlock::livein_iterator I = MBB->livein_begin(),
1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           E = MBB->livein_end(); I != E; ++I) {
1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        unsigned Reg = *I;
1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (Reg < X86::FP0 || Reg > X86::FP6)
1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          continue;
1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        Mask |= 1 << (Reg - X86::FP0);
1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return Mask;
1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Partition all the CFG edges into LiveBundles.
1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void bundleCFG(MachineFunction &MF);
1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    MachineBasicBlock *MBB;     // Current basic block
1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // The hardware keeps track of how many FP registers are live, so we have
1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // to model that exactly. Usually, each live register corresponds to an
1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // FP<n> register, but when dealing with calls, returns, and inline
1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // assembly, it is sometimes necessary to have live scratch registers.
1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned Stack[8];          // FP<n> Registers in each stack slot...
1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    unsigned StackTop;          // The current top of the FP stack.
1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    enum {
139c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      NumFPRegs = 16            // Including scratch pseudo-registers.
140424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    };
141c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // For each live FP<n> register, point to its Stack[] entry.
1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // The first entries correspond to FP0-FP6, the rest are scratch registers
1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // used when we need slightly different live registers than what the
1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // register allocator thinks.
1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned RegMap[NumFPRegs];
1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Pending fixed registers - Inline assembly needs FP registers to appear
1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // in fixed stack slot positions. This is handled by copying FP registers
1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // to ST registers before the instruction, and copying back after the
1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // instruction.
1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //
1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // This is modeled with pending ST registers. NumPendingSTs is the number
1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // of ST registers (ST0-STn) we are tracking. PendingST[n] points to an FP
1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // register that holds the ST value. The ST registers are not moved into
1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // place until immediately before the instruction that needs them.
1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //
1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // It can happen that we need an ST register to be live when no FP register
1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // holds the value:
1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //
1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //   %ST0 = COPY %FP4<kill>
1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    //
1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // When that happens, we allocate a scratch FP register to hold the ST
1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // value. That means every register in PendingST must be live.
1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned NumPendingSTs;
1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned char PendingST[8];
1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Set up our stack model to match the incoming registers to MBB.
1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void setupBlockStack();
1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Shuffle live registers to match the expectations of successor blocks.
1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void finishBlockStack();
1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void dumpStack() const {
1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      dbgs() << "Stack contents:";
1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (unsigned i = 0; i != StackTop; ++i) {
1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        dbgs() << " FP" << Stack[i];
1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        assert(RegMap[Stack[i]] == i && "Stack[] doesn't match RegMap[]!");
1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (unsigned i = 0; i != NumPendingSTs; ++i)
1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        dbgs() << ", ST" << i << " in FP" << unsigned(PendingST[i]);
1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      dbgs() << "\n";
1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// getSlot - Return the stack slot number a particular register number is
1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// in.
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned getSlot(unsigned RegNo) const {
1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      assert(RegNo < NumFPRegs && "Regno out of range!");
1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return RegMap[RegNo];
1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// isLive - Is RegNo currently live in the stack?
1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool isLive(unsigned RegNo) const {
1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned Slot = getSlot(RegNo);
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return Slot < StackTop && Stack[Slot] == RegNo;
1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// getScratchReg - Return an FP register that is not currently in use.
2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned getScratchReg() const {
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (int i = NumFPRegs - 1; i >= 8; --i)
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (!isLive(i))
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          return i;
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      llvm_unreachable("Ran out of scratch FP registers");
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// isScratchReg - Returns trus if RegNo is a scratch FP register.
2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    static bool isScratchReg(unsigned RegNo) {
2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return RegNo > 8 && RegNo < NumFPRegs;
2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// getStackEntry - Return the X86::FP<n> register in register ST(i).
2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned getStackEntry(unsigned STi) const {
2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (STi >= StackTop)
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        report_fatal_error("Access past stack top!");
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return Stack[StackTop-1-STi];
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// getSTReg - Return the X86::ST(i) register which contains the specified
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// FP<RegNo> register.
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned getSTReg(unsigned RegNo) const {
2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return StackTop - 1 - getSlot(RegNo) + X86::ST0;
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // pushReg - Push the specified FP<n> register onto the stack.
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void pushReg(unsigned Reg) {
2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      assert(Reg < NumFPRegs && "Register number out of range!");
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (StackTop >= 8)
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        report_fatal_error("Stack overflow!");
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      Stack[StackTop] = Reg;
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      RegMap[Reg] = StackTop++;
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool isAtTop(unsigned RegNo) const { return getSlot(RegNo) == StackTop-1; }
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void moveToTop(unsigned RegNo, MachineBasicBlock::iterator I) {
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DebugLoc dl = I == MBB->end() ? DebugLoc() : I->getDebugLoc();
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (isAtTop(RegNo)) return;
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      unsigned STReg = getSTReg(RegNo);
2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned RegOnTop = getStackEntry(0);
2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Swap the slots the regs are in.
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      std::swap(RegMap[RegNo], RegMap[RegOnTop]);
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Swap stack slot contents.
2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (RegMap[RegOnTop] >= StackTop)
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        report_fatal_error("Access past stack top!");
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      std::swap(Stack[RegMap[RegOnTop]], Stack[StackTop-1]);
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Emit an fxch to update the runtime processors version of the state.
2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      BuildMI(*MBB, I, dl, TII->get(X86::XCH_F)).addReg(STReg);
2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      ++NumFXCH;
2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void duplicateToTop(unsigned RegNo, unsigned AsReg, MachineInstr *I) {
2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DebugLoc dl = I == MBB->end() ? DebugLoc() : I->getDebugLoc();
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned STReg = getSTReg(RegNo);
2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      pushReg(AsReg);   // New register on top of stack
2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      BuildMI(*MBB, I, dl, TII->get(X86::LD_Frr)).addReg(STReg);
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// duplicatePendingSTBeforeKill - The instruction at I is about to kill
2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// RegNo. If any PendingST registers still need the RegNo value, duplicate
2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// them to new scratch registers.
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void duplicatePendingSTBeforeKill(unsigned RegNo, MachineInstr *I) {
2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (unsigned i = 0; i != NumPendingSTs; ++i) {
2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (PendingST[i] != RegNo)
2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)          continue;
2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        unsigned SR = getScratchReg();
2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        DEBUG(dbgs() << "Duplicating pending ST" << i
2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                     << " in FP" << RegNo << " to FP" << SR << '\n');
2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        duplicateToTop(RegNo, SR, I);
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        PendingST[i] = SR;
2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// popStackAfter - Pop the current value off of the top of the FP stack
2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// after the specified instruction.
2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void popStackAfter(MachineBasicBlock::iterator &I);
2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// freeStackSlotAfter - Free the specified register from the register
2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    /// stack, so that it is no longer in a register.  If the register is
2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// currently at the top of the stack, we just pop the current instruction,
2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// otherwise we store the current top-of-stack into the specified slot,
2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// then pop the top of stack.
2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void freeStackSlotAfter(MachineBasicBlock::iterator &I, unsigned Reg);
2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// freeStackSlotBefore - Just the pop, no folding. Return the inserted
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// instruction.
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    MachineBasicBlock::iterator
2947d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)    freeStackSlotBefore(MachineBasicBlock::iterator I, unsigned FPRegNo);
2957d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// Adjust the live registers to be the set in Mask.
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void adjustLiveRegs(unsigned Mask, MachineBasicBlock::iterator I);
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// Shuffle the top FixCount stack entries such that FP reg FixStack[0] is
3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    /// st(0), FP reg FixStack[1] is st(1) etc.
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void shuffleStackTop(const unsigned char *FixStack, unsigned FixCount,
3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                         MachineBasicBlock::iterator I);
3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool processBasicBlock(MachineFunction &MF, MachineBasicBlock &MBB);
3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleZeroArgFP(MachineBasicBlock::iterator &I);
3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleOneArgFP(MachineBasicBlock::iterator &I);
3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleOneArgFPRW(MachineBasicBlock::iterator &I);
3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleTwoArgFP(MachineBasicBlock::iterator &I);
3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleCompareFP(MachineBasicBlock::iterator &I);
3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleCondMovFP(MachineBasicBlock::iterator &I);
3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void handleSpecialFP(MachineBasicBlock::iterator &I);
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Check if a COPY instruction is using FP registers.
3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    static bool isFPCopy(MachineInstr *MI) {
3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned DstReg = MI->getOperand(0).getReg();
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned SrcReg = MI->getOperand(1).getReg();
3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return X86::RFP80RegClass.contains(DstReg) ||
3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        X86::RFP80RegClass.contains(SrcReg);
3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  };
3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  char FPS::ID = 0;
3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)FunctionPass *llvm::createX86FloatingPointStackifierPass() { return new FPS(); }
3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// getFPReg - Return the X86::FPx register number for the specified operand.
3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// For example, this returns 3 for X86::FP3.
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static unsigned getFPReg(const MachineOperand &MO) {
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  assert(MO.isReg() && "Expected an FP register!");
3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned Reg = MO.getReg();
3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  assert(Reg >= X86::FP0 && Reg <= X86::FP6 && "Expected FP register!");
3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return Reg - X86::FP0;
3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// runOnMachineFunction - Loop over all of the basic blocks, transforming FP
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// register references into FP stack references.
3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)///
3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool FPS::runOnMachineFunction(MachineFunction &MF) {
3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // We only need to run this pass if there are any FP registers used in this
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // function.  If it is all integer, there is nothing for us to do!
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool FPIsUsed = false;
3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  assert(X86::FP6 == X86::FP0+6 && "Register enums aren't sorted right!");
3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (unsigned i = 0; i <= 6; ++i)
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (MF.getRegInfo().isPhysRegUsed(X86::FP0+i)) {
3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FPIsUsed = true;
3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      break;
3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Early exit.
3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!FPIsUsed) return false;
3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  Bundles = &getAnalysis<EdgeBundles>();
3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  TII = MF.getTarget().getInstrInfo();
3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Prepare cross-MBB liveness.
3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bundleCFG(MF);
3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  StackTop = 0;
3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Process the function in depth first order so that we process at least one
3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // of the predecessors for every reachable block in the function.
3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  SmallPtrSet<MachineBasicBlock*, 8> Processed;
3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  MachineBasicBlock *Entry = MF.begin();
3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool Changed = false;
3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (df_ext_iterator<MachineBasicBlock*, SmallPtrSet<MachineBasicBlock*, 8> >
3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         I = df_ext_begin(Entry, Processed), E = df_ext_end(Entry, Processed);
3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       I != E; ++I)
3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    Changed |= processBasicBlock(MF, **I);
3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Process any unreachable blocks in arbitrary order now.
3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (MF.size() != Processed.size())
3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (MachineFunction::iterator BB = MF.begin(), E = MF.end(); BB != E; ++BB)
3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (Processed.insert(BB))
3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        Changed |= processBasicBlock(MF, *BB);
3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LiveBundles.clear();
3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return Changed;
3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// bundleCFG - Scan all the basic blocks to determine consistent live-in and
3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// live-out sets for the FP registers. Consistent means that the set of
3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)/// registers live-out from a block is identical to the live-in set of all
3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)/// successors. This is not enforced by the normal live-in lists since
3892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)/// registers may be implicitly defined, or not used by all successors.
3902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void FPS::bundleCFG(MachineFunction &MF) {
3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  assert(LiveBundles.empty() && "Stale data in LiveBundles");
3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  LiveBundles.resize(Bundles->getNumBundles());
3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Gather the actual live-in masks for all MBBs.
3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (MachineFunction::iterator I = MF.begin(), E = MF.end(); I != E; ++I) {
3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    MachineBasicBlock *MBB = I;
3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const unsigned Mask = calcLiveInMask(MBB);
3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!Mask)
3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Update MBB ingoing bundle mask.
4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    LiveBundles[Bundles->getBundle(MBB->getNumber(), false)].Mask |= Mask;
4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// processBasicBlock - Loop over all of the instructions in the basic block,
4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// transforming FP instructions into their stack form.
4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)///
4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool FPS::processBasicBlock(MachineFunction &MF, MachineBasicBlock &BB) {
4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  bool Changed = false;
4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  MBB = &BB;
4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  NumPendingSTs = 0;
4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  setupBlockStack();
4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (MachineBasicBlock::iterator I = BB.begin(); I != BB.end(); ++I) {
4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    MachineInstr *MI = I;
4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    uint64_t Flags = MI->getDesc().TSFlags;
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    unsigned FPInstClass = Flags & X86II::FPTypeMask;
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (MI->isInlineAsm())
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FPInstClass = X86II::SpecialFP;
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (MI->isCopy() && isFPCopy(MI))
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FPInstClass = X86II::SpecialFP;
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (MI->isImplicitDef() &&
4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        X86::RFP80RegClass.contains(MI->getOperand(0).getReg()))
4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      FPInstClass = X86II::SpecialFP;
4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (FPInstClass == X86II::NotFP)
4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      continue;  // Efficiently ignore non-fp insts!
4322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    MachineInstr *PrevMI = 0;
4342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (I != BB.begin())
4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      PrevMI = std::prev(I);
4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ++NumFP;  // Keep track of # of pseudo instrs
4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(dbgs() << "\nFPInst:\t" << *MI);
4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Get dead variables list now because the MI pointer may be deleted as part
4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // of processing!
4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    SmallVector<unsigned, 8> DeadRegs;
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const MachineOperand &MO = MI->getOperand(i);
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (MO.isReg() && MO.isDead())
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        DeadRegs.push_back(MO.getReg());
4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
448424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)
4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    switch (FPInstClass) {
4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case X86II::ZeroArgFP:  handleZeroArgFP(I); break;
4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case X86II::OneArgFP:   handleOneArgFP(I);  break;  // fstp ST(0)
4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case X86II::OneArgFPRW: handleOneArgFPRW(I); break; // ST(0) = fsqrt(ST(0))
4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case X86II::TwoArgFP:   handleTwoArgFP(I);  break;
4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case X86II::CompareFP:  handleCompareFP(I); break;
4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case X86II::CondMovFP:  handleCondMovFP(I); break;
456424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    case X86II::SpecialFP:  handleSpecialFP(I); break;
4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default: llvm_unreachable("Unknown FP Type!");
4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Check to see if any of the values defined by this instruction are dead
461424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    // after definition.  If so, pop them.
462424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    for (unsigned i = 0, e = DeadRegs.size(); i != e; ++i) {
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      unsigned Reg = DeadRegs[i];
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (Reg >= X86::FP0 && Reg <= X86::FP6) {
465424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)        DEBUG(dbgs() << "Register FP#" << Reg-X86::FP0 << " is dead!\n");
466424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)        freeStackSlotAfter(I, Reg-X86::FP0);
467424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)      }
468424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    }
4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Print out all of the instructions expanded to if -debug
4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(
4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      MachineBasicBlock::iterator PrevI(PrevMI);
4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (I == PrevI) {
4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        dbgs() << "Just deleted pseudo instruction\n";
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      } else {
4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        MachineBasicBlock::iterator Start = I;
4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Rewind to first instruction newly inserted.
4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        while (Start != BB.begin() && std::prev(Start) != PrevI) --Start;
4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        dbgs() << "Inserted instructions:\n\t";
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        Start->print(dbgs(), &MF.getTarget());
4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        while (++Start != std::next(I)) {}
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      dumpStack();
4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    );
4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    (void)PrevMI;
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    Changed = true;
4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  finishBlockStack();
4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return Changed;
4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// setupBlockStack - Use the live bundles to set up our model of the stack
4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// to match predecessors' live out stack.
4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void FPS::setupBlockStack() {
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DEBUG(dbgs() << "\nSetting up live-ins for BB#" << MBB->getNumber()
4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               << " derived from " << MBB->getName() << ".\n");
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  StackTop = 0;
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Get the live-in bundle for MBB.
5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const LiveBundle &Bundle =
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    LiveBundles[Bundles->getBundle(MBB->getNumber(), false)];
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!Bundle.Mask) {
5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(dbgs() << "Block has no FP live-ins.\n");
5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Depth-first iteration should ensure that we always have an assigned stack.
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  assert(Bundle.isFixed() && "Reached block before any predecessors");
5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  // Push the fixed live-in registers.
5142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  for (unsigned i = Bundle.FixCount; i > 0; --i) {
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    MBB->addLiveIn(X86::ST0+i-1);
5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(dbgs() << "Live-in st(" << (i-1) << "): %FP"
5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                 << unsigned(Bundle.FixStack[i-1]) << '\n');
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    pushReg(Bundle.FixStack[i-1]);
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Kill off unwanted live-ins. This can happen with a critical edge.
5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // FIXME: We could keep these live registers around as zombies. They may need
5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // to be revived at the end of a short block. It might save a few instrs.
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  adjustLiveRegs(calcLiveInMask(MBB), MBB->begin());
5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DEBUG(MBB->dump());
5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// finishBlockStack - Revive live-outs that are implicitly defined out of
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// MBB. Shuffle live registers to match the expected fixed stack of any
5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// predecessors, and ensure that all predecessors are expecting the same
5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/// stack.
5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void FPS::finishBlockStack() {
5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // The RET handling below takes care of return blocks for us.
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (MBB->succ_empty())
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DEBUG(dbgs() << "Setting up live-outs for BB#" << MBB->getNumber()
5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               << " derived from " << MBB->getName() << ".\n");
5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Get MBB's live-out bundle.
5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned BundleIdx = Bundles->getBundle(MBB->getNumber(), true);
5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LiveBundle &Bundle = LiveBundles[BundleIdx];
5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // We may need to kill and define some registers to match successors.
5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // FIXME: This can probably be combined with the shuffle below.
5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  MachineBasicBlock::iterator Term = MBB->getFirstTerminator();
5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  adjustLiveRegs(Bundle.Mask, Term);
5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!Bundle.Mask) {
5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(dbgs() << "No live-outs.\n");
5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Has the stack order been fixed yet?
5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DEBUG(dbgs() << "LB#" << BundleIdx << ": ");
5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (Bundle.isFixed()) {
5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(dbgs() << "Shuffling stack to match.\n");
5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    shuffleStackTop(Bundle.FixStack, Bundle.FixCount, Term);
5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Not fixed yet, we get to choose.
5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DEBUG(dbgs() << "Fixing stack order now.\n");
5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    Bundle.FixCount = StackTop;
563424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    for (unsigned i = 0; i < StackTop; ++i)
564424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)      Bundle.FixStack[i] = getStackEntry(i);
565424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)  }
5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//===----------------------------------------------------------------------===//
5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Efficient Lookup Table Support
5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//===----------------------------------------------------------------------===//
5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  struct TableEntry {
5752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    uint16_t from;
5762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    uint16_t to;
5772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    bool operator<(const TableEntry &TE) const { return from < TE.from; }
578c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    friend bool operator<(const TableEntry &TE, unsigned V) {
579c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)      return TE.from < V;
580c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    }
581424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    friend bool LLVM_ATTRIBUTE_UNUSED operator<(unsigned V,
582424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)                                                const TableEntry &TE) {
583424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)      return V < TE.from;
584424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles)    }
5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  };
5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef NDEBUG
5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static bool TableIsSorted(const TableEntry *Table, unsigned NumEntries) {
5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (unsigned i = 0; i != NumEntries-1; ++i)
591    if (!(Table[i] < Table[i+1])) return false;
592  return true;
593}
594#endif
595
596static int Lookup(const TableEntry *Table, unsigned N, unsigned Opcode) {
597  const TableEntry *I = std::lower_bound(Table, Table+N, Opcode);
598  if (I != Table+N && I->from == Opcode)
599    return I->to;
600  return -1;
601}
602
603#ifdef NDEBUG
604#define ASSERT_SORTED(TABLE)
605#else
606#define ASSERT_SORTED(TABLE)                                              \
607  { static bool TABLE##Checked = false;                                   \
608    if (!TABLE##Checked) {                                                \
609       assert(TableIsSorted(TABLE, array_lengthof(TABLE)) &&              \
610              "All lookup tables must be sorted for efficient access!");  \
611       TABLE##Checked = true;                                             \
612    }                                                                     \
613  }
614#endif
615
616//===----------------------------------------------------------------------===//
617// Register File -> Register Stack Mapping Methods
618//===----------------------------------------------------------------------===//
619
620// OpcodeTable - Sorted map of register instructions to their stack version.
621// The first element is an register file pseudo instruction, the second is the
622// concrete X86 instruction which uses the register stack.
623//
624static const TableEntry OpcodeTable[] = {
625  { X86::ABS_Fp32     , X86::ABS_F     },
626  { X86::ABS_Fp64     , X86::ABS_F     },
627  { X86::ABS_Fp80     , X86::ABS_F     },
628  { X86::ADD_Fp32m    , X86::ADD_F32m  },
629  { X86::ADD_Fp64m    , X86::ADD_F64m  },
630  { X86::ADD_Fp64m32  , X86::ADD_F32m  },
631  { X86::ADD_Fp80m32  , X86::ADD_F32m  },
632  { X86::ADD_Fp80m64  , X86::ADD_F64m  },
633  { X86::ADD_FpI16m32 , X86::ADD_FI16m },
634  { X86::ADD_FpI16m64 , X86::ADD_FI16m },
635  { X86::ADD_FpI16m80 , X86::ADD_FI16m },
636  { X86::ADD_FpI32m32 , X86::ADD_FI32m },
637  { X86::ADD_FpI32m64 , X86::ADD_FI32m },
638  { X86::ADD_FpI32m80 , X86::ADD_FI32m },
639  { X86::CHS_Fp32     , X86::CHS_F     },
640  { X86::CHS_Fp64     , X86::CHS_F     },
641  { X86::CHS_Fp80     , X86::CHS_F     },
642  { X86::CMOVBE_Fp32  , X86::CMOVBE_F  },
643  { X86::CMOVBE_Fp64  , X86::CMOVBE_F  },
644  { X86::CMOVBE_Fp80  , X86::CMOVBE_F  },
645  { X86::CMOVB_Fp32   , X86::CMOVB_F   },
646  { X86::CMOVB_Fp64   , X86::CMOVB_F  },
647  { X86::CMOVB_Fp80   , X86::CMOVB_F  },
648  { X86::CMOVE_Fp32   , X86::CMOVE_F  },
649  { X86::CMOVE_Fp64   , X86::CMOVE_F   },
650  { X86::CMOVE_Fp80   , X86::CMOVE_F   },
651  { X86::CMOVNBE_Fp32 , X86::CMOVNBE_F },
652  { X86::CMOVNBE_Fp64 , X86::CMOVNBE_F },
653  { X86::CMOVNBE_Fp80 , X86::CMOVNBE_F },
654  { X86::CMOVNB_Fp32  , X86::CMOVNB_F  },
655  { X86::CMOVNB_Fp64  , X86::CMOVNB_F  },
656  { X86::CMOVNB_Fp80  , X86::CMOVNB_F  },
657  { X86::CMOVNE_Fp32  , X86::CMOVNE_F  },
658  { X86::CMOVNE_Fp64  , X86::CMOVNE_F  },
659  { X86::CMOVNE_Fp80  , X86::CMOVNE_F  },
660  { X86::CMOVNP_Fp32  , X86::CMOVNP_F  },
661  { X86::CMOVNP_Fp64  , X86::CMOVNP_F  },
662  { X86::CMOVNP_Fp80  , X86::CMOVNP_F  },
663  { X86::CMOVP_Fp32   , X86::CMOVP_F   },
664  { X86::CMOVP_Fp64   , X86::CMOVP_F   },
665  { X86::CMOVP_Fp80   , X86::CMOVP_F   },
666  { X86::COS_Fp32     , X86::COS_F     },
667  { X86::COS_Fp64     , X86::COS_F     },
668  { X86::COS_Fp80     , X86::COS_F     },
669  { X86::DIVR_Fp32m   , X86::DIVR_F32m },
670  { X86::DIVR_Fp64m   , X86::DIVR_F64m },
671  { X86::DIVR_Fp64m32 , X86::DIVR_F32m },
672  { X86::DIVR_Fp80m32 , X86::DIVR_F32m },
673  { X86::DIVR_Fp80m64 , X86::DIVR_F64m },
674  { X86::DIVR_FpI16m32, X86::DIVR_FI16m},
675  { X86::DIVR_FpI16m64, X86::DIVR_FI16m},
676  { X86::DIVR_FpI16m80, X86::DIVR_FI16m},
677  { X86::DIVR_FpI32m32, X86::DIVR_FI32m},
678  { X86::DIVR_FpI32m64, X86::DIVR_FI32m},
679  { X86::DIVR_FpI32m80, X86::DIVR_FI32m},
680  { X86::DIV_Fp32m    , X86::DIV_F32m  },
681  { X86::DIV_Fp64m    , X86::DIV_F64m  },
682  { X86::DIV_Fp64m32  , X86::DIV_F32m  },
683  { X86::DIV_Fp80m32  , X86::DIV_F32m  },
684  { X86::DIV_Fp80m64  , X86::DIV_F64m  },
685  { X86::DIV_FpI16m32 , X86::DIV_FI16m },
686  { X86::DIV_FpI16m64 , X86::DIV_FI16m },
687  { X86::DIV_FpI16m80 , X86::DIV_FI16m },
688  { X86::DIV_FpI32m32 , X86::DIV_FI32m },
689  { X86::DIV_FpI32m64 , X86::DIV_FI32m },
690  { X86::DIV_FpI32m80 , X86::DIV_FI32m },
691  { X86::ILD_Fp16m32  , X86::ILD_F16m  },
692  { X86::ILD_Fp16m64  , X86::ILD_F16m  },
693  { X86::ILD_Fp16m80  , X86::ILD_F16m  },
694  { X86::ILD_Fp32m32  , X86::ILD_F32m  },
695  { X86::ILD_Fp32m64  , X86::ILD_F32m  },
696  { X86::ILD_Fp32m80  , X86::ILD_F32m  },
697  { X86::ILD_Fp64m32  , X86::ILD_F64m  },
698  { X86::ILD_Fp64m64  , X86::ILD_F64m  },
699  { X86::ILD_Fp64m80  , X86::ILD_F64m  },
700  { X86::ISTT_Fp16m32 , X86::ISTT_FP16m},
701  { X86::ISTT_Fp16m64 , X86::ISTT_FP16m},
702  { X86::ISTT_Fp16m80 , X86::ISTT_FP16m},
703  { X86::ISTT_Fp32m32 , X86::ISTT_FP32m},
704  { X86::ISTT_Fp32m64 , X86::ISTT_FP32m},
705  { X86::ISTT_Fp32m80 , X86::ISTT_FP32m},
706  { X86::ISTT_Fp64m32 , X86::ISTT_FP64m},
707  { X86::ISTT_Fp64m64 , X86::ISTT_FP64m},
708  { X86::ISTT_Fp64m80 , X86::ISTT_FP64m},
709  { X86::IST_Fp16m32  , X86::IST_F16m  },
710  { X86::IST_Fp16m64  , X86::IST_F16m  },
711  { X86::IST_Fp16m80  , X86::IST_F16m  },
712  { X86::IST_Fp32m32  , X86::IST_F32m  },
713  { X86::IST_Fp32m64  , X86::IST_F32m  },
714  { X86::IST_Fp32m80  , X86::IST_F32m  },
715  { X86::IST_Fp64m32  , X86::IST_FP64m },
716  { X86::IST_Fp64m64  , X86::IST_FP64m },
717  { X86::IST_Fp64m80  , X86::IST_FP64m },
718  { X86::LD_Fp032     , X86::LD_F0     },
719  { X86::LD_Fp064     , X86::LD_F0     },
720  { X86::LD_Fp080     , X86::LD_F0     },
721  { X86::LD_Fp132     , X86::LD_F1     },
722  { X86::LD_Fp164     , X86::LD_F1     },
723  { X86::LD_Fp180     , X86::LD_F1     },
724  { X86::LD_Fp32m     , X86::LD_F32m   },
725  { X86::LD_Fp32m64   , X86::LD_F32m   },
726  { X86::LD_Fp32m80   , X86::LD_F32m   },
727  { X86::LD_Fp64m     , X86::LD_F64m   },
728  { X86::LD_Fp64m80   , X86::LD_F64m   },
729  { X86::LD_Fp80m     , X86::LD_F80m   },
730  { X86::MUL_Fp32m    , X86::MUL_F32m  },
731  { X86::MUL_Fp64m    , X86::MUL_F64m  },
732  { X86::MUL_Fp64m32  , X86::MUL_F32m  },
733  { X86::MUL_Fp80m32  , X86::MUL_F32m  },
734  { X86::MUL_Fp80m64  , X86::MUL_F64m  },
735  { X86::MUL_FpI16m32 , X86::MUL_FI16m },
736  { X86::MUL_FpI16m64 , X86::MUL_FI16m },
737  { X86::MUL_FpI16m80 , X86::MUL_FI16m },
738  { X86::MUL_FpI32m32 , X86::MUL_FI32m },
739  { X86::MUL_FpI32m64 , X86::MUL_FI32m },
740  { X86::MUL_FpI32m80 , X86::MUL_FI32m },
741  { X86::SIN_Fp32     , X86::SIN_F     },
742  { X86::SIN_Fp64     , X86::SIN_F     },
743  { X86::SIN_Fp80     , X86::SIN_F     },
744  { X86::SQRT_Fp32    , X86::SQRT_F    },
745  { X86::SQRT_Fp64    , X86::SQRT_F    },
746  { X86::SQRT_Fp80    , X86::SQRT_F    },
747  { X86::ST_Fp32m     , X86::ST_F32m   },
748  { X86::ST_Fp64m     , X86::ST_F64m   },
749  { X86::ST_Fp64m32   , X86::ST_F32m   },
750  { X86::ST_Fp80m32   , X86::ST_F32m   },
751  { X86::ST_Fp80m64   , X86::ST_F64m   },
752  { X86::ST_FpP80m    , X86::ST_FP80m  },
753  { X86::SUBR_Fp32m   , X86::SUBR_F32m },
754  { X86::SUBR_Fp64m   , X86::SUBR_F64m },
755  { X86::SUBR_Fp64m32 , X86::SUBR_F32m },
756  { X86::SUBR_Fp80m32 , X86::SUBR_F32m },
757  { X86::SUBR_Fp80m64 , X86::SUBR_F64m },
758  { X86::SUBR_FpI16m32, X86::SUBR_FI16m},
759  { X86::SUBR_FpI16m64, X86::SUBR_FI16m},
760  { X86::SUBR_FpI16m80, X86::SUBR_FI16m},
761  { X86::SUBR_FpI32m32, X86::SUBR_FI32m},
762  { X86::SUBR_FpI32m64, X86::SUBR_FI32m},
763  { X86::SUBR_FpI32m80, X86::SUBR_FI32m},
764  { X86::SUB_Fp32m    , X86::SUB_F32m  },
765  { X86::SUB_Fp64m    , X86::SUB_F64m  },
766  { X86::SUB_Fp64m32  , X86::SUB_F32m  },
767  { X86::SUB_Fp80m32  , X86::SUB_F32m  },
768  { X86::SUB_Fp80m64  , X86::SUB_F64m  },
769  { X86::SUB_FpI16m32 , X86::SUB_FI16m },
770  { X86::SUB_FpI16m64 , X86::SUB_FI16m },
771  { X86::SUB_FpI16m80 , X86::SUB_FI16m },
772  { X86::SUB_FpI32m32 , X86::SUB_FI32m },
773  { X86::SUB_FpI32m64 , X86::SUB_FI32m },
774  { X86::SUB_FpI32m80 , X86::SUB_FI32m },
775  { X86::TST_Fp32     , X86::TST_F     },
776  { X86::TST_Fp64     , X86::TST_F     },
777  { X86::TST_Fp80     , X86::TST_F     },
778  { X86::UCOM_FpIr32  , X86::UCOM_FIr  },
779  { X86::UCOM_FpIr64  , X86::UCOM_FIr  },
780  { X86::UCOM_FpIr80  , X86::UCOM_FIr  },
781  { X86::UCOM_Fpr32   , X86::UCOM_Fr   },
782  { X86::UCOM_Fpr64   , X86::UCOM_Fr   },
783  { X86::UCOM_Fpr80   , X86::UCOM_Fr   },
784};
785
786static unsigned getConcreteOpcode(unsigned Opcode) {
787  ASSERT_SORTED(OpcodeTable);
788  int Opc = Lookup(OpcodeTable, array_lengthof(OpcodeTable), Opcode);
789  assert(Opc != -1 && "FP Stack instruction not in OpcodeTable!");
790  return Opc;
791}
792
793//===----------------------------------------------------------------------===//
794// Helper Methods
795//===----------------------------------------------------------------------===//
796
797// PopTable - Sorted map of instructions to their popping version.  The first
798// element is an instruction, the second is the version which pops.
799//
800static const TableEntry PopTable[] = {
801  { X86::ADD_FrST0 , X86::ADD_FPrST0  },
802
803  { X86::DIVR_FrST0, X86::DIVR_FPrST0 },
804  { X86::DIV_FrST0 , X86::DIV_FPrST0  },
805
806  { X86::IST_F16m  , X86::IST_FP16m   },
807  { X86::IST_F32m  , X86::IST_FP32m   },
808
809  { X86::MUL_FrST0 , X86::MUL_FPrST0  },
810
811  { X86::ST_F32m   , X86::ST_FP32m    },
812  { X86::ST_F64m   , X86::ST_FP64m    },
813  { X86::ST_Frr    , X86::ST_FPrr     },
814
815  { X86::SUBR_FrST0, X86::SUBR_FPrST0 },
816  { X86::SUB_FrST0 , X86::SUB_FPrST0  },
817
818  { X86::UCOM_FIr  , X86::UCOM_FIPr   },
819
820  { X86::UCOM_FPr  , X86::UCOM_FPPr   },
821  { X86::UCOM_Fr   , X86::UCOM_FPr    },
822};
823
824/// popStackAfter - Pop the current value off of the top of the FP stack after
825/// the specified instruction.  This attempts to be sneaky and combine the pop
826/// into the instruction itself if possible.  The iterator is left pointing to
827/// the last instruction, be it a new pop instruction inserted, or the old
828/// instruction if it was modified in place.
829///
830void FPS::popStackAfter(MachineBasicBlock::iterator &I) {
831  MachineInstr* MI = I;
832  DebugLoc dl = MI->getDebugLoc();
833  ASSERT_SORTED(PopTable);
834  if (StackTop == 0)
835    report_fatal_error("Cannot pop empty stack!");
836  RegMap[Stack[--StackTop]] = ~0;     // Update state
837
838  // Check to see if there is a popping version of this instruction...
839  int Opcode = Lookup(PopTable, array_lengthof(PopTable), I->getOpcode());
840  if (Opcode != -1) {
841    I->setDesc(TII->get(Opcode));
842    if (Opcode == X86::UCOM_FPPr)
843      I->RemoveOperand(0);
844  } else {    // Insert an explicit pop
845    I = BuildMI(*MBB, ++I, dl, TII->get(X86::ST_FPrr)).addReg(X86::ST0);
846  }
847}
848
849/// freeStackSlotAfter - Free the specified register from the register stack, so
850/// that it is no longer in a register.  If the register is currently at the top
851/// of the stack, we just pop the current instruction, otherwise we store the
852/// current top-of-stack into the specified slot, then pop the top of stack.
853void FPS::freeStackSlotAfter(MachineBasicBlock::iterator &I, unsigned FPRegNo) {
854  if (getStackEntry(0) == FPRegNo) {  // already at the top of stack? easy.
855    popStackAfter(I);
856    return;
857  }
858
859  // Otherwise, store the top of stack into the dead slot, killing the operand
860  // without having to add in an explicit xchg then pop.
861  //
862  I = freeStackSlotBefore(++I, FPRegNo);
863}
864
865/// freeStackSlotBefore - Free the specified register without trying any
866/// folding.
867MachineBasicBlock::iterator
868FPS::freeStackSlotBefore(MachineBasicBlock::iterator I, unsigned FPRegNo) {
869  unsigned STReg    = getSTReg(FPRegNo);
870  unsigned OldSlot  = getSlot(FPRegNo);
871  unsigned TopReg   = Stack[StackTop-1];
872  Stack[OldSlot]    = TopReg;
873  RegMap[TopReg]    = OldSlot;
874  RegMap[FPRegNo]   = ~0;
875  Stack[--StackTop] = ~0;
876  return BuildMI(*MBB, I, DebugLoc(), TII->get(X86::ST_FPrr)).addReg(STReg);
877}
878
879/// adjustLiveRegs - Kill and revive registers such that exactly the FP
880/// registers with a bit in Mask are live.
881void FPS::adjustLiveRegs(unsigned Mask, MachineBasicBlock::iterator I) {
882  unsigned Defs = Mask;
883  unsigned Kills = 0;
884  for (unsigned i = 0; i < StackTop; ++i) {
885    unsigned RegNo = Stack[i];
886    if (!(Defs & (1 << RegNo)))
887      // This register is live, but we don't want it.
888      Kills |= (1 << RegNo);
889    else
890      // We don't need to imp-def this live register.
891      Defs &= ~(1 << RegNo);
892  }
893  assert((Kills & Defs) == 0 && "Register needs killing and def'ing?");
894
895  // Produce implicit-defs for free by using killed registers.
896  while (Kills && Defs) {
897    unsigned KReg = countTrailingZeros(Kills);
898    unsigned DReg = countTrailingZeros(Defs);
899    DEBUG(dbgs() << "Renaming %FP" << KReg << " as imp %FP" << DReg << "\n");
900    std::swap(Stack[getSlot(KReg)], Stack[getSlot(DReg)]);
901    std::swap(RegMap[KReg], RegMap[DReg]);
902    Kills &= ~(1 << KReg);
903    Defs &= ~(1 << DReg);
904  }
905
906  // Kill registers by popping.
907  if (Kills && I != MBB->begin()) {
908    MachineBasicBlock::iterator I2 = std::prev(I);
909    while (StackTop) {
910      unsigned KReg = getStackEntry(0);
911      if (!(Kills & (1 << KReg)))
912        break;
913      DEBUG(dbgs() << "Popping %FP" << KReg << "\n");
914      popStackAfter(I2);
915      Kills &= ~(1 << KReg);
916    }
917  }
918
919  // Manually kill the rest.
920  while (Kills) {
921    unsigned KReg = countTrailingZeros(Kills);
922    DEBUG(dbgs() << "Killing %FP" << KReg << "\n");
923    freeStackSlotBefore(I, KReg);
924    Kills &= ~(1 << KReg);
925  }
926
927  // Load zeros for all the imp-defs.
928  while(Defs) {
929    unsigned DReg = countTrailingZeros(Defs);
930    DEBUG(dbgs() << "Defining %FP" << DReg << " as 0\n");
931    BuildMI(*MBB, I, DebugLoc(), TII->get(X86::LD_F0));
932    pushReg(DReg);
933    Defs &= ~(1 << DReg);
934  }
935
936  // Now we should have the correct registers live.
937  DEBUG(dumpStack());
938  assert(StackTop == CountPopulation_32(Mask) && "Live count mismatch");
939}
940
941/// shuffleStackTop - emit fxch instructions before I to shuffle the top
942/// FixCount entries into the order given by FixStack.
943/// FIXME: Is there a better algorithm than insertion sort?
944void FPS::shuffleStackTop(const unsigned char *FixStack,
945                          unsigned FixCount,
946                          MachineBasicBlock::iterator I) {
947  // Move items into place, starting from the desired stack bottom.
948  while (FixCount--) {
949    // Old register at position FixCount.
950    unsigned OldReg = getStackEntry(FixCount);
951    // Desired register at position FixCount.
952    unsigned Reg = FixStack[FixCount];
953    if (Reg == OldReg)
954      continue;
955    // (Reg st0) (OldReg st0) = (Reg OldReg st0)
956    moveToTop(Reg, I);
957    if (FixCount > 0)
958      moveToTop(OldReg, I);
959  }
960  DEBUG(dumpStack());
961}
962
963
964//===----------------------------------------------------------------------===//
965// Instruction transformation implementation
966//===----------------------------------------------------------------------===//
967
968/// handleZeroArgFP - ST(0) = fld0    ST(0) = flds <mem>
969///
970void FPS::handleZeroArgFP(MachineBasicBlock::iterator &I) {
971  MachineInstr *MI = I;
972  unsigned DestReg = getFPReg(MI->getOperand(0));
973
974  // Change from the pseudo instruction to the concrete instruction.
975  MI->RemoveOperand(0);   // Remove the explicit ST(0) operand
976  MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
977
978  // Result gets pushed on the stack.
979  pushReg(DestReg);
980}
981
982/// handleOneArgFP - fst <mem>, ST(0)
983///
984void FPS::handleOneArgFP(MachineBasicBlock::iterator &I) {
985  MachineInstr *MI = I;
986  unsigned NumOps = MI->getDesc().getNumOperands();
987  assert((NumOps == X86::AddrNumOperands + 1 || NumOps == 1) &&
988         "Can only handle fst* & ftst instructions!");
989
990  // Is this the last use of the source register?
991  unsigned Reg = getFPReg(MI->getOperand(NumOps-1));
992  bool KillsSrc = MI->killsRegister(X86::FP0+Reg);
993
994  if (KillsSrc)
995    duplicatePendingSTBeforeKill(Reg, I);
996
997  // FISTP64m is strange because there isn't a non-popping versions.
998  // If we have one _and_ we don't want to pop the operand, duplicate the value
999  // on the stack instead of moving it.  This ensure that popping the value is
1000  // always ok.
1001  // Ditto FISTTP16m, FISTTP32m, FISTTP64m, ST_FpP80m.
1002  //
1003  if (!KillsSrc &&
1004      (MI->getOpcode() == X86::IST_Fp64m32 ||
1005       MI->getOpcode() == X86::ISTT_Fp16m32 ||
1006       MI->getOpcode() == X86::ISTT_Fp32m32 ||
1007       MI->getOpcode() == X86::ISTT_Fp64m32 ||
1008       MI->getOpcode() == X86::IST_Fp64m64 ||
1009       MI->getOpcode() == X86::ISTT_Fp16m64 ||
1010       MI->getOpcode() == X86::ISTT_Fp32m64 ||
1011       MI->getOpcode() == X86::ISTT_Fp64m64 ||
1012       MI->getOpcode() == X86::IST_Fp64m80 ||
1013       MI->getOpcode() == X86::ISTT_Fp16m80 ||
1014       MI->getOpcode() == X86::ISTT_Fp32m80 ||
1015       MI->getOpcode() == X86::ISTT_Fp64m80 ||
1016       MI->getOpcode() == X86::ST_FpP80m)) {
1017    duplicateToTop(Reg, getScratchReg(), I);
1018  } else {
1019    moveToTop(Reg, I);            // Move to the top of the stack...
1020  }
1021
1022  // Convert from the pseudo instruction to the concrete instruction.
1023  MI->RemoveOperand(NumOps-1);    // Remove explicit ST(0) operand
1024  MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1025
1026  if (MI->getOpcode() == X86::IST_FP64m ||
1027      MI->getOpcode() == X86::ISTT_FP16m ||
1028      MI->getOpcode() == X86::ISTT_FP32m ||
1029      MI->getOpcode() == X86::ISTT_FP64m ||
1030      MI->getOpcode() == X86::ST_FP80m) {
1031    if (StackTop == 0)
1032      report_fatal_error("Stack empty??");
1033    --StackTop;
1034  } else if (KillsSrc) { // Last use of operand?
1035    popStackAfter(I);
1036  }
1037}
1038
1039
1040/// handleOneArgFPRW: Handle instructions that read from the top of stack and
1041/// replace the value with a newly computed value.  These instructions may have
1042/// non-fp operands after their FP operands.
1043///
1044///  Examples:
1045///     R1 = fchs R2
1046///     R1 = fadd R2, [mem]
1047///
1048void FPS::handleOneArgFPRW(MachineBasicBlock::iterator &I) {
1049  MachineInstr *MI = I;
1050#ifndef NDEBUG
1051  unsigned NumOps = MI->getDesc().getNumOperands();
1052  assert(NumOps >= 2 && "FPRW instructions must have 2 ops!!");
1053#endif
1054
1055  // Is this the last use of the source register?
1056  unsigned Reg = getFPReg(MI->getOperand(1));
1057  bool KillsSrc = MI->killsRegister(X86::FP0+Reg);
1058
1059  if (KillsSrc) {
1060    duplicatePendingSTBeforeKill(Reg, I);
1061    // If this is the last use of the source register, just make sure it's on
1062    // the top of the stack.
1063    moveToTop(Reg, I);
1064    if (StackTop == 0)
1065      report_fatal_error("Stack cannot be empty!");
1066    --StackTop;
1067    pushReg(getFPReg(MI->getOperand(0)));
1068  } else {
1069    // If this is not the last use of the source register, _copy_ it to the top
1070    // of the stack.
1071    duplicateToTop(Reg, getFPReg(MI->getOperand(0)), I);
1072  }
1073
1074  // Change from the pseudo instruction to the concrete instruction.
1075  MI->RemoveOperand(1);   // Drop the source operand.
1076  MI->RemoveOperand(0);   // Drop the destination operand.
1077  MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1078}
1079
1080
1081//===----------------------------------------------------------------------===//
1082// Define tables of various ways to map pseudo instructions
1083//
1084
1085// ForwardST0Table - Map: A = B op C  into: ST(0) = ST(0) op ST(i)
1086static const TableEntry ForwardST0Table[] = {
1087  { X86::ADD_Fp32  , X86::ADD_FST0r },
1088  { X86::ADD_Fp64  , X86::ADD_FST0r },
1089  { X86::ADD_Fp80  , X86::ADD_FST0r },
1090  { X86::DIV_Fp32  , X86::DIV_FST0r },
1091  { X86::DIV_Fp64  , X86::DIV_FST0r },
1092  { X86::DIV_Fp80  , X86::DIV_FST0r },
1093  { X86::MUL_Fp32  , X86::MUL_FST0r },
1094  { X86::MUL_Fp64  , X86::MUL_FST0r },
1095  { X86::MUL_Fp80  , X86::MUL_FST0r },
1096  { X86::SUB_Fp32  , X86::SUB_FST0r },
1097  { X86::SUB_Fp64  , X86::SUB_FST0r },
1098  { X86::SUB_Fp80  , X86::SUB_FST0r },
1099};
1100
1101// ReverseST0Table - Map: A = B op C  into: ST(0) = ST(i) op ST(0)
1102static const TableEntry ReverseST0Table[] = {
1103  { X86::ADD_Fp32  , X86::ADD_FST0r  },   // commutative
1104  { X86::ADD_Fp64  , X86::ADD_FST0r  },   // commutative
1105  { X86::ADD_Fp80  , X86::ADD_FST0r  },   // commutative
1106  { X86::DIV_Fp32  , X86::DIVR_FST0r },
1107  { X86::DIV_Fp64  , X86::DIVR_FST0r },
1108  { X86::DIV_Fp80  , X86::DIVR_FST0r },
1109  { X86::MUL_Fp32  , X86::MUL_FST0r  },   // commutative
1110  { X86::MUL_Fp64  , X86::MUL_FST0r  },   // commutative
1111  { X86::MUL_Fp80  , X86::MUL_FST0r  },   // commutative
1112  { X86::SUB_Fp32  , X86::SUBR_FST0r },
1113  { X86::SUB_Fp64  , X86::SUBR_FST0r },
1114  { X86::SUB_Fp80  , X86::SUBR_FST0r },
1115};
1116
1117// ForwardSTiTable - Map: A = B op C  into: ST(i) = ST(0) op ST(i)
1118static const TableEntry ForwardSTiTable[] = {
1119  { X86::ADD_Fp32  , X86::ADD_FrST0  },   // commutative
1120  { X86::ADD_Fp64  , X86::ADD_FrST0  },   // commutative
1121  { X86::ADD_Fp80  , X86::ADD_FrST0  },   // commutative
1122  { X86::DIV_Fp32  , X86::DIVR_FrST0 },
1123  { X86::DIV_Fp64  , X86::DIVR_FrST0 },
1124  { X86::DIV_Fp80  , X86::DIVR_FrST0 },
1125  { X86::MUL_Fp32  , X86::MUL_FrST0  },   // commutative
1126  { X86::MUL_Fp64  , X86::MUL_FrST0  },   // commutative
1127  { X86::MUL_Fp80  , X86::MUL_FrST0  },   // commutative
1128  { X86::SUB_Fp32  , X86::SUBR_FrST0 },
1129  { X86::SUB_Fp64  , X86::SUBR_FrST0 },
1130  { X86::SUB_Fp80  , X86::SUBR_FrST0 },
1131};
1132
1133// ReverseSTiTable - Map: A = B op C  into: ST(i) = ST(i) op ST(0)
1134static const TableEntry ReverseSTiTable[] = {
1135  { X86::ADD_Fp32  , X86::ADD_FrST0 },
1136  { X86::ADD_Fp64  , X86::ADD_FrST0 },
1137  { X86::ADD_Fp80  , X86::ADD_FrST0 },
1138  { X86::DIV_Fp32  , X86::DIV_FrST0 },
1139  { X86::DIV_Fp64  , X86::DIV_FrST0 },
1140  { X86::DIV_Fp80  , X86::DIV_FrST0 },
1141  { X86::MUL_Fp32  , X86::MUL_FrST0 },
1142  { X86::MUL_Fp64  , X86::MUL_FrST0 },
1143  { X86::MUL_Fp80  , X86::MUL_FrST0 },
1144  { X86::SUB_Fp32  , X86::SUB_FrST0 },
1145  { X86::SUB_Fp64  , X86::SUB_FrST0 },
1146  { X86::SUB_Fp80  , X86::SUB_FrST0 },
1147};
1148
1149
1150/// handleTwoArgFP - Handle instructions like FADD and friends which are virtual
1151/// instructions which need to be simplified and possibly transformed.
1152///
1153/// Result: ST(0) = fsub  ST(0), ST(i)
1154///         ST(i) = fsub  ST(0), ST(i)
1155///         ST(0) = fsubr ST(0), ST(i)
1156///         ST(i) = fsubr ST(0), ST(i)
1157///
1158void FPS::handleTwoArgFP(MachineBasicBlock::iterator &I) {
1159  ASSERT_SORTED(ForwardST0Table); ASSERT_SORTED(ReverseST0Table);
1160  ASSERT_SORTED(ForwardSTiTable); ASSERT_SORTED(ReverseSTiTable);
1161  MachineInstr *MI = I;
1162
1163  unsigned NumOperands = MI->getDesc().getNumOperands();
1164  assert(NumOperands == 3 && "Illegal TwoArgFP instruction!");
1165  unsigned Dest = getFPReg(MI->getOperand(0));
1166  unsigned Op0 = getFPReg(MI->getOperand(NumOperands-2));
1167  unsigned Op1 = getFPReg(MI->getOperand(NumOperands-1));
1168  bool KillsOp0 = MI->killsRegister(X86::FP0+Op0);
1169  bool KillsOp1 = MI->killsRegister(X86::FP0+Op1);
1170  DebugLoc dl = MI->getDebugLoc();
1171
1172  unsigned TOS = getStackEntry(0);
1173
1174  // One of our operands must be on the top of the stack.  If neither is yet, we
1175  // need to move one.
1176  if (Op0 != TOS && Op1 != TOS) {   // No operand at TOS?
1177    // We can choose to move either operand to the top of the stack.  If one of
1178    // the operands is killed by this instruction, we want that one so that we
1179    // can update right on top of the old version.
1180    if (KillsOp0) {
1181      moveToTop(Op0, I);         // Move dead operand to TOS.
1182      TOS = Op0;
1183    } else if (KillsOp1) {
1184      moveToTop(Op1, I);
1185      TOS = Op1;
1186    } else {
1187      // All of the operands are live after this instruction executes, so we
1188      // cannot update on top of any operand.  Because of this, we must
1189      // duplicate one of the stack elements to the top.  It doesn't matter
1190      // which one we pick.
1191      //
1192      duplicateToTop(Op0, Dest, I);
1193      Op0 = TOS = Dest;
1194      KillsOp0 = true;
1195    }
1196  } else if (!KillsOp0 && !KillsOp1) {
1197    // If we DO have one of our operands at the top of the stack, but we don't
1198    // have a dead operand, we must duplicate one of the operands to a new slot
1199    // on the stack.
1200    duplicateToTop(Op0, Dest, I);
1201    Op0 = TOS = Dest;
1202    KillsOp0 = true;
1203  }
1204
1205  // Now we know that one of our operands is on the top of the stack, and at
1206  // least one of our operands is killed by this instruction.
1207  assert((TOS == Op0 || TOS == Op1) && (KillsOp0 || KillsOp1) &&
1208         "Stack conditions not set up right!");
1209
1210  // We decide which form to use based on what is on the top of the stack, and
1211  // which operand is killed by this instruction.
1212  const TableEntry *InstTable;
1213  bool isForward = TOS == Op0;
1214  bool updateST0 = (TOS == Op0 && !KillsOp1) || (TOS == Op1 && !KillsOp0);
1215  if (updateST0) {
1216    if (isForward)
1217      InstTable = ForwardST0Table;
1218    else
1219      InstTable = ReverseST0Table;
1220  } else {
1221    if (isForward)
1222      InstTable = ForwardSTiTable;
1223    else
1224      InstTable = ReverseSTiTable;
1225  }
1226
1227  int Opcode = Lookup(InstTable, array_lengthof(ForwardST0Table),
1228                      MI->getOpcode());
1229  assert(Opcode != -1 && "Unknown TwoArgFP pseudo instruction!");
1230
1231  // NotTOS - The register which is not on the top of stack...
1232  unsigned NotTOS = (TOS == Op0) ? Op1 : Op0;
1233
1234  // Replace the old instruction with a new instruction
1235  MBB->remove(I++);
1236  I = BuildMI(*MBB, I, dl, TII->get(Opcode)).addReg(getSTReg(NotTOS));
1237
1238  // If both operands are killed, pop one off of the stack in addition to
1239  // overwriting the other one.
1240  if (KillsOp0 && KillsOp1 && Op0 != Op1) {
1241    assert(!updateST0 && "Should have updated other operand!");
1242    popStackAfter(I);   // Pop the top of stack
1243  }
1244
1245  // Update stack information so that we know the destination register is now on
1246  // the stack.
1247  unsigned UpdatedSlot = getSlot(updateST0 ? TOS : NotTOS);
1248  assert(UpdatedSlot < StackTop && Dest < 7);
1249  Stack[UpdatedSlot]   = Dest;
1250  RegMap[Dest]         = UpdatedSlot;
1251  MBB->getParent()->DeleteMachineInstr(MI); // Remove the old instruction
1252}
1253
1254/// handleCompareFP - Handle FUCOM and FUCOMI instructions, which have two FP
1255/// register arguments and no explicit destinations.
1256///
1257void FPS::handleCompareFP(MachineBasicBlock::iterator &I) {
1258  ASSERT_SORTED(ForwardST0Table); ASSERT_SORTED(ReverseST0Table);
1259  ASSERT_SORTED(ForwardSTiTable); ASSERT_SORTED(ReverseSTiTable);
1260  MachineInstr *MI = I;
1261
1262  unsigned NumOperands = MI->getDesc().getNumOperands();
1263  assert(NumOperands == 2 && "Illegal FUCOM* instruction!");
1264  unsigned Op0 = getFPReg(MI->getOperand(NumOperands-2));
1265  unsigned Op1 = getFPReg(MI->getOperand(NumOperands-1));
1266  bool KillsOp0 = MI->killsRegister(X86::FP0+Op0);
1267  bool KillsOp1 = MI->killsRegister(X86::FP0+Op1);
1268
1269  // Make sure the first operand is on the top of stack, the other one can be
1270  // anywhere.
1271  moveToTop(Op0, I);
1272
1273  // Change from the pseudo instruction to the concrete instruction.
1274  MI->getOperand(0).setReg(getSTReg(Op1));
1275  MI->RemoveOperand(1);
1276  MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1277
1278  // If any of the operands are killed by this instruction, free them.
1279  if (KillsOp0) freeStackSlotAfter(I, Op0);
1280  if (KillsOp1 && Op0 != Op1) freeStackSlotAfter(I, Op1);
1281}
1282
1283/// handleCondMovFP - Handle two address conditional move instructions.  These
1284/// instructions move a st(i) register to st(0) iff a condition is true.  These
1285/// instructions require that the first operand is at the top of the stack, but
1286/// otherwise don't modify the stack at all.
1287void FPS::handleCondMovFP(MachineBasicBlock::iterator &I) {
1288  MachineInstr *MI = I;
1289
1290  unsigned Op0 = getFPReg(MI->getOperand(0));
1291  unsigned Op1 = getFPReg(MI->getOperand(2));
1292  bool KillsOp1 = MI->killsRegister(X86::FP0+Op1);
1293
1294  // The first operand *must* be on the top of the stack.
1295  moveToTop(Op0, I);
1296
1297  // Change the second operand to the stack register that the operand is in.
1298  // Change from the pseudo instruction to the concrete instruction.
1299  MI->RemoveOperand(0);
1300  MI->RemoveOperand(1);
1301  MI->getOperand(0).setReg(getSTReg(Op1));
1302  MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1303
1304  // If we kill the second operand, make sure to pop it from the stack.
1305  if (Op0 != Op1 && KillsOp1) {
1306    // Get this value off of the register stack.
1307    freeStackSlotAfter(I, Op1);
1308  }
1309}
1310
1311
1312/// handleSpecialFP - Handle special instructions which behave unlike other
1313/// floating point instructions.  This is primarily intended for use by pseudo
1314/// instructions.
1315///
1316void FPS::handleSpecialFP(MachineBasicBlock::iterator &I) {
1317  MachineInstr *MI = I;
1318  switch (MI->getOpcode()) {
1319  default: llvm_unreachable("Unknown SpecialFP instruction!");
1320  case TargetOpcode::COPY: {
1321    // We handle three kinds of copies: FP <- FP, FP <- ST, and ST <- FP.
1322    const MachineOperand &MO1 = MI->getOperand(1);
1323    const MachineOperand &MO0 = MI->getOperand(0);
1324    unsigned DstST = MO0.getReg() - X86::ST0;
1325    unsigned SrcST = MO1.getReg() - X86::ST0;
1326    bool KillsSrc = MI->killsRegister(MO1.getReg());
1327
1328    // ST = COPY FP. Set up a pending ST register.
1329    if (DstST < 8) {
1330      unsigned SrcFP = getFPReg(MO1);
1331      assert(isLive(SrcFP) && "Cannot copy dead register");
1332      assert(!MO0.isDead() && "Cannot copy to dead ST register");
1333
1334      // Unallocated STs are marked as the nonexistent FP255.
1335      while (NumPendingSTs <= DstST)
1336        PendingST[NumPendingSTs++] = NumFPRegs;
1337
1338      // STi could still be live from a previous inline asm.
1339      if (isScratchReg(PendingST[DstST])) {
1340        DEBUG(dbgs() << "Clobbering old ST in FP" << unsigned(PendingST[DstST])
1341                     << '\n');
1342        freeStackSlotBefore(MI, PendingST[DstST]);
1343      }
1344
1345      // When the source is killed, allocate a scratch FP register.
1346      if (KillsSrc) {
1347        duplicatePendingSTBeforeKill(SrcFP, I);
1348        unsigned Slot = getSlot(SrcFP);
1349        unsigned SR = getScratchReg();
1350        PendingST[DstST] = SR;
1351        Stack[Slot] = SR;
1352        RegMap[SR] = Slot;
1353      } else
1354        PendingST[DstST] = SrcFP;
1355      break;
1356    }
1357
1358    // FP = COPY ST. Extract fixed stack value.
1359    // Any instruction defining ST registers must have assigned them to a
1360    // scratch register.
1361    if (SrcST < 8) {
1362      unsigned DstFP = getFPReg(MO0);
1363      assert(!isLive(DstFP) && "Cannot copy ST to live FP register");
1364      assert(NumPendingSTs > SrcST && "Cannot copy from dead ST register");
1365      unsigned SrcFP = PendingST[SrcST];
1366      assert(isScratchReg(SrcFP) && "Expected ST in a scratch register");
1367      assert(isLive(SrcFP) && "Scratch holding ST is dead");
1368
1369      // DstFP steals the stack slot from SrcFP.
1370      unsigned Slot = getSlot(SrcFP);
1371      Stack[Slot] = DstFP;
1372      RegMap[DstFP] = Slot;
1373
1374      // Always treat the ST as killed.
1375      PendingST[SrcST] = NumFPRegs;
1376      while (NumPendingSTs && PendingST[NumPendingSTs - 1] == NumFPRegs)
1377        --NumPendingSTs;
1378      break;
1379    }
1380
1381    // FP <- FP copy.
1382    unsigned DstFP = getFPReg(MO0);
1383    unsigned SrcFP = getFPReg(MO1);
1384    assert(isLive(SrcFP) && "Cannot copy dead register");
1385    if (KillsSrc) {
1386      // If the input operand is killed, we can just change the owner of the
1387      // incoming stack slot into the result.
1388      unsigned Slot = getSlot(SrcFP);
1389      Stack[Slot] = DstFP;
1390      RegMap[DstFP] = Slot;
1391    } else {
1392      // For COPY we just duplicate the specified value to a new stack slot.
1393      // This could be made better, but would require substantial changes.
1394      duplicateToTop(SrcFP, DstFP, I);
1395    }
1396    break;
1397  }
1398
1399  case TargetOpcode::IMPLICIT_DEF: {
1400    // All FP registers must be explicitly defined, so load a 0 instead.
1401    unsigned Reg = MI->getOperand(0).getReg() - X86::FP0;
1402    DEBUG(dbgs() << "Emitting LD_F0 for implicit FP" << Reg << '\n');
1403    BuildMI(*MBB, I, MI->getDebugLoc(), TII->get(X86::LD_F0));
1404    pushReg(Reg);
1405    break;
1406  }
1407
1408  case X86::FpPOP_RETVAL: {
1409    // The FpPOP_RETVAL instruction is used after calls that return a value on
1410    // the floating point stack. We cannot model this with ST defs since CALL
1411    // instructions have fixed clobber lists. This instruction is interpreted
1412    // to mean that there is one more live register on the stack than we
1413    // thought.
1414    //
1415    // This means that StackTop does not match the hardware stack between a
1416    // call and the FpPOP_RETVAL instructions.  We do tolerate FP instructions
1417    // between CALL and FpPOP_RETVAL as long as they don't overflow the
1418    // hardware stack.
1419    unsigned DstFP = getFPReg(MI->getOperand(0));
1420
1421    // Move existing stack elements up to reflect reality.
1422    assert(StackTop < 8 && "Stack overflowed before FpPOP_RETVAL");
1423    if (StackTop) {
1424      std::copy_backward(Stack, Stack + StackTop, Stack + StackTop + 1);
1425      for (unsigned i = 0; i != NumFPRegs; ++i)
1426        ++RegMap[i];
1427    }
1428    ++StackTop;
1429
1430    // DstFP is the new bottom of the stack.
1431    Stack[0] = DstFP;
1432    RegMap[DstFP] = 0;
1433
1434    // DstFP will be killed by processBasicBlock if this was a dead def.
1435    break;
1436  }
1437
1438  case TargetOpcode::INLINEASM: {
1439    // The inline asm MachineInstr currently only *uses* FP registers for the
1440    // 'f' constraint.  These should be turned into the current ST(x) register
1441    // in the machine instr.
1442    //
1443    // There are special rules for x87 inline assembly. The compiler must know
1444    // exactly how many registers are popped and pushed implicitly by the asm.
1445    // Otherwise it is not possible to restore the stack state after the inline
1446    // asm.
1447    //
1448    // There are 3 kinds of input operands:
1449    //
1450    // 1. Popped inputs. These must appear at the stack top in ST0-STn. A
1451    //    popped input operand must be in a fixed stack slot, and it is either
1452    //    tied to an output operand, or in the clobber list. The MI has ST use
1453    //    and def operands for these inputs.
1454    //
1455    // 2. Fixed inputs. These inputs appear in fixed stack slots, but are
1456    //    preserved by the inline asm. The fixed stack slots must be STn-STm
1457    //    following the popped inputs. A fixed input operand cannot be tied to
1458    //    an output or appear in the clobber list. The MI has ST use operands
1459    //    and no defs for these inputs.
1460    //
1461    // 3. Preserved inputs. These inputs use the "f" constraint which is
1462    //    represented as an FP register. The inline asm won't change these
1463    //    stack slots.
1464    //
1465    // Outputs must be in ST registers, FP outputs are not allowed. Clobbered
1466    // registers do not count as output operands. The inline asm changes the
1467    // stack as if it popped all the popped inputs and then pushed all the
1468    // output operands.
1469
1470    // Scan the assembly for ST registers used, defined and clobbered. We can
1471    // only tell clobbers from defs by looking at the asm descriptor.
1472    unsigned STUses = 0, STDefs = 0, STClobbers = 0, STDeadDefs = 0;
1473    unsigned NumOps = 0;
1474    for (unsigned i = InlineAsm::MIOp_FirstOperand, e = MI->getNumOperands();
1475         i != e && MI->getOperand(i).isImm(); i += 1 + NumOps) {
1476      unsigned Flags = MI->getOperand(i).getImm();
1477      NumOps = InlineAsm::getNumOperandRegisters(Flags);
1478      if (NumOps != 1)
1479        continue;
1480      const MachineOperand &MO = MI->getOperand(i + 1);
1481      if (!MO.isReg())
1482        continue;
1483      unsigned STReg = MO.getReg() - X86::ST0;
1484      if (STReg >= 8)
1485        continue;
1486
1487      switch (InlineAsm::getKind(Flags)) {
1488      case InlineAsm::Kind_RegUse:
1489        STUses |= (1u << STReg);
1490        break;
1491      case InlineAsm::Kind_RegDef:
1492      case InlineAsm::Kind_RegDefEarlyClobber:
1493        STDefs |= (1u << STReg);
1494        if (MO.isDead())
1495          STDeadDefs |= (1u << STReg);
1496        break;
1497      case InlineAsm::Kind_Clobber:
1498        STClobbers |= (1u << STReg);
1499        break;
1500      default:
1501        break;
1502      }
1503    }
1504
1505    if (STUses && !isMask_32(STUses))
1506      MI->emitError("fixed input regs must be last on the x87 stack");
1507    unsigned NumSTUses = CountTrailingOnes_32(STUses);
1508
1509    // Defs must be contiguous from the stack top. ST0-STn.
1510    if (STDefs && !isMask_32(STDefs)) {
1511      MI->emitError("output regs must be last on the x87 stack");
1512      STDefs = NextPowerOf2(STDefs) - 1;
1513    }
1514    unsigned NumSTDefs = CountTrailingOnes_32(STDefs);
1515
1516    // So must the clobbered stack slots. ST0-STm, m >= n.
1517    if (STClobbers && !isMask_32(STDefs | STClobbers))
1518      MI->emitError("clobbers must be last on the x87 stack");
1519
1520    // Popped inputs are the ones that are also clobbered or defined.
1521    unsigned STPopped = STUses & (STDefs | STClobbers);
1522    if (STPopped && !isMask_32(STPopped))
1523      MI->emitError("implicitly popped regs must be last on the x87 stack");
1524    unsigned NumSTPopped = CountTrailingOnes_32(STPopped);
1525
1526    DEBUG(dbgs() << "Asm uses " << NumSTUses << " fixed regs, pops "
1527                 << NumSTPopped << ", and defines " << NumSTDefs << " regs.\n");
1528
1529    // Scan the instruction for FP uses corresponding to "f" constraints.
1530    // Collect FP registers to kill afer the instruction.
1531    // Always kill all the scratch regs.
1532    unsigned FPKills = ((1u << NumFPRegs) - 1) & ~0xff;
1533    unsigned FPUsed = 0;
1534    for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1535      MachineOperand &Op = MI->getOperand(i);
1536      if (!Op.isReg() || Op.getReg() < X86::FP0 || Op.getReg() > X86::FP6)
1537        continue;
1538      if (!Op.isUse())
1539        MI->emitError("illegal \"f\" output constraint");
1540      unsigned FPReg = getFPReg(Op);
1541      FPUsed |= 1U << FPReg;
1542
1543      // If we kill this operand, make sure to pop it from the stack after the
1544      // asm.  We just remember it for now, and pop them all off at the end in
1545      // a batch.
1546      if (Op.isKill())
1547        FPKills |= 1U << FPReg;
1548    }
1549
1550    // The popped inputs will be killed by the instruction, so duplicate them
1551    // if the FP register needs to be live after the instruction, or if it is
1552    // used in the instruction itself. We effectively treat the popped inputs
1553    // as early clobbers.
1554    for (unsigned i = 0; i < NumSTPopped; ++i) {
1555      if ((FPKills & ~FPUsed) & (1u << PendingST[i]))
1556        continue;
1557      unsigned SR = getScratchReg();
1558      duplicateToTop(PendingST[i], SR, I);
1559      DEBUG(dbgs() << "Duplicating ST" << i << " in FP"
1560                   << unsigned(PendingST[i]) << " to avoid clobbering it.\n");
1561      PendingST[i] = SR;
1562    }
1563
1564    // Make sure we have a unique live register for every fixed use. Some of
1565    // them could be undef uses, and we need to emit LD_F0 instructions.
1566    for (unsigned i = 0; i < NumSTUses; ++i) {
1567      if (i < NumPendingSTs && PendingST[i] < NumFPRegs) {
1568        // Check for shared assignments.
1569        for (unsigned j = 0; j < i; ++j) {
1570          if (PendingST[j] != PendingST[i])
1571            continue;
1572          // STi and STj are inn the same register, create a copy.
1573          unsigned SR = getScratchReg();
1574          duplicateToTop(PendingST[i], SR, I);
1575          DEBUG(dbgs() << "Duplicating ST" << i << " in FP"
1576                       << unsigned(PendingST[i])
1577                       << " to avoid collision with ST" << j << '\n');
1578          PendingST[i] = SR;
1579        }
1580        continue;
1581      }
1582      unsigned SR = getScratchReg();
1583      DEBUG(dbgs() << "Emitting LD_F0 for ST" << i << " in FP" << SR << '\n');
1584      BuildMI(*MBB, I, MI->getDebugLoc(), TII->get(X86::LD_F0));
1585      pushReg(SR);
1586      PendingST[i] = SR;
1587      if (NumPendingSTs == i)
1588        ++NumPendingSTs;
1589    }
1590    assert(NumPendingSTs >= NumSTUses && "Fixed registers should be assigned");
1591
1592    // Now we can rearrange the live registers to match what was requested.
1593    shuffleStackTop(PendingST, NumPendingSTs, I);
1594    DEBUG({dbgs() << "Before asm: "; dumpStack();});
1595
1596    // With the stack layout fixed, rewrite the FP registers.
1597    for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1598      MachineOperand &Op = MI->getOperand(i);
1599      if (!Op.isReg() || Op.getReg() < X86::FP0 || Op.getReg() > X86::FP6)
1600        continue;
1601      unsigned FPReg = getFPReg(Op);
1602      Op.setReg(getSTReg(FPReg));
1603    }
1604
1605    // Simulate the inline asm popping its inputs and pushing its outputs.
1606    StackTop -= NumSTPopped;
1607
1608    // Hold the fixed output registers in scratch FP registers. They will be
1609    // transferred to real FP registers by copies.
1610    NumPendingSTs = 0;
1611    for (unsigned i = 0; i < NumSTDefs; ++i) {
1612      unsigned SR = getScratchReg();
1613      pushReg(SR);
1614      FPKills &= ~(1u << SR);
1615    }
1616    for (unsigned i = 0; i < NumSTDefs; ++i)
1617      PendingST[NumPendingSTs++] = getStackEntry(i);
1618    DEBUG({dbgs() << "After asm: "; dumpStack();});
1619
1620    // If any of the ST defs were dead, pop them immediately. Our caller only
1621    // handles dead FP defs.
1622    MachineBasicBlock::iterator InsertPt = MI;
1623    for (unsigned i = 0; STDefs & (1u << i); ++i) {
1624      if (!(STDeadDefs & (1u << i)))
1625        continue;
1626      freeStackSlotAfter(InsertPt, PendingST[i]);
1627      PendingST[i] = NumFPRegs;
1628    }
1629    while (NumPendingSTs && PendingST[NumPendingSTs - 1] == NumFPRegs)
1630      --NumPendingSTs;
1631
1632    // If this asm kills any FP registers (is the last use of them) we must
1633    // explicitly emit pop instructions for them.  Do this now after the asm has
1634    // executed so that the ST(x) numbers are not off (which would happen if we
1635    // did this inline with operand rewriting).
1636    //
1637    // Note: this might be a non-optimal pop sequence.  We might be able to do
1638    // better by trying to pop in stack order or something.
1639    while (FPKills) {
1640      unsigned FPReg = countTrailingZeros(FPKills);
1641      if (isLive(FPReg))
1642        freeStackSlotAfter(InsertPt, FPReg);
1643      FPKills &= ~(1U << FPReg);
1644    }
1645    // Don't delete the inline asm!
1646    return;
1647  }
1648
1649  case X86::WIN_FTOL_32:
1650  case X86::WIN_FTOL_64: {
1651    // Push the operand into ST0.
1652    MachineOperand &Op = MI->getOperand(0);
1653    assert(Op.isUse() && Op.isReg() &&
1654      Op.getReg() >= X86::FP0 && Op.getReg() <= X86::FP6);
1655    unsigned FPReg = getFPReg(Op);
1656    if (Op.isKill())
1657      moveToTop(FPReg, I);
1658    else
1659      duplicateToTop(FPReg, FPReg, I);
1660
1661    // Emit the call. This will pop the operand.
1662    BuildMI(*MBB, I, MI->getDebugLoc(), TII->get(X86::CALLpcrel32))
1663      .addExternalSymbol("_ftol2")
1664      .addReg(X86::ST0, RegState::ImplicitKill)
1665      .addReg(X86::ECX, RegState::ImplicitDefine)
1666      .addReg(X86::EAX, RegState::Define | RegState::Implicit)
1667      .addReg(X86::EDX, RegState::Define | RegState::Implicit)
1668      .addReg(X86::EFLAGS, RegState::Define | RegState::Implicit);
1669    --StackTop;
1670
1671    break;
1672  }
1673
1674  case X86::RETQ:
1675  case X86::RETL:
1676  case X86::RETIL:
1677  case X86::RETIQ:
1678    // If RET has an FP register use operand, pass the first one in ST(0) and
1679    // the second one in ST(1).
1680
1681    // Find the register operands.
1682    unsigned FirstFPRegOp = ~0U, SecondFPRegOp = ~0U;
1683    unsigned LiveMask = 0;
1684
1685    for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1686      MachineOperand &Op = MI->getOperand(i);
1687      if (!Op.isReg() || Op.getReg() < X86::FP0 || Op.getReg() > X86::FP6)
1688        continue;
1689      // FP Register uses must be kills unless there are two uses of the same
1690      // register, in which case only one will be a kill.
1691      assert(Op.isUse() &&
1692             (Op.isKill() ||                        // Marked kill.
1693              getFPReg(Op) == FirstFPRegOp ||       // Second instance.
1694              MI->killsRegister(Op.getReg())) &&    // Later use is marked kill.
1695             "Ret only defs operands, and values aren't live beyond it");
1696
1697      if (FirstFPRegOp == ~0U)
1698        FirstFPRegOp = getFPReg(Op);
1699      else {
1700        assert(SecondFPRegOp == ~0U && "More than two fp operands!");
1701        SecondFPRegOp = getFPReg(Op);
1702      }
1703      LiveMask |= (1 << getFPReg(Op));
1704
1705      // Remove the operand so that later passes don't see it.
1706      MI->RemoveOperand(i);
1707      --i, --e;
1708    }
1709
1710    // We may have been carrying spurious live-ins, so make sure only the returned
1711    // registers are left live.
1712    adjustLiveRegs(LiveMask, MI);
1713    if (!LiveMask) return;  // Quick check to see if any are possible.
1714
1715    // There are only four possibilities here:
1716    // 1) we are returning a single FP value.  In this case, it has to be in
1717    //    ST(0) already, so just declare success by removing the value from the
1718    //    FP Stack.
1719    if (SecondFPRegOp == ~0U) {
1720      // Assert that the top of stack contains the right FP register.
1721      assert(StackTop == 1 && FirstFPRegOp == getStackEntry(0) &&
1722             "Top of stack not the right register for RET!");
1723
1724      // Ok, everything is good, mark the value as not being on the stack
1725      // anymore so that our assertion about the stack being empty at end of
1726      // block doesn't fire.
1727      StackTop = 0;
1728      return;
1729    }
1730
1731    // Otherwise, we are returning two values:
1732    // 2) If returning the same value for both, we only have one thing in the FP
1733    //    stack.  Consider:  RET FP1, FP1
1734    if (StackTop == 1) {
1735      assert(FirstFPRegOp == SecondFPRegOp && FirstFPRegOp == getStackEntry(0)&&
1736             "Stack misconfiguration for RET!");
1737
1738      // Duplicate the TOS so that we return it twice.  Just pick some other FPx
1739      // register to hold it.
1740      unsigned NewReg = getScratchReg();
1741      duplicateToTop(FirstFPRegOp, NewReg, MI);
1742      FirstFPRegOp = NewReg;
1743    }
1744
1745    /// Okay we know we have two different FPx operands now:
1746    assert(StackTop == 2 && "Must have two values live!");
1747
1748    /// 3) If SecondFPRegOp is currently in ST(0) and FirstFPRegOp is currently
1749    ///    in ST(1).  In this case, emit an fxch.
1750    if (getStackEntry(0) == SecondFPRegOp) {
1751      assert(getStackEntry(1) == FirstFPRegOp && "Unknown regs live");
1752      moveToTop(FirstFPRegOp, MI);
1753    }
1754
1755    /// 4) Finally, FirstFPRegOp must be in ST(0) and SecondFPRegOp must be in
1756    /// ST(1).  Just remove both from our understanding of the stack and return.
1757    assert(getStackEntry(0) == FirstFPRegOp && "Unknown regs live");
1758    assert(getStackEntry(1) == SecondFPRegOp && "Unknown regs live");
1759    StackTop = 0;
1760    return;
1761  }
1762
1763  I = MBB->erase(I);  // Remove the pseudo instruction
1764
1765  // We want to leave I pointing to the previous instruction, but what if we
1766  // just erased the first instruction?
1767  if (I == MBB->begin()) {
1768    DEBUG(dbgs() << "Inserting dummy KILL\n");
1769    I = BuildMI(*MBB, I, DebugLoc(), TII->get(TargetOpcode::KILL));
1770  } else
1771    --I;
1772}
1773