• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===-- X86FloatingPoint.cpp - Floating point Reg -> Stack converter ------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file defines the pass which converts floating point instructions from
11 // pseudo registers into register stack instructions.  This pass uses live
12 // variable information to indicate where the FPn registers are used and their
13 // lifetimes.
14 //
15 // The x87 hardware tracks liveness of the stack registers, so it is necessary
16 // to implement exact liveness tracking between basic blocks. The CFG edges are
17 // partitioned into bundles where the same FP registers must be live in
18 // identical stack positions. Instructions are inserted at the end of each basic
19 // block to rearrange the live registers to match the outgoing bundle.
20 //
21 // This approach avoids splitting critical edges at the potential cost of more
22 // live register shuffling instructions when critical edges are present.
23 //
24 //===----------------------------------------------------------------------===//
25 
26 #include "X86.h"
27 #include "X86InstrInfo.h"
28 #include "llvm/ADT/DepthFirstIterator.h"
29 #include "llvm/ADT/STLExtras.h"
30 #include "llvm/ADT/SmallPtrSet.h"
31 #include "llvm/ADT/SmallSet.h"
32 #include "llvm/ADT/SmallVector.h"
33 #include "llvm/ADT/Statistic.h"
34 #include "llvm/CodeGen/EdgeBundles.h"
35 #include "llvm/CodeGen/LivePhysRegs.h"
36 #include "llvm/CodeGen/MachineFunctionPass.h"
37 #include "llvm/CodeGen/MachineInstrBuilder.h"
38 #include "llvm/CodeGen/MachineRegisterInfo.h"
39 #include "llvm/CodeGen/Passes.h"
40 #include "llvm/IR/InlineAsm.h"
41 #include "llvm/Support/Debug.h"
42 #include "llvm/Support/ErrorHandling.h"
43 #include "llvm/Support/raw_ostream.h"
44 #include "llvm/Target/TargetInstrInfo.h"
45 #include "llvm/Target/TargetMachine.h"
46 #include "llvm/Target/TargetSubtargetInfo.h"
47 #include <algorithm>
48 #include <bitset>
49 using namespace llvm;
50 
51 #define DEBUG_TYPE "x86-codegen"
52 
53 STATISTIC(NumFXCH, "Number of fxch instructions inserted");
54 STATISTIC(NumFP  , "Number of floating point instructions");
55 
56 namespace {
57   const unsigned ScratchFPReg = 7;
58 
59   struct FPS : public MachineFunctionPass {
60     static char ID;
FPS__anon351e13210111::FPS61     FPS() : MachineFunctionPass(ID) {
62       initializeEdgeBundlesPass(*PassRegistry::getPassRegistry());
63       // This is really only to keep valgrind quiet.
64       // The logic in isLive() is too much for it.
65       memset(Stack, 0, sizeof(Stack));
66       memset(RegMap, 0, sizeof(RegMap));
67     }
68 
getAnalysisUsage__anon351e13210111::FPS69     void getAnalysisUsage(AnalysisUsage &AU) const override {
70       AU.setPreservesCFG();
71       AU.addRequired<EdgeBundles>();
72       AU.addPreservedID(MachineLoopInfoID);
73       AU.addPreservedID(MachineDominatorsID);
74       MachineFunctionPass::getAnalysisUsage(AU);
75     }
76 
77     bool runOnMachineFunction(MachineFunction &MF) override;
78 
getPassName__anon351e13210111::FPS79     const char *getPassName() const override { return "X86 FP Stackifier"; }
80 
81   private:
82     const TargetInstrInfo *TII; // Machine instruction info.
83 
84     // Two CFG edges are related if they leave the same block, or enter the same
85     // block. The transitive closure of an edge under this relation is a
86     // LiveBundle. It represents a set of CFG edges where the live FP stack
87     // registers must be allocated identically in the x87 stack.
88     //
89     // A LiveBundle is usually all the edges leaving a block, or all the edges
90     // entering a block, but it can contain more edges if critical edges are
91     // present.
92     //
93     // The set of live FP registers in a LiveBundle is calculated by bundleCFG,
94     // but the exact mapping of FP registers to stack slots is fixed later.
95     struct LiveBundle {
96       // Bit mask of live FP registers. Bit 0 = FP0, bit 1 = FP1, &c.
97       unsigned Mask;
98 
99       // Number of pre-assigned live registers in FixStack. This is 0 when the
100       // stack order has not yet been fixed.
101       unsigned FixCount;
102 
103       // Assigned stack order for live-in registers.
104       // FixStack[i] == getStackEntry(i) for all i < FixCount.
105       unsigned char FixStack[8];
106 
LiveBundle__anon351e13210111::FPS::LiveBundle107       LiveBundle() : Mask(0), FixCount(0) {}
108 
109       // Have the live registers been assigned a stack order yet?
isFixed__anon351e13210111::FPS::LiveBundle110       bool isFixed() const { return !Mask || FixCount; }
111     };
112 
113     // Numbered LiveBundle structs. LiveBundles[0] is used for all CFG edges
114     // with no live FP registers.
115     SmallVector<LiveBundle, 8> LiveBundles;
116 
117     // The edge bundle analysis provides indices into the LiveBundles vector.
118     EdgeBundles *Bundles;
119 
120     // Return a bitmask of FP registers in block's live-in list.
calcLiveInMask__anon351e13210111::FPS121     static unsigned calcLiveInMask(MachineBasicBlock *MBB) {
122       unsigned Mask = 0;
123       for (MachineBasicBlock::livein_iterator I = MBB->livein_begin(),
124            E = MBB->livein_end(); I != E; ++I) {
125         unsigned Reg = *I;
126         if (Reg < X86::FP0 || Reg > X86::FP6)
127           continue;
128         Mask |= 1 << (Reg - X86::FP0);
129       }
130       return Mask;
131     }
132 
133     // Partition all the CFG edges into LiveBundles.
134     void bundleCFG(MachineFunction &MF);
135 
136     MachineBasicBlock *MBB;     // Current basic block
137 
138     // The hardware keeps track of how many FP registers are live, so we have
139     // to model that exactly. Usually, each live register corresponds to an
140     // FP<n> register, but when dealing with calls, returns, and inline
141     // assembly, it is sometimes necessary to have live scratch registers.
142     unsigned Stack[8];          // FP<n> Registers in each stack slot...
143     unsigned StackTop;          // The current top of the FP stack.
144 
145     enum {
146       NumFPRegs = 8             // Including scratch pseudo-registers.
147     };
148 
149     // For each live FP<n> register, point to its Stack[] entry.
150     // The first entries correspond to FP0-FP6, the rest are scratch registers
151     // used when we need slightly different live registers than what the
152     // register allocator thinks.
153     unsigned RegMap[NumFPRegs];
154 
155     // Set up our stack model to match the incoming registers to MBB.
156     void setupBlockStack();
157 
158     // Shuffle live registers to match the expectations of successor blocks.
159     void finishBlockStack();
160 
161 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
dumpStack__anon351e13210111::FPS162     void dumpStack() const {
163       dbgs() << "Stack contents:";
164       for (unsigned i = 0; i != StackTop; ++i) {
165         dbgs() << " FP" << Stack[i];
166         assert(RegMap[Stack[i]] == i && "Stack[] doesn't match RegMap[]!");
167       }
168     }
169 #endif
170 
171     /// getSlot - Return the stack slot number a particular register number is
172     /// in.
getSlot__anon351e13210111::FPS173     unsigned getSlot(unsigned RegNo) const {
174       assert(RegNo < NumFPRegs && "Regno out of range!");
175       return RegMap[RegNo];
176     }
177 
178     /// isLive - Is RegNo currently live in the stack?
isLive__anon351e13210111::FPS179     bool isLive(unsigned RegNo) const {
180       unsigned Slot = getSlot(RegNo);
181       return Slot < StackTop && Stack[Slot] == RegNo;
182     }
183 
184     /// getStackEntry - Return the X86::FP<n> register in register ST(i).
getStackEntry__anon351e13210111::FPS185     unsigned getStackEntry(unsigned STi) const {
186       if (STi >= StackTop)
187         report_fatal_error("Access past stack top!");
188       return Stack[StackTop-1-STi];
189     }
190 
191     /// getSTReg - Return the X86::ST(i) register which contains the specified
192     /// FP<RegNo> register.
getSTReg__anon351e13210111::FPS193     unsigned getSTReg(unsigned RegNo) const {
194       return StackTop - 1 - getSlot(RegNo) + X86::ST0;
195     }
196 
197     // pushReg - Push the specified FP<n> register onto the stack.
pushReg__anon351e13210111::FPS198     void pushReg(unsigned Reg) {
199       assert(Reg < NumFPRegs && "Register number out of range!");
200       if (StackTop >= 8)
201         report_fatal_error("Stack overflow!");
202       Stack[StackTop] = Reg;
203       RegMap[Reg] = StackTop++;
204     }
205 
isAtTop__anon351e13210111::FPS206     bool isAtTop(unsigned RegNo) const { return getSlot(RegNo) == StackTop-1; }
moveToTop__anon351e13210111::FPS207     void moveToTop(unsigned RegNo, MachineBasicBlock::iterator I) {
208       DebugLoc dl = I == MBB->end() ? DebugLoc() : I->getDebugLoc();
209       if (isAtTop(RegNo)) return;
210 
211       unsigned STReg = getSTReg(RegNo);
212       unsigned RegOnTop = getStackEntry(0);
213 
214       // Swap the slots the regs are in.
215       std::swap(RegMap[RegNo], RegMap[RegOnTop]);
216 
217       // Swap stack slot contents.
218       if (RegMap[RegOnTop] >= StackTop)
219         report_fatal_error("Access past stack top!");
220       std::swap(Stack[RegMap[RegOnTop]], Stack[StackTop-1]);
221 
222       // Emit an fxch to update the runtime processors version of the state.
223       BuildMI(*MBB, I, dl, TII->get(X86::XCH_F)).addReg(STReg);
224       ++NumFXCH;
225     }
226 
duplicateToTop__anon351e13210111::FPS227     void duplicateToTop(unsigned RegNo, unsigned AsReg, MachineInstr *I) {
228       DebugLoc dl = I == MBB->end() ? DebugLoc() : I->getDebugLoc();
229       unsigned STReg = getSTReg(RegNo);
230       pushReg(AsReg);   // New register on top of stack
231 
232       BuildMI(*MBB, I, dl, TII->get(X86::LD_Frr)).addReg(STReg);
233     }
234 
235     /// popStackAfter - Pop the current value off of the top of the FP stack
236     /// after the specified instruction.
237     void popStackAfter(MachineBasicBlock::iterator &I);
238 
239     /// freeStackSlotAfter - Free the specified register from the register
240     /// stack, so that it is no longer in a register.  If the register is
241     /// currently at the top of the stack, we just pop the current instruction,
242     /// otherwise we store the current top-of-stack into the specified slot,
243     /// then pop the top of stack.
244     void freeStackSlotAfter(MachineBasicBlock::iterator &I, unsigned Reg);
245 
246     /// freeStackSlotBefore - Just the pop, no folding. Return the inserted
247     /// instruction.
248     MachineBasicBlock::iterator
249     freeStackSlotBefore(MachineBasicBlock::iterator I, unsigned FPRegNo);
250 
251     /// Adjust the live registers to be the set in Mask.
252     void adjustLiveRegs(unsigned Mask, MachineBasicBlock::iterator I);
253 
254     /// Shuffle the top FixCount stack entries such that FP reg FixStack[0] is
255     /// st(0), FP reg FixStack[1] is st(1) etc.
256     void shuffleStackTop(const unsigned char *FixStack, unsigned FixCount,
257                          MachineBasicBlock::iterator I);
258 
259     bool processBasicBlock(MachineFunction &MF, MachineBasicBlock &MBB);
260 
261     void handleCall(MachineBasicBlock::iterator &I);
262     void handleZeroArgFP(MachineBasicBlock::iterator &I);
263     void handleOneArgFP(MachineBasicBlock::iterator &I);
264     void handleOneArgFPRW(MachineBasicBlock::iterator &I);
265     void handleTwoArgFP(MachineBasicBlock::iterator &I);
266     void handleCompareFP(MachineBasicBlock::iterator &I);
267     void handleCondMovFP(MachineBasicBlock::iterator &I);
268     void handleSpecialFP(MachineBasicBlock::iterator &I);
269 
270     // Check if a COPY instruction is using FP registers.
isFPCopy__anon351e13210111::FPS271     static bool isFPCopy(MachineInstr *MI) {
272       unsigned DstReg = MI->getOperand(0).getReg();
273       unsigned SrcReg = MI->getOperand(1).getReg();
274 
275       return X86::RFP80RegClass.contains(DstReg) ||
276         X86::RFP80RegClass.contains(SrcReg);
277     }
278 
279     void setKillFlags(MachineBasicBlock &MBB) const;
280   };
281   char FPS::ID = 0;
282 }
283 
createX86FloatingPointStackifierPass()284 FunctionPass *llvm::createX86FloatingPointStackifierPass() { return new FPS(); }
285 
286 /// getFPReg - Return the X86::FPx register number for the specified operand.
287 /// For example, this returns 3 for X86::FP3.
getFPReg(const MachineOperand & MO)288 static unsigned getFPReg(const MachineOperand &MO) {
289   assert(MO.isReg() && "Expected an FP register!");
290   unsigned Reg = MO.getReg();
291   assert(Reg >= X86::FP0 && Reg <= X86::FP6 && "Expected FP register!");
292   return Reg - X86::FP0;
293 }
294 
295 /// runOnMachineFunction - Loop over all of the basic blocks, transforming FP
296 /// register references into FP stack references.
297 ///
runOnMachineFunction(MachineFunction & MF)298 bool FPS::runOnMachineFunction(MachineFunction &MF) {
299   // We only need to run this pass if there are any FP registers used in this
300   // function.  If it is all integer, there is nothing for us to do!
301   bool FPIsUsed = false;
302 
303   static_assert(X86::FP6 == X86::FP0+6, "Register enums aren't sorted right!");
304   for (unsigned i = 0; i <= 6; ++i)
305     if (MF.getRegInfo().isPhysRegUsed(X86::FP0+i)) {
306       FPIsUsed = true;
307       break;
308     }
309 
310   // Early exit.
311   if (!FPIsUsed) return false;
312 
313   Bundles = &getAnalysis<EdgeBundles>();
314   TII = MF.getSubtarget().getInstrInfo();
315 
316   // Prepare cross-MBB liveness.
317   bundleCFG(MF);
318 
319   StackTop = 0;
320 
321   // Process the function in depth first order so that we process at least one
322   // of the predecessors for every reachable block in the function.
323   SmallPtrSet<MachineBasicBlock*, 8> Processed;
324   MachineBasicBlock *Entry = MF.begin();
325 
326   bool Changed = false;
327   for (MachineBasicBlock *BB : depth_first_ext(Entry, Processed))
328     Changed |= processBasicBlock(MF, *BB);
329 
330   // Process any unreachable blocks in arbitrary order now.
331   if (MF.size() != Processed.size())
332     for (MachineFunction::iterator BB = MF.begin(), E = MF.end(); BB != E; ++BB)
333       if (Processed.insert(BB).second)
334         Changed |= processBasicBlock(MF, *BB);
335 
336   LiveBundles.clear();
337 
338   return Changed;
339 }
340 
341 /// bundleCFG - Scan all the basic blocks to determine consistent live-in and
342 /// live-out sets for the FP registers. Consistent means that the set of
343 /// registers live-out from a block is identical to the live-in set of all
344 /// successors. This is not enforced by the normal live-in lists since
345 /// registers may be implicitly defined, or not used by all successors.
bundleCFG(MachineFunction & MF)346 void FPS::bundleCFG(MachineFunction &MF) {
347   assert(LiveBundles.empty() && "Stale data in LiveBundles");
348   LiveBundles.resize(Bundles->getNumBundles());
349 
350   // Gather the actual live-in masks for all MBBs.
351   for (MachineFunction::iterator I = MF.begin(), E = MF.end(); I != E; ++I) {
352     MachineBasicBlock *MBB = I;
353     const unsigned Mask = calcLiveInMask(MBB);
354     if (!Mask)
355       continue;
356     // Update MBB ingoing bundle mask.
357     LiveBundles[Bundles->getBundle(MBB->getNumber(), false)].Mask |= Mask;
358   }
359 }
360 
361 /// processBasicBlock - Loop over all of the instructions in the basic block,
362 /// transforming FP instructions into their stack form.
363 ///
processBasicBlock(MachineFunction & MF,MachineBasicBlock & BB)364 bool FPS::processBasicBlock(MachineFunction &MF, MachineBasicBlock &BB) {
365   bool Changed = false;
366   MBB = &BB;
367 
368   setKillFlags(BB);
369   setupBlockStack();
370 
371   for (MachineBasicBlock::iterator I = BB.begin(); I != BB.end(); ++I) {
372     MachineInstr *MI = I;
373     uint64_t Flags = MI->getDesc().TSFlags;
374 
375     unsigned FPInstClass = Flags & X86II::FPTypeMask;
376     if (MI->isInlineAsm())
377       FPInstClass = X86II::SpecialFP;
378 
379     if (MI->isCopy() && isFPCopy(MI))
380       FPInstClass = X86II::SpecialFP;
381 
382     if (MI->isImplicitDef() &&
383         X86::RFP80RegClass.contains(MI->getOperand(0).getReg()))
384       FPInstClass = X86II::SpecialFP;
385 
386     if (MI->isCall())
387       FPInstClass = X86II::SpecialFP;
388 
389     if (FPInstClass == X86II::NotFP)
390       continue;  // Efficiently ignore non-fp insts!
391 
392     MachineInstr *PrevMI = nullptr;
393     if (I != BB.begin())
394       PrevMI = std::prev(I);
395 
396     ++NumFP;  // Keep track of # of pseudo instrs
397     DEBUG(dbgs() << "\nFPInst:\t" << *MI);
398 
399     // Get dead variables list now because the MI pointer may be deleted as part
400     // of processing!
401     SmallVector<unsigned, 8> DeadRegs;
402     for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
403       const MachineOperand &MO = MI->getOperand(i);
404       if (MO.isReg() && MO.isDead())
405         DeadRegs.push_back(MO.getReg());
406     }
407 
408     switch (FPInstClass) {
409     case X86II::ZeroArgFP:  handleZeroArgFP(I); break;
410     case X86II::OneArgFP:   handleOneArgFP(I);  break;  // fstp ST(0)
411     case X86II::OneArgFPRW: handleOneArgFPRW(I); break; // ST(0) = fsqrt(ST(0))
412     case X86II::TwoArgFP:   handleTwoArgFP(I);  break;
413     case X86II::CompareFP:  handleCompareFP(I); break;
414     case X86II::CondMovFP:  handleCondMovFP(I); break;
415     case X86II::SpecialFP:  handleSpecialFP(I); break;
416     default: llvm_unreachable("Unknown FP Type!");
417     }
418 
419     // Check to see if any of the values defined by this instruction are dead
420     // after definition.  If so, pop them.
421     for (unsigned i = 0, e = DeadRegs.size(); i != e; ++i) {
422       unsigned Reg = DeadRegs[i];
423       // Check if Reg is live on the stack. An inline-asm register operand that
424       // is in the clobber list and marked dead might not be live on the stack.
425       if (Reg >= X86::FP0 && Reg <= X86::FP6 && isLive(Reg-X86::FP0)) {
426         DEBUG(dbgs() << "Register FP#" << Reg-X86::FP0 << " is dead!\n");
427         freeStackSlotAfter(I, Reg-X86::FP0);
428       }
429     }
430 
431     // Print out all of the instructions expanded to if -debug
432     DEBUG(
433       MachineBasicBlock::iterator PrevI(PrevMI);
434       if (I == PrevI) {
435         dbgs() << "Just deleted pseudo instruction\n";
436       } else {
437         MachineBasicBlock::iterator Start = I;
438         // Rewind to first instruction newly inserted.
439         while (Start != BB.begin() && std::prev(Start) != PrevI) --Start;
440         dbgs() << "Inserted instructions:\n\t";
441         Start->print(dbgs());
442         while (++Start != std::next(I)) {}
443       }
444       dumpStack();
445     );
446     (void)PrevMI;
447 
448     Changed = true;
449   }
450 
451   finishBlockStack();
452 
453   return Changed;
454 }
455 
456 /// setupBlockStack - Use the live bundles to set up our model of the stack
457 /// to match predecessors' live out stack.
setupBlockStack()458 void FPS::setupBlockStack() {
459   DEBUG(dbgs() << "\nSetting up live-ins for BB#" << MBB->getNumber()
460                << " derived from " << MBB->getName() << ".\n");
461   StackTop = 0;
462   // Get the live-in bundle for MBB.
463   const LiveBundle &Bundle =
464     LiveBundles[Bundles->getBundle(MBB->getNumber(), false)];
465 
466   if (!Bundle.Mask) {
467     DEBUG(dbgs() << "Block has no FP live-ins.\n");
468     return;
469   }
470 
471   // Depth-first iteration should ensure that we always have an assigned stack.
472   assert(Bundle.isFixed() && "Reached block before any predecessors");
473 
474   // Push the fixed live-in registers.
475   for (unsigned i = Bundle.FixCount; i > 0; --i) {
476     MBB->addLiveIn(X86::ST0+i-1);
477     DEBUG(dbgs() << "Live-in st(" << (i-1) << "): %FP"
478                  << unsigned(Bundle.FixStack[i-1]) << '\n');
479     pushReg(Bundle.FixStack[i-1]);
480   }
481 
482   // Kill off unwanted live-ins. This can happen with a critical edge.
483   // FIXME: We could keep these live registers around as zombies. They may need
484   // to be revived at the end of a short block. It might save a few instrs.
485   adjustLiveRegs(calcLiveInMask(MBB), MBB->begin());
486   DEBUG(MBB->dump());
487 }
488 
489 /// finishBlockStack - Revive live-outs that are implicitly defined out of
490 /// MBB. Shuffle live registers to match the expected fixed stack of any
491 /// predecessors, and ensure that all predecessors are expecting the same
492 /// stack.
finishBlockStack()493 void FPS::finishBlockStack() {
494   // The RET handling below takes care of return blocks for us.
495   if (MBB->succ_empty())
496     return;
497 
498   DEBUG(dbgs() << "Setting up live-outs for BB#" << MBB->getNumber()
499                << " derived from " << MBB->getName() << ".\n");
500 
501   // Get MBB's live-out bundle.
502   unsigned BundleIdx = Bundles->getBundle(MBB->getNumber(), true);
503   LiveBundle &Bundle = LiveBundles[BundleIdx];
504 
505   // We may need to kill and define some registers to match successors.
506   // FIXME: This can probably be combined with the shuffle below.
507   MachineBasicBlock::iterator Term = MBB->getFirstTerminator();
508   adjustLiveRegs(Bundle.Mask, Term);
509 
510   if (!Bundle.Mask) {
511     DEBUG(dbgs() << "No live-outs.\n");
512     return;
513   }
514 
515   // Has the stack order been fixed yet?
516   DEBUG(dbgs() << "LB#" << BundleIdx << ": ");
517   if (Bundle.isFixed()) {
518     DEBUG(dbgs() << "Shuffling stack to match.\n");
519     shuffleStackTop(Bundle.FixStack, Bundle.FixCount, Term);
520   } else {
521     // Not fixed yet, we get to choose.
522     DEBUG(dbgs() << "Fixing stack order now.\n");
523     Bundle.FixCount = StackTop;
524     for (unsigned i = 0; i < StackTop; ++i)
525       Bundle.FixStack[i] = getStackEntry(i);
526   }
527 }
528 
529 
530 //===----------------------------------------------------------------------===//
531 // Efficient Lookup Table Support
532 //===----------------------------------------------------------------------===//
533 
534 namespace {
535   struct TableEntry {
536     uint16_t from;
537     uint16_t to;
operator <__anon351e13210311::TableEntry538     bool operator<(const TableEntry &TE) const { return from < TE.from; }
operator <(const TableEntry & TE,unsigned V)539     friend bool operator<(const TableEntry &TE, unsigned V) {
540       return TE.from < V;
541     }
operator <(unsigned V,const TableEntry & TE)542     friend bool LLVM_ATTRIBUTE_UNUSED operator<(unsigned V,
543                                                 const TableEntry &TE) {
544       return V < TE.from;
545     }
546   };
547 }
548 
549 #ifndef NDEBUG
TableIsSorted(const TableEntry * Table,unsigned NumEntries)550 static bool TableIsSorted(const TableEntry *Table, unsigned NumEntries) {
551   for (unsigned i = 0; i != NumEntries-1; ++i)
552     if (!(Table[i] < Table[i+1])) return false;
553   return true;
554 }
555 #endif
556 
Lookup(const TableEntry * Table,unsigned N,unsigned Opcode)557 static int Lookup(const TableEntry *Table, unsigned N, unsigned Opcode) {
558   const TableEntry *I = std::lower_bound(Table, Table+N, Opcode);
559   if (I != Table+N && I->from == Opcode)
560     return I->to;
561   return -1;
562 }
563 
564 #ifdef NDEBUG
565 #define ASSERT_SORTED(TABLE)
566 #else
567 #define ASSERT_SORTED(TABLE)                                              \
568   { static bool TABLE##Checked = false;                                   \
569     if (!TABLE##Checked) {                                                \
570        assert(TableIsSorted(TABLE, array_lengthof(TABLE)) &&              \
571               "All lookup tables must be sorted for efficient access!");  \
572        TABLE##Checked = true;                                             \
573     }                                                                     \
574   }
575 #endif
576 
577 //===----------------------------------------------------------------------===//
578 // Register File -> Register Stack Mapping Methods
579 //===----------------------------------------------------------------------===//
580 
581 // OpcodeTable - Sorted map of register instructions to their stack version.
582 // The first element is an register file pseudo instruction, the second is the
583 // concrete X86 instruction which uses the register stack.
584 //
585 static const TableEntry OpcodeTable[] = {
586   { X86::ABS_Fp32     , X86::ABS_F     },
587   { X86::ABS_Fp64     , X86::ABS_F     },
588   { X86::ABS_Fp80     , X86::ABS_F     },
589   { X86::ADD_Fp32m    , X86::ADD_F32m  },
590   { X86::ADD_Fp64m    , X86::ADD_F64m  },
591   { X86::ADD_Fp64m32  , X86::ADD_F32m  },
592   { X86::ADD_Fp80m32  , X86::ADD_F32m  },
593   { X86::ADD_Fp80m64  , X86::ADD_F64m  },
594   { X86::ADD_FpI16m32 , X86::ADD_FI16m },
595   { X86::ADD_FpI16m64 , X86::ADD_FI16m },
596   { X86::ADD_FpI16m80 , X86::ADD_FI16m },
597   { X86::ADD_FpI32m32 , X86::ADD_FI32m },
598   { X86::ADD_FpI32m64 , X86::ADD_FI32m },
599   { X86::ADD_FpI32m80 , X86::ADD_FI32m },
600   { X86::CHS_Fp32     , X86::CHS_F     },
601   { X86::CHS_Fp64     , X86::CHS_F     },
602   { X86::CHS_Fp80     , X86::CHS_F     },
603   { X86::CMOVBE_Fp32  , X86::CMOVBE_F  },
604   { X86::CMOVBE_Fp64  , X86::CMOVBE_F  },
605   { X86::CMOVBE_Fp80  , X86::CMOVBE_F  },
606   { X86::CMOVB_Fp32   , X86::CMOVB_F   },
607   { X86::CMOVB_Fp64   , X86::CMOVB_F  },
608   { X86::CMOVB_Fp80   , X86::CMOVB_F  },
609   { X86::CMOVE_Fp32   , X86::CMOVE_F  },
610   { X86::CMOVE_Fp64   , X86::CMOVE_F   },
611   { X86::CMOVE_Fp80   , X86::CMOVE_F   },
612   { X86::CMOVNBE_Fp32 , X86::CMOVNBE_F },
613   { X86::CMOVNBE_Fp64 , X86::CMOVNBE_F },
614   { X86::CMOVNBE_Fp80 , X86::CMOVNBE_F },
615   { X86::CMOVNB_Fp32  , X86::CMOVNB_F  },
616   { X86::CMOVNB_Fp64  , X86::CMOVNB_F  },
617   { X86::CMOVNB_Fp80  , X86::CMOVNB_F  },
618   { X86::CMOVNE_Fp32  , X86::CMOVNE_F  },
619   { X86::CMOVNE_Fp64  , X86::CMOVNE_F  },
620   { X86::CMOVNE_Fp80  , X86::CMOVNE_F  },
621   { X86::CMOVNP_Fp32  , X86::CMOVNP_F  },
622   { X86::CMOVNP_Fp64  , X86::CMOVNP_F  },
623   { X86::CMOVNP_Fp80  , X86::CMOVNP_F  },
624   { X86::CMOVP_Fp32   , X86::CMOVP_F   },
625   { X86::CMOVP_Fp64   , X86::CMOVP_F   },
626   { X86::CMOVP_Fp80   , X86::CMOVP_F   },
627   { X86::COS_Fp32     , X86::COS_F     },
628   { X86::COS_Fp64     , X86::COS_F     },
629   { X86::COS_Fp80     , X86::COS_F     },
630   { X86::DIVR_Fp32m   , X86::DIVR_F32m },
631   { X86::DIVR_Fp64m   , X86::DIVR_F64m },
632   { X86::DIVR_Fp64m32 , X86::DIVR_F32m },
633   { X86::DIVR_Fp80m32 , X86::DIVR_F32m },
634   { X86::DIVR_Fp80m64 , X86::DIVR_F64m },
635   { X86::DIVR_FpI16m32, X86::DIVR_FI16m},
636   { X86::DIVR_FpI16m64, X86::DIVR_FI16m},
637   { X86::DIVR_FpI16m80, X86::DIVR_FI16m},
638   { X86::DIVR_FpI32m32, X86::DIVR_FI32m},
639   { X86::DIVR_FpI32m64, X86::DIVR_FI32m},
640   { X86::DIVR_FpI32m80, X86::DIVR_FI32m},
641   { X86::DIV_Fp32m    , X86::DIV_F32m  },
642   { X86::DIV_Fp64m    , X86::DIV_F64m  },
643   { X86::DIV_Fp64m32  , X86::DIV_F32m  },
644   { X86::DIV_Fp80m32  , X86::DIV_F32m  },
645   { X86::DIV_Fp80m64  , X86::DIV_F64m  },
646   { X86::DIV_FpI16m32 , X86::DIV_FI16m },
647   { X86::DIV_FpI16m64 , X86::DIV_FI16m },
648   { X86::DIV_FpI16m80 , X86::DIV_FI16m },
649   { X86::DIV_FpI32m32 , X86::DIV_FI32m },
650   { X86::DIV_FpI32m64 , X86::DIV_FI32m },
651   { X86::DIV_FpI32m80 , X86::DIV_FI32m },
652   { X86::ILD_Fp16m32  , X86::ILD_F16m  },
653   { X86::ILD_Fp16m64  , X86::ILD_F16m  },
654   { X86::ILD_Fp16m80  , X86::ILD_F16m  },
655   { X86::ILD_Fp32m32  , X86::ILD_F32m  },
656   { X86::ILD_Fp32m64  , X86::ILD_F32m  },
657   { X86::ILD_Fp32m80  , X86::ILD_F32m  },
658   { X86::ILD_Fp64m32  , X86::ILD_F64m  },
659   { X86::ILD_Fp64m64  , X86::ILD_F64m  },
660   { X86::ILD_Fp64m80  , X86::ILD_F64m  },
661   { X86::ISTT_Fp16m32 , X86::ISTT_FP16m},
662   { X86::ISTT_Fp16m64 , X86::ISTT_FP16m},
663   { X86::ISTT_Fp16m80 , X86::ISTT_FP16m},
664   { X86::ISTT_Fp32m32 , X86::ISTT_FP32m},
665   { X86::ISTT_Fp32m64 , X86::ISTT_FP32m},
666   { X86::ISTT_Fp32m80 , X86::ISTT_FP32m},
667   { X86::ISTT_Fp64m32 , X86::ISTT_FP64m},
668   { X86::ISTT_Fp64m64 , X86::ISTT_FP64m},
669   { X86::ISTT_Fp64m80 , X86::ISTT_FP64m},
670   { X86::IST_Fp16m32  , X86::IST_F16m  },
671   { X86::IST_Fp16m64  , X86::IST_F16m  },
672   { X86::IST_Fp16m80  , X86::IST_F16m  },
673   { X86::IST_Fp32m32  , X86::IST_F32m  },
674   { X86::IST_Fp32m64  , X86::IST_F32m  },
675   { X86::IST_Fp32m80  , X86::IST_F32m  },
676   { X86::IST_Fp64m32  , X86::IST_FP64m },
677   { X86::IST_Fp64m64  , X86::IST_FP64m },
678   { X86::IST_Fp64m80  , X86::IST_FP64m },
679   { X86::LD_Fp032     , X86::LD_F0     },
680   { X86::LD_Fp064     , X86::LD_F0     },
681   { X86::LD_Fp080     , X86::LD_F0     },
682   { X86::LD_Fp132     , X86::LD_F1     },
683   { X86::LD_Fp164     , X86::LD_F1     },
684   { X86::LD_Fp180     , X86::LD_F1     },
685   { X86::LD_Fp32m     , X86::LD_F32m   },
686   { X86::LD_Fp32m64   , X86::LD_F32m   },
687   { X86::LD_Fp32m80   , X86::LD_F32m   },
688   { X86::LD_Fp64m     , X86::LD_F64m   },
689   { X86::LD_Fp64m80   , X86::LD_F64m   },
690   { X86::LD_Fp80m     , X86::LD_F80m   },
691   { X86::MUL_Fp32m    , X86::MUL_F32m  },
692   { X86::MUL_Fp64m    , X86::MUL_F64m  },
693   { X86::MUL_Fp64m32  , X86::MUL_F32m  },
694   { X86::MUL_Fp80m32  , X86::MUL_F32m  },
695   { X86::MUL_Fp80m64  , X86::MUL_F64m  },
696   { X86::MUL_FpI16m32 , X86::MUL_FI16m },
697   { X86::MUL_FpI16m64 , X86::MUL_FI16m },
698   { X86::MUL_FpI16m80 , X86::MUL_FI16m },
699   { X86::MUL_FpI32m32 , X86::MUL_FI32m },
700   { X86::MUL_FpI32m64 , X86::MUL_FI32m },
701   { X86::MUL_FpI32m80 , X86::MUL_FI32m },
702   { X86::SIN_Fp32     , X86::SIN_F     },
703   { X86::SIN_Fp64     , X86::SIN_F     },
704   { X86::SIN_Fp80     , X86::SIN_F     },
705   { X86::SQRT_Fp32    , X86::SQRT_F    },
706   { X86::SQRT_Fp64    , X86::SQRT_F    },
707   { X86::SQRT_Fp80    , X86::SQRT_F    },
708   { X86::ST_Fp32m     , X86::ST_F32m   },
709   { X86::ST_Fp64m     , X86::ST_F64m   },
710   { X86::ST_Fp64m32   , X86::ST_F32m   },
711   { X86::ST_Fp80m32   , X86::ST_F32m   },
712   { X86::ST_Fp80m64   , X86::ST_F64m   },
713   { X86::ST_FpP80m    , X86::ST_FP80m  },
714   { X86::SUBR_Fp32m   , X86::SUBR_F32m },
715   { X86::SUBR_Fp64m   , X86::SUBR_F64m },
716   { X86::SUBR_Fp64m32 , X86::SUBR_F32m },
717   { X86::SUBR_Fp80m32 , X86::SUBR_F32m },
718   { X86::SUBR_Fp80m64 , X86::SUBR_F64m },
719   { X86::SUBR_FpI16m32, X86::SUBR_FI16m},
720   { X86::SUBR_FpI16m64, X86::SUBR_FI16m},
721   { X86::SUBR_FpI16m80, X86::SUBR_FI16m},
722   { X86::SUBR_FpI32m32, X86::SUBR_FI32m},
723   { X86::SUBR_FpI32m64, X86::SUBR_FI32m},
724   { X86::SUBR_FpI32m80, X86::SUBR_FI32m},
725   { X86::SUB_Fp32m    , X86::SUB_F32m  },
726   { X86::SUB_Fp64m    , X86::SUB_F64m  },
727   { X86::SUB_Fp64m32  , X86::SUB_F32m  },
728   { X86::SUB_Fp80m32  , X86::SUB_F32m  },
729   { X86::SUB_Fp80m64  , X86::SUB_F64m  },
730   { X86::SUB_FpI16m32 , X86::SUB_FI16m },
731   { X86::SUB_FpI16m64 , X86::SUB_FI16m },
732   { X86::SUB_FpI16m80 , X86::SUB_FI16m },
733   { X86::SUB_FpI32m32 , X86::SUB_FI32m },
734   { X86::SUB_FpI32m64 , X86::SUB_FI32m },
735   { X86::SUB_FpI32m80 , X86::SUB_FI32m },
736   { X86::TST_Fp32     , X86::TST_F     },
737   { X86::TST_Fp64     , X86::TST_F     },
738   { X86::TST_Fp80     , X86::TST_F     },
739   { X86::UCOM_FpIr32  , X86::UCOM_FIr  },
740   { X86::UCOM_FpIr64  , X86::UCOM_FIr  },
741   { X86::UCOM_FpIr80  , X86::UCOM_FIr  },
742   { X86::UCOM_Fpr32   , X86::UCOM_Fr   },
743   { X86::UCOM_Fpr64   , X86::UCOM_Fr   },
744   { X86::UCOM_Fpr80   , X86::UCOM_Fr   },
745 };
746 
getConcreteOpcode(unsigned Opcode)747 static unsigned getConcreteOpcode(unsigned Opcode) {
748   ASSERT_SORTED(OpcodeTable);
749   int Opc = Lookup(OpcodeTable, array_lengthof(OpcodeTable), Opcode);
750   assert(Opc != -1 && "FP Stack instruction not in OpcodeTable!");
751   return Opc;
752 }
753 
754 //===----------------------------------------------------------------------===//
755 // Helper Methods
756 //===----------------------------------------------------------------------===//
757 
758 // PopTable - Sorted map of instructions to their popping version.  The first
759 // element is an instruction, the second is the version which pops.
760 //
761 static const TableEntry PopTable[] = {
762   { X86::ADD_FrST0 , X86::ADD_FPrST0  },
763 
764   { X86::DIVR_FrST0, X86::DIVR_FPrST0 },
765   { X86::DIV_FrST0 , X86::DIV_FPrST0  },
766 
767   { X86::IST_F16m  , X86::IST_FP16m   },
768   { X86::IST_F32m  , X86::IST_FP32m   },
769 
770   { X86::MUL_FrST0 , X86::MUL_FPrST0  },
771 
772   { X86::ST_F32m   , X86::ST_FP32m    },
773   { X86::ST_F64m   , X86::ST_FP64m    },
774   { X86::ST_Frr    , X86::ST_FPrr     },
775 
776   { X86::SUBR_FrST0, X86::SUBR_FPrST0 },
777   { X86::SUB_FrST0 , X86::SUB_FPrST0  },
778 
779   { X86::UCOM_FIr  , X86::UCOM_FIPr   },
780 
781   { X86::UCOM_FPr  , X86::UCOM_FPPr   },
782   { X86::UCOM_Fr   , X86::UCOM_FPr    },
783 };
784 
785 /// popStackAfter - Pop the current value off of the top of the FP stack after
786 /// the specified instruction.  This attempts to be sneaky and combine the pop
787 /// into the instruction itself if possible.  The iterator is left pointing to
788 /// the last instruction, be it a new pop instruction inserted, or the old
789 /// instruction if it was modified in place.
790 ///
popStackAfter(MachineBasicBlock::iterator & I)791 void FPS::popStackAfter(MachineBasicBlock::iterator &I) {
792   MachineInstr* MI = I;
793   DebugLoc dl = MI->getDebugLoc();
794   ASSERT_SORTED(PopTable);
795   if (StackTop == 0)
796     report_fatal_error("Cannot pop empty stack!");
797   RegMap[Stack[--StackTop]] = ~0;     // Update state
798 
799   // Check to see if there is a popping version of this instruction...
800   int Opcode = Lookup(PopTable, array_lengthof(PopTable), I->getOpcode());
801   if (Opcode != -1) {
802     I->setDesc(TII->get(Opcode));
803     if (Opcode == X86::UCOM_FPPr)
804       I->RemoveOperand(0);
805   } else {    // Insert an explicit pop
806     I = BuildMI(*MBB, ++I, dl, TII->get(X86::ST_FPrr)).addReg(X86::ST0);
807   }
808 }
809 
810 /// freeStackSlotAfter - Free the specified register from the register stack, so
811 /// that it is no longer in a register.  If the register is currently at the top
812 /// of the stack, we just pop the current instruction, otherwise we store the
813 /// current top-of-stack into the specified slot, then pop the top of stack.
freeStackSlotAfter(MachineBasicBlock::iterator & I,unsigned FPRegNo)814 void FPS::freeStackSlotAfter(MachineBasicBlock::iterator &I, unsigned FPRegNo) {
815   if (getStackEntry(0) == FPRegNo) {  // already at the top of stack? easy.
816     popStackAfter(I);
817     return;
818   }
819 
820   // Otherwise, store the top of stack into the dead slot, killing the operand
821   // without having to add in an explicit xchg then pop.
822   //
823   I = freeStackSlotBefore(++I, FPRegNo);
824 }
825 
826 /// freeStackSlotBefore - Free the specified register without trying any
827 /// folding.
828 MachineBasicBlock::iterator
freeStackSlotBefore(MachineBasicBlock::iterator I,unsigned FPRegNo)829 FPS::freeStackSlotBefore(MachineBasicBlock::iterator I, unsigned FPRegNo) {
830   unsigned STReg    = getSTReg(FPRegNo);
831   unsigned OldSlot  = getSlot(FPRegNo);
832   unsigned TopReg   = Stack[StackTop-1];
833   Stack[OldSlot]    = TopReg;
834   RegMap[TopReg]    = OldSlot;
835   RegMap[FPRegNo]   = ~0;
836   Stack[--StackTop] = ~0;
837   return BuildMI(*MBB, I, DebugLoc(), TII->get(X86::ST_FPrr))
838       .addReg(STReg)
839       .getInstr();
840 }
841 
842 /// adjustLiveRegs - Kill and revive registers such that exactly the FP
843 /// registers with a bit in Mask are live.
adjustLiveRegs(unsigned Mask,MachineBasicBlock::iterator I)844 void FPS::adjustLiveRegs(unsigned Mask, MachineBasicBlock::iterator I) {
845   unsigned Defs = Mask;
846   unsigned Kills = 0;
847   for (unsigned i = 0; i < StackTop; ++i) {
848     unsigned RegNo = Stack[i];
849     if (!(Defs & (1 << RegNo)))
850       // This register is live, but we don't want it.
851       Kills |= (1 << RegNo);
852     else
853       // We don't need to imp-def this live register.
854       Defs &= ~(1 << RegNo);
855   }
856   assert((Kills & Defs) == 0 && "Register needs killing and def'ing?");
857 
858   // Produce implicit-defs for free by using killed registers.
859   while (Kills && Defs) {
860     unsigned KReg = countTrailingZeros(Kills);
861     unsigned DReg = countTrailingZeros(Defs);
862     DEBUG(dbgs() << "Renaming %FP" << KReg << " as imp %FP" << DReg << "\n");
863     std::swap(Stack[getSlot(KReg)], Stack[getSlot(DReg)]);
864     std::swap(RegMap[KReg], RegMap[DReg]);
865     Kills &= ~(1 << KReg);
866     Defs &= ~(1 << DReg);
867   }
868 
869   // Kill registers by popping.
870   if (Kills && I != MBB->begin()) {
871     MachineBasicBlock::iterator I2 = std::prev(I);
872     while (StackTop) {
873       unsigned KReg = getStackEntry(0);
874       if (!(Kills & (1 << KReg)))
875         break;
876       DEBUG(dbgs() << "Popping %FP" << KReg << "\n");
877       popStackAfter(I2);
878       Kills &= ~(1 << KReg);
879     }
880   }
881 
882   // Manually kill the rest.
883   while (Kills) {
884     unsigned KReg = countTrailingZeros(Kills);
885     DEBUG(dbgs() << "Killing %FP" << KReg << "\n");
886     freeStackSlotBefore(I, KReg);
887     Kills &= ~(1 << KReg);
888   }
889 
890   // Load zeros for all the imp-defs.
891   while(Defs) {
892     unsigned DReg = countTrailingZeros(Defs);
893     DEBUG(dbgs() << "Defining %FP" << DReg << " as 0\n");
894     BuildMI(*MBB, I, DebugLoc(), TII->get(X86::LD_F0));
895     pushReg(DReg);
896     Defs &= ~(1 << DReg);
897   }
898 
899   // Now we should have the correct registers live.
900   DEBUG(dumpStack());
901   assert(StackTop == countPopulation(Mask) && "Live count mismatch");
902 }
903 
904 /// shuffleStackTop - emit fxch instructions before I to shuffle the top
905 /// FixCount entries into the order given by FixStack.
906 /// FIXME: Is there a better algorithm than insertion sort?
shuffleStackTop(const unsigned char * FixStack,unsigned FixCount,MachineBasicBlock::iterator I)907 void FPS::shuffleStackTop(const unsigned char *FixStack,
908                           unsigned FixCount,
909                           MachineBasicBlock::iterator I) {
910   // Move items into place, starting from the desired stack bottom.
911   while (FixCount--) {
912     // Old register at position FixCount.
913     unsigned OldReg = getStackEntry(FixCount);
914     // Desired register at position FixCount.
915     unsigned Reg = FixStack[FixCount];
916     if (Reg == OldReg)
917       continue;
918     // (Reg st0) (OldReg st0) = (Reg OldReg st0)
919     moveToTop(Reg, I);
920     if (FixCount > 0)
921       moveToTop(OldReg, I);
922   }
923   DEBUG(dumpStack());
924 }
925 
926 
927 //===----------------------------------------------------------------------===//
928 // Instruction transformation implementation
929 //===----------------------------------------------------------------------===//
930 
handleCall(MachineBasicBlock::iterator & I)931 void FPS::handleCall(MachineBasicBlock::iterator &I) {
932   unsigned STReturns = 0;
933 
934   for (const auto &MO : I->operands()) {
935     if (!MO.isReg())
936       continue;
937 
938     unsigned R = MO.getReg() - X86::FP0;
939 
940     if (R < 8) {
941       assert(MO.isDef() && MO.isImplicit());
942       STReturns |= 1 << R;
943     }
944   }
945 
946   unsigned N = countTrailingOnes(STReturns);
947 
948   // FP registers used for function return must be consecutive starting at
949   // FP0.
950   assert(STReturns == 0 || (isMask_32(STReturns) && N <= 2));
951 
952   for (unsigned I = 0; I < N; ++I)
953     pushReg(N - I - 1);
954 }
955 
956 /// handleZeroArgFP - ST(0) = fld0    ST(0) = flds <mem>
957 ///
handleZeroArgFP(MachineBasicBlock::iterator & I)958 void FPS::handleZeroArgFP(MachineBasicBlock::iterator &I) {
959   MachineInstr *MI = I;
960   unsigned DestReg = getFPReg(MI->getOperand(0));
961 
962   // Change from the pseudo instruction to the concrete instruction.
963   MI->RemoveOperand(0);   // Remove the explicit ST(0) operand
964   MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
965 
966   // Result gets pushed on the stack.
967   pushReg(DestReg);
968 }
969 
970 /// handleOneArgFP - fst <mem>, ST(0)
971 ///
handleOneArgFP(MachineBasicBlock::iterator & I)972 void FPS::handleOneArgFP(MachineBasicBlock::iterator &I) {
973   MachineInstr *MI = I;
974   unsigned NumOps = MI->getDesc().getNumOperands();
975   assert((NumOps == X86::AddrNumOperands + 1 || NumOps == 1) &&
976          "Can only handle fst* & ftst instructions!");
977 
978   // Is this the last use of the source register?
979   unsigned Reg = getFPReg(MI->getOperand(NumOps-1));
980   bool KillsSrc = MI->killsRegister(X86::FP0+Reg);
981 
982   // FISTP64m is strange because there isn't a non-popping versions.
983   // If we have one _and_ we don't want to pop the operand, duplicate the value
984   // on the stack instead of moving it.  This ensure that popping the value is
985   // always ok.
986   // Ditto FISTTP16m, FISTTP32m, FISTTP64m, ST_FpP80m.
987   //
988   if (!KillsSrc &&
989       (MI->getOpcode() == X86::IST_Fp64m32 ||
990        MI->getOpcode() == X86::ISTT_Fp16m32 ||
991        MI->getOpcode() == X86::ISTT_Fp32m32 ||
992        MI->getOpcode() == X86::ISTT_Fp64m32 ||
993        MI->getOpcode() == X86::IST_Fp64m64 ||
994        MI->getOpcode() == X86::ISTT_Fp16m64 ||
995        MI->getOpcode() == X86::ISTT_Fp32m64 ||
996        MI->getOpcode() == X86::ISTT_Fp64m64 ||
997        MI->getOpcode() == X86::IST_Fp64m80 ||
998        MI->getOpcode() == X86::ISTT_Fp16m80 ||
999        MI->getOpcode() == X86::ISTT_Fp32m80 ||
1000        MI->getOpcode() == X86::ISTT_Fp64m80 ||
1001        MI->getOpcode() == X86::ST_FpP80m)) {
1002     duplicateToTop(Reg, ScratchFPReg, I);
1003   } else {
1004     moveToTop(Reg, I);            // Move to the top of the stack...
1005   }
1006 
1007   // Convert from the pseudo instruction to the concrete instruction.
1008   MI->RemoveOperand(NumOps-1);    // Remove explicit ST(0) operand
1009   MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1010 
1011   if (MI->getOpcode() == X86::IST_FP64m ||
1012       MI->getOpcode() == X86::ISTT_FP16m ||
1013       MI->getOpcode() == X86::ISTT_FP32m ||
1014       MI->getOpcode() == X86::ISTT_FP64m ||
1015       MI->getOpcode() == X86::ST_FP80m) {
1016     if (StackTop == 0)
1017       report_fatal_error("Stack empty??");
1018     --StackTop;
1019   } else if (KillsSrc) { // Last use of operand?
1020     popStackAfter(I);
1021   }
1022 }
1023 
1024 
1025 /// handleOneArgFPRW: Handle instructions that read from the top of stack and
1026 /// replace the value with a newly computed value.  These instructions may have
1027 /// non-fp operands after their FP operands.
1028 ///
1029 ///  Examples:
1030 ///     R1 = fchs R2
1031 ///     R1 = fadd R2, [mem]
1032 ///
handleOneArgFPRW(MachineBasicBlock::iterator & I)1033 void FPS::handleOneArgFPRW(MachineBasicBlock::iterator &I) {
1034   MachineInstr *MI = I;
1035 #ifndef NDEBUG
1036   unsigned NumOps = MI->getDesc().getNumOperands();
1037   assert(NumOps >= 2 && "FPRW instructions must have 2 ops!!");
1038 #endif
1039 
1040   // Is this the last use of the source register?
1041   unsigned Reg = getFPReg(MI->getOperand(1));
1042   bool KillsSrc = MI->killsRegister(X86::FP0+Reg);
1043 
1044   if (KillsSrc) {
1045     // If this is the last use of the source register, just make sure it's on
1046     // the top of the stack.
1047     moveToTop(Reg, I);
1048     if (StackTop == 0)
1049       report_fatal_error("Stack cannot be empty!");
1050     --StackTop;
1051     pushReg(getFPReg(MI->getOperand(0)));
1052   } else {
1053     // If this is not the last use of the source register, _copy_ it to the top
1054     // of the stack.
1055     duplicateToTop(Reg, getFPReg(MI->getOperand(0)), I);
1056   }
1057 
1058   // Change from the pseudo instruction to the concrete instruction.
1059   MI->RemoveOperand(1);   // Drop the source operand.
1060   MI->RemoveOperand(0);   // Drop the destination operand.
1061   MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1062 }
1063 
1064 
1065 //===----------------------------------------------------------------------===//
1066 // Define tables of various ways to map pseudo instructions
1067 //
1068 
1069 // ForwardST0Table - Map: A = B op C  into: ST(0) = ST(0) op ST(i)
1070 static const TableEntry ForwardST0Table[] = {
1071   { X86::ADD_Fp32  , X86::ADD_FST0r },
1072   { X86::ADD_Fp64  , X86::ADD_FST0r },
1073   { X86::ADD_Fp80  , X86::ADD_FST0r },
1074   { X86::DIV_Fp32  , X86::DIV_FST0r },
1075   { X86::DIV_Fp64  , X86::DIV_FST0r },
1076   { X86::DIV_Fp80  , X86::DIV_FST0r },
1077   { X86::MUL_Fp32  , X86::MUL_FST0r },
1078   { X86::MUL_Fp64  , X86::MUL_FST0r },
1079   { X86::MUL_Fp80  , X86::MUL_FST0r },
1080   { X86::SUB_Fp32  , X86::SUB_FST0r },
1081   { X86::SUB_Fp64  , X86::SUB_FST0r },
1082   { X86::SUB_Fp80  , X86::SUB_FST0r },
1083 };
1084 
1085 // ReverseST0Table - Map: A = B op C  into: ST(0) = ST(i) op ST(0)
1086 static const TableEntry ReverseST0Table[] = {
1087   { X86::ADD_Fp32  , X86::ADD_FST0r  },   // commutative
1088   { X86::ADD_Fp64  , X86::ADD_FST0r  },   // commutative
1089   { X86::ADD_Fp80  , X86::ADD_FST0r  },   // commutative
1090   { X86::DIV_Fp32  , X86::DIVR_FST0r },
1091   { X86::DIV_Fp64  , X86::DIVR_FST0r },
1092   { X86::DIV_Fp80  , X86::DIVR_FST0r },
1093   { X86::MUL_Fp32  , X86::MUL_FST0r  },   // commutative
1094   { X86::MUL_Fp64  , X86::MUL_FST0r  },   // commutative
1095   { X86::MUL_Fp80  , X86::MUL_FST0r  },   // commutative
1096   { X86::SUB_Fp32  , X86::SUBR_FST0r },
1097   { X86::SUB_Fp64  , X86::SUBR_FST0r },
1098   { X86::SUB_Fp80  , X86::SUBR_FST0r },
1099 };
1100 
1101 // ForwardSTiTable - Map: A = B op C  into: ST(i) = ST(0) op ST(i)
1102 static const TableEntry ForwardSTiTable[] = {
1103   { X86::ADD_Fp32  , X86::ADD_FrST0  },   // commutative
1104   { X86::ADD_Fp64  , X86::ADD_FrST0  },   // commutative
1105   { X86::ADD_Fp80  , X86::ADD_FrST0  },   // commutative
1106   { X86::DIV_Fp32  , X86::DIVR_FrST0 },
1107   { X86::DIV_Fp64  , X86::DIVR_FrST0 },
1108   { X86::DIV_Fp80  , X86::DIVR_FrST0 },
1109   { X86::MUL_Fp32  , X86::MUL_FrST0  },   // commutative
1110   { X86::MUL_Fp64  , X86::MUL_FrST0  },   // commutative
1111   { X86::MUL_Fp80  , X86::MUL_FrST0  },   // commutative
1112   { X86::SUB_Fp32  , X86::SUBR_FrST0 },
1113   { X86::SUB_Fp64  , X86::SUBR_FrST0 },
1114   { X86::SUB_Fp80  , X86::SUBR_FrST0 },
1115 };
1116 
1117 // ReverseSTiTable - Map: A = B op C  into: ST(i) = ST(i) op ST(0)
1118 static const TableEntry ReverseSTiTable[] = {
1119   { X86::ADD_Fp32  , X86::ADD_FrST0 },
1120   { X86::ADD_Fp64  , X86::ADD_FrST0 },
1121   { X86::ADD_Fp80  , X86::ADD_FrST0 },
1122   { X86::DIV_Fp32  , X86::DIV_FrST0 },
1123   { X86::DIV_Fp64  , X86::DIV_FrST0 },
1124   { X86::DIV_Fp80  , X86::DIV_FrST0 },
1125   { X86::MUL_Fp32  , X86::MUL_FrST0 },
1126   { X86::MUL_Fp64  , X86::MUL_FrST0 },
1127   { X86::MUL_Fp80  , X86::MUL_FrST0 },
1128   { X86::SUB_Fp32  , X86::SUB_FrST0 },
1129   { X86::SUB_Fp64  , X86::SUB_FrST0 },
1130   { X86::SUB_Fp80  , X86::SUB_FrST0 },
1131 };
1132 
1133 
1134 /// handleTwoArgFP - Handle instructions like FADD and friends which are virtual
1135 /// instructions which need to be simplified and possibly transformed.
1136 ///
1137 /// Result: ST(0) = fsub  ST(0), ST(i)
1138 ///         ST(i) = fsub  ST(0), ST(i)
1139 ///         ST(0) = fsubr ST(0), ST(i)
1140 ///         ST(i) = fsubr ST(0), ST(i)
1141 ///
handleTwoArgFP(MachineBasicBlock::iterator & I)1142 void FPS::handleTwoArgFP(MachineBasicBlock::iterator &I) {
1143   ASSERT_SORTED(ForwardST0Table); ASSERT_SORTED(ReverseST0Table);
1144   ASSERT_SORTED(ForwardSTiTable); ASSERT_SORTED(ReverseSTiTable);
1145   MachineInstr *MI = I;
1146 
1147   unsigned NumOperands = MI->getDesc().getNumOperands();
1148   assert(NumOperands == 3 && "Illegal TwoArgFP instruction!");
1149   unsigned Dest = getFPReg(MI->getOperand(0));
1150   unsigned Op0 = getFPReg(MI->getOperand(NumOperands-2));
1151   unsigned Op1 = getFPReg(MI->getOperand(NumOperands-1));
1152   bool KillsOp0 = MI->killsRegister(X86::FP0+Op0);
1153   bool KillsOp1 = MI->killsRegister(X86::FP0+Op1);
1154   DebugLoc dl = MI->getDebugLoc();
1155 
1156   unsigned TOS = getStackEntry(0);
1157 
1158   // One of our operands must be on the top of the stack.  If neither is yet, we
1159   // need to move one.
1160   if (Op0 != TOS && Op1 != TOS) {   // No operand at TOS?
1161     // We can choose to move either operand to the top of the stack.  If one of
1162     // the operands is killed by this instruction, we want that one so that we
1163     // can update right on top of the old version.
1164     if (KillsOp0) {
1165       moveToTop(Op0, I);         // Move dead operand to TOS.
1166       TOS = Op0;
1167     } else if (KillsOp1) {
1168       moveToTop(Op1, I);
1169       TOS = Op1;
1170     } else {
1171       // All of the operands are live after this instruction executes, so we
1172       // cannot update on top of any operand.  Because of this, we must
1173       // duplicate one of the stack elements to the top.  It doesn't matter
1174       // which one we pick.
1175       //
1176       duplicateToTop(Op0, Dest, I);
1177       Op0 = TOS = Dest;
1178       KillsOp0 = true;
1179     }
1180   } else if (!KillsOp0 && !KillsOp1) {
1181     // If we DO have one of our operands at the top of the stack, but we don't
1182     // have a dead operand, we must duplicate one of the operands to a new slot
1183     // on the stack.
1184     duplicateToTop(Op0, Dest, I);
1185     Op0 = TOS = Dest;
1186     KillsOp0 = true;
1187   }
1188 
1189   // Now we know that one of our operands is on the top of the stack, and at
1190   // least one of our operands is killed by this instruction.
1191   assert((TOS == Op0 || TOS == Op1) && (KillsOp0 || KillsOp1) &&
1192          "Stack conditions not set up right!");
1193 
1194   // We decide which form to use based on what is on the top of the stack, and
1195   // which operand is killed by this instruction.
1196   const TableEntry *InstTable;
1197   bool isForward = TOS == Op0;
1198   bool updateST0 = (TOS == Op0 && !KillsOp1) || (TOS == Op1 && !KillsOp0);
1199   if (updateST0) {
1200     if (isForward)
1201       InstTable = ForwardST0Table;
1202     else
1203       InstTable = ReverseST0Table;
1204   } else {
1205     if (isForward)
1206       InstTable = ForwardSTiTable;
1207     else
1208       InstTable = ReverseSTiTable;
1209   }
1210 
1211   int Opcode = Lookup(InstTable, array_lengthof(ForwardST0Table),
1212                       MI->getOpcode());
1213   assert(Opcode != -1 && "Unknown TwoArgFP pseudo instruction!");
1214 
1215   // NotTOS - The register which is not on the top of stack...
1216   unsigned NotTOS = (TOS == Op0) ? Op1 : Op0;
1217 
1218   // Replace the old instruction with a new instruction
1219   MBB->remove(I++);
1220   I = BuildMI(*MBB, I, dl, TII->get(Opcode)).addReg(getSTReg(NotTOS));
1221 
1222   // If both operands are killed, pop one off of the stack in addition to
1223   // overwriting the other one.
1224   if (KillsOp0 && KillsOp1 && Op0 != Op1) {
1225     assert(!updateST0 && "Should have updated other operand!");
1226     popStackAfter(I);   // Pop the top of stack
1227   }
1228 
1229   // Update stack information so that we know the destination register is now on
1230   // the stack.
1231   unsigned UpdatedSlot = getSlot(updateST0 ? TOS : NotTOS);
1232   assert(UpdatedSlot < StackTop && Dest < 7);
1233   Stack[UpdatedSlot]   = Dest;
1234   RegMap[Dest]         = UpdatedSlot;
1235   MBB->getParent()->DeleteMachineInstr(MI); // Remove the old instruction
1236 }
1237 
1238 /// handleCompareFP - Handle FUCOM and FUCOMI instructions, which have two FP
1239 /// register arguments and no explicit destinations.
1240 ///
handleCompareFP(MachineBasicBlock::iterator & I)1241 void FPS::handleCompareFP(MachineBasicBlock::iterator &I) {
1242   ASSERT_SORTED(ForwardST0Table); ASSERT_SORTED(ReverseST0Table);
1243   ASSERT_SORTED(ForwardSTiTable); ASSERT_SORTED(ReverseSTiTable);
1244   MachineInstr *MI = I;
1245 
1246   unsigned NumOperands = MI->getDesc().getNumOperands();
1247   assert(NumOperands == 2 && "Illegal FUCOM* instruction!");
1248   unsigned Op0 = getFPReg(MI->getOperand(NumOperands-2));
1249   unsigned Op1 = getFPReg(MI->getOperand(NumOperands-1));
1250   bool KillsOp0 = MI->killsRegister(X86::FP0+Op0);
1251   bool KillsOp1 = MI->killsRegister(X86::FP0+Op1);
1252 
1253   // Make sure the first operand is on the top of stack, the other one can be
1254   // anywhere.
1255   moveToTop(Op0, I);
1256 
1257   // Change from the pseudo instruction to the concrete instruction.
1258   MI->getOperand(0).setReg(getSTReg(Op1));
1259   MI->RemoveOperand(1);
1260   MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1261 
1262   // If any of the operands are killed by this instruction, free them.
1263   if (KillsOp0) freeStackSlotAfter(I, Op0);
1264   if (KillsOp1 && Op0 != Op1) freeStackSlotAfter(I, Op1);
1265 }
1266 
1267 /// handleCondMovFP - Handle two address conditional move instructions.  These
1268 /// instructions move a st(i) register to st(0) iff a condition is true.  These
1269 /// instructions require that the first operand is at the top of the stack, but
1270 /// otherwise don't modify the stack at all.
handleCondMovFP(MachineBasicBlock::iterator & I)1271 void FPS::handleCondMovFP(MachineBasicBlock::iterator &I) {
1272   MachineInstr *MI = I;
1273 
1274   unsigned Op0 = getFPReg(MI->getOperand(0));
1275   unsigned Op1 = getFPReg(MI->getOperand(2));
1276   bool KillsOp1 = MI->killsRegister(X86::FP0+Op1);
1277 
1278   // The first operand *must* be on the top of the stack.
1279   moveToTop(Op0, I);
1280 
1281   // Change the second operand to the stack register that the operand is in.
1282   // Change from the pseudo instruction to the concrete instruction.
1283   MI->RemoveOperand(0);
1284   MI->RemoveOperand(1);
1285   MI->getOperand(0).setReg(getSTReg(Op1));
1286   MI->setDesc(TII->get(getConcreteOpcode(MI->getOpcode())));
1287 
1288   // If we kill the second operand, make sure to pop it from the stack.
1289   if (Op0 != Op1 && KillsOp1) {
1290     // Get this value off of the register stack.
1291     freeStackSlotAfter(I, Op1);
1292   }
1293 }
1294 
1295 
1296 /// handleSpecialFP - Handle special instructions which behave unlike other
1297 /// floating point instructions.  This is primarily intended for use by pseudo
1298 /// instructions.
1299 ///
handleSpecialFP(MachineBasicBlock::iterator & Inst)1300 void FPS::handleSpecialFP(MachineBasicBlock::iterator &Inst) {
1301   MachineInstr *MI = Inst;
1302 
1303   if (MI->isCall()) {
1304     handleCall(Inst);
1305     return;
1306   }
1307 
1308   switch (MI->getOpcode()) {
1309   default: llvm_unreachable("Unknown SpecialFP instruction!");
1310   case TargetOpcode::COPY: {
1311     // We handle three kinds of copies: FP <- FP, FP <- ST, and ST <- FP.
1312     const MachineOperand &MO1 = MI->getOperand(1);
1313     const MachineOperand &MO0 = MI->getOperand(0);
1314     bool KillsSrc = MI->killsRegister(MO1.getReg());
1315 
1316     // FP <- FP copy.
1317     unsigned DstFP = getFPReg(MO0);
1318     unsigned SrcFP = getFPReg(MO1);
1319     assert(isLive(SrcFP) && "Cannot copy dead register");
1320     if (KillsSrc) {
1321       // If the input operand is killed, we can just change the owner of the
1322       // incoming stack slot into the result.
1323       unsigned Slot = getSlot(SrcFP);
1324       Stack[Slot] = DstFP;
1325       RegMap[DstFP] = Slot;
1326     } else {
1327       // For COPY we just duplicate the specified value to a new stack slot.
1328       // This could be made better, but would require substantial changes.
1329       duplicateToTop(SrcFP, DstFP, Inst);
1330     }
1331     break;
1332   }
1333 
1334   case TargetOpcode::IMPLICIT_DEF: {
1335     // All FP registers must be explicitly defined, so load a 0 instead.
1336     unsigned Reg = MI->getOperand(0).getReg() - X86::FP0;
1337     DEBUG(dbgs() << "Emitting LD_F0 for implicit FP" << Reg << '\n');
1338     BuildMI(*MBB, Inst, MI->getDebugLoc(), TII->get(X86::LD_F0));
1339     pushReg(Reg);
1340     break;
1341   }
1342 
1343   case TargetOpcode::INLINEASM: {
1344     // The inline asm MachineInstr currently only *uses* FP registers for the
1345     // 'f' constraint.  These should be turned into the current ST(x) register
1346     // in the machine instr.
1347     //
1348     // There are special rules for x87 inline assembly. The compiler must know
1349     // exactly how many registers are popped and pushed implicitly by the asm.
1350     // Otherwise it is not possible to restore the stack state after the inline
1351     // asm.
1352     //
1353     // There are 3 kinds of input operands:
1354     //
1355     // 1. Popped inputs. These must appear at the stack top in ST0-STn. A
1356     //    popped input operand must be in a fixed stack slot, and it is either
1357     //    tied to an output operand, or in the clobber list. The MI has ST use
1358     //    and def operands for these inputs.
1359     //
1360     // 2. Fixed inputs. These inputs appear in fixed stack slots, but are
1361     //    preserved by the inline asm. The fixed stack slots must be STn-STm
1362     //    following the popped inputs. A fixed input operand cannot be tied to
1363     //    an output or appear in the clobber list. The MI has ST use operands
1364     //    and no defs for these inputs.
1365     //
1366     // 3. Preserved inputs. These inputs use the "f" constraint which is
1367     //    represented as an FP register. The inline asm won't change these
1368     //    stack slots.
1369     //
1370     // Outputs must be in ST registers, FP outputs are not allowed. Clobbered
1371     // registers do not count as output operands. The inline asm changes the
1372     // stack as if it popped all the popped inputs and then pushed all the
1373     // output operands.
1374 
1375     // Scan the assembly for ST registers used, defined and clobbered. We can
1376     // only tell clobbers from defs by looking at the asm descriptor.
1377     unsigned STUses = 0, STDefs = 0, STClobbers = 0, STDeadDefs = 0;
1378     unsigned NumOps = 0;
1379     SmallSet<unsigned, 1> FRegIdx;
1380     unsigned RCID;
1381 
1382     for (unsigned i = InlineAsm::MIOp_FirstOperand, e = MI->getNumOperands();
1383          i != e && MI->getOperand(i).isImm(); i += 1 + NumOps) {
1384       unsigned Flags = MI->getOperand(i).getImm();
1385 
1386       NumOps = InlineAsm::getNumOperandRegisters(Flags);
1387       if (NumOps != 1)
1388         continue;
1389       const MachineOperand &MO = MI->getOperand(i + 1);
1390       if (!MO.isReg())
1391         continue;
1392       unsigned STReg = MO.getReg() - X86::FP0;
1393       if (STReg >= 8)
1394         continue;
1395 
1396       // If the flag has a register class constraint, this must be an operand
1397       // with constraint "f". Record its index and continue.
1398       if (InlineAsm::hasRegClassConstraint(Flags, RCID)) {
1399         FRegIdx.insert(i + 1);
1400         continue;
1401       }
1402 
1403       switch (InlineAsm::getKind(Flags)) {
1404       case InlineAsm::Kind_RegUse:
1405         STUses |= (1u << STReg);
1406         break;
1407       case InlineAsm::Kind_RegDef:
1408       case InlineAsm::Kind_RegDefEarlyClobber:
1409         STDefs |= (1u << STReg);
1410         if (MO.isDead())
1411           STDeadDefs |= (1u << STReg);
1412         break;
1413       case InlineAsm::Kind_Clobber:
1414         STClobbers |= (1u << STReg);
1415         break;
1416       default:
1417         break;
1418       }
1419     }
1420 
1421     if (STUses && !isMask_32(STUses))
1422       MI->emitError("fixed input regs must be last on the x87 stack");
1423     unsigned NumSTUses = countTrailingOnes(STUses);
1424 
1425     // Defs must be contiguous from the stack top. ST0-STn.
1426     if (STDefs && !isMask_32(STDefs)) {
1427       MI->emitError("output regs must be last on the x87 stack");
1428       STDefs = NextPowerOf2(STDefs) - 1;
1429     }
1430     unsigned NumSTDefs = countTrailingOnes(STDefs);
1431 
1432     // So must the clobbered stack slots. ST0-STm, m >= n.
1433     if (STClobbers && !isMask_32(STDefs | STClobbers))
1434       MI->emitError("clobbers must be last on the x87 stack");
1435 
1436     // Popped inputs are the ones that are also clobbered or defined.
1437     unsigned STPopped = STUses & (STDefs | STClobbers);
1438     if (STPopped && !isMask_32(STPopped))
1439       MI->emitError("implicitly popped regs must be last on the x87 stack");
1440     unsigned NumSTPopped = countTrailingOnes(STPopped);
1441 
1442     DEBUG(dbgs() << "Asm uses " << NumSTUses << " fixed regs, pops "
1443                  << NumSTPopped << ", and defines " << NumSTDefs << " regs.\n");
1444 
1445 #ifndef NDEBUG
1446     // If any input operand uses constraint "f", all output register
1447     // constraints must be early-clobber defs.
1448     for (unsigned I = 0, E = MI->getNumOperands(); I < E; ++I)
1449       if (FRegIdx.count(I)) {
1450         assert((1 << getFPReg(MI->getOperand(I)) & STDefs) == 0 &&
1451                "Operands with constraint \"f\" cannot overlap with defs");
1452       }
1453 #endif
1454 
1455     // Collect all FP registers (register operands with constraints "t", "u",
1456     // and "f") to kill afer the instruction.
1457     unsigned FPKills = ((1u << NumFPRegs) - 1) & ~0xff;
1458     for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1459       MachineOperand &Op = MI->getOperand(i);
1460       if (!Op.isReg() || Op.getReg() < X86::FP0 || Op.getReg() > X86::FP6)
1461         continue;
1462       unsigned FPReg = getFPReg(Op);
1463 
1464       // If we kill this operand, make sure to pop it from the stack after the
1465       // asm.  We just remember it for now, and pop them all off at the end in
1466       // a batch.
1467       if (Op.isUse() && Op.isKill())
1468         FPKills |= 1U << FPReg;
1469     }
1470 
1471     // Do not include registers that are implicitly popped by defs/clobbers.
1472     FPKills &= ~(STDefs | STClobbers);
1473 
1474     // Now we can rearrange the live registers to match what was requested.
1475     unsigned char STUsesArray[8];
1476 
1477     for (unsigned I = 0; I < NumSTUses; ++I)
1478       STUsesArray[I] = I;
1479 
1480     shuffleStackTop(STUsesArray, NumSTUses, Inst);
1481     DEBUG({dbgs() << "Before asm: "; dumpStack();});
1482 
1483     // With the stack layout fixed, rewrite the FP registers.
1484     for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1485       MachineOperand &Op = MI->getOperand(i);
1486       if (!Op.isReg() || Op.getReg() < X86::FP0 || Op.getReg() > X86::FP6)
1487         continue;
1488 
1489       unsigned FPReg = getFPReg(Op);
1490 
1491       if (FRegIdx.count(i))
1492         // Operand with constraint "f".
1493         Op.setReg(getSTReg(FPReg));
1494       else
1495         // Operand with a single register class constraint ("t" or "u").
1496         Op.setReg(X86::ST0 + FPReg);
1497     }
1498 
1499     // Simulate the inline asm popping its inputs and pushing its outputs.
1500     StackTop -= NumSTPopped;
1501 
1502     for (unsigned i = 0; i < NumSTDefs; ++i)
1503       pushReg(NumSTDefs - i - 1);
1504 
1505     // If this asm kills any FP registers (is the last use of them) we must
1506     // explicitly emit pop instructions for them.  Do this now after the asm has
1507     // executed so that the ST(x) numbers are not off (which would happen if we
1508     // did this inline with operand rewriting).
1509     //
1510     // Note: this might be a non-optimal pop sequence.  We might be able to do
1511     // better by trying to pop in stack order or something.
1512     while (FPKills) {
1513       unsigned FPReg = countTrailingZeros(FPKills);
1514       if (isLive(FPReg))
1515         freeStackSlotAfter(Inst, FPReg);
1516       FPKills &= ~(1U << FPReg);
1517     }
1518 
1519     // Don't delete the inline asm!
1520     return;
1521   }
1522 
1523   case X86::WIN_FTOL_32:
1524   case X86::WIN_FTOL_64: {
1525     // Push the operand into ST0.
1526     MachineOperand &Op = MI->getOperand(0);
1527     assert(Op.isUse() && Op.isReg() &&
1528       Op.getReg() >= X86::FP0 && Op.getReg() <= X86::FP6);
1529     unsigned FPReg = getFPReg(Op);
1530     if (Op.isKill())
1531       moveToTop(FPReg, Inst);
1532     else
1533       duplicateToTop(FPReg, FPReg, Inst);
1534 
1535     // Emit the call. This will pop the operand.
1536     BuildMI(*MBB, Inst, MI->getDebugLoc(), TII->get(X86::CALLpcrel32))
1537       .addExternalSymbol("_ftol2")
1538       .addReg(X86::ST0, RegState::ImplicitKill)
1539       .addReg(X86::ECX, RegState::ImplicitDefine)
1540       .addReg(X86::EAX, RegState::Define | RegState::Implicit)
1541       .addReg(X86::EDX, RegState::Define | RegState::Implicit)
1542       .addReg(X86::EFLAGS, RegState::Define | RegState::Implicit);
1543     --StackTop;
1544 
1545     break;
1546   }
1547 
1548   case X86::RETQ:
1549   case X86::RETL:
1550   case X86::RETIL:
1551   case X86::RETIQ:
1552     // If RET has an FP register use operand, pass the first one in ST(0) and
1553     // the second one in ST(1).
1554 
1555     // Find the register operands.
1556     unsigned FirstFPRegOp = ~0U, SecondFPRegOp = ~0U;
1557     unsigned LiveMask = 0;
1558 
1559     for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1560       MachineOperand &Op = MI->getOperand(i);
1561       if (!Op.isReg() || Op.getReg() < X86::FP0 || Op.getReg() > X86::FP6)
1562         continue;
1563       // FP Register uses must be kills unless there are two uses of the same
1564       // register, in which case only one will be a kill.
1565       assert(Op.isUse() &&
1566              (Op.isKill() ||                        // Marked kill.
1567               getFPReg(Op) == FirstFPRegOp ||       // Second instance.
1568               MI->killsRegister(Op.getReg())) &&    // Later use is marked kill.
1569              "Ret only defs operands, and values aren't live beyond it");
1570 
1571       if (FirstFPRegOp == ~0U)
1572         FirstFPRegOp = getFPReg(Op);
1573       else {
1574         assert(SecondFPRegOp == ~0U && "More than two fp operands!");
1575         SecondFPRegOp = getFPReg(Op);
1576       }
1577       LiveMask |= (1 << getFPReg(Op));
1578 
1579       // Remove the operand so that later passes don't see it.
1580       MI->RemoveOperand(i);
1581       --i, --e;
1582     }
1583 
1584     // We may have been carrying spurious live-ins, so make sure only the returned
1585     // registers are left live.
1586     adjustLiveRegs(LiveMask, MI);
1587     if (!LiveMask) return;  // Quick check to see if any are possible.
1588 
1589     // There are only four possibilities here:
1590     // 1) we are returning a single FP value.  In this case, it has to be in
1591     //    ST(0) already, so just declare success by removing the value from the
1592     //    FP Stack.
1593     if (SecondFPRegOp == ~0U) {
1594       // Assert that the top of stack contains the right FP register.
1595       assert(StackTop == 1 && FirstFPRegOp == getStackEntry(0) &&
1596              "Top of stack not the right register for RET!");
1597 
1598       // Ok, everything is good, mark the value as not being on the stack
1599       // anymore so that our assertion about the stack being empty at end of
1600       // block doesn't fire.
1601       StackTop = 0;
1602       return;
1603     }
1604 
1605     // Otherwise, we are returning two values:
1606     // 2) If returning the same value for both, we only have one thing in the FP
1607     //    stack.  Consider:  RET FP1, FP1
1608     if (StackTop == 1) {
1609       assert(FirstFPRegOp == SecondFPRegOp && FirstFPRegOp == getStackEntry(0)&&
1610              "Stack misconfiguration for RET!");
1611 
1612       // Duplicate the TOS so that we return it twice.  Just pick some other FPx
1613       // register to hold it.
1614       unsigned NewReg = ScratchFPReg;
1615       duplicateToTop(FirstFPRegOp, NewReg, MI);
1616       FirstFPRegOp = NewReg;
1617     }
1618 
1619     /// Okay we know we have two different FPx operands now:
1620     assert(StackTop == 2 && "Must have two values live!");
1621 
1622     /// 3) If SecondFPRegOp is currently in ST(0) and FirstFPRegOp is currently
1623     ///    in ST(1).  In this case, emit an fxch.
1624     if (getStackEntry(0) == SecondFPRegOp) {
1625       assert(getStackEntry(1) == FirstFPRegOp && "Unknown regs live");
1626       moveToTop(FirstFPRegOp, MI);
1627     }
1628 
1629     /// 4) Finally, FirstFPRegOp must be in ST(0) and SecondFPRegOp must be in
1630     /// ST(1).  Just remove both from our understanding of the stack and return.
1631     assert(getStackEntry(0) == FirstFPRegOp && "Unknown regs live");
1632     assert(getStackEntry(1) == SecondFPRegOp && "Unknown regs live");
1633     StackTop = 0;
1634     return;
1635   }
1636 
1637   Inst = MBB->erase(Inst);  // Remove the pseudo instruction
1638 
1639   // We want to leave I pointing to the previous instruction, but what if we
1640   // just erased the first instruction?
1641   if (Inst == MBB->begin()) {
1642     DEBUG(dbgs() << "Inserting dummy KILL\n");
1643     Inst = BuildMI(*MBB, Inst, DebugLoc(), TII->get(TargetOpcode::KILL));
1644   } else
1645     --Inst;
1646 }
1647 
setKillFlags(MachineBasicBlock & MBB) const1648 void FPS::setKillFlags(MachineBasicBlock &MBB) const {
1649   const TargetRegisterInfo *TRI =
1650       MBB.getParent()->getSubtarget().getRegisterInfo();
1651   LivePhysRegs LPR(TRI);
1652 
1653   LPR.addLiveOuts(&MBB);
1654 
1655   for (MachineBasicBlock::reverse_iterator I = MBB.rbegin(), E = MBB.rend();
1656        I != E; ++I) {
1657     if (I->isDebugValue())
1658       continue;
1659 
1660     std::bitset<8> Defs;
1661     SmallVector<MachineOperand *, 2> Uses;
1662     MachineInstr &MI = *I;
1663 
1664     for (auto &MO : I->operands()) {
1665       if (!MO.isReg())
1666         continue;
1667 
1668       unsigned Reg = MO.getReg() - X86::FP0;
1669 
1670       if (Reg >= 8)
1671         continue;
1672 
1673       if (MO.isDef()) {
1674         Defs.set(Reg);
1675         if (!LPR.contains(MO.getReg()))
1676           MO.setIsDead();
1677       } else
1678         Uses.push_back(&MO);
1679     }
1680 
1681     for (auto *MO : Uses)
1682       if (Defs.test(getFPReg(*MO)) || !LPR.contains(MO->getReg()))
1683         MO->setIsKill();
1684 
1685     LPR.stepBackward(MI);
1686   }
1687 }
1688