blob: 53f6ec0feb45c5478793248e28795adeb152ace4 [file] [log] [blame]
Chris Lattnera960d952003-01-13 01:01:59 +00001//===-- FloatingPoint.cpp - Floating point Reg -> Stack converter ---------===//
John Criswellb576c942003-10-20 19:43:21 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
Chris Lattnera960d952003-01-13 01:01:59 +00009//
10// This file defines the pass which converts floating point instructions from
Chris Lattner847df252004-01-30 22:25:18 +000011// virtual 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// This pass is hampered by the lack of decent CFG manipulation routines for
16// machine code. In particular, this wants to be able to split critical edges
17// as necessary, traverse the machine basic block CFG in depth-first order, and
18// allow there to be multiple machine basic blocks for each LLVM basicblock
19// (needed for critical edge splitting).
20//
21// In particular, this pass currently barfs on critical edges. Because of this,
22// it requires the instruction selector to insert FP_REG_KILL instructions on
23// the exits of any basic block that has critical edges going from it, or which
24// branch to a critical basic block.
25//
26// FIXME: this is not implemented yet. The stackifier pass only works on local
27// basic blocks.
Chris Lattnera960d952003-01-13 01:01:59 +000028//
29//===----------------------------------------------------------------------===//
30
Chris Lattnercb533582003-08-03 21:14:38 +000031#define DEBUG_TYPE "fp"
Chris Lattnera960d952003-01-13 01:01:59 +000032#include "X86.h"
33#include "X86InstrInfo.h"
34#include "llvm/CodeGen/MachineFunctionPass.h"
35#include "llvm/CodeGen/MachineInstrBuilder.h"
36#include "llvm/CodeGen/LiveVariables.h"
Alkis Evlogimenos359b65f2003-12-13 05:36:22 +000037#include "llvm/CodeGen/Passes.h"
Chris Lattner3501fea2003-01-14 22:00:31 +000038#include "llvm/Target/TargetInstrInfo.h"
Chris Lattnera960d952003-01-13 01:01:59 +000039#include "llvm/Target/TargetMachine.h"
Chris Lattner847df252004-01-30 22:25:18 +000040#include "llvm/Function.h" // FIXME: remove when using MBB CFG!
41#include "llvm/Support/CFG.h" // FIXME: remove when using MBB CFG!
Chris Lattnera11136b2003-08-01 22:21:34 +000042#include "Support/Debug.h"
Chris Lattner847df252004-01-30 22:25:18 +000043#include "Support/DepthFirstIterator.h"
Chris Lattnera960d952003-01-13 01:01:59 +000044#include "Support/Statistic.h"
45#include <algorithm>
Chris Lattner847df252004-01-30 22:25:18 +000046#include <set>
Chris Lattnerf2e49d42003-12-20 09:58:55 +000047using namespace llvm;
Brian Gaeked0fde302003-11-11 22:41:34 +000048
Chris Lattnera960d952003-01-13 01:01:59 +000049namespace {
50 Statistic<> NumFXCH("x86-codegen", "Number of fxch instructions inserted");
51 Statistic<> NumFP ("x86-codegen", "Number of floating point instructions");
52
53 struct FPS : public MachineFunctionPass {
54 virtual bool runOnMachineFunction(MachineFunction &MF);
55
56 virtual const char *getPassName() const { return "X86 FP Stackifier"; }
57
58 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
59 AU.addRequired<LiveVariables>();
60 MachineFunctionPass::getAnalysisUsage(AU);
61 }
62 private:
63 LiveVariables *LV; // Live variable info for current function...
64 MachineBasicBlock *MBB; // Current basic block
65 unsigned Stack[8]; // FP<n> Registers in each stack slot...
66 unsigned RegMap[8]; // Track which stack slot contains each register
67 unsigned StackTop; // The current top of the FP stack.
68
69 void dumpStack() const {
70 std::cerr << "Stack contents:";
71 for (unsigned i = 0; i != StackTop; ++i) {
72 std::cerr << " FP" << Stack[i];
73 assert(RegMap[Stack[i]] == i && "Stack[] doesn't match RegMap[]!");
74 }
75 std::cerr << "\n";
76 }
77 private:
78 // getSlot - Return the stack slot number a particular register number is
79 // in...
80 unsigned getSlot(unsigned RegNo) const {
81 assert(RegNo < 8 && "Regno out of range!");
82 return RegMap[RegNo];
83 }
84
85 // getStackEntry - Return the X86::FP<n> register in register ST(i)
86 unsigned getStackEntry(unsigned STi) const {
87 assert(STi < StackTop && "Access past stack top!");
88 return Stack[StackTop-1-STi];
89 }
90
91 // getSTReg - Return the X86::ST(i) register which contains the specified
92 // FP<RegNo> register
93 unsigned getSTReg(unsigned RegNo) const {
Brian Gaeked0fde302003-11-11 22:41:34 +000094 return StackTop - 1 - getSlot(RegNo) + llvm::X86::ST0;
Chris Lattnera960d952003-01-13 01:01:59 +000095 }
96
Chris Lattner4a06f352004-02-02 19:23:15 +000097 // pushReg - Push the specified FP<n> register onto the stack
Chris Lattnera960d952003-01-13 01:01:59 +000098 void pushReg(unsigned Reg) {
99 assert(Reg < 8 && "Register number out of range!");
100 assert(StackTop < 8 && "Stack overflow!");
101 Stack[StackTop] = Reg;
102 RegMap[Reg] = StackTop++;
103 }
104
105 bool isAtTop(unsigned RegNo) const { return getSlot(RegNo) == StackTop-1; }
106 void moveToTop(unsigned RegNo, MachineBasicBlock::iterator &I) {
107 if (!isAtTop(RegNo)) {
108 unsigned Slot = getSlot(RegNo);
109 unsigned STReg = getSTReg(RegNo);
110 unsigned RegOnTop = getStackEntry(0);
111
112 // Swap the slots the regs are in
113 std::swap(RegMap[RegNo], RegMap[RegOnTop]);
114
115 // Swap stack slot contents
116 assert(RegMap[RegOnTop] < StackTop);
117 std::swap(Stack[RegMap[RegOnTop]], Stack[StackTop-1]);
118
119 // Emit an fxch to update the runtime processors version of the state
120 MachineInstr *MI = BuildMI(X86::FXCH, 1).addReg(STReg);
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000121 MBB->insert(I, MI);
Chris Lattnera960d952003-01-13 01:01:59 +0000122 NumFXCH++;
123 }
124 }
125
126 void duplicateToTop(unsigned RegNo, unsigned AsReg,
127 MachineBasicBlock::iterator &I) {
128 unsigned STReg = getSTReg(RegNo);
129 pushReg(AsReg); // New register on top of stack
130
131 MachineInstr *MI = BuildMI(X86::FLDrr, 1).addReg(STReg);
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000132 MBB->insert(I, MI);
Chris Lattnera960d952003-01-13 01:01:59 +0000133 }
134
135 // popStackAfter - Pop the current value off of the top of the FP stack
136 // after the specified instruction.
137 void popStackAfter(MachineBasicBlock::iterator &I);
138
139 bool processBasicBlock(MachineFunction &MF, MachineBasicBlock &MBB);
140
141 void handleZeroArgFP(MachineBasicBlock::iterator &I);
142 void handleOneArgFP(MachineBasicBlock::iterator &I);
Chris Lattner4a06f352004-02-02 19:23:15 +0000143 void handleOneArgFPRW(MachineBasicBlock::iterator &I);
Chris Lattnera960d952003-01-13 01:01:59 +0000144 void handleTwoArgFP(MachineBasicBlock::iterator &I);
145 void handleSpecialFP(MachineBasicBlock::iterator &I);
146 };
147}
148
Chris Lattnerf2e49d42003-12-20 09:58:55 +0000149FunctionPass *llvm::createX86FloatingPointStackifierPass() { return new FPS(); }
Chris Lattnera960d952003-01-13 01:01:59 +0000150
151/// runOnMachineFunction - Loop over all of the basic blocks, transforming FP
152/// register references into FP stack references.
153///
154bool FPS::runOnMachineFunction(MachineFunction &MF) {
155 LV = &getAnalysis<LiveVariables>();
156 StackTop = 0;
157
Chris Lattner847df252004-01-30 22:25:18 +0000158 // Figure out the mapping of MBB's to BB's.
159 //
160 // FIXME: Eventually we should be able to traverse the MBB CFG directly, and
161 // we will need to extend this when one llvm basic block can codegen to
162 // multiple MBBs.
163 //
164 // FIXME again: Just use the mapping established by LiveVariables!
165 //
166 std::map<const BasicBlock*, MachineBasicBlock *> MBBMap;
Chris Lattnera960d952003-01-13 01:01:59 +0000167 for (MachineFunction::iterator I = MF.begin(), E = MF.end(); I != E; ++I)
Chris Lattner847df252004-01-30 22:25:18 +0000168 MBBMap[I->getBasicBlock()] = I;
169
170 // Process the function in depth first order so that we process at least one
171 // of the predecessors for every reachable block in the function.
172 std::set<const BasicBlock*> Processed;
173 const BasicBlock *Entry = MF.getFunction()->begin();
174
175 bool Changed = false;
176 for (df_ext_iterator<const BasicBlock*, std::set<const BasicBlock*> >
177 I = df_ext_begin(Entry, Processed), E = df_ext_end(Entry, Processed);
178 I != E; ++I)
179 Changed |= processBasicBlock(MF, *MBBMap[*I]);
180
181 assert(MBBMap.size() == Processed.size() &&
182 "Doesn't handle unreachable code yet!");
183
Chris Lattnera960d952003-01-13 01:01:59 +0000184 return Changed;
185}
186
187/// processBasicBlock - Loop over all of the instructions in the basic block,
188/// transforming FP instructions into their stack form.
189///
190bool FPS::processBasicBlock(MachineFunction &MF, MachineBasicBlock &BB) {
191 const TargetInstrInfo &TII = MF.getTarget().getInstrInfo();
192 bool Changed = false;
193 MBB = &BB;
194
195 for (MachineBasicBlock::iterator I = BB.begin(); I != BB.end(); ++I) {
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000196 MachineInstr *MI = I;
Chris Lattnera960d952003-01-13 01:01:59 +0000197 unsigned Flags = TII.get(MI->getOpcode()).TSFlags;
Chris Lattner847df252004-01-30 22:25:18 +0000198 if ((Flags & X86II::FPTypeMask) == X86II::NotFP)
199 continue; // Efficiently ignore non-fp insts!
Chris Lattnera960d952003-01-13 01:01:59 +0000200
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000201 MachineInstr *PrevMI = 0;
202 if (I != BB.begin()) {
203 MachineBasicBlock::iterator tmp = I;
204 --tmp;
205 PrevMI = tmp;
206 }
Chris Lattnera960d952003-01-13 01:01:59 +0000207
208 ++NumFP; // Keep track of # of pseudo instrs
209 DEBUG(std::cerr << "\nFPInst:\t";
210 MI->print(std::cerr, MF.getTarget()));
211
212 // Get dead variables list now because the MI pointer may be deleted as part
213 // of processing!
214 LiveVariables::killed_iterator IB = LV->dead_begin(MI);
215 LiveVariables::killed_iterator IE = LV->dead_end(MI);
216
217 DEBUG(const MRegisterInfo *MRI = MF.getTarget().getRegisterInfo();
218 LiveVariables::killed_iterator I = LV->killed_begin(MI);
219 LiveVariables::killed_iterator E = LV->killed_end(MI);
220 if (I != E) {
221 std::cerr << "Killed Operands:";
222 for (; I != E; ++I)
223 std::cerr << " %" << MRI->getName(I->second);
224 std::cerr << "\n";
225 });
226
227 switch (Flags & X86II::FPTypeMask) {
Chris Lattner4a06f352004-02-02 19:23:15 +0000228 case X86II::ZeroArgFP: handleZeroArgFP(I); break;
229 case X86II::OneArgFP: handleOneArgFP(I); break; // fstp ST(0)
230 case X86II::OneArgFPRW: handleOneArgFPRW(I); break; // ST(0) = fsqrt(ST(0))
231 case X86II::TwoArgFP: handleTwoArgFP(I); break;
232 case X86II::SpecialFP: handleSpecialFP(I); break;
Chris Lattnera960d952003-01-13 01:01:59 +0000233 default: assert(0 && "Unknown FP Type!");
234 }
235
236 // Check to see if any of the values defined by this instruction are dead
237 // after definition. If so, pop them.
238 for (; IB != IE; ++IB) {
239 unsigned Reg = IB->second;
240 if (Reg >= X86::FP0 && Reg <= X86::FP6) {
241 DEBUG(std::cerr << "Register FP#" << Reg-X86::FP0 << " is dead!\n");
242 ++I; // Insert fxch AFTER the instruction
Misha Brukman5560c9d2003-08-18 14:43:39 +0000243 moveToTop(Reg-X86::FP0, I); // Insert fxch if necessary
Chris Lattnera960d952003-01-13 01:01:59 +0000244 --I; // Move to fxch or old instruction
245 popStackAfter(I); // Pop the top of the stack, killing value
246 }
247 }
248
249 // Print out all of the instructions expanded to if -debug
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000250 DEBUG(if (&*I == PrevMI) {
Chris Lattnera960d952003-01-13 01:01:59 +0000251 std::cerr<< "Just deleted pseudo instruction\n";
252 } else {
253 MachineBasicBlock::iterator Start = I;
254 // Rewind to first instruction newly inserted.
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000255 while (Start != BB.begin() &&
256 --Start != MachineBasicBlock::iterator(PrevMI));
257 ++Start;
Brian Gaeked7908f62003-06-27 00:00:48 +0000258 std::cerr << "Inserted instructions:\n\t";
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000259 Start->print(std::cerr, MF.getTarget());
260 while (++Start != I); ++Start;
Chris Lattnera960d952003-01-13 01:01:59 +0000261 }
262 dumpStack();
263 );
264
265 Changed = true;
266 }
267
268 assert(StackTop == 0 && "Stack not empty at end of basic block?");
269 return Changed;
270}
271
272//===----------------------------------------------------------------------===//
273// Efficient Lookup Table Support
274//===----------------------------------------------------------------------===//
275
Chris Lattnerf2e49d42003-12-20 09:58:55 +0000276namespace {
277 struct TableEntry {
278 unsigned from;
279 unsigned to;
280 bool operator<(const TableEntry &TE) const { return from < TE.from; }
281 bool operator<(unsigned V) const { return from < V; }
282 };
283}
Chris Lattnera960d952003-01-13 01:01:59 +0000284
285static bool TableIsSorted(const TableEntry *Table, unsigned NumEntries) {
286 for (unsigned i = 0; i != NumEntries-1; ++i)
287 if (!(Table[i] < Table[i+1])) return false;
288 return true;
289}
290
291static int Lookup(const TableEntry *Table, unsigned N, unsigned Opcode) {
292 const TableEntry *I = std::lower_bound(Table, Table+N, Opcode);
293 if (I != Table+N && I->from == Opcode)
294 return I->to;
295 return -1;
296}
297
298#define ARRAY_SIZE(TABLE) \
299 (sizeof(TABLE)/sizeof(TABLE[0]))
300
301#ifdef NDEBUG
302#define ASSERT_SORTED(TABLE)
303#else
304#define ASSERT_SORTED(TABLE) \
305 { static bool TABLE##Checked = false; \
306 if (!TABLE##Checked) \
307 assert(TableIsSorted(TABLE, ARRAY_SIZE(TABLE)) && \
308 "All lookup tables must be sorted for efficient access!"); \
309 }
310#endif
311
312
313//===----------------------------------------------------------------------===//
314// Helper Methods
315//===----------------------------------------------------------------------===//
316
317// PopTable - Sorted map of instructions to their popping version. The first
318// element is an instruction, the second is the version which pops.
319//
320static const TableEntry PopTable[] = {
Chris Lattner113455b2003-08-03 21:56:36 +0000321 { X86::FADDrST0 , X86::FADDPrST0 },
322
323 { X86::FDIVRrST0, X86::FDIVRPrST0 },
324 { X86::FDIVrST0 , X86::FDIVPrST0 },
325
Chris Lattnera960d952003-01-13 01:01:59 +0000326 { X86::FISTr16 , X86::FISTPr16 },
327 { X86::FISTr32 , X86::FISTPr32 },
328
Chris Lattnera960d952003-01-13 01:01:59 +0000329 { X86::FMULrST0 , X86::FMULPrST0 },
Chris Lattnera960d952003-01-13 01:01:59 +0000330
Chris Lattner113455b2003-08-03 21:56:36 +0000331 { X86::FSTr32 , X86::FSTPr32 },
332 { X86::FSTr64 , X86::FSTPr64 },
333 { X86::FSTrr , X86::FSTPrr },
334
335 { X86::FSUBRrST0, X86::FSUBRPrST0 },
336 { X86::FSUBrST0 , X86::FSUBPrST0 },
337
Chris Lattnera960d952003-01-13 01:01:59 +0000338 { X86::FUCOMPr , X86::FUCOMPPr },
Chris Lattner113455b2003-08-03 21:56:36 +0000339 { X86::FUCOMr , X86::FUCOMPr },
Chris Lattnera960d952003-01-13 01:01:59 +0000340};
341
342/// popStackAfter - Pop the current value off of the top of the FP stack after
343/// the specified instruction. This attempts to be sneaky and combine the pop
344/// into the instruction itself if possible. The iterator is left pointing to
345/// the last instruction, be it a new pop instruction inserted, or the old
346/// instruction if it was modified in place.
347///
348void FPS::popStackAfter(MachineBasicBlock::iterator &I) {
349 ASSERT_SORTED(PopTable);
350 assert(StackTop > 0 && "Cannot pop empty stack!");
351 RegMap[Stack[--StackTop]] = ~0; // Update state
352
353 // Check to see if there is a popping version of this instruction...
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000354 int Opcode = Lookup(PopTable, ARRAY_SIZE(PopTable), I->getOpcode());
Chris Lattnera960d952003-01-13 01:01:59 +0000355 if (Opcode != -1) {
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000356 I->setOpcode(Opcode);
Chris Lattnera960d952003-01-13 01:01:59 +0000357 if (Opcode == X86::FUCOMPPr)
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000358 I->RemoveOperand(0);
Chris Lattnera960d952003-01-13 01:01:59 +0000359
360 } else { // Insert an explicit pop
361 MachineInstr *MI = BuildMI(X86::FSTPrr, 1).addReg(X86::ST0);
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000362 I = MBB->insert(++I, MI);
Chris Lattnera960d952003-01-13 01:01:59 +0000363 }
364}
365
366static unsigned getFPReg(const MachineOperand &MO) {
Chris Lattner6d215182004-02-10 20:31:28 +0000367 assert(MO.isRegister() && "Expected an FP register!");
Chris Lattnera960d952003-01-13 01:01:59 +0000368 unsigned Reg = MO.getReg();
369 assert(Reg >= X86::FP0 && Reg <= X86::FP6 && "Expected FP register!");
370 return Reg - X86::FP0;
371}
372
373
374//===----------------------------------------------------------------------===//
375// Instruction transformation implementation
376//===----------------------------------------------------------------------===//
377
378/// handleZeroArgFP - ST(0) = fld0 ST(0) = flds <mem>
Chris Lattner4a06f352004-02-02 19:23:15 +0000379///
Chris Lattnera960d952003-01-13 01:01:59 +0000380void FPS::handleZeroArgFP(MachineBasicBlock::iterator &I) {
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000381 MachineInstr *MI = I;
Chris Lattnera960d952003-01-13 01:01:59 +0000382 unsigned DestReg = getFPReg(MI->getOperand(0));
383 MI->RemoveOperand(0); // Remove the explicit ST(0) operand
384
385 // Result gets pushed on the stack...
386 pushReg(DestReg);
387}
388
Chris Lattner4a06f352004-02-02 19:23:15 +0000389/// handleOneArgFP - fst <mem>, ST(0)
390///
Chris Lattnera960d952003-01-13 01:01:59 +0000391void FPS::handleOneArgFP(MachineBasicBlock::iterator &I) {
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000392 MachineInstr *MI = I;
Chris Lattnerb97046a2004-02-03 07:27:34 +0000393 assert((MI->getNumOperands() == 5 || MI->getNumOperands() == 1) &&
394 "Can only handle fst* & ftst instructions!");
Chris Lattnera960d952003-01-13 01:01:59 +0000395
Chris Lattner4a06f352004-02-02 19:23:15 +0000396 // Is this the last use of the source register?
Chris Lattnerb97046a2004-02-03 07:27:34 +0000397 unsigned Reg = getFPReg(MI->getOperand(MI->getNumOperands()-1));
Chris Lattnera960d952003-01-13 01:01:59 +0000398 bool KillsSrc = false;
399 for (LiveVariables::killed_iterator KI = LV->killed_begin(MI),
400 E = LV->killed_end(MI); KI != E; ++KI)
401 KillsSrc |= KI->second == X86::FP0+Reg;
402
403 // FSTPr80 and FISTPr64 are strange because there are no non-popping versions.
404 // If we have one _and_ we don't want to pop the operand, duplicate the value
405 // on the stack instead of moving it. This ensure that popping the value is
406 // always ok.
407 //
408 if ((MI->getOpcode() == X86::FSTPr80 ||
409 MI->getOpcode() == X86::FISTPr64) && !KillsSrc) {
410 duplicateToTop(Reg, 7 /*temp register*/, I);
411 } else {
412 moveToTop(Reg, I); // Move to the top of the stack...
413 }
Chris Lattnerb97046a2004-02-03 07:27:34 +0000414 MI->RemoveOperand(MI->getNumOperands()-1); // Remove explicit ST(0) operand
Chris Lattnera960d952003-01-13 01:01:59 +0000415
416 if (MI->getOpcode() == X86::FSTPr80 || MI->getOpcode() == X86::FISTPr64) {
417 assert(StackTop > 0 && "Stack empty??");
418 --StackTop;
419 } else if (KillsSrc) { // Last use of operand?
420 popStackAfter(I);
421 }
422}
423
Chris Lattner4a06f352004-02-02 19:23:15 +0000424
425/// handleOneArgFPRW - fchs - ST(0) = -ST(0)
426///
427void FPS::handleOneArgFPRW(MachineBasicBlock::iterator &I) {
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000428 MachineInstr *MI = I;
Chris Lattner4a06f352004-02-02 19:23:15 +0000429 assert(MI->getNumOperands() == 2 && "Can only handle fst* instructions!");
430
431 // Is this the last use of the source register?
432 unsigned Reg = getFPReg(MI->getOperand(1));
433 bool KillsSrc = false;
434 for (LiveVariables::killed_iterator KI = LV->killed_begin(MI),
435 E = LV->killed_end(MI); KI != E; ++KI)
436 KillsSrc |= KI->second == X86::FP0+Reg;
437
438 if (KillsSrc) {
439 // If this is the last use of the source register, just make sure it's on
440 // the top of the stack.
441 moveToTop(Reg, I);
442 assert(StackTop > 0 && "Stack cannot be empty!");
443 --StackTop;
444 pushReg(getFPReg(MI->getOperand(0)));
445 } else {
446 // If this is not the last use of the source register, _copy_ it to the top
447 // of the stack.
448 duplicateToTop(Reg, getFPReg(MI->getOperand(0)), I);
449 }
450
451 MI->RemoveOperand(1); // Drop the source operand.
452 MI->RemoveOperand(0); // Drop the destination operand.
453}
454
455
Chris Lattnera960d952003-01-13 01:01:59 +0000456//===----------------------------------------------------------------------===//
457// Define tables of various ways to map pseudo instructions
458//
459
460// ForwardST0Table - Map: A = B op C into: ST(0) = ST(0) op ST(i)
461static const TableEntry ForwardST0Table[] = {
462 { X86::FpADD, X86::FADDST0r },
Chris Lattnera960d952003-01-13 01:01:59 +0000463 { X86::FpDIV, X86::FDIVST0r },
Chris Lattner113455b2003-08-03 21:56:36 +0000464 { X86::FpMUL, X86::FMULST0r },
465 { X86::FpSUB, X86::FSUBST0r },
Chris Lattnera960d952003-01-13 01:01:59 +0000466 { X86::FpUCOM, X86::FUCOMr },
467};
468
469// ReverseST0Table - Map: A = B op C into: ST(0) = ST(i) op ST(0)
470static const TableEntry ReverseST0Table[] = {
471 { X86::FpADD, X86::FADDST0r }, // commutative
Chris Lattnera960d952003-01-13 01:01:59 +0000472 { X86::FpDIV, X86::FDIVRST0r },
Chris Lattner113455b2003-08-03 21:56:36 +0000473 { X86::FpMUL, X86::FMULST0r }, // commutative
474 { X86::FpSUB, X86::FSUBRST0r },
Chris Lattnera960d952003-01-13 01:01:59 +0000475 { X86::FpUCOM, ~0 },
476};
477
478// ForwardSTiTable - Map: A = B op C into: ST(i) = ST(0) op ST(i)
479static const TableEntry ForwardSTiTable[] = {
480 { X86::FpADD, X86::FADDrST0 }, // commutative
Chris Lattnera960d952003-01-13 01:01:59 +0000481 { X86::FpDIV, X86::FDIVRrST0 },
Chris Lattner113455b2003-08-03 21:56:36 +0000482 { X86::FpMUL, X86::FMULrST0 }, // commutative
483 { X86::FpSUB, X86::FSUBRrST0 },
Chris Lattnera960d952003-01-13 01:01:59 +0000484 { X86::FpUCOM, X86::FUCOMr },
485};
486
487// ReverseSTiTable - Map: A = B op C into: ST(i) = ST(i) op ST(0)
488static const TableEntry ReverseSTiTable[] = {
489 { X86::FpADD, X86::FADDrST0 },
Chris Lattnera960d952003-01-13 01:01:59 +0000490 { X86::FpDIV, X86::FDIVrST0 },
Chris Lattner113455b2003-08-03 21:56:36 +0000491 { X86::FpMUL, X86::FMULrST0 },
492 { X86::FpSUB, X86::FSUBrST0 },
Chris Lattnera960d952003-01-13 01:01:59 +0000493 { X86::FpUCOM, ~0 },
494};
495
496
497/// handleTwoArgFP - Handle instructions like FADD and friends which are virtual
498/// instructions which need to be simplified and possibly transformed.
499///
500/// Result: ST(0) = fsub ST(0), ST(i)
501/// ST(i) = fsub ST(0), ST(i)
502/// ST(0) = fsubr ST(0), ST(i)
503/// ST(i) = fsubr ST(0), ST(i)
504///
505/// In addition to three address instructions, this also handles the FpUCOM
506/// instruction which only has two operands, but no destination. This
507/// instruction is also annoying because there is no "reverse" form of it
508/// available.
509///
510void FPS::handleTwoArgFP(MachineBasicBlock::iterator &I) {
511 ASSERT_SORTED(ForwardST0Table); ASSERT_SORTED(ReverseST0Table);
512 ASSERT_SORTED(ForwardSTiTable); ASSERT_SORTED(ReverseSTiTable);
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000513 MachineInstr *MI = I;
Chris Lattnera960d952003-01-13 01:01:59 +0000514
515 unsigned NumOperands = MI->getNumOperands();
516 assert(NumOperands == 3 ||
517 (NumOperands == 2 && MI->getOpcode() == X86::FpUCOM) &&
518 "Illegal TwoArgFP instruction!");
519 unsigned Dest = getFPReg(MI->getOperand(0));
520 unsigned Op0 = getFPReg(MI->getOperand(NumOperands-2));
521 unsigned Op1 = getFPReg(MI->getOperand(NumOperands-1));
522 bool KillsOp0 = false, KillsOp1 = false;
523
524 for (LiveVariables::killed_iterator KI = LV->killed_begin(MI),
525 E = LV->killed_end(MI); KI != E; ++KI) {
526 KillsOp0 |= (KI->second == X86::FP0+Op0);
527 KillsOp1 |= (KI->second == X86::FP0+Op1);
528 }
529
530 // If this is an FpUCOM instruction, we must make sure the first operand is on
531 // the top of stack, the other one can be anywhere...
532 if (MI->getOpcode() == X86::FpUCOM)
533 moveToTop(Op0, I);
534
535 unsigned TOS = getStackEntry(0);
536
537 // One of our operands must be on the top of the stack. If neither is yet, we
538 // need to move one.
539 if (Op0 != TOS && Op1 != TOS) { // No operand at TOS?
540 // We can choose to move either operand to the top of the stack. If one of
541 // the operands is killed by this instruction, we want that one so that we
542 // can update right on top of the old version.
543 if (KillsOp0) {
544 moveToTop(Op0, I); // Move dead operand to TOS.
545 TOS = Op0;
546 } else if (KillsOp1) {
547 moveToTop(Op1, I);
548 TOS = Op1;
549 } else {
550 // All of the operands are live after this instruction executes, so we
551 // cannot update on top of any operand. Because of this, we must
552 // duplicate one of the stack elements to the top. It doesn't matter
553 // which one we pick.
554 //
555 duplicateToTop(Op0, Dest, I);
556 Op0 = TOS = Dest;
557 KillsOp0 = true;
558 }
559 } else if (!KillsOp0 && !KillsOp1 && MI->getOpcode() != X86::FpUCOM) {
560 // If we DO have one of our operands at the top of the stack, but we don't
561 // have a dead operand, we must duplicate one of the operands to a new slot
562 // on the stack.
563 duplicateToTop(Op0, Dest, I);
564 Op0 = TOS = Dest;
565 KillsOp0 = true;
566 }
567
568 // Now we know that one of our operands is on the top of the stack, and at
569 // least one of our operands is killed by this instruction.
570 assert((TOS == Op0 || TOS == Op1) &&
571 (KillsOp0 || KillsOp1 || MI->getOpcode() == X86::FpUCOM) &&
572 "Stack conditions not set up right!");
573
574 // We decide which form to use based on what is on the top of the stack, and
575 // which operand is killed by this instruction.
576 const TableEntry *InstTable;
577 bool isForward = TOS == Op0;
578 bool updateST0 = (TOS == Op0 && !KillsOp1) || (TOS == Op1 && !KillsOp0);
579 if (updateST0) {
580 if (isForward)
581 InstTable = ForwardST0Table;
582 else
583 InstTable = ReverseST0Table;
584 } else {
585 if (isForward)
586 InstTable = ForwardSTiTable;
587 else
588 InstTable = ReverseSTiTable;
589 }
590
591 int Opcode = Lookup(InstTable, ARRAY_SIZE(ForwardST0Table), MI->getOpcode());
592 assert(Opcode != -1 && "Unknown TwoArgFP pseudo instruction!");
593
594 // NotTOS - The register which is not on the top of stack...
595 unsigned NotTOS = (TOS == Op0) ? Op1 : Op0;
596
597 // Replace the old instruction with a new instruction
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000598 MBB->remove(I);
599 I = MBB->insert(I, BuildMI(Opcode, 1).addReg(getSTReg(NotTOS)));
Chris Lattnera960d952003-01-13 01:01:59 +0000600
601 // If both operands are killed, pop one off of the stack in addition to
602 // overwriting the other one.
603 if (KillsOp0 && KillsOp1 && Op0 != Op1) {
604 assert(!updateST0 && "Should have updated other operand!");
605 popStackAfter(I); // Pop the top of stack
606 }
607
608 // Insert an explicit pop of the "updated" operand for FUCOM
609 if (MI->getOpcode() == X86::FpUCOM) {
610 if (KillsOp0 && !KillsOp1)
611 popStackAfter(I); // If we kill the first operand, pop it!
612 else if (KillsOp1 && Op0 != Op1) {
613 if (getStackEntry(0) == Op1) {
614 popStackAfter(I); // If it's right at the top of stack, just pop it
615 } else {
616 // Otherwise, move the top of stack into the dead slot, killing the
617 // operand without having to add in an explicit xchg then pop.
618 //
619 unsigned STReg = getSTReg(Op1);
620 unsigned OldSlot = getSlot(Op1);
621 unsigned TopReg = Stack[StackTop-1];
622 Stack[OldSlot] = TopReg;
623 RegMap[TopReg] = OldSlot;
624 RegMap[Op1] = ~0;
625 Stack[--StackTop] = ~0;
626
627 MachineInstr *MI = BuildMI(X86::FSTPrr, 1).addReg(STReg);
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000628 I = MBB->insert(++I, MI);
Chris Lattnera960d952003-01-13 01:01:59 +0000629 }
630 }
631 }
632
633 // Update stack information so that we know the destination register is now on
634 // the stack.
635 if (MI->getOpcode() != X86::FpUCOM) {
636 unsigned UpdatedSlot = getSlot(updateST0 ? TOS : NotTOS);
637 assert(UpdatedSlot < StackTop && Dest < 7);
638 Stack[UpdatedSlot] = Dest;
639 RegMap[Dest] = UpdatedSlot;
640 }
641 delete MI; // Remove the old instruction
642}
643
644
645/// handleSpecialFP - Handle special instructions which behave unlike other
Misha Brukmancf00c4a2003-10-10 17:57:28 +0000646/// floating point instructions. This is primarily intended for use by pseudo
Chris Lattnera960d952003-01-13 01:01:59 +0000647/// instructions.
648///
649void FPS::handleSpecialFP(MachineBasicBlock::iterator &I) {
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000650 MachineInstr *MI = I;
Chris Lattnera960d952003-01-13 01:01:59 +0000651 switch (MI->getOpcode()) {
652 default: assert(0 && "Unknown SpecialFP instruction!");
653 case X86::FpGETRESULT: // Appears immediately after a call returning FP type!
654 assert(StackTop == 0 && "Stack should be empty after a call!");
655 pushReg(getFPReg(MI->getOperand(0)));
656 break;
657 case X86::FpSETRESULT:
658 assert(StackTop == 1 && "Stack should have one element on it to return!");
659 --StackTop; // "Forget" we have something on the top of stack!
660 break;
661 case X86::FpMOV: {
662 unsigned SrcReg = getFPReg(MI->getOperand(1));
663 unsigned DestReg = getFPReg(MI->getOperand(0));
664 bool KillsSrc = false;
665 for (LiveVariables::killed_iterator KI = LV->killed_begin(MI),
666 E = LV->killed_end(MI); KI != E; ++KI)
667 KillsSrc |= KI->second == X86::FP0+SrcReg;
668
669 if (KillsSrc) {
670 // If the input operand is killed, we can just change the owner of the
671 // incoming stack slot into the result.
672 unsigned Slot = getSlot(SrcReg);
673 assert(Slot < 7 && DestReg < 7 && "FpMOV operands invalid!");
674 Stack[Slot] = DestReg;
675 RegMap[DestReg] = Slot;
676
677 } else {
678 // For FMOV we just duplicate the specified value to a new stack slot.
679 // This could be made better, but would require substantial changes.
680 duplicateToTop(SrcReg, DestReg, I);
681 }
682 break;
683 }
684 }
685
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000686 I = MBB->erase(I); // Remove the pseudo instruction
687 --I;
Chris Lattnera960d952003-01-13 01:01:59 +0000688}