1 //===-- SystemZElimCompare.cpp - Eliminate comparison instructions --------===// 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 pass: 10 // (1) tries to remove compares if CC already contains the required information 11 // (2) fuses compares and branches into COMPARE AND BRANCH instructions 12 // 13 //===----------------------------------------------------------------------===// 14 15 #include "SystemZ.h" 16 #include "SystemZInstrInfo.h" 17 #include "SystemZTargetMachine.h" 18 #include "llvm/ADT/SmallVector.h" 19 #include "llvm/ADT/Statistic.h" 20 #include "llvm/ADT/StringRef.h" 21 #include "llvm/CodeGen/LivePhysRegs.h" 22 #include "llvm/CodeGen/MachineBasicBlock.h" 23 #include "llvm/CodeGen/MachineFunction.h" 24 #include "llvm/CodeGen/MachineFunctionPass.h" 25 #include "llvm/CodeGen/MachineInstr.h" 26 #include "llvm/CodeGen/MachineInstrBuilder.h" 27 #include "llvm/CodeGen/MachineOperand.h" 28 #include "llvm/CodeGen/TargetRegisterInfo.h" 29 #include "llvm/CodeGen/TargetSubtargetInfo.h" 30 #include "llvm/MC/MCInstrDesc.h" 31 #include <cassert> 32 #include <cstdint> 33 34 using namespace llvm; 35 36 #define DEBUG_TYPE "systemz-elim-compare" 37 38 STATISTIC(BranchOnCounts, "Number of branch-on-count instructions"); 39 STATISTIC(LoadAndTraps, "Number of load-and-trap instructions"); 40 STATISTIC(EliminatedComparisons, "Number of eliminated comparisons"); 41 STATISTIC(FusedComparisons, "Number of fused compare-and-branch instructions"); 42 43 namespace { 44 45 // Represents the references to a particular register in one or more 46 // instructions. 47 struct Reference { 48 Reference() = default; 49 50 Reference &operator|=(const Reference &Other) { 51 Def |= Other.Def; 52 Use |= Other.Use; 53 return *this; 54 } 55 56 explicit operator bool() const { return Def || Use; } 57 58 // True if the register is defined or used in some form, either directly or 59 // via a sub- or super-register. 60 bool Def = false; 61 bool Use = false; 62 }; 63 64 class SystemZElimCompare : public MachineFunctionPass { 65 public: 66 static char ID; 67 68 SystemZElimCompare(const SystemZTargetMachine &tm) 69 : MachineFunctionPass(ID) {} 70 71 StringRef getPassName() const override { 72 return "SystemZ Comparison Elimination"; 73 } 74 75 bool processBlock(MachineBasicBlock &MBB); 76 bool runOnMachineFunction(MachineFunction &F) override; 77 78 MachineFunctionProperties getRequiredProperties() const override { 79 return MachineFunctionProperties().set( 80 MachineFunctionProperties::Property::NoVRegs); 81 } 82 83 private: 84 Reference getRegReferences(MachineInstr &MI, unsigned Reg); 85 bool convertToBRCT(MachineInstr &MI, MachineInstr &Compare, 86 SmallVectorImpl<MachineInstr *> &CCUsers); 87 bool convertToLoadAndTrap(MachineInstr &MI, MachineInstr &Compare, 88 SmallVectorImpl<MachineInstr *> &CCUsers); 89 bool convertToLoadAndTest(MachineInstr &MI, MachineInstr &Compare, 90 SmallVectorImpl<MachineInstr *> &CCUsers); 91 bool adjustCCMasksForInstr(MachineInstr &MI, MachineInstr &Compare, 92 SmallVectorImpl<MachineInstr *> &CCUsers, 93 unsigned ConvOpc = 0); 94 bool optimizeCompareZero(MachineInstr &Compare, 95 SmallVectorImpl<MachineInstr *> &CCUsers); 96 bool fuseCompareOperations(MachineInstr &Compare, 97 SmallVectorImpl<MachineInstr *> &CCUsers); 98 99 const SystemZInstrInfo *TII = nullptr; 100 const TargetRegisterInfo *TRI = nullptr; 101 }; 102 103 char SystemZElimCompare::ID = 0; 104 105 } // end anonymous namespace 106 107 // Returns true if MI is an instruction whose output equals the value in Reg. 108 static bool preservesValueOf(MachineInstr &MI, unsigned Reg) { 109 switch (MI.getOpcode()) { 110 case SystemZ::LR: 111 case SystemZ::LGR: 112 case SystemZ::LGFR: 113 case SystemZ::LTR: 114 case SystemZ::LTGR: 115 case SystemZ::LTGFR: 116 case SystemZ::LER: 117 case SystemZ::LDR: 118 case SystemZ::LXR: 119 case SystemZ::LTEBR: 120 case SystemZ::LTDBR: 121 case SystemZ::LTXBR: 122 if (MI.getOperand(1).getReg() == Reg) 123 return true; 124 } 125 126 return false; 127 } 128 129 // Return true if any CC result of MI would (perhaps after conversion) 130 // reflect the value of Reg. 131 static bool resultTests(MachineInstr &MI, unsigned Reg) { 132 if (MI.getNumOperands() > 0 && MI.getOperand(0).isReg() && 133 MI.getOperand(0).isDef() && MI.getOperand(0).getReg() == Reg) 134 return true; 135 136 return (preservesValueOf(MI, Reg)); 137 } 138 139 // Describe the references to Reg or any of its aliases in MI. 140 Reference SystemZElimCompare::getRegReferences(MachineInstr &MI, unsigned Reg) { 141 Reference Ref; 142 if (MI.isDebugInstr()) 143 return Ref; 144 145 for (unsigned I = 0, E = MI.getNumOperands(); I != E; ++I) { 146 const MachineOperand &MO = MI.getOperand(I); 147 if (MO.isReg()) { 148 if (Register MOReg = MO.getReg()) { 149 if (TRI->regsOverlap(MOReg, Reg)) { 150 if (MO.isUse()) 151 Ref.Use = true; 152 else if (MO.isDef()) 153 Ref.Def = true; 154 } 155 } 156 } 157 } 158 return Ref; 159 } 160 161 // Return true if this is a load and test which can be optimized the 162 // same way as compare instruction. 163 static bool isLoadAndTestAsCmp(MachineInstr &MI) { 164 // If we during isel used a load-and-test as a compare with 0, the 165 // def operand is dead. 166 return (MI.getOpcode() == SystemZ::LTEBR || 167 MI.getOpcode() == SystemZ::LTDBR || 168 MI.getOpcode() == SystemZ::LTXBR) && 169 MI.getOperand(0).isDead(); 170 } 171 172 // Return the source register of Compare, which is the unknown value 173 // being tested. 174 static unsigned getCompareSourceReg(MachineInstr &Compare) { 175 unsigned reg = 0; 176 if (Compare.isCompare()) 177 reg = Compare.getOperand(0).getReg(); 178 else if (isLoadAndTestAsCmp(Compare)) 179 reg = Compare.getOperand(1).getReg(); 180 assert(reg); 181 182 return reg; 183 } 184 185 // Compare compares the result of MI against zero. If MI is an addition 186 // of -1 and if CCUsers is a single branch on nonzero, eliminate the addition 187 // and convert the branch to a BRCT(G) or BRCTH. Return true on success. 188 bool SystemZElimCompare::convertToBRCT( 189 MachineInstr &MI, MachineInstr &Compare, 190 SmallVectorImpl<MachineInstr *> &CCUsers) { 191 // Check whether we have an addition of -1. 192 unsigned Opcode = MI.getOpcode(); 193 unsigned BRCT; 194 if (Opcode == SystemZ::AHI) 195 BRCT = SystemZ::BRCT; 196 else if (Opcode == SystemZ::AGHI) 197 BRCT = SystemZ::BRCTG; 198 else if (Opcode == SystemZ::AIH) 199 BRCT = SystemZ::BRCTH; 200 else 201 return false; 202 if (MI.getOperand(2).getImm() != -1) 203 return false; 204 205 // Check whether we have a single JLH. 206 if (CCUsers.size() != 1) 207 return false; 208 MachineInstr *Branch = CCUsers[0]; 209 if (Branch->getOpcode() != SystemZ::BRC || 210 Branch->getOperand(0).getImm() != SystemZ::CCMASK_ICMP || 211 Branch->getOperand(1).getImm() != SystemZ::CCMASK_CMP_NE) 212 return false; 213 214 // We already know that there are no references to the register between 215 // MI and Compare. Make sure that there are also no references between 216 // Compare and Branch. 217 unsigned SrcReg = getCompareSourceReg(Compare); 218 MachineBasicBlock::iterator MBBI = Compare, MBBE = Branch; 219 for (++MBBI; MBBI != MBBE; ++MBBI) 220 if (getRegReferences(*MBBI, SrcReg)) 221 return false; 222 223 // The transformation is OK. Rebuild Branch as a BRCT(G) or BRCTH. 224 MachineOperand Target(Branch->getOperand(2)); 225 while (Branch->getNumOperands()) 226 Branch->RemoveOperand(0); 227 Branch->setDesc(TII->get(BRCT)); 228 MachineInstrBuilder MIB(*Branch->getParent()->getParent(), Branch); 229 MIB.add(MI.getOperand(0)).add(MI.getOperand(1)).add(Target); 230 // Add a CC def to BRCT(G), since we may have to split them again if the 231 // branch displacement overflows. BRCTH has a 32-bit displacement, so 232 // this is not necessary there. 233 if (BRCT != SystemZ::BRCTH) 234 MIB.addReg(SystemZ::CC, RegState::ImplicitDefine | RegState::Dead); 235 MI.eraseFromParent(); 236 return true; 237 } 238 239 // Compare compares the result of MI against zero. If MI is a suitable load 240 // instruction and if CCUsers is a single conditional trap on zero, eliminate 241 // the load and convert the branch to a load-and-trap. Return true on success. 242 bool SystemZElimCompare::convertToLoadAndTrap( 243 MachineInstr &MI, MachineInstr &Compare, 244 SmallVectorImpl<MachineInstr *> &CCUsers) { 245 unsigned LATOpcode = TII->getLoadAndTrap(MI.getOpcode()); 246 if (!LATOpcode) 247 return false; 248 249 // Check whether we have a single CondTrap that traps on zero. 250 if (CCUsers.size() != 1) 251 return false; 252 MachineInstr *Branch = CCUsers[0]; 253 if (Branch->getOpcode() != SystemZ::CondTrap || 254 Branch->getOperand(0).getImm() != SystemZ::CCMASK_ICMP || 255 Branch->getOperand(1).getImm() != SystemZ::CCMASK_CMP_EQ) 256 return false; 257 258 // We already know that there are no references to the register between 259 // MI and Compare. Make sure that there are also no references between 260 // Compare and Branch. 261 unsigned SrcReg = getCompareSourceReg(Compare); 262 MachineBasicBlock::iterator MBBI = Compare, MBBE = Branch; 263 for (++MBBI; MBBI != MBBE; ++MBBI) 264 if (getRegReferences(*MBBI, SrcReg)) 265 return false; 266 267 // The transformation is OK. Rebuild Branch as a load-and-trap. 268 while (Branch->getNumOperands()) 269 Branch->RemoveOperand(0); 270 Branch->setDesc(TII->get(LATOpcode)); 271 MachineInstrBuilder(*Branch->getParent()->getParent(), Branch) 272 .add(MI.getOperand(0)) 273 .add(MI.getOperand(1)) 274 .add(MI.getOperand(2)) 275 .add(MI.getOperand(3)); 276 MI.eraseFromParent(); 277 return true; 278 } 279 280 // If MI is a load instruction, try to convert it into a LOAD AND TEST. 281 // Return true on success. 282 bool SystemZElimCompare::convertToLoadAndTest( 283 MachineInstr &MI, MachineInstr &Compare, 284 SmallVectorImpl<MachineInstr *> &CCUsers) { 285 286 // Try to adjust CC masks for the LOAD AND TEST opcode that could replace MI. 287 unsigned Opcode = TII->getLoadAndTest(MI.getOpcode()); 288 if (!Opcode || !adjustCCMasksForInstr(MI, Compare, CCUsers, Opcode)) 289 return false; 290 291 // Rebuild to get the CC operand in the right place. 292 auto MIB = BuildMI(*MI.getParent(), MI, MI.getDebugLoc(), TII->get(Opcode)); 293 for (const auto &MO : MI.operands()) 294 MIB.add(MO); 295 MIB.setMemRefs(MI.memoperands()); 296 MI.eraseFromParent(); 297 298 return true; 299 } 300 301 // The CC users in CCUsers are testing the result of a comparison of some 302 // value X against zero and we know that any CC value produced by MI would 303 // also reflect the value of X. ConvOpc may be used to pass the transfomed 304 // opcode MI will have if this succeeds. Try to adjust CCUsers so that they 305 // test the result of MI directly, returning true on success. Leave 306 // everything unchanged on failure. 307 bool SystemZElimCompare::adjustCCMasksForInstr( 308 MachineInstr &MI, MachineInstr &Compare, 309 SmallVectorImpl<MachineInstr *> &CCUsers, 310 unsigned ConvOpc) { 311 int Opcode = (ConvOpc ? ConvOpc : MI.getOpcode()); 312 const MCInstrDesc &Desc = TII->get(Opcode); 313 unsigned MIFlags = Desc.TSFlags; 314 315 // See which compare-style condition codes are available. 316 unsigned ReusableCCMask = SystemZII::getCompareZeroCCMask(MIFlags); 317 318 // For unsigned comparisons with zero, only equality makes sense. 319 unsigned CompareFlags = Compare.getDesc().TSFlags; 320 if (CompareFlags & SystemZII::IsLogical) 321 ReusableCCMask &= SystemZ::CCMASK_CMP_EQ; 322 323 if (ReusableCCMask == 0) 324 return false; 325 326 unsigned CCValues = SystemZII::getCCValues(MIFlags); 327 assert((ReusableCCMask & ~CCValues) == 0 && "Invalid CCValues"); 328 329 bool MIEquivalentToCmp = 330 (ReusableCCMask == CCValues && 331 CCValues == SystemZII::getCCValues(CompareFlags)); 332 333 if (!MIEquivalentToCmp) { 334 // Now check whether these flags are enough for all users. 335 SmallVector<MachineOperand *, 4> AlterMasks; 336 for (unsigned int I = 0, E = CCUsers.size(); I != E; ++I) { 337 MachineInstr *MI = CCUsers[I]; 338 339 // Fail if this isn't a use of CC that we understand. 340 unsigned Flags = MI->getDesc().TSFlags; 341 unsigned FirstOpNum; 342 if (Flags & SystemZII::CCMaskFirst) 343 FirstOpNum = 0; 344 else if (Flags & SystemZII::CCMaskLast) 345 FirstOpNum = MI->getNumExplicitOperands() - 2; 346 else 347 return false; 348 349 // Check whether the instruction predicate treats all CC values 350 // outside of ReusableCCMask in the same way. In that case it 351 // doesn't matter what those CC values mean. 352 unsigned CCValid = MI->getOperand(FirstOpNum).getImm(); 353 unsigned CCMask = MI->getOperand(FirstOpNum + 1).getImm(); 354 unsigned OutValid = ~ReusableCCMask & CCValid; 355 unsigned OutMask = ~ReusableCCMask & CCMask; 356 if (OutMask != 0 && OutMask != OutValid) 357 return false; 358 359 AlterMasks.push_back(&MI->getOperand(FirstOpNum)); 360 AlterMasks.push_back(&MI->getOperand(FirstOpNum + 1)); 361 } 362 363 // All users are OK. Adjust the masks for MI. 364 for (unsigned I = 0, E = AlterMasks.size(); I != E; I += 2) { 365 AlterMasks[I]->setImm(CCValues); 366 unsigned CCMask = AlterMasks[I + 1]->getImm(); 367 if (CCMask & ~ReusableCCMask) 368 AlterMasks[I + 1]->setImm((CCMask & ReusableCCMask) | 369 (CCValues & ~ReusableCCMask)); 370 } 371 } 372 373 // CC is now live after MI. 374 if (!ConvOpc) 375 MI.clearRegisterDeads(SystemZ::CC); 376 377 // Check if MI lies before Compare. 378 bool BeforeCmp = false; 379 MachineBasicBlock::iterator MBBI = MI, MBBE = MI.getParent()->end(); 380 for (++MBBI; MBBI != MBBE; ++MBBI) 381 if (MBBI == Compare) { 382 BeforeCmp = true; 383 break; 384 } 385 386 // Clear any intervening kills of CC. 387 if (BeforeCmp) { 388 MachineBasicBlock::iterator MBBI = MI, MBBE = Compare; 389 for (++MBBI; MBBI != MBBE; ++MBBI) 390 MBBI->clearRegisterKills(SystemZ::CC, TRI); 391 } 392 393 return true; 394 } 395 396 // Return true if Compare is a comparison against zero. 397 static bool isCompareZero(MachineInstr &Compare) { 398 switch (Compare.getOpcode()) { 399 case SystemZ::LTEBRCompare: 400 case SystemZ::LTDBRCompare: 401 case SystemZ::LTXBRCompare: 402 return true; 403 404 default: 405 if (isLoadAndTestAsCmp(Compare)) 406 return true; 407 return Compare.getNumExplicitOperands() == 2 && 408 Compare.getOperand(1).isImm() && Compare.getOperand(1).getImm() == 0; 409 } 410 } 411 412 // Try to optimize cases where comparison instruction Compare is testing 413 // a value against zero. Return true on success and if Compare should be 414 // deleted as dead. CCUsers is the list of instructions that use the CC 415 // value produced by Compare. 416 bool SystemZElimCompare::optimizeCompareZero( 417 MachineInstr &Compare, SmallVectorImpl<MachineInstr *> &CCUsers) { 418 if (!isCompareZero(Compare)) 419 return false; 420 421 // Search back for CC results that are based on the first operand. 422 unsigned SrcReg = getCompareSourceReg(Compare); 423 MachineBasicBlock &MBB = *Compare.getParent(); 424 Reference CCRefs; 425 Reference SrcRefs; 426 for (MachineBasicBlock::reverse_iterator MBBI = 427 std::next(MachineBasicBlock::reverse_iterator(&Compare)), 428 MBBE = MBB.rend(); MBBI != MBBE;) { 429 MachineInstr &MI = *MBBI++; 430 if (resultTests(MI, SrcReg)) { 431 // Try to remove both MI and Compare by converting a branch to BRCT(G). 432 // or a load-and-trap instruction. We don't care in this case whether 433 // CC is modified between MI and Compare. 434 if (!CCRefs.Use && !SrcRefs) { 435 if (convertToBRCT(MI, Compare, CCUsers)) { 436 BranchOnCounts += 1; 437 return true; 438 } 439 if (convertToLoadAndTrap(MI, Compare, CCUsers)) { 440 LoadAndTraps += 1; 441 return true; 442 } 443 } 444 // Try to eliminate Compare by reusing a CC result from MI. 445 if ((!CCRefs && convertToLoadAndTest(MI, Compare, CCUsers)) || 446 (!CCRefs.Def && adjustCCMasksForInstr(MI, Compare, CCUsers))) { 447 EliminatedComparisons += 1; 448 return true; 449 } 450 } 451 SrcRefs |= getRegReferences(MI, SrcReg); 452 if (SrcRefs.Def) 453 break; 454 CCRefs |= getRegReferences(MI, SystemZ::CC); 455 if (CCRefs.Use && CCRefs.Def) 456 break; 457 } 458 459 // Also do a forward search to handle cases where an instruction after the 460 // compare can be converted, like 461 // LTEBRCompare %f0s, %f0s; %f2s = LER %f0s => LTEBRCompare %f2s, %f0s 462 for (MachineBasicBlock::iterator MBBI = 463 std::next(MachineBasicBlock::iterator(&Compare)), MBBE = MBB.end(); 464 MBBI != MBBE;) { 465 MachineInstr &MI = *MBBI++; 466 if (preservesValueOf(MI, SrcReg)) { 467 // Try to eliminate Compare by reusing a CC result from MI. 468 if (convertToLoadAndTest(MI, Compare, CCUsers)) { 469 EliminatedComparisons += 1; 470 return true; 471 } 472 } 473 if (getRegReferences(MI, SrcReg).Def) 474 return false; 475 if (getRegReferences(MI, SystemZ::CC)) 476 return false; 477 } 478 479 return false; 480 } 481 482 // Try to fuse comparison instruction Compare into a later branch. 483 // Return true on success and if Compare is therefore redundant. 484 bool SystemZElimCompare::fuseCompareOperations( 485 MachineInstr &Compare, SmallVectorImpl<MachineInstr *> &CCUsers) { 486 // See whether we have a single branch with which to fuse. 487 if (CCUsers.size() != 1) 488 return false; 489 MachineInstr *Branch = CCUsers[0]; 490 SystemZII::FusedCompareType Type; 491 switch (Branch->getOpcode()) { 492 case SystemZ::BRC: 493 Type = SystemZII::CompareAndBranch; 494 break; 495 case SystemZ::CondReturn: 496 Type = SystemZII::CompareAndReturn; 497 break; 498 case SystemZ::CallBCR: 499 Type = SystemZII::CompareAndSibcall; 500 break; 501 case SystemZ::CondTrap: 502 Type = SystemZII::CompareAndTrap; 503 break; 504 default: 505 return false; 506 } 507 508 // See whether we have a comparison that can be fused. 509 unsigned FusedOpcode = 510 TII->getFusedCompare(Compare.getOpcode(), Type, &Compare); 511 if (!FusedOpcode) 512 return false; 513 514 // Make sure that the operands are available at the branch. 515 // SrcReg2 is the register if the source operand is a register, 516 // 0 if the source operand is immediate, and the base register 517 // if the source operand is memory (index is not supported). 518 Register SrcReg = Compare.getOperand(0).getReg(); 519 Register SrcReg2 = 520 Compare.getOperand(1).isReg() ? Compare.getOperand(1).getReg() : Register(); 521 MachineBasicBlock::iterator MBBI = Compare, MBBE = Branch; 522 for (++MBBI; MBBI != MBBE; ++MBBI) 523 if (MBBI->modifiesRegister(SrcReg, TRI) || 524 (SrcReg2 && MBBI->modifiesRegister(SrcReg2, TRI))) 525 return false; 526 527 // Read the branch mask, target (if applicable), regmask (if applicable). 528 MachineOperand CCMask(MBBI->getOperand(1)); 529 assert((CCMask.getImm() & ~SystemZ::CCMASK_ICMP) == 0 && 530 "Invalid condition-code mask for integer comparison"); 531 // This is only valid for CompareAndBranch. 532 MachineOperand Target(MBBI->getOperand( 533 Type == SystemZII::CompareAndBranch ? 2 : 0)); 534 const uint32_t *RegMask; 535 if (Type == SystemZII::CompareAndSibcall) 536 RegMask = MBBI->getOperand(2).getRegMask(); 537 538 // Clear out all current operands. 539 int CCUse = MBBI->findRegisterUseOperandIdx(SystemZ::CC, false, TRI); 540 assert(CCUse >= 0 && "BRC/BCR must use CC"); 541 Branch->RemoveOperand(CCUse); 542 // Remove target (branch) or regmask (sibcall). 543 if (Type == SystemZII::CompareAndBranch || 544 Type == SystemZII::CompareAndSibcall) 545 Branch->RemoveOperand(2); 546 Branch->RemoveOperand(1); 547 Branch->RemoveOperand(0); 548 549 // Rebuild Branch as a fused compare and branch. 550 // SrcNOps is the number of MI operands of the compare instruction 551 // that we need to copy over. 552 unsigned SrcNOps = 2; 553 if (FusedOpcode == SystemZ::CLT || FusedOpcode == SystemZ::CLGT) 554 SrcNOps = 3; 555 Branch->setDesc(TII->get(FusedOpcode)); 556 MachineInstrBuilder MIB(*Branch->getParent()->getParent(), Branch); 557 for (unsigned I = 0; I < SrcNOps; I++) 558 MIB.add(Compare.getOperand(I)); 559 MIB.add(CCMask); 560 561 if (Type == SystemZII::CompareAndBranch) { 562 // Only conditional branches define CC, as they may be converted back 563 // to a non-fused branch because of a long displacement. Conditional 564 // returns don't have that problem. 565 MIB.add(Target).addReg(SystemZ::CC, 566 RegState::ImplicitDefine | RegState::Dead); 567 } 568 569 if (Type == SystemZII::CompareAndSibcall) 570 MIB.addRegMask(RegMask); 571 572 // Clear any intervening kills of SrcReg and SrcReg2. 573 MBBI = Compare; 574 for (++MBBI; MBBI != MBBE; ++MBBI) { 575 MBBI->clearRegisterKills(SrcReg, TRI); 576 if (SrcReg2) 577 MBBI->clearRegisterKills(SrcReg2, TRI); 578 } 579 FusedComparisons += 1; 580 return true; 581 } 582 583 // Process all comparison instructions in MBB. Return true if something 584 // changed. 585 bool SystemZElimCompare::processBlock(MachineBasicBlock &MBB) { 586 bool Changed = false; 587 588 // Walk backwards through the block looking for comparisons, recording 589 // all CC users as we go. The subroutines can delete Compare and 590 // instructions before it. 591 LivePhysRegs LiveRegs(*TRI); 592 LiveRegs.addLiveOuts(MBB); 593 bool CompleteCCUsers = !LiveRegs.contains(SystemZ::CC); 594 SmallVector<MachineInstr *, 4> CCUsers; 595 MachineBasicBlock::iterator MBBI = MBB.end(); 596 while (MBBI != MBB.begin()) { 597 MachineInstr &MI = *--MBBI; 598 if (CompleteCCUsers && (MI.isCompare() || isLoadAndTestAsCmp(MI)) && 599 (optimizeCompareZero(MI, CCUsers) || 600 fuseCompareOperations(MI, CCUsers))) { 601 ++MBBI; 602 MI.eraseFromParent(); 603 Changed = true; 604 CCUsers.clear(); 605 continue; 606 } 607 608 if (MI.definesRegister(SystemZ::CC)) { 609 CCUsers.clear(); 610 CompleteCCUsers = true; 611 } 612 if (MI.readsRegister(SystemZ::CC) && CompleteCCUsers) 613 CCUsers.push_back(&MI); 614 } 615 return Changed; 616 } 617 618 bool SystemZElimCompare::runOnMachineFunction(MachineFunction &F) { 619 if (skipFunction(F.getFunction())) 620 return false; 621 622 TII = static_cast<const SystemZInstrInfo *>(F.getSubtarget().getInstrInfo()); 623 TRI = &TII->getRegisterInfo(); 624 625 bool Changed = false; 626 for (auto &MBB : F) 627 Changed |= processBlock(MBB); 628 629 return Changed; 630 } 631 632 FunctionPass *llvm::createSystemZElimComparePass(SystemZTargetMachine &TM) { 633 return new SystemZElimCompare(TM); 634 } 635