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